summaryrefslogtreecommitdiffstatsabout
path: root/utility/database.c
blob: c4e4bc9941267fc4eefe9d8ca51ccb3110411718 (plain)
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
#include "database.h"
#include "apu.h"
#include "apr_dbd.h"
#include "apr_strings.h"

#include "util.h"
#include "mysql/mysql.h"

struct config_dbd_t {
    const apr_dbd_driver_t *driver;
    apr_dbd_t *dbd;
    apr_dbd_prepared_t *stmt;
    const char **args;
};

void database_init(apr_pool_t *p)
{
    apr_dbd_init(p);
}

apr_status_t database_connect(config_t *cfg)
{
    apr_status_t rv;
    if (!cfg->dbdriver || !cfg->dbparams)
        return APR_EINVAL;
    if (!cfg->dbconn) {
        cfg->dbconn = apr_pcalloc(cfg->pool, sizeof(config_dbd_t));
    }
    rv = apr_dbd_get_driver(cfg->pool, cfg->dbdriver, &(cfg->dbconn->driver));
    if (rv) {

        logging_log(cfg, LOGLEVEL_ERROR,
                "Could not load database driver %s. Error %s", cfg->dbdriver,
                logging_strerror(rv));
        return rv;
    }

    rv = apr_dbd_open(cfg->dbconn->driver, cfg->pool, cfg->dbparams,
            &(cfg->dbconn->dbd));
    if (rv) {
        logging_log(cfg, LOGLEVEL_ERROR,
                "Could not connect to database. Error %s", logging_strerror(rv));
        return rv;
    }

    return APR_SUCCESS;
}

apr_status_t database_disconnect(config_t *cfg)
{
    return apr_dbd_close(cfg->dbconn->driver, cfg->dbconn->dbd);
}

static apr_dbd_prepared_t *database_prepare_insert(config_t *cfg, apr_pool_t *p)
{
    apr_status_t rv;
    char *sql;
    int i, f;
    struct iovec *vec;
    apr_dbd_prepared_t *stmt = NULL;
    int nfs = cfg->output_fields->nelts;
    config_output_field_t *ofields;

    ofields = (config_output_field_t *)cfg->output_fields->elts;

    vec = apr_palloc(p, (nfs*2 + 5) * sizeof(struct iovec));
    sql = apr_palloc(p, (nfs*3));

    vec[0].iov_base = "INSERT INTO ";
    vec[0].iov_len = 12;
    vec[1].iov_base = (void *)cfg->table;
    vec[1].iov_len = strlen(cfg->table);
    vec[2].iov_base = " (";
    vec[2].iov_len = 2;
    for (i=3, f=0; f<nfs; f++, i+=2) {
        vec[i].iov_base = (void *)ofields[f].field;
        vec[i].iov_len = strlen(vec[i].iov_base);
        vec[i+1].iov_base = ",";
        vec[i+1].iov_len = 1;
        memcpy(&sql[f*3], "%s,", 3);
    }
    sql[nfs*3-1] = '\0';
    vec[i-1].iov_base = ") VALUES (";
    vec[i-1].iov_len = 10;
    vec[i].iov_base = sql;
    vec[i].iov_len = nfs*3-1;
    vec[i+1].iov_base = ")";
    vec[i+1].iov_len = 1;

    sql = apr_pstrcatv(p, vec, i+2, NULL);

    logging_log(cfg, LOGLEVEL_DEBUG, "Generated SQL: %s", sql);

    rv = apr_dbd_prepare(cfg->dbconn->driver, cfg->pool, cfg->dbconn->dbd, sql,
            "INSERT", &stmt);

    if (rv) {
        logging_log(cfg, LOGLEVEL_ERROR, "Unable to Prepare SQL insert: %s",
                apr_dbd_error(cfg->dbconn->driver, cfg->dbconn->dbd, rv));
        return NULL;
    }
    return stmt;
}
apr_status_t database_insert(config_t *cfg, apr_pool_t *p, apr_table_t *data)
{
    apr_status_t rv;
    int f, nfs;
    config_output_field_t *ofields;
    ofields = (config_output_field_t *)cfg->output_fields->elts;
    nfs = cfg->output_fields->nelts;
    // Prepare statement
    if (!cfg->dbconn->stmt) {
        cfg->dbconn->stmt = database_prepare_insert(cfg, p);
        if (!cfg->dbconn->stmt) {
            logging_log(cfg, LOGLEVEL_NOISE, "Unable to prepare SQL statement");
            return APR_EINVAL;
        }
        cfg->dbconn->args = apr_palloc(cfg->pool, nfs * sizeof(char *));
    }
    for (f=0; f<nfs; f++) {
        cfg->dbconn->args[f] = apr_table_get(data, ofields[f].field);
    }
    rv = apr_dbd_pquery(cfg->dbconn->driver, p, cfg->dbconn->dbd, &f,
            cfg->dbconn->stmt, nfs, cfg->dbconn->args);
    if (rv) {
        logging_log(cfg, LOGLEVEL_ERROR, "Unable to Insert SQL: %s",
                apr_dbd_error(cfg->dbconn->driver, cfg->dbconn->dbd, rv));
        return rv;
    }
    return APR_SUCCESS;
}

/** @todo implement transactions */