X-Git-Url: http://git.vrable.net/?a=blobdiff_plain;f=localdb.cc;h=251148b27711d1afed984bdf497ce3ad0b4da4d1;hb=2a08b08c0b0ddc64d0be49a3d8c503feb8b4058f;hp=951f35ceb67063f8a627f2d564a39a97071ea317;hpb=58a0d3f8749111c15e9afa9d929016d65ed32250;p=cumulus.git diff --git a/localdb.cc b/localdb.cc index 951f35c..251148b 100644 --- a/localdb.cc +++ b/localdb.cc @@ -30,7 +30,7 @@ sqlite3_stmt *LocalDb::Prepare(const char *sql) int rc; const char *tail; - rc = sqlite3_prepare(db, sql, strlen(sql), &stmt, &tail); + rc = sqlite3_prepare_v2(db, sql, strlen(sql), &stmt, &tail); if (rc != SQLITE_OK) { throw IOException(string("Error preparing statement: ") + sql); } @@ -38,7 +38,14 @@ sqlite3_stmt *LocalDb::Prepare(const char *sql) return stmt; } -void LocalDb::Open(const char *path, const char *snapshot_name) +void LocalDb::ReportError(int rc) +{ + fprintf(stderr, "Result code: %d\n", rc); + fprintf(stderr, "Error message: %s\n", sqlite3_errmsg(db)); +} + +void LocalDb::Open(const char *path, const char *snapshot_name, + const char *snapshot_scheme) { int rc; @@ -56,15 +63,24 @@ void LocalDb::Open(const char *path, const char *snapshot_name) throw IOException("Error starting transaction"); } + sqlite3_extended_result_codes(db, 1); + /* 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_stmt *stmt = Prepare("insert into " + "snapshots(name, scheme, timestamp) " + "values (?, ?, julianday('now'))"); sqlite3_bind_text(stmt, 1, snapshot_name, strlen(snapshot_name), SQLITE_TRANSIENT); + if (snapshot_scheme == NULL) + sqlite3_bind_null(stmt, 2); + else + sqlite3_bind_text(stmt, 2, snapshot_scheme, strlen(snapshot_scheme), + SQLITE_TRANSIENT); rc = sqlite3_step(stmt); if (rc != SQLITE_DONE) { + ReportError(rc); sqlite3_close(db); throw IOException("Database execution error!"); } @@ -72,6 +88,7 @@ void LocalDb::Open(const char *path, const char *snapshot_name) snapshotid = sqlite3_last_insert_rowid(db); sqlite3_finalize(stmt); if (snapshotid == 0) { + ReportError(rc); sqlite3_close(db); throw IOException("Find snapshot id"); } @@ -82,7 +99,8 @@ void LocalDb::Close() int rc; rc = sqlite3_exec(db, "commit", NULL, NULL, NULL); if (rc != SQLITE_OK) { - fprintf(stderr, "Can't commit database!\n"); + fprintf(stderr, "DATABASE ERROR: Can't commit database!\n"); + ReportError(rc); } sqlite3_close(db); } @@ -144,14 +162,21 @@ string LocalDb::IdToSegment(int64_t segmentid) } void LocalDb::StoreObject(const ObjectReference& ref, - const string &checksum, int64_t size) + const string &checksum, int64_t size, + double age) { int rc; sqlite3_stmt *stmt; - stmt = Prepare("insert into " - "block_index(segmentid, object, checksum, size, timestamp) " - "values (?, ?, ?, ?, julianday('now'))"); + if (age == 0.0) { + stmt = Prepare("insert into block_index(" + "segmentid, object, checksum, size, timestamp) " + "values (?, ?, ?, ?, julianday('now'))"); + } else { + stmt = Prepare("insert into block_index(" + "segmentid, object, checksum, size, timestamp) " + "values (?, ?, ?, ?, ?)"); + } sqlite3_bind_int64(stmt, 1, SegmentToId(ref.get_segment())); string obj = ref.get_sequence(); @@ -159,10 +184,13 @@ void LocalDb::StoreObject(const ObjectReference& ref, sqlite3_bind_text(stmt, 3, checksum.c_str(), checksum.size(), SQLITE_TRANSIENT); sqlite3_bind_int64(stmt, 4, size); + if (age != 0.0) + sqlite3_bind_double(stmt, 5, age); rc = sqlite3_step(stmt); if (rc != SQLITE_DONE) { fprintf(stderr, "Could not execute INSERT statement!\n"); + ReportError(rc); } sqlite3_finalize(stmt); @@ -187,6 +215,7 @@ ObjectReference LocalDb::FindObject(const string &checksum, int64_t size) (const char *)sqlite3_column_text(stmt, 1)); } else { fprintf(stderr, "Could not execute SELECT statement!\n"); + ReportError(rc); } sqlite3_finalize(stmt); @@ -194,14 +223,15 @@ ObjectReference LocalDb::FindObject(const string &checksum, int64_t size) return ref; } -bool LocalDb::IsOldObject(const string &checksum, int64_t size) +bool LocalDb::IsOldObject(const string &checksum, int64_t size, double *age, + int *group) { int rc; sqlite3_stmt *stmt; bool found = false; - stmt = Prepare("select segmentid, object from block_index " - "where checksum = ? and size = ?"); + stmt = Prepare("select segmentid, object, timestamp, expired " + "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); @@ -211,8 +241,40 @@ bool LocalDb::IsOldObject(const string &checksum, int64_t size) found = false; } else if (rc == SQLITE_ROW) { found = true; + *age = sqlite3_column_double(stmt, 2); + *group = sqlite3_column_int(stmt, 3); } else { fprintf(stderr, "Could not execute SELECT statement!\n"); + ReportError(rc); + } + + sqlite3_finalize(stmt); + + return found; +} + +/* Does this object still exist in the database (and not expired)? */ +bool LocalDb::IsAvailable(const ObjectReference &ref) +{ + int rc; + sqlite3_stmt *stmt; + bool found = false; + + stmt = Prepare("select count(*) from block_index " + "where segmentid = ? and object = ? and expired is null"); + sqlite3_bind_int64(stmt, 1, SegmentToId(ref.get_segment())); + sqlite3_bind_text(stmt, 2, ref.get_sequence().c_str(), + ref.get_sequence().size(), SQLITE_TRANSIENT); + + rc = sqlite3_step(stmt); + if (rc == SQLITE_DONE) { + found = false; + } else if (rc == SQLITE_ROW) { + if (sqlite3_column_int(stmt, 0) > 0) + found = true; + } else { + fprintf(stderr, "Could not execute SELECT statement!\n"); + ReportError(rc); } sqlite3_finalize(stmt); @@ -236,7 +298,72 @@ void LocalDb::UseObject(const ObjectReference& ref) rc = sqlite3_step(stmt); if (rc != SQLITE_DONE) { fprintf(stderr, "Could not execute INSERT statement!\n"); + ReportError(rc); + } + + sqlite3_finalize(stmt); +} + +void LocalDb::SetSegmentChecksum(const std::string &segment, + const std::string &path, + const std::string &checksum) +{ + int rc; + sqlite3_stmt *stmt; + + stmt = Prepare("update segments set path = ?, checksum = ? " + "where segmentid = ?"); + sqlite3_bind_text(stmt, 1, path.c_str(), path.size(), + SQLITE_TRANSIENT); + sqlite3_bind_text(stmt, 2, checksum.c_str(), checksum.size(), + SQLITE_TRANSIENT); + sqlite3_bind_int64(stmt, 3, SegmentToId(segment)); + + rc = sqlite3_step(stmt); + if (rc != SQLITE_DONE) { + fprintf(stderr, "Could not update segment checksum in database!\n"); + ReportError(rc); } sqlite3_finalize(stmt); } + +bool LocalDb::GetSegmentChecksum(const string &segment, + string *seg_path, + string *seg_checksum) +{ + int rc; + sqlite3_stmt *stmt; + ObjectReference ref; + int found = false; + + stmt = Prepare("select path, checksum from segments where segment = ?"); + sqlite3_bind_text(stmt, 1, segment.c_str(), segment.size(), + SQLITE_TRANSIENT); + + rc = sqlite3_step(stmt); + if (rc == SQLITE_DONE) { + } else if (rc == SQLITE_ROW) { + found = true; + const char *val; + + val = (const char *)sqlite3_column_text(stmt, 0); + if (val == NULL) + found = false; + else + *seg_path = val; + + val = (const char *)sqlite3_column_text(stmt, 1); + if (val == NULL) + found = false; + else + *seg_checksum = val; + } else { + fprintf(stderr, "Could not execute SELECT statement!\n"); + ReportError(rc); + } + + sqlite3_finalize(stmt); + + return found; +}