/* Remove DLR */
static void dlr_mongodb_remove(const Octstr *smsc, const Octstr *ts, const Octstr *dst)
{
DBPoolConn *pconn;
bson cond;
bson_buffer cond_buf;
mongo_connection *conn = NULL;
pconn = dbpool_conn_consume(pool);
if (pconn == NULL) {
return;
}
conn = (mongo_connection*)pconn->conn;
bson_buffer_init(&cond_buf);
bson_append_string(&cond_buf, octstr_get_cstr(fields->field_smsc), octstr_get_cstr(smsc));
bson_append_string(&cond_buf, octstr_get_cstr(fields->field_ts), octstr_get_cstr(ts));
if (dst) {
bson_append_string(&cond_buf, octstr_get_cstr(fields->field_dst), octstr_get_cstr(dst));
}
bson_from_buffer(&cond, &cond_buf);
MONGO_TRY {
mongo_remove(conn, mongodb_namespace, &cond);
} MONGO_CATCH {
mongodb_error("dlr_mongodb_remove", conn->exception.type);
}
dbpool_conn_produce(pconn);
bson_destroy(&cond);
}
/* Create index on smsc and ts fields, as these are used for retrieving the DLR */
static void dlr_mongodb_ensure_index(void)
{
DBPoolConn *pconn;
mongo_connection *conn = NULL;
bson_buffer bb;
bson key;
pconn = dbpool_conn_consume(pool);
if (pconn == NULL) {
return;
}
conn = (mongo_connection*)pconn->conn;
bson_buffer_init(&bb);
bson_append_int(&bb, octstr_get_cstr(fields->field_smsc), 1);
bson_append_int(&bb, octstr_get_cstr(fields->field_ts), 1);
bson_from_buffer(&key, &bb);
MONGO_TRY {
mongo_create_index(conn, mongodb_namespace, &key, 0, NULL);
} MONGO_CATCH {
mongodb_error("dlr_mongodb_ensure_index", conn->exception.type);
}
dbpool_conn_produce(pconn);
bson_destroy(&key);
}
/* Remove all DLRs from our namespace */
static void dlr_mongodb_flush(void)
{
DBPoolConn *pconn;
bson b;
mongo_connection *conn = NULL;
pconn = dbpool_conn_consume(pool);
if (pconn == NULL) {
return;
}
conn = (mongo_connection*)pconn->conn;
MONGO_TRY {
mongo_remove(conn, mongodb_namespace, bson_empty(&b));
} MONGO_CATCH {
mongodb_error("dlr_mongodb_flush", conn->exception.type);
}
dbpool_conn_produce(pconn);
}
/* Add a new DLR entry to MongoDB */
static void dlr_mongodb_add(struct dlr_entry *entry)
{
DBPoolConn *pconn;
bson b;
bson_buffer buf;
mongo_connection *conn = NULL;
pconn = dbpool_conn_consume(pool);
if (pconn == NULL) {
dlr_entry_destroy(entry);
return;
}
conn = (mongo_connection*)pconn->conn;
bson_buffer_init(&buf);
bson_append_new_oid(&buf, "_id");
bson_append_string(&buf, octstr_get_cstr(fields->field_smsc), octstr_get_cstr(entry->smsc));
bson_append_string(&buf, octstr_get_cstr(fields->field_ts), octstr_get_cstr(entry->timestamp));
bson_append_string(&buf, octstr_get_cstr(fields->field_src), octstr_get_cstr(entry->source));
bson_append_string(&buf, octstr_get_cstr(fields->field_dst), octstr_get_cstr(entry->destination));
bson_append_string(&buf, octstr_get_cstr(fields->field_serv), octstr_get_cstr(entry->service));
bson_append_string(&buf, octstr_get_cstr(fields->field_url), octstr_get_cstr(entry->url));
bson_append_string(&buf, octstr_get_cstr(fields->field_account), octstr_get_cstr(entry->account));
bson_append_string(&buf, octstr_get_cstr(fields->field_binfo), octstr_get_cstr(entry->binfo));
bson_append_int(&buf, octstr_get_cstr(fields->field_mask), entry->mask);
bson_append_string(&buf, octstr_get_cstr(fields->field_boxc), octstr_get_cstr(entry->boxc_id));
bson_append_int(&buf, octstr_get_cstr(fields->field_status), 0);
bson_from_buffer(&b, &buf);
/* TODO: namespace support */
MONGO_TRY {
mongo_insert(conn, mongodb_namespace, &b);
} MONGO_CATCH {
mongodb_error("dlr_mongodb_insert", conn->exception.type);
}
dbpool_conn_produce(pconn);
bson_destroy(&b);
dlr_entry_destroy(entry);
}
/* Update DLR */
static void dlr_mongodb_update(const Octstr *smsc, const Octstr *ts, const Octstr *dst, int status)
{
DBPoolConn *pconn;
bson cond, op;
bson_buffer cond_buf, op_buf;
mongo_connection *conn = NULL;
pconn = dbpool_conn_consume(pool);
if (pconn == NULL) {
return;
}
conn = (mongo_connection*)pconn->conn;
bson_buffer_init(&cond_buf);
bson_append_string(&cond_buf, octstr_get_cstr(fields->field_smsc), octstr_get_cstr(smsc));
bson_append_string(&cond_buf, octstr_get_cstr(fields->field_ts), octstr_get_cstr(ts));
if (dst) {
bson_append_string(&cond_buf, octstr_get_cstr(fields->field_dst), octstr_get_cstr(dst));
}
bson_from_buffer(&cond, &cond_buf);
bson_buffer_init(&op_buf);
{
bson_buffer *sub = bson_append_start_object(&op_buf, "$set");
bson_append_int(sub, octstr_get_cstr(fields->field_status), status);
bson_append_finish_object(sub);
}
bson_from_buffer(&op, &op_buf);
MONGO_TRY {
mongo_update(conn, mongodb_namespace, &cond, &op, 0);
} MONGO_CATCH {
mongodb_error("dlr_mongodb_update", conn->exception.type);
}
dbpool_conn_produce(pconn);
bson_destroy(&cond);
bson_destroy(&op);
}
/* Number of DLRs in our namespace */
static long dlr_mongodb_messages(void)
{
DBPoolConn *pconn;
long count = 0;
mongo_connection *conn = NULL;
pconn = dbpool_conn_consume(pool);
if (pconn == NULL) {
return -1;
}
conn = (mongo_connection*)pconn->conn;
/* TODO: namespace support */
MONGO_TRY {
count = mongo_count(conn, mongodb_database, mongodb_table, NULL);
} MONGO_CATCH {
mongodb_error("dlr_mongodb_messages", conn->exception.type);
}
dbpool_conn_produce(pconn);
return count;
}
static struct dlr_entry* dlr_mongodb_get(const Octstr *smsc, const Octstr *ts, const Octstr *dst)
{
DBPoolConn *pconn;
bson cond, obj;
bson_buffer cond_buf;
bson_iterator it;
struct dlr_entry *res = NULL;
bson_bool_t found = 0;
mongo_connection *conn = NULL;
pconn = dbpool_conn_consume(pool);
if (pconn == NULL) {
return NULL;
}
conn = (mongo_connection*)pconn->conn;
bson_buffer_init(&cond_buf);
bson_append_string(&cond_buf, octstr_get_cstr(fields->field_smsc), octstr_get_cstr(smsc));
bson_append_string(&cond_buf, octstr_get_cstr(fields->field_ts), octstr_get_cstr(ts));
if (dst) {
bson_append_string(&cond_buf, octstr_get_cstr(fields->field_dst), octstr_get_cstr(dst));
}
bson_from_buffer(&cond, &cond_buf);
memset(&obj, 0, sizeof(bson));
MONGO_TRY {
found = mongo_find_one(conn, mongodb_namespace, &cond, NULL, &obj);
} MONGO_CATCH {
mongodb_error("dlr_mongodb_get", conn->exception.type);
found = 0;
}
if (found) {
res = dlr_entry_create();
gw_assert(res != NULL);
bson_find(&it, &obj, octstr_get_cstr(fields->field_mask));
res->mask = bson_iterator_int(&it);
bson_find(&it, &obj, octstr_get_cstr(fields->field_serv));
res->service = octstr_create(bson_iterator_string(&it));
bson_find(&it, &obj, octstr_get_cstr(fields->field_url));
res->url = octstr_create(bson_iterator_string(&it));
bson_find(&it, &obj, octstr_get_cstr(fields->field_src));
res->source = octstr_create(bson_iterator_string(&it));
bson_find(&it, &obj, octstr_get_cstr(fields->field_dst));
res->destination = octstr_create(bson_iterator_string(&it));
bson_find(&it, &obj, octstr_get_cstr(fields->field_boxc));
res->boxc_id = octstr_create(bson_iterator_string(&it));
bson_find(&it, &obj, octstr_get_cstr(fields->field_smsc));
res->smsc = octstr_create(bson_iterator_string(&it));
}
dbpool_conn_produce(pconn);
bson_destroy(&cond);
bson_destroy(&obj);
return res;
}
