using std::string;
-void LocalDb::Open(const char *path)
+/* Helper function to prepare a statement for execution in the current
+ * database. */
+sqlite3_stmt *LocalDb::Prepare(const char *sql)
+{
+ sqlite3_stmt *stmt;
+ int rc;
+ const char *tail;
+
+ rc = sqlite3_prepare(db, sql, strlen(sql), &stmt, &tail);
+ if (rc != SQLITE_OK) {
+ throw IOException(string("Error preparing statement: ") + sql);
+ }
+
+ return stmt;
+}
+
+void LocalDb::Open(const char *path, const char *snapshot_name)
{
int rc;
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 = Prepare("insert into snapshots(name, timestamp) "
+ "values (?, julianday('now'))");
+ 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()
sqlite3_close(db);
}
+int64_t LocalDb::SegmentToId(const string &segment)
+{
+ int rc;
+ sqlite3_stmt *stmt;
+ int64_t result;
+
+ stmt = Prepare("insert or ignore into segments(segment) values (?)");
+ 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);
+
+ stmt = Prepare("select segmentid from segments where segment = ?");
+ 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;
+ string result;
+
+ stmt = Prepare("select segment from segments where segmentid = ?");
+ 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 (?, ?, ?, ?)";
- const char *tail;
- rc = sqlite3_prepare_v2(db, s, strlen(s), &stmt, &tail);
- if (rc != SQLITE_OK) {
- return;
- }
+ stmt = Prepare("insert into "
+ "block_index(segmentid, object, checksum, size, timestamp) "
+ "values (?, ?, ?, ?, julianday('now'))");
- 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(),
sqlite3_finalize(stmt);
}
+
+ObjectReference LocalDb::FindObject(const string &checksum, int64_t size)
+{
+ int rc;
+ sqlite3_stmt *stmt;
+ ObjectReference ref;
+
+ stmt = Prepare("select segmentid, object from block_index "
+ "where checksum = ? and size = ? and expired is null");
+ 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;
+ bool found = false;
+
+ stmt = Prepare("select segmentid, object from block_index "
+ "where checksum = ? and size = ?");
+ 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;
+
+ stmt = Prepare("insert or ignore into snapshot_contents "
+ "select blockid, ? as snapshotid from block_index "
+ "where segmentid = ? and object = ?");
+ 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);
+}