Project: DBSlayer

Revision: 60

Author: dgottfrid

Date: 19 Mar 2008 19:08:36

• /branches/refactor-08/server/dbslayer_server.c

  ......	@@ -0,0 +1,134 @@
  	1	#include "slayer_http_server.h"
  	2	#include "dbaccess.h"
  	3	#include "simplejson.h"
  	4	#include "slayer_util.h"
  	5	#include "json2xml.h"
  	6
  	7	#include <apr_portable.h>
  	8
  	9	typedef struct _dbslayer_config_t {
  	10	char *server;
  	11	char *multiserver;
  	12	char *configure;
  	13	char *username;
  	14	char *password;
  	15	} dbslayer_config_t;
  	16
  	17	void * db_global_init(apr_pool_t *pmpool, int argc, char **argv) {
  	18	int i = 0;
  	19	mysql_library_init(0,NULL,NULL); //setup init
  	20	dbslayer_config_t *config = apr_pcalloc(pmpool,sizeof(dbslayer_config_t));
  	21	//looking for -> -s server -c config -u username -x password
  	22	for( i = 0; i < argc; i++) {
  	23	if( i+1 < argc && argv[i][0] == '-' ) {
  	24	switch(argv[i][1]) {
  	25	case 's': config->server = apr_pstrdup(pmpool,argv[i+1]); break;
  	26	case 'c': config->configure = apr_pstrdup(pmpool,argv[i+1]); break;
  	27	case 'u': config->username = apr_pstrdup(pmpool,argv[i+1]); break;
  	28	case 'x': config->password = apr_pstrdup(pmpool,argv[i+1]); break;
  	29	case 'm': config->multiserver= apr_pstrdup(pmpool,argv[i+1]); break;
  	30	}
  	31	}
  	32	}
  	33	if((config->server == NULL && config->multiserver == NULL) || config->configure == NULL ) {
  	34	fprintf(stderr,"Failed to configure dbslayer - [-s server[:server] or -m server[:server] ] and -c configure are required arguments\n");
  	35	exit(0);
  	36	}
  	37	return config;
  	38	}
  	39	void db_global_destroy(void *_config) {
  	40	mysql_library_end();
  	41	}
  	42	void * db_thread_init(apr_pool_t *mpool, void *_global_config) {
  	43	dbslayer_config_t *config = (dbslayer_config_t*) _global_config;
  	44	unsigned int id = apr_os_thread_current();
  	45	return db_handle_init(config->username,config->password,config->multiserver == NULL ? config->server : config->multiserver,config->configure,&id,config->multiserver == NULL ? 0 : 1);
  	46	}
  	47	void db_thread_destroy(void *x) {
  	48	db_handle_destroy((db_handle_t*)x);
  	49	}
  	50
  	51	void *dbjson_handler(slayer_http_server_t *server, void *_global_config, slayer_http_connection_t *client, void *local_config) {
  	52	db_handle_t *dbhandle = (db_handle_t*)local_config;
  	53	char *output = "{\"ERROR\" : \"we got problems - couldn't parse your incoming json\"}";
  	54	json_value *stmt = NULL;
  	55	json_value *output_format = NULL;
  	56	char *content_type = SLAYER_MT_TEXT_PLAIN;
  	57
  	58	if(strcmp(client->request->uri.path,"/dbform") == 0) {
  	59	json_skip_head_t *form = parse_qstring(client->request->mpool,client->request->uri.query);
  	60	stmt = json_object_create(client->request->mpool);
  	61	if(form->node_count) {
  	62	json_skip_node_t *list = form->node->next_list[0];
  	63	while(list !=NULL) {
  	64	const char *k = (const char*)list->key;
  	65	const char *v = (const char*)list->data;
  	66	json_object_add(stmt,k,json_string_create(client->request->mpool,v));
  	67	list = list->next_list[0];
  	68	}
  	69	}
  	70	} else {
  	71	char *cquery = urldecode(client->request->mpool,client->request->uri.query);
  	72	stmt = decode_json(cquery,strlen(cquery),client->request->mpool);
  	73	}
  	74
  	75	if(dbhandle && stmt) {
  	76	output_format = json_skip_get(stmt->value.object,"FORMAT");
  	77	json_value *result = dbexecute(dbhandle,stmt,client->request->mpool);
  	78	json_value *errors = json_skip_get(result->value.object,"MYSQL_ERROR");
  	79	if(errors) {
  	80	char *http_request = apr_pstrcat(client->request->mpool,client->request->parse->method == HTTP_METHOD_GET ? "GET ": "POST ",client->request->parse->uri_data,client->request->parse->version == HTTP_10 ? " HTTP/1.0" : " HTTP/1.0",NULL);
  	81	slayer_server_log_err_message(server->elmanager,client->request->mpool,client->conn,http_request,apr_pstrcat(client->request->mpool,"ERROR: ",errors->value.string,NULL));
  	82	}
  	83	if(output_format && output_format->type == JSON_STRING && strcmp(output_format->value.string,"xml")==0) {
  	84	output = xml_serialize(client->request->mpool,result);
  	85	content_type = "text/xml";
  	86	} else {
  	87	output = json_serialize(client->request->mpool,result);
  	88	}
  	89	} else {
  	90	char *http_request = apr_pstrcat(client->request->mpool,client->request->parse->method == HTTP_METHOD_GET ? "GET ": "POST ",client->request->parse->uri_data,client->request->parse->version == HTTP_10 ? " HTTP/1.0" : " HTTP/1.0",NULL);
  	91	slayer_server_log_err_message(server->elmanager,client->request->mpool,client->conn,http_request,"ERROR: Couldn't parse incoming JSON");
  	92	}
  	93
  	94	client->request->response_code = 200;
  	95	slayer_http_handle_response(server, client, content_type,output,-1);
  	96	return NULL;
  	97	}
  	98
  	99	int main(int argc, char **argv) {
  	100
  	101	slayer_http_service_map_t **service_map;
  	102	char *urls[] = { "/db","/dbform",NULL};
  	103
  	104	slayer_http_service_t handler;
  	105	handler.help_string = "-s server[:server] -c config [-u username -x password]";
  	106	handler.service_global_init_func = db_global_init;
  	107	handler.service_global_destroy_func = db_global_destroy;
  	108	handler.service_thread_init_func = db_thread_init;
  	109	handler.service_thread_destroy_func = db_thread_destroy;
  	110	handler.service_handler_func = dbjson_handler;
  	111
  	112
  	113	service_map = malloc(sizeof(slayer_http_service_map_t*) * 1);
  	114	service_map[0] = malloc(sizeof(slayer_http_service_map_t));
  	115	service_map[0]->urls = urls;
  	116	service_map[0]->service = &handler;
  	117
  	118	/*
  	119	slayer_http_service_t dummy_handler;
  	120	dummy_handler.help_string = "dummy don't need to be configured";
  	121	dummy_handler.service_global_init_func = NULL;
  	122	dummy_handler.service_global_destroy_func = NULL;
  	123	dummy_handler.service_thread_init_func = NULL;
  	124	dummy_handler.service_thread_destroy_func = NULL;
  	125	dummy_handler.service_handler_func = xdummy_handler;
  	126
  	127	char *dummy_urls[] = {"/dummy",NULL};
  	128	service_map[1] = malloc(sizeof(slayer_http_service_map_t));
  	129	service_map[1]->urls = dummy_urls;
  	130	service_map[1]->service = &dummy_handler;
  	131	*/
  	132
  	133	return slayer_server_run(1,service_map,argc,argv,1024 * 1000 ,"dbslayer/beta-12");
  	134	}