freeswitch/src/mod/applications/mod_mongo/mod_mongo.cpp

#include <switch.h>
#include "mod_mongo.h"


static struct {
	mongo_connection_pool_t *conn_pool;
} globals;


static const char *SYNTAX = "mongo_find_one ns; query; fields";

SWITCH_STANDARD_API(mongo_find_one_function) 
{
  switch_status_t status = SWITCH_STATUS_SUCCESS;
  char *ns = NULL, *json_query = NULL, *json_fields = NULL;
  char delim = ';';

  ns = strdup(cmd);
  switch_assert(ns != NULL);

  if ((json_query = strchr(ns, delim))) {
	  *json_query++ = '\0';
	  if ((json_fields = strchr(json_query, delim))) {
		  *json_fields++ = '\0';
	  }
  }

  if (!zstr(ns) && !zstr(json_query) && !zstr(json_fields)) {

	  DBClientConnection *conn = NULL;

	  try {
		  BSONObj query = fromjson(json_query);
		  BSONObj fields = fromjson(json_fields);

		  conn = mongo_connection_pool_get(globals.conn_pool);
		  if (conn) {
			  BSONObj res = conn->findOne(ns, Query(query), &fields);
			  mongo_connection_pool_put(globals.conn_pool, conn);

			  stream->write_function(stream, "-OK\n%s\n", res.toString().c_str());
		  } else {
			  stream->write_function(stream, "-ERR\nNo connection\n");
		  }
	  } catch (DBException &e) {
		  if (conn) {
			  mongo_connection_destroy(&conn);
		  }
		  stream->write_function(stream, "-ERR\n%s\n", e.toString().c_str());
	  }


  } else {
	  stream->write_function(stream, "-ERR\n%s\n", SYNTAX);	  
  }

  switch_safe_free(ns);

  return status;
}

static switch_status_t config(void)
{
	const char *cf = "mongo.conf";
	switch_xml_t cfg, xml, settings, param;
	switch_status_t status = SWITCH_STATUS_SUCCESS;
	const char *host = "127.0.0.1";
	switch_size_t min_connections = 1, max_connections = 1;

	if (!(xml = switch_xml_open_cfg(cf, &cfg, NULL))) {
		switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Open of %s failed\n", cf);
		return SWITCH_STATUS_GENERR;
	}

	if ((settings = switch_xml_child(cfg, "settings"))) {
		for (param = switch_xml_child(settings, "param"); param; param = param->next) {
			char *var = (char *) switch_xml_attr_soft(param, "name");
			char *val = (char *) switch_xml_attr_soft(param, "value");
			int tmp;

			if (!strcmp(var, "host")) {
				host = val;
			} else if (!strcmp(var, "min-connections")) {
				if ((tmp = atoi(val)) > 0) {
					min_connections = tmp;
				}
			} else if (!strcmp(var, "max-connections")) {
				if ((tmp = atoi(val)) > 0) {
					max_connections = tmp;
				}
			}
		}
	}

	if (mongo_connection_pool_create(&globals.conn_pool, min_connections, max_connections, host) != SWITCH_STATUS_SUCCESS) {
		switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_CRIT, "Can't create connection pool\n");
		status = SWITCH_STATUS_GENERR;
	} else {
		switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "Mongo connection pool created [%s %d/%d]\n", host, (int)min_connections, (int)max_connections);
	}

	switch_xml_free(xml);

	return status;
}


SWITCH_BEGIN_EXTERN_C

SWITCH_MODULE_LOAD_FUNCTION(mod_mongo_load);
SWITCH_MODULE_SHUTDOWN_FUNCTION(mod_mongo_shutdown);
SWITCH_MODULE_DEFINITION(mod_mongo, mod_mongo_load, mod_mongo_shutdown, NULL);


SWITCH_MODULE_LOAD_FUNCTION(mod_mongo_load)
{
  switch_api_interface_t *api_interface;
  switch_application_interface_t *app_interface;

  *module_interface = switch_loadable_module_create_module_interface(pool, modname);

  memset(&globals, 0, sizeof(globals));

  if (config() != SWITCH_STATUS_SUCCESS) {
	  return SWITCH_STATUS_TERM;
  }
  
  SWITCH_ADD_API(api_interface, "mongo_find_one", "mongo", mongo_find_one_function, SYNTAX);

  return SWITCH_STATUS_SUCCESS;
}

SWITCH_MODULE_SHUTDOWN_FUNCTION(mod_mongo_shutdown)
{
	mongo_connection_pool_destroy(&globals.conn_pool);
	return SWITCH_STATUS_SUCCESS;
}

SWITCH_END_EXTERN_C

/* For Emacs:
 * Local Variables:
 * mode:c
 * indent-tabs-mode:t
 * tab-width:4
 * c-basic-offset:4
 * End:
 * For VIM:
 * vim:set softtabstop=4 shiftwidth=4 tabstop=4
 */