diff --git a/src/utils/dbi.c b/src/utils/dbi.c
index 3aef35e2213514839e9b12eb3d0625270e49b008..ee2280ce939e6bb9f61cdde27ba3cfa5415d5028 100644 (file)
--- a/src/utils/dbi.c
+++ b/src/utils/dbi.c
/*
/*
- * syscollector - src/utils/dbi.c
+ * SysDB - src/utils/dbi.c
* Copyright (C) 2012 Sebastian 'tokkee' Harl <sh@tokkee.org>
* All rights reserved.
*
* Copyright (C) 2012 Sebastian 'tokkee' Harl <sh@tokkee.org>
* All rights reserved.
*
typedef struct {
char *key;
char *value;
typedef struct {
char *key;
char *value;
-} sc_dbi_option_t;
+} sdb_dbi_option_t;
-struct sc_dbi_options {
- sc_dbi_option_t *options;
+struct sdb_dbi_options {
+ sdb_dbi_option_t *options;
size_t options_num;
};
size_t options_num;
};
-struct sc_dbi_client {
+struct sdb_dbi_client {
char *driver;
char *database;
dbi_conn conn;
char *driver;
char *database;
dbi_conn conn;
- sc_dbi_options_t *options;
+ sdb_dbi_options_t *options;
};
/*
};
/*
*/
static const char *
*/
static const char *
-sc_dbi_strerror(dbi_conn conn)
+sdb_dbi_strerror(dbi_conn conn)
{
const char *errmsg = NULL;
dbi_conn_error(conn, &errmsg);
return errmsg;
{
const char *errmsg = NULL;
dbi_conn_error(conn, &errmsg);
return errmsg;
-} /* sc_dbi_strerror */
+} /* sdb_dbi_strerror */
static int
static int
-sc_dbi_get_field(dbi_result res, unsigned int i,
- int type, sc_dbi_data_t *data)
+sdb_dbi_get_field(dbi_result res, unsigned int i,
+ int type, sdb_data_t *data)
{
switch (type) {
{
switch (type) {
- case DBI_TYPE_INTEGER:
+ case SDB_TYPE_INTEGER:
data->data.integer = dbi_result_get_longlong_idx(res, i);
break;
data->data.integer = dbi_result_get_longlong_idx(res, i);
break;
- case DBI_TYPE_DECIMAL:
+ case SDB_TYPE_DECIMAL:
data->data.decimal = dbi_result_get_double_idx(res, i);
break;
data->data.decimal = dbi_result_get_double_idx(res, i);
break;
- case DBI_TYPE_STRING:
+ case SDB_TYPE_STRING:
data->data.string = dbi_result_get_string_idx(res, i);
break;
data->data.string = dbi_result_get_string_idx(res, i);
break;
- case DBI_TYPE_DATETIME:
+ case SDB_TYPE_DATETIME:
{
/* libdbi does not provide any higher resolutions than that */
time_t datetime = dbi_result_get_datetime_idx(res, i);
{
/* libdbi does not provide any higher resolutions than that */
time_t datetime = dbi_result_get_datetime_idx(res, i);
- data->data.datetime = SECS_TO_SC_TIME(datetime);
+ data->data.datetime = SECS_TO_SDB_TIME(datetime);
}
break;
}
break;
- case DBI_TYPE_BINARY:
+ case SDB_TYPE_BINARY:
{
size_t length = dbi_result_get_field_length_idx(res, i);
const unsigned char *datum = dbi_result_get_binary_idx(res, i);
{
size_t length = dbi_result_get_field_length_idx(res, i);
const unsigned char *datum = dbi_result_get_binary_idx(res, i);
data->type = type;
return 0;
data->type = type;
return 0;
-} /* sc_dbi_get_field */
+} /* sdb_dbi_get_field */
static int
static int
-sc_dbi_get_data(sc_dbi_client_t *client, dbi_result res,
- unsigned int num_fields, sc_dbi_data_cb callback)
+sdb_dbi_get_data(sdb_dbi_client_t *client, dbi_result res,
+ unsigned int num_fields, sdb_dbi_data_cb callback,
+ sdb_object_t *user_data)
{
{
- sc_dbi_data_t data[num_fields];
+ sdb_data_t data[num_fields];
int types[num_fields];
unsigned int i;
int types[num_fields];
unsigned int i;
types[i] = dbi_result_get_field_type_idx(res, i + 1);
if (types[i] == DBI_TYPE_ERROR) {
fprintf(stderr, "dbi: failed to fetch data: %s\n",
types[i] = dbi_result_get_field_type_idx(res, i + 1);
if (types[i] == DBI_TYPE_ERROR) {
fprintf(stderr, "dbi: failed to fetch data: %s\n",
- sc_dbi_strerror(client->conn));
+ sdb_dbi_strerror(client->conn));
return -1;
}
return -1;
}
+ types[i] = DBI_TYPE_TO_SC(types[i]);
}
num_rows = dbi_result_get_numrows(res);
}
num_rows = dbi_result_get_numrows(res);
for (n = 0; n < num_rows; ++n) {
if (! dbi_result_seek_row(res, n + 1)) {
fprintf(stderr, "dbi: Failed to retrieve row %llu: %s\n",
for (n = 0; n < num_rows; ++n) {
if (! dbi_result_seek_row(res, n + 1)) {
fprintf(stderr, "dbi: Failed to retrieve row %llu: %s\n",
- n, sc_dbi_strerror(client->conn));
+ n, sdb_dbi_strerror(client->conn));
continue;
}
for (i = 0; i < num_fields; ++i)
continue;
}
for (i = 0; i < num_fields; ++i)
- if (sc_dbi_get_field(res, (unsigned int)(i + 1),
+ if (sdb_dbi_get_field(res, (unsigned int)(i + 1),
types[i], &data[i]))
continue;
types[i], &data[i]))
continue;
- if (callback(client, num_fields, data))
+ if (callback(client, num_fields, data, user_data))
continue;
++success;
continue;
++success;
if (! success)
return -1;
return 0;
if (! success)
return -1;
return 0;
-} /* sc_dbi_get_data */
+} /* sdb_dbi_get_data */
/*
* public API
*/
/*
* public API
*/
-sc_dbi_options_t *
-sc_dbi_options_create(void)
+sdb_dbi_options_t *
+sdb_dbi_options_create(void)
{
{
- sc_dbi_options_t *options;
+ sdb_dbi_options_t *options;
options = malloc(sizeof(options));
if (! options)
options = malloc(sizeof(options));
if (! options)
options->options = NULL;
options->options_num = 0;
return options;
options->options = NULL;
options->options_num = 0;
return options;
-} /* sc_dbi_options_create */
+} /* sdb_dbi_options_create */
int
int
-sc_dbi_options_add(sc_dbi_options_t *options,
+sdb_dbi_options_add(sdb_dbi_options_t *options,
const char *key, const char *value)
{
const char *key, const char *value)
{
- sc_dbi_option_t *new;
+ sdb_dbi_option_t *new;
if ((! options) || (! key) || (! value))
return -1;
if ((! options) || (! key) || (! value))
return -1;
++options->options_num;
return 0;
++options->options_num;
return 0;
-} /* sc_dbi_options_add */
+} /* sdb_dbi_options_add */
void
void
-sc_dbi_options_destroy(sc_dbi_options_t *options)
+sdb_dbi_options_destroy(sdb_dbi_options_t *options)
{
size_t i;
{
size_t i;
return;
for (i = 0; i < options->options_num; ++i) {
return;
for (i = 0; i < options->options_num; ++i) {
- sc_dbi_option_t *opt = options->options + i;
+ sdb_dbi_option_t *opt = options->options + i;
if (opt->key)
free(opt->key);
if (opt->key)
free(opt->key);
options->options = NULL;
options->options_num = 0;
free(options);
options->options = NULL;
options->options_num = 0;
free(options);
-} /* sc_dbi_options_destroy */
+} /* sdb_dbi_options_destroy */
-sc_dbi_client_t *
-sc_dbi_client_create(const char *driver, const char *database)
+sdb_dbi_client_t *
+sdb_dbi_client_create(const char *driver, const char *database)
{
{
- sc_dbi_client_t *client;
+ sdb_dbi_client_t *client;
if ((! driver) || (! database))
return NULL;
if ((! driver) || (! database))
return NULL;
client->driver = strdup(driver);
client->database = strdup(database);
if ((! client->driver) || (! client->database)) {
client->driver = strdup(driver);
client->database = strdup(database);
if ((! client->driver) || (! client->database)) {
- sc_dbi_client_destroy(client);
+ sdb_dbi_client_destroy(client);
return NULL;
}
return client;
return NULL;
}
return client;
-} /* sc_dbi_client_create */
+} /* sdb_dbi_client_create */
int
int
-sc_dbi_client_set_options(sc_dbi_client_t *client,
- sc_dbi_options_t *options)
+sdb_dbi_client_set_options(sdb_dbi_client_t *client,
+ sdb_dbi_options_t *options)
{
if (! client)
return -1;
if (client->options)
{
if (! client)
return -1;
if (client->options)
- sc_dbi_options_destroy(client->options);
+ sdb_dbi_options_destroy(client->options);
client->options = options;
return 0;
client->options = options;
return 0;
-} /* sc_dbi_client_set_options */
+} /* sdb_dbi_client_set_options */
int
int
-sc_dbi_client_connect(sc_dbi_client_t *client)
+sdb_dbi_client_connect(sdb_dbi_client_t *client)
{
dbi_driver driver;
size_t i;
{
dbi_driver driver;
size_t i;
fprintf(stderr, "dbi: failed to set option '%s': %s\n",
client->options->options[i].key,
fprintf(stderr, "dbi: failed to set option '%s': %s\n",
client->options->options[i].key,
- sc_dbi_strerror(client->conn));
+ sdb_dbi_strerror(client->conn));
fprintf(stderr, "dbi: known driver options:\n");
for (opt = dbi_conn_get_option_list(client->conn, NULL); opt;
fprintf(stderr, "dbi: known driver options:\n");
for (opt = dbi_conn_get_option_list(client->conn, NULL); opt;
if (dbi_conn_set_option(client->conn, "dbname", client->database)) {
fprintf(stderr, "dbi: failed to set option 'dbname': %s\n",
if (dbi_conn_set_option(client->conn, "dbname", client->database)) {
fprintf(stderr, "dbi: failed to set option 'dbname': %s\n",
- sc_dbi_strerror(client->conn));
+ sdb_dbi_strerror(client->conn));
dbi_conn_close(client->conn);
return -1;
}
if (dbi_conn_connect(client->conn) < 0) {
fprintf(stderr, "dbi: failed to connect to database '%s': %s\n",
dbi_conn_close(client->conn);
return -1;
}
if (dbi_conn_connect(client->conn) < 0) {
fprintf(stderr, "dbi: failed to connect to database '%s': %s\n",
- client->database, sc_dbi_strerror(client->conn));
+ client->database, sdb_dbi_strerror(client->conn));
dbi_conn_close(client->conn);
return -1;
}
return 0;
dbi_conn_close(client->conn);
return -1;
}
return 0;
-} /* sc_dbi_client_connect */
+} /* sdb_dbi_client_connect */
int
int
-sc_dbi_exec_query(sc_dbi_client_t *client, const char *query,
- sc_dbi_data_cb callback, int n, ...)
+sdb_dbi_exec_query(sdb_dbi_client_t *client, const char *query,
+ sdb_dbi_data_cb callback, sdb_object_t *user_data, int n, ...)
{
dbi_result res;
unsigned int num_fields;
{
dbi_result res;
unsigned int num_fields;
res = dbi_conn_query(client->conn, query);
if (! res) {
fprintf(stderr, "dbi: failed to execute query '%s': %s\n",
res = dbi_conn_query(client->conn, query);
if (! res) {
fprintf(stderr, "dbi: failed to execute query '%s': %s\n",
- query, sc_dbi_strerror(client->conn));
+ query, sdb_dbi_strerror(client->conn));
return -1;
}
if (dbi_result_get_numrows(res) == DBI_ROW_ERROR) {
fprintf(stderr, "dbi: failed to fetch rows for query '%s': %s\n",
return -1;
}
if (dbi_result_get_numrows(res) == DBI_ROW_ERROR) {
fprintf(stderr, "dbi: failed to fetch rows for query '%s': %s\n",
- query, sc_dbi_strerror(client->conn));
+ query, sdb_dbi_strerror(client->conn));
dbi_result_free(res);
return -1;
}
dbi_result_free(res);
return -1;
}
unsigned int type = va_arg(types, unsigned int);
unsigned int type = va_arg(types, unsigned int);
+ field_type = DBI_TYPE_TO_SC(field_type);
+
/* column count starts at 1 */
if ((unsigned int)field_type != type) {
fprintf(stderr, "dbi: type of column '%s' (%u) does not match "
/* column count starts at 1 */
if ((unsigned int)field_type != type) {
fprintf(stderr, "dbi: type of column '%s' (%u) does not match "
return 0;
}
return 0;
}
- status = sc_dbi_get_data(client, res, num_fields, callback);
+ status = sdb_dbi_get_data(client, res, num_fields, callback, user_data);
dbi_result_free(res);
return status;
dbi_result_free(res);
return status;
-} /* sc_dbi_exec_query */
+} /* sdb_dbi_exec_query */
void
void
-sc_dbi_client_destroy(sc_dbi_client_t *client)
+sdb_dbi_client_destroy(sdb_dbi_client_t *client)
{
if (! client)
return;
{
if (! client)
return;
dbi_conn_close(client->conn);
if (client->options)
dbi_conn_close(client->conn);
if (client->options)
- sc_dbi_options_destroy(client->options);
+ sdb_dbi_options_destroy(client->options);
client->options = NULL;
free(client);
client->options = NULL;
free(client);
-} /* sc_dbi_client_destroy */
+} /* sdb_dbi_client_destroy */
/* vim: set tw=78 sw=4 ts=4 noexpandtab : */
/* vim: set tw=78 sw=4 ts=4 noexpandtab : */