summaryrefslogtreecommitdiff
path: root/pg.pgc
diff options
context:
space:
mode:
authorJon Bergli Heier <snakebite@jvnv.net>2010-11-15 20:19:06 +0100
committerJon Bergli Heier <snakebite@jvnv.net>2010-11-15 20:19:06 +0100
commit3ff195398245d187f0477ef6c83bd5ed5eb50b03 (patch)
tree8a3ce664c11df3e693dd4d0c18fcb87443490463 /pg.pgc
parent14e35e4d1f2f8927685b0c55bdc1f5d6bd5a2e9f (diff)
Added some ecpg stuff.
Diffstat (limited to 'pg.pgc')
-rw-r--r--pg.pgc114
1 files changed, 114 insertions, 0 deletions
diff --git a/pg.pgc b/pg.pgc
new file mode 100644
index 0000000..e004dc0
--- /dev/null
+++ b/pg.pgc
@@ -0,0 +1,114 @@
+EXEC SQL WHENEVER SQLERROR CONTINUE;
+
+static inline int iserror() {
+ return !(SQLSTATE[0] == '0' && SQLSTATE[1] == '0');
+}
+
+static inline void printerror() {
+ printf("%c%c%c%c%c\n", SQLSTATE[0], SQLSTATE[1], SQLSTATE[2], SQLSTATE[3], SQLSTATE[4]);
+}
+
+static inline int testerror(const char *code) {
+ for(int i = 0; i < 5; i++) {
+ if(code[i] != SQLSTATE[i])
+ return 0;
+ }
+ return 1;
+}
+
+int pg_connect(const char *string) {
+ EXEC SQL BEGIN DECLARE SECTION;
+ const char *target = string;
+ EXEC SQL END DECLARE SECTION;
+
+ EXEC SQL CONNECT TO :target;
+
+ if(iserror()) {
+ sqlprint();
+ printerror();
+ }
+
+ return !iserror();
+}
+
+void pg_disconnect() {
+ EXEC SQL DISCONNECT;
+}
+
+int pg_init() {
+ EXEC SQL WHENEVER SQLERROR GOTO pg_init_error;
+ EXEC SQL CREATE TABLE schema (version INTEGER);
+ EXEC SQL INSERT INTO schema (version) VALUES (1);
+
+ EXEC SQL CREATE TABLE channel (
+ id serial,
+ name varchar,
+ PRIMARY KEY (id)
+ );
+ EXEC SQL CREATE TABLE channel_hours (
+ channel_id integer,
+ time_index integer,
+ lines integer,
+ PRIMARY KEY (channel_id, time_index),
+ FOREIGN KEY (channel_id) REFERENCES channel (id)
+ );
+
+ /* "user" is a reserved word */
+ EXEC SQL CREATE TABLE users (
+ id serial,
+ nick varchar,
+ words integer,
+ characters integer,
+ kicks integer,
+ kicked integer,
+ monolog_lines integer,
+ monologs integer,
+ seen_first integer,
+ seen_last integer,
+ PRIMARY KEY (id)
+ );
+ EXEC SQL CREATE TABLE user_hours (
+ user_id integer,
+ time_index integer,
+ lines integer,
+ PRIMARY KEY (user_id, time_index),
+ FOREIGN KEY (user_id) REFERENCES users (id)
+ );
+
+ EXEC SQL CREATE TABLE word (
+ id serial PRIMARY KEY,
+ name varchar,
+ count integer
+ );
+
+ EXEC SQL COMMIT;
+
+ return 1;
+
+pg_init_error:
+
+ sqlprint();
+ return 0;
+}
+
+EXEC SQL WHENEVER SQLERROR CONTINUE;
+int pg_upgrade() {
+ EXEC SQL BEGIN DECLARE SECTION;
+ int version = 0;
+ EXEC SQL END DECLARE SECTION;
+
+ EXEC SQL SELECT version INTO :version FROM schema;
+ if(testerror("02000")) {
+ EXEC SQL INSERT INTO schema (version) VALUES (1);
+ } else if(iserror()) {
+ sqlprint();
+ return 0;
+ }
+
+ printf("current schema version: %d\n", version);
+
+ EXEC SQL COMMIT;
+
+ return !iserror();
+}
+/* vim: set syn=c: */