chiark / gitweb /
Fix spelling error in intro
[ypp-sc-tools.db-live.git] / yarrg / rssql.c
1 /*
2  * Route searcher - database helper functions
3  */
4 /*
5  *  This is part of the YARRG website, a tool for assisting
6  *  players of Yohoho Puzzle Pirates.
7  * 
8  *  Copyright (C) 2009 Ian Jackson <ijackson@chiark.greenend.org.uk>
9  *
10  *  This program is free software: you can redistribute it and/or modify
11  *  it under the terms of the GNU Affero General Public License as
12  *  published by the Free Software Foundation, either version 3 of the
13  *  License, or (at your option) any later version.
14  *  
15  *  This program is distributed in the hope that it will be useful,
16  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
17  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  *  GNU Affero General Public License for more details.
19  *  
20  *  You should have received a copy of the GNU Affero General Public License
21  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
22  *  
23  *  Yohoho and Puzzle Pirates are probably trademarks of Three Rings and
24  *  are used without permission.  This program is not endorsed or
25  *  sponsored by Three Rings.
26  */
27
28 #include "rscommon.h"
29
30 sqlite3 *db;
31 sqlite3_stmt *ss_ipair;
32
33 int islandtablesz;
34
35 DEBUG_DEFINE_DEBUGF(sql);
36 DEBUG_DEFINE_SOME_DEBUGF(sql,debug2f);
37
38 static int busy_handler(void *u, int previous) {
39   debugf("[[DB BUSY %d]]",previous);
40   sysassert(! usleep(5000) );
41   return 1;
42 }
43
44 void setup_sql(const char *database) {
45   sqlite3_stmt *sst;
46   
47   SQL_MUST( sqlite3_open(database, &db) );
48   SQL_MUST( sqlite3_busy_handler(db, busy_handler, 0) );
49
50   sst= sql_prepare("BEGIN","(begin)");
51   assert( !SQL_STEP(sst) );
52   sqlite3_finalize(sst);
53
54   islandtablesz= 1 + sql_single_int("SELECT max(islandid) FROM islands");
55   debugf("SQL islandtablesz=%d\n",islandtablesz);
56 }
57
58 int sql_single_int(const char *stmt) {
59   sqlite3_stmt *sst;
60   sst= sql_prepare(stmt,"(single int)");
61   assert( SQL_STEP(sst) );
62   int rv= sqlite3_column_int(sst,0);
63   sqlite3_finalize(sst);
64   return rv;
65 }
66
67 void sql_fatal(const char *stmt_what, int sqr, const char *act_what) {
68   fatal("SQL call failed, stmt %s code %d: %s: %s",
69         stmt_what, sqr, sqlite3_errmsg(db), act_what);
70 }
71
72 void sql_bind(sqlite3_stmt *ss, int index, int value,
73               const char *ss_what, const char *val_what) {
74   debug2f("SQL BIND %s #%d = %d = %s\n", ss_what, index, value, val_what);
75   int sqr= sqlite3_bind_int(ss, index, value);
76   if (sqr) sql_fatal(ss_what, sqr,
77                      masprintf("bind #%d (%s)", index, val_what));
78 }
79   
80 sqlite3_stmt *sql_prepare(const char *stmt, const char *what) {
81   sqlite3_stmt *ssr;
82   debugf("SQL PREPARE %s [[\n%s\n]]\n", what, stmt);
83   SQL_MUST( sqlite3_prepare(db, stmt, -1, &ssr, 0) );
84   return ssr;
85 }
86
87 int sql_step_distinct(sqlite3_stmt *ssh, const char *ssh_string,
88                       const char *file, int line,
89                       int *cols, int ncols, int nkeycols) {
90   for (;;) {
91     if (!sql_step(ssh, ssh_string, file, line)) return 0;
92
93     int i;
94     for (i=0; i<ncols; i++) {
95       int v= sqlite3_column_int(ssh, i);
96       if (v == cols[i]) continue;
97       
98       assert(i<nkeycols);
99       cols[i++]= v;
100       for ( ; i<ncols; i++)
101         cols[i]= sqlite3_column_int(ssh, i);
102       return 1;
103     }
104   }
105 }
106
107 int sql_step(sqlite3_stmt *ssh, const char *ssh_string,
108              const char *file, int line) {
109   for (;;) {
110     int sqr;
111     sqr= sqlite3_step((ssh));
112     switch (sqr) {
113     case SQLITE_DONE:
114       debug2f("SQL %s DONE\n",ssh_string);
115       return 0;
116     case SQLITE_ROW:
117       if (DEBUGP(sql2)) {
118         int i;
119         fprintf(debug,"SQL %s R",ssh_string);
120         for (i=0; i<sqlite3_column_count(ssh); i++) {
121           fputc('\t',debug);
122           const char *txt= (const char*)sqlite3_column_text(ssh,i);
123           fputs(txt ? txt : "<null>", debug);
124         }
125         fputs("\n",debug);
126       }
127       return 1;
128     default: fatal("SQL step failed at %s:%d: code %d: %s: %s",
129                    file, line, sqr, sqlite3_errmsg(db), ssh_string);
130     }
131   }
132 }