yarrg/rssql.c

   1
   2 #include "rscommon.h"
   3
   4 sqlite3 *db;
   5 sqlite3_stmt *ss_ipair;
   6
   7 int islandtablesz;
   8
   9 DEBUG_DEFINE_DEBUGF(sql);
  10
  11 static int busy_handler(void *u, int previous) {
  12   debugf("[[DB BUSY %d]]",previous);
  13   sysassert(! usleep(5000) );
  14   return 1;
  15 }
  16
  17 void setup_sql(void) {
  18   sqlite3_stmt *sst;
  19
  20   SQL_MUST( sqlite3_open("OCEAN-Midnight.db", &db) );
  21   SQL_MUST( sqlite3_busy_handler(db, busy_handler, 0) );
  22
  23   sst= sql_prepare("BEGIN","(begin)");
  24   assert( !SQL_STEP(sst) );
  25   sqlite3_finalize(sst);
  26
  27   islandtablesz= 1 + sql_single_int("SELECT max(islandid) FROM islands");
  28   debugf("SQL islandtablesz=%d\n",islandtablesz);
  29 }
  30
  31 int sql_single_int(const char *stmt) {
  32   sqlite3_stmt *sst;
  33   sst= sql_prepare(stmt,"(single int)");
  34   assert( SQL_STEP(sst) );
  35   int rv= sqlite3_column_int(sst,0);
  36   sqlite3_finalize(sst);
  37   return rv;
  38 }
  39
  40 void sql_fatal(const char *stmt_what, int sqr, const char *act_what) {
  41   fatal("SQL call failed, stmt %s code %d: %s: %s",
  42         stmt_what, sqr, sqlite3_errmsg(db), act_what);
  43 }
  44
  45 void sql_bind(sqlite3_stmt *ss, int index, int value,
  46               const char *ss_what, const char *val_what) {
  47   debugf("SQL BIND %s #%d = %d = %s\n", ss_what, index, value, val_what);
  48   int sqr= sqlite3_bind_int(ss, index, value);
  49   if (sqr) sql_fatal(ss_what, sqr,
  50                      masprintf("bind #%d (%s)", index, val_what));
  51 }
  52
  53 sqlite3_stmt *sql_prepare(const char *stmt, const char *what) {
  54   sqlite3_stmt *ssr;
  55   debugf("SQL PREPARE %s [[\n%s\n]]\n", what, stmt);
  56   SQL_MUST( sqlite3_prepare(db, stmt, -1, &ssr, 0) );
  57   return ssr;
  58 }
  59
  60 int sql_step_wrap(sqlite3_stmt *ssh, const char *ssh_string,
  61                   const char *file, int line) {
  62   for (;;) {
  63     int sqr;
  64     sqr= sqlite3_step((ssh));
  65     switch (sqr) {
  66     case SQLITE_DONE:
  67       debugf("SQL %s DONE\n",ssh_string);
  68       return 0;
  69     case SQLITE_ROW:
  70       if (DEBUGP(sql)) {
  71         int i;
  72         fprintf(debug,"SQL %s R",ssh_string);
  73         for (i=0; i<sqlite3_column_count(ssh); i++) {
  74           fputc('\t',debug);
  75           const char *txt= (const char*)sqlite3_column_text(ssh,i);
  76           fputs(txt ? txt : "<null>", debug);
  77         }
  78         fputs("\n",debug);
  79       }
  80       return 1;
  81     default: fatal("SQL step failed at %s:%d: code %d: %s: %s",
  82                    file, line, sqr, sqlite3_errmsg(db), ssh_string);
  83     }
  84   }
  85 }