diff --git a/src/db.c b/src/db.c index 0ee076e..2456232 100644 --- a/src/db.c +++ b/src/db.c @@ -4,6 +4,19 @@ #include "include/db.h" +static int get_num_rows(sqlite3* db) +{ + int rows = 0; + + sqlite3_stmt* msg; + sqlite3_prepare(db, "SELECT * FROM books;", -1, &msg, NULL); + while (sqlite3_step(msg) == SQLITE_ROW) + rows++; + sqlite3_finalize(msg); + + return rows; +} + void setup_db(sqlite3* db) { int err;