1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
/* $Id$
*
* This file is part of pgets.
*
* pgets is free software; you can redistribute it and/or modify it under
* the terms of the GNU General Public License as published by the Free
* Software Foundation; either version 2 of the License, or (at your
* option) any later version.
*
* pgets is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* for more details.
*
* You should have received a copy of the GNU General Public License
* along with pgets; if not, write to the Free Software Foundation,
* Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
*/
#include <time.h>
#include <assert.h>
#include <stdio.h>
#include <postgresql/libpq-fe.h>
#include "db.h"
void* db_connect(const char*t) {
PGconn *pg;
if (!t) {
fprintf(stderr, "Database specification required.\n");
return NULL;
}
if (!(pg = PQconnectdb(t)))
return NULL;
if (PQstatus(pg) != CONNECTION_OK) {
fprintf(stderr, "Could not connect to database: %s\n", PQerrorMessage(pg));
PQfinish(pg);
return NULL;
}
return pg;
}
void db_disconnect(void *db) {
assert(db);
PQfinish((PGconn*) db);
}
int db_write_entry(void *db, const struct entry *entry) {
static time_t t = 0;
char query[512];
struct tm tm;
PGresult* r;
PGconn *pg = db;
int year;
assert(pg);
if (t == 0) {
t = time(NULL);
localtime_r(&t, &tm);
}
if (entry->month < tm.tm_mon+1 || (entry->month == tm.tm_mon+1 && entry->day <= tm.tm_mday))
year = 1900+tm.tm_year;
else
year = 1900+tm.tm_year-1;
snprintf(query, sizeof(query), "INSERT INTO pgets_accounting (remote_msn, local_mm, participant, incoming, _timestamp, duration) VALUES ('%s', %i, %i, '%c', TIMESTAMP '%04i-%02i-%02i %02i:%02i:00', %i)",
entry->remote_msn, entry->local_mm, entry->participant, entry->incoming ? 't' : 'f', year, entry->month, entry->day, entry->hour, entry->minute, entry->duration);
if (!(r = PQexec(pg, query)) || PQstatus(pg) != CONNECTION_OK || PQresultStatus(r) != PGRES_COMMAND_OK) {
if (r) {
if (PQresultStatus(r) == PGRES_FATAL_ERROR) {
PQclear(r);
return 1;
}
fprintf(stderr, "Query [%s] failed (#1), reason given: %s\n", query, PQresultErrorMessage(r));
PQclear(r);
} else
fprintf(stderr, "Query [%s] failed (#2), reason given: %s\n", query, PQerrorMessage(pg));
return -1;
}
PQclear(r);
return 0;
}
|