add fetch-task-db and integrate it with fetch-manager
diff --git a/src/fetch-task-db.cc b/src/fetch-task-db.cc
new file mode 100644
index 0000000..11ae6c9
--- /dev/null
+++ b/src/fetch-task-db.cc
@@ -0,0 +1,126 @@
+/* -*- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil -*- */
+/*
+ * Copyright (c) 2013 University of California, Los Angeles
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 as
+ * published by the Free Software Foundation;
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ *
+ * Author: Zhenkai Zhu <zhenkai@cs.ucla.edu>
+ * Alexander Afanasyev <alexander.afanasyev@ucla.edu>
+ */
+#include "fetch-task-db.h"
+#include "db-helper.h"
+
+using namespace std;
+using namespace boost;
+using namespace Ccnx;
+namespace fs = boost::filesystem;
+
+const string INIT_DATABASE = "\
+CREATE TABLE IF NOT EXISTS \n\
+ Task( \n\
+ deviceName BLOB NOT NULL, \n\
+ baseName BLOB NOT NULL, \n\
+ minSeqNo INTEGER, \n\
+ maxSeqNo INTEGER, \n\
+ priority INTEGER, \n\
+ PRIMARY KEY (deviceName, baseName) \n\
+ ); \n\
+CREATE INDEX identifier ON Task (deviceName, baseName); \n\
+";
+
+FetchTaskDb::FetchTaskDb(const boost::filesystem::path &folder, const std::string &tag)
+{
+ fs::path actualFolder = folder / ".chronoshare" / "fetch_tasks";
+ fs::create_directories (actualFolder);
+
+ int res = sqlite3_open((actualFolder / tag).c_str(), &m_db);
+ if (res != SQLITE_OK)
+ {
+ BOOST_THROW_EXCEPTION(Error::Db() << errmsg_info_str("Cannot open database: " + (actualFolder / tag).string()));
+ }
+
+ char *errmsg = 0;
+ res = sqlite3_exec(m_db, INIT_DATABASE.c_str(), NULL, NULL, &errmsg);
+ if (res != SQLITE_OK && errmsg != 0)
+ {
+ // _LOG_TRACE ("Init \"error\": " << errmsg);
+ sqlite3_free (errmsg);
+ }
+ else
+ {
+ }
+}
+
+FetchTaskDb::~FetchTaskDb()
+{
+ int res = sqlite3_close(m_db);
+ if (res != SQLITE_OK)
+ {
+ // _LOG_ERROR
+ }
+}
+
+void
+FetchTaskDb::addTask(const Name &deviceName, const Name &baseName, uint64_t minSeqNo, uint64_t maxSeqNo, int priority)
+{
+ sqlite3_stmt *stmt;
+ sqlite3_prepare_v2(m_db, "INSERT OR IGNORE INTO Task (deviceName, baseName, minSeqNo, maxSeqNo, priority) VALUES (?, ?, ?, ?, ?)", -1, &stmt, 0);
+ CcnxCharbufPtr deviceBuf = CcnxCharbufPtr(deviceName);
+ CcnxCharbufPtr baseBuf = CcnxCharbufPtr(baseName);
+ sqlite3_bind_blob(stmt, 1, deviceBuf->buf(), deviceBuf->length(), SQLITE_STATIC);
+ sqlite3_bind_blob(stmt, 2, baseBuf->buf(), baseBuf->length(), SQLITE_STATIC);
+ sqlite3_bind_int64(stmt, 3, minSeqNo);
+ sqlite3_bind_int64(stmt, 4, maxSeqNo);
+ sqlite3_bind_int(stmt, 5, priority);
+ int res = sqlite3_step(stmt);
+
+ if (res == SQLITE_OK)
+ {
+ }
+ sqlite3_finalize(stmt);
+}
+
+void
+FetchTaskDb::deleteTask(const Name &deviceName, const Name &baseName)
+{
+ sqlite3_stmt *stmt;
+ sqlite3_prepare_v2(m_db, "DELETE FROM Task WHERE deviceName = ? AND baseName = ?;", -1, &stmt, 0);
+ CcnxCharbufPtr deviceBuf = CcnxCharbufPtr(deviceName);
+ CcnxCharbufPtr baseBuf = CcnxCharbufPtr(baseName);
+ sqlite3_bind_blob(stmt, 1, deviceBuf->buf(), deviceBuf->length(), SQLITE_STATIC);
+ sqlite3_bind_blob(stmt, 2, baseBuf->buf(), baseBuf->length(), SQLITE_STATIC);
+ int res = sqlite3_step(stmt);
+ if (res == SQLITE_OK)
+ {
+ }
+ sqlite3_finalize(stmt);
+}
+
+void
+FetchTaskDb::foreachTask(const FetchTaskCallback &callback)
+{
+ sqlite3_stmt *stmt;
+ sqlite3_prepare_v2(m_db, "SELECT * FROM Task;", -1, &stmt, 0);
+ while (sqlite3_step(stmt) == SQLITE_ROW)
+ {
+ Name deviceName(sqlite3_column_blob(stmt, 0), sqlite3_column_bytes(stmt, 0));
+ Name baseName(sqlite3_column_blob(stmt, 1), sqlite3_column_bytes(stmt, 1));
+ uint64_t minSeqNo = sqlite3_column_int64(stmt, 2);
+ uint64_t maxSeqNo = sqlite3_column_int64(stmt, 3);
+ int priority = sqlite3_column_int(stmt, 4);
+ callback(deviceName, baseName, minSeqNo, maxSeqNo, priority);
+ }
+
+ sqlite3_finalize(stmt);
+}