drizzle-nginx-module-master.zip_drizzle_keepalive_nginx_nginx my

/* Copyright (C) agentzh */ #ifndef DDEBUG #define DDEBUG 0 #endif #include "ddebug.h" #include "ngx_http_drizzle_module.h" #include "ngx_http_drizzle_upstream.h" #include "ngx_http_drizzle_keepalive.h" #include "ngx_http_drizzle_processor.h" #include "ngx_http_drizzle_util.h" enum { ngx_http_drizzle_default_port = 3306 }; static void ngx_http_upstream_drizzle_cleanup(void *data); static ngx_int_t ngx_http_upstream_drizzle_init(ngx_conf_t *cf, ngx_http_upstream_srv_conf_t *uscf); static ngx_int_t ngx_http_upstream_drizzle_init_peer(ngx_http_request_t *r, ngx_http_upstream_srv_conf_t *uscf); static ngx_int_t ngx_http_upstream_drizzle_get_peer(ngx_peer_connection_t *pc, void *data); static void ngx_http_upstream_drizzle_free_peer(ngx_peer_connection_t *pc, void *data, ngx_uint_t state); /* just a work-around to override the default u->output_filter */ static ngx_int_t ngx_http_drizzle_output_filter(void *data, ngx_chain_t *in); void * ngx_http_upstream_drizzle_create_srv_conf(ngx_conf_t *cf) { ngx_pool_cleanup_t *cln; ngx_http_upstream_drizzle_srv_conf_t *conf; dd("drizzle create srv conf"); conf = ngx_pcalloc(cf->pool, sizeof(ngx_http_upstream_drizzle_srv_conf_t)); if (conf == NULL) { return NULL; } /* set by ngx_pcalloc: * conf->peers = NULL * conf->current = 0 * conf->servers = NULL * conf->single = 0 * conf->max_cached = 0 * conf->overflow = 0 (drizzle_keepalive_overflow_ignore) */ conf->pool = cf->pool; cln = ngx_pool_cleanup_add(cf->pool, 0); (void) drizzle_create(&conf->drizzle); cln->handler = ngx_http_upstream_drizzle_cleanup; cln->data = &conf->drizzle; drizzle_add_options(&conf->drizzle, DRIZZLE_NON_BLOCKING); /* we use 0 timeout for the underlying poll event model * used by libdrizzle itself. */ drizzle_set_timeout(&conf->drizzle, 0); return conf; } /* mostly based on ngx_http_upstream_server in * ngx_http_upstream.c of nginx 0.8.30. * Copyright (C) Igor Sysoev */ char * ngx_http_upstream_drizzle_server(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) { ngx_http_upstream_drizzle_srv_conf_t *dscf = conf; ngx_http_upstream_drizzle_server_t *ds; ngx_str_t *value; ngx_url_t u; ngx_uint_t i, j; ngx_http_upstream_srv_conf_t *uscf; ngx_str_t protocol; ngx_str_t charset; u_char *p; size_t len; dd("entered drizzle_server directive handler..."); uscf = ngx_http_conf_get_module_srv_conf(cf, ngx_http_upstream_module); if (dscf->servers == NULL) { dscf->servers = ngx_array_create(cf->pool, 4, sizeof(ngx_http_upstream_drizzle_server_t)); if (dscf->servers == NULL) { return NGX_CONF_ERROR; } uscf->servers = dscf->servers; } ds = ngx_array_push(dscf->servers); if (ds == NULL) { return NGX_CONF_ERROR; } ngx_memzero(ds, sizeof(ngx_http_upstream_drizzle_server_t)); value = cf->args->elts; /* parse the first name:port argument */ ngx_memzero(&u, sizeof(ngx_url_t)); u.url = value[1]; u.default_port = ngx_http_drizzle_default_port; if (ngx_parse_url(cf->pool, &u) != NGX_OK) { if (u.err) { ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, "drizzle: %s in upstream \"%V\"", u.err, &u.url); } return NGX_CONF_ERROR; } ds->addrs = u.addrs; ds->naddrs = u.naddrs; ds->port = u.port; ds->protocol = ngx_http_drizzle_protocol; /* parse various options */ for (i = 2; i < cf->args->nelts; i++) { if (ngx_strncmp(value[i].data, "dbname=", sizeof("dbname=") - 1) == 0) { ds->dbname.len = value[i].len - (sizeof("dbname=") - 1); if (ds->dbname.len >= DRIZZLE_MAX_DB_SIZE) { ngx_log_error(NGX_LOG_EMERG, cf->log, 0, "drizzle: \"dbname\" value too large in upstream \"%V\"" " (at most %d bytes)", dscf->peers->name, (int) DRIZZLE_MAX_DB_SIZE); return NGX_CONF_ERROR; } ds->dbname.data = &value[i].data[sizeof("dbname=") - 1]; continue; } if (ngx_strncmp(value[i].data, "user=", sizeof("user=") - 1) == 0) { ds->user.len = value[i].len - (sizeof("user=") - 1); if (ds->user.len >= DRIZZLE_MAX_USER_SIZE) { ngx_log_error(NGX_LOG_EMERG, cf->log, 0, "drizzle: \"user\" value too large in upstream \"%V\"" " (at most %d bytes)", dscf->peers->name, (int) DRIZZLE_MAX_USER_SIZE); return NGX_CONF_ERROR; } ds->user.data = &value[i].data[sizeof("user=") - 1]; continue; } if (ngx_strncmp(value[i].data, "password=", sizeof("password=") - 1) == 0) { ds->password.len = value[i].len - (sizeof("password=") - 1); if (ds->password.len >= DRIZZLE_MAX_PASSWORD_SIZE) { ngx_log_error(NGX_LOG_EMERG, cf->log, 0, "drizzle: \"password\" value too large in upstream " "\"%V\" (at most %d bytes)", dscf->peers->name, (int) DRIZZLE_MAX_PASSWORD_SIZE); return NGX_CONF_ERROR; } ds->password.data = &value[i].data[sizeof("password=") - 1]; continue; } if (ngx_strncmp(value[i].data, "protocol=", sizeof("protocol=") - 1) == 0) { protocol.len = value[i].len - (sizeof("protocol=") - 1); protocol.data = &value[i].data[sizeof("protocol=") - 1]; switch (protocol.len) { case 5: if (ngx_http_drizzle_strcmp_const(protocol.data, "mysql") == 0) { ds->protocol = ngx_http_mysql_protocol; } else { continue; } break; case 7: if (ngx_http_drizzle_strcmp_const(protocol.data, "drizzle") != 0) { continue; } break; default: ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, "drizzle: invalid protocol \"%V\"" " in drizzle_server", &protocol); return NGX_CONF_ERROR; } continue; } if (ngx_strncmp(value[i].data, "charset=", sizeof("charset=") - 1) == 0) { charset.len = value[i].len - (sizeof("charset=") - 1); charset.data = &value[i].data[sizeof("charset=") - 1]; dd("charset: %.*s", (int) charset.len, charset.data); if (charset.len == 0) { continue; } for (j = 0; j < charset.len; j++) { if (charset.data[j] == '\'') { ngx_conf_log_error(NGX_LOG_EMERG, cf, 0, "bad charste value \"%V\" in" " drizzle_server", &charset); return NGX_CONF_ERROR; } } len = sizeof("set names ''") - 1 + charset.len; p = ngx_palloc(cf->pool, len); if (p == NUL