X-Git-Url: http://git.vrable.net/?a=blobdiff_plain;f=localdb.cc;h=4683f4d83c567dca863fe42886a0a5f2ceac348b;hb=cac644d993d130efd8d29539de6557b18c9b737e;hp=63a9f8349531e0fb13531b8ba7931dc4300c112d;hpb=ed5a047c1b8a8f5c43b0192e8e774b91497c9706;p=cumulus.git

diff --git a/localdb.cc b/localdb.cc
index 63a9f83..4683f4d 100644
--- a/localdb.cc
+++ b/localdb.cc
@@ -22,7 +22,7 @@
 
 using std::string;
 
-void LocalDb::Open(const char *path)
+void LocalDb::Open(const char *path, const char *snapshot_name)
 {
     int rc;
 
@@ -39,6 +39,36 @@ void LocalDb::Open(const char *path)
         sqlite3_close(db);
         throw IOException("Error starting transaction");
     }
+
+    /* Insert this snapshot into the database, and determine the integer key
+     * which will be used to identify it. */
+    sqlite3_stmt *stmt;
+    static const char s[] =
+        "insert into snapshots(name, timestamp) "
+        "values (?, julianday('now'))";
+    const char *tail;
+
+    rc = sqlite3_prepare_v2(db, s, strlen(s), &stmt, &tail);
+    if (rc != SQLITE_OK) {
+        sqlite3_close(db);
+        throw IOException("Error adding snapshot");
+    }
+
+    sqlite3_bind_text(stmt, 1, snapshot_name, strlen(snapshot_name),
+                      SQLITE_TRANSIENT);
+
+    rc = sqlite3_step(stmt);
+    if (rc != SQLITE_DONE) {
+        sqlite3_close(db);
+        throw IOException("Database execution error!");
+    }
+
+    snapshotid = sqlite3_last_insert_rowid(db);
+    sqlite3_finalize(stmt);
+    if (snapshotid == 0) {
+        sqlite3_close(db);
+        throw IOException("Find snapshot id");
+    }
 }
 
 void LocalDb::Close()
@@ -51,14 +81,90 @@ void LocalDb::Close()
     sqlite3_close(db);
 }
 
+int64_t LocalDb::SegmentToId(const string &segment)
+{
+    int rc;
+    sqlite3_stmt *stmt;
+    static const char s1[] =
+        "insert or ignore into segments(segment) values (?);";
+    static const char s2[] =
+        "select segmentid from segments where segment = ?";
+    const char *tail;
+    int64_t result;
+
+    rc = sqlite3_prepare_v2(db, s1, strlen(s1), &stmt, &tail);
+    if (rc != SQLITE_OK) {
+        throw IOException("Find id by segment name");
+    }
+    sqlite3_bind_text(stmt, 1, segment.c_str(), segment.size(),
+                      SQLITE_TRANSIENT);
+    rc = sqlite3_step(stmt);
+    if (rc != SQLITE_DONE) {
+        throw IOException("Could not execute INSERT statement!");
+    }
+    sqlite3_finalize(stmt);
+
+    rc = sqlite3_prepare_v2(db, s2, strlen(s2), &stmt, &tail);
+    if (rc != SQLITE_OK) {
+        throw IOException("Find id by segment name");
+    }
+
+    sqlite3_bind_text(stmt, 1, segment.c_str(), segment.size(),
+                      SQLITE_TRANSIENT);
+
+    rc = sqlite3_step(stmt);
+    if (rc == SQLITE_DONE) {
+        throw IOException("No segment found by id");
+    } else if (rc == SQLITE_ROW) {
+        result = sqlite3_column_int64(stmt, 0);
+    } else {
+        throw IOException("Error executing find segment by id query");
+    }
+
+    sqlite3_finalize(stmt);
+
+    return result;
+}
+
+string LocalDb::IdToSegment(int64_t segmentid)
+{
+    int rc;
+    sqlite3_stmt *stmt;
+    static const char s[] =
+        "select segment from segments where segmentid = ?";
+    const char *tail;
+    string result;
+
+    rc = sqlite3_prepare_v2(db, s, strlen(s), &stmt, &tail);
+    if (rc != SQLITE_OK) {
+        throw IOException("Find segment by id");
+    }
+
+    sqlite3_bind_int64(stmt, 1, segmentid);
+
+    rc = sqlite3_step(stmt);
+    if (rc == SQLITE_DONE) {
+        throw IOException("No segment found by id");
+    } else if (rc == SQLITE_ROW) {
+        result = (const char *)sqlite3_column_text(stmt, 0);
+    } else {
+        throw IOException("Error executing find segment by id query");
+    }
+
+    sqlite3_finalize(stmt);
+
+    return result;
+}
+
 void LocalDb::StoreObject(const ObjectReference& ref,
                           const string &checksum, int64_t size)
 {
     int rc;
     sqlite3_stmt *stmt;
     static const char s[] =
-        "insert into block_index(segment, object, checksum, size) "
-        "values (?, ?, ?, ?)";
+        "insert into "
+        "block_index(segmentid, object, checksum, size, timestamp) "
+        "values (?, ?, ?, ?, julianday('now'))";
     const char *tail;
 
     rc = sqlite3_prepare_v2(db, s, strlen(s), &stmt, &tail);
@@ -66,8 +172,7 @@ void LocalDb::StoreObject(const ObjectReference& ref,
         return;
     }
 
-    string seg = ref.get_segment();
-    sqlite3_bind_text(stmt, 1, seg.c_str(), seg.size(), SQLITE_TRANSIENT);
+    sqlite3_bind_int64(stmt, 1, SegmentToId(ref.get_segment()));
     string obj = ref.get_sequence();
     sqlite3_bind_text(stmt, 2, obj.c_str(), obj.size(), SQLITE_TRANSIENT);
     sqlite3_bind_text(stmt, 3, checksum.c_str(), checksum.size(),
@@ -81,3 +186,99 @@ void LocalDb::StoreObject(const ObjectReference& ref,
 
     sqlite3_finalize(stmt);
 }
+
+ObjectReference LocalDb::FindObject(const string &checksum, int64_t size)
+{
+    int rc;
+    sqlite3_stmt *stmt;
+    static const char s[] =
+        "select segmentid, object from block_index "
+        "where checksum = ? and size = ? and expired is null";
+    const char *tail;
+
+    ObjectReference ref;
+
+    rc = sqlite3_prepare_v2(db, s, strlen(s), &stmt, &tail);
+    if (rc != SQLITE_OK) {
+        return ref;
+    }
+
+    sqlite3_bind_text(stmt, 1, checksum.c_str(), checksum.size(),
+                      SQLITE_TRANSIENT);
+    sqlite3_bind_int64(stmt, 2, size);
+
+    rc = sqlite3_step(stmt);
+    if (rc == SQLITE_DONE) {
+    } else if (rc == SQLITE_ROW) {
+        ref = ObjectReference(IdToSegment(sqlite3_column_int64(stmt, 0)),
+                              (const char *)sqlite3_column_text(stmt, 1));
+    } else {
+        fprintf(stderr, "Could not execute SELECT statement!\n");
+    }
+
+    sqlite3_finalize(stmt);
+
+    return ref;
+}
+
+bool LocalDb::IsOldObject(const string &checksum, int64_t size)
+{
+    int rc;
+    sqlite3_stmt *stmt;
+    static const char s[] =
+        "select segmentid, object from block_index "
+        "where checksum = ? and size = ?";
+    const char *tail;
+
+    bool found = false;
+
+    rc = sqlite3_prepare_v2(db, s, strlen(s), &stmt, &tail);
+    if (rc != SQLITE_OK) {
+        return false;
+    }
+
+    sqlite3_bind_text(stmt, 1, checksum.c_str(), checksum.size(),
+                      SQLITE_TRANSIENT);
+    sqlite3_bind_int64(stmt, 2, size);
+
+    rc = sqlite3_step(stmt);
+    if (rc == SQLITE_DONE) {
+        found = false;
+    } else if (rc == SQLITE_ROW) {
+        found = true;
+    } else {
+        fprintf(stderr, "Could not execute SELECT statement!\n");
+    }
+
+    sqlite3_finalize(stmt);
+
+    return found;
+}
+
+void LocalDb::UseObject(const ObjectReference& ref)
+{
+    int rc;
+    sqlite3_stmt *stmt;
+    static const char s[] =
+        "insert or ignore into snapshot_contents "
+        "select blockid, ? as snapshotid from block_index "
+        "where segmentid = ? and object = ?";
+    const char *tail;
+
+    rc = sqlite3_prepare_v2(db, s, strlen(s), &stmt, &tail);
+    if (rc != SQLITE_OK) {
+        return;
+    }
+
+    sqlite3_bind_int64(stmt, 1, snapshotid);
+    sqlite3_bind_int64(stmt, 2, SegmentToId(ref.get_segment()));
+    string obj = ref.get_sequence();
+    sqlite3_bind_text(stmt, 3, obj.c_str(), obj.size(), SQLITE_TRANSIENT);
+
+    rc = sqlite3_step(stmt);
+    if (rc != SQLITE_DONE) {
+        fprintf(stderr, "Could not execute INSERT statement!\n");
+    }
+
+    sqlite3_finalize(stmt);
+}