nginx subrequest 프로그램 예제

#user  nobody;
worker_processes  1;

#error_log  logs/error.log;
#error_log  logs/error.log  notice;
#error_log  logs/error.log  info;

#pid        logs/nginx.pid;


events {
    worker_connections  1024;
}


http {
        include       mime.types;
        default_type  application/octet-stream;

    #log_format  main  '$remote_addr - $remote_user [$time_local] "$request" '
    #                  '$status $body_bytes_sent "$http_referer" '
    #                  '"$http_user_agent" "$http_x_forwarded_for"';

    #access_log  logs/access.log  main;
    sendfile        on;
    #tcp_nopush     on;

    #keepalive_timeout  0;
        keepalive_timeout  65;

    #gzip  on;

        server {
                listen       80;
                server_name  localhost;

        #charset koi8-r;

        #access_log  logs/host.access.log  main;

                location / {
                        root   html;
                        index  index.html index.htm;
                }

#               location /test2 {  
#                       test_str "hello my dear HUST!";  
#                       test_flag on;  
#                       test_num 10;  
#                       test_size 1000;  
#                       mytest;  
#               }
                location /list {
                        proxy_pass http://hq.sinajs.cn;
                        proxy_set_header Accept-Encoding  "";
                }

                location /query {
                        mytest;
                }
#               location /test2 {
#                        test_str  "hello my dear ye.";
#                        test_flag  on;
#                        test_num   10;
#                        test_size  1000;
#                        mytest;
#                }

        #error_page  404              /404.html;

        # redirect server error pages to the static page /50x.html
        #
               error_page   500 502 503 504  /50x.html;
                location = /50x.html {
                        root   html;
                }


        }

        server {
                listen    80 default_server;
                server_name  www8.test.com ;
                root  /home/src/index;
                index  index.html;
                location /list {
                        proxy_pass http://hq.sinajs.cn;
                        proxy_set_header Accept-Encoding  "";
                }

                location /query {
                        mytest;
                }

#               location /test2 {
#                       test_str  "hello my dear ye.";
                   #                       test_flag  on;
#                       test_num   10;
#                       test_size  1000;
#                       mytest;
#               }

                error_page   500 502 503 504  /50x.html;
                location = /50x.html {
                       root   html;
                }

        }

}
                                       

ngx_addon_name=ngx_http_mytest_module
HTTP_MODULES="$HTTP_MODULES ngx_http_mytest_module"
NGX_ADDON_SRCS="$NGX_ADDON_SRCS $ngx_addon_dir/ngx_http_mytest_subrequest_module.c"

#include
#include
#include

static void mytest_post_handler(ngx_http_request_t *r);
static ngx_int_t mytest_subrequest_post_handler(ngx_http_request_t *r,void *    data,ngx_int_t rc);
static ngx_int_t ngx_http_mytest_handler(ngx_http_request_t *r);
//save gupiao data
typedef struct {
        ngx_str_t stock[6];
}ngx_http_mytest_ctx_t;

static ngx_http_module_t ngx_http_mytest_module_ctx = {
        NULL,NULL,
        NULL,NULL,
        NULL,NULL,
        NULL,NULL
};
static char * ngx_http_mytest(ngx_conf_t *cf,ngx_command_t *cmd,void *conf){
        ngx_http_core_loc_conf_t *clcf;
        clcf = ngx_http_conf_get_module_loc_conf(cf,ngx_http_core_module);
        clcf->handler = ngx_http_mytest_handler;
        return NGX_CONF_OK;
}

static ngx_command_t ngx_http_mytest_commands[] = {
        {
                ngx_string("mytest"),
                NGX_HTTP_MAIN_CONF|NGX_HTTP_SRV_CONF|NGX_HTTP_LOC_CONF|NGX_HTTP_LMT_CONF|NGX_CONF_NOARGS,
                ngx_http_mytest,
                NGX_HTTP_LOC_CONF_OFFSET,
                0,
                NULL
        },
        ngx_null_command
};

ngx_module_t ngx_http_mytest_module = {
        NGX_MODULE_V1,
        &ngx_http_mytest_module_ctx,
        ngx_http_mytest_commands,
        NGX_HTTP_MODULE,
        NULL,NULL,
        NULL,NULL,
        NULL,NULL,
        NULL,
        NGX_MODULE_V1_PADDING
};

//start subrequest 
static ngx_int_t ngx_http_mytest_handler(ngx_http_request_t *r){
        //creat HTTP ctx
        ngx_http_mytest_ctx_t *myctx = ngx_http_get_module_ctx(r,ngx_http_mytest_module);
        if(myctx == NULL){
                myctx = ngx_palloc(r->pool,sizeof(ngx_http_mytest_ctx_t));
                if(myctx == NULL) return NGX_ERROR;
                ngx_http_set_ctx(r,myctx,ngx_http_mytest_module);
        }
        ngx_http_post_subrequest_t *psr = ngx_palloc(r->pool,sizeof(ngx_http_post_subrequest_t));
        if(psr == NULL) return NGX_HTTP_INTERNAL_SERVER_ERROR;
        psr->handler = mytest_subrequest_post_handler;
        psr->data = myctx;

        ngx_str_t sub_prefix = ngx_string("/list=");
        ngx_str_t sub_location;
        sub_location.len = sub_prefix.len + r->args.len;
        sub_location.data = ngx_palloc(r->pool,sub_location.len);
        ngx_snprintf(sub_location.data,sub_location.len,"%V%V",&sub_prefix,&r->args);
        ngx_http_request_t *sr;
        ngx_int_t rc = ngx_http_subrequest(r,&sub_location,NULL,&sr,psr,NGX_HTTP_SUBREQUEST_IN_MEMORY);
        if(rc != NGX_OK) return NGX_ERROR;
        return NGX_DONE;
}
// subrequest end handler 
static ngx_int_t mytest_subrequest_post_handler(ngx_http_request_t *r,void *data,ngx_int_t rc){
        ngx_http_request_t *pr = r->parent;
        ngx_http_mytest_ctx_t *myctx = ngx_http_get_module_ctx(pr,ngx_http_mytest_module);
        pr->headers_out.status = r->headers_out.status;
        if(r->headers_out.status == NGX_HTTP_OK){
                int flag = 0;
                ngx_buf_t *pRecvBuf = &r->upstream->buffer;
                for(;pRecvBuf->pos != pRecvBuf->last;pRecvBuf->pos++){
                        if(*pRecvBuf->pos == ','||*pRecvBuf->pos == '\"'){
                                if(flag>0) myctx->stock[flag-1].len = pRecvBuf->pos - myctx->stock[flag-1].data;
                                flag++;
                                myctx->stock[flag-1].data = pRecvBuf->pos + 1;
                        }

                        if(flag>6) break;
                }
        }
        pr->write_event_handler = mytest_post_handler;
        return NGX_OK;
}
static void mytest_post_handler(ngx_http_request_t *r){
        if(r->headers_out.status != NGX_HTTP_OK){
                ngx_http_finalize_request(r,r->headers_out.status);
                return;
        }
        ngx_http_mytest_ctx_t *myctx = ngx_http_get_module_ctx(r,ngx_http_mytest_module);
        ngx_str_t output_format = ngx_string("stock[%V],Today current price: %V,volum: %V");
        int bodylen = output_format.len + myctx->stock[0].len + myctx->stock[1].len+myctx->stock[4].len - 6;
        r->headers_out.content_length_n = bodylen;
        ngx_buf_t *b = ngx_create_temp_buf(r->pool,bodylen);
        ngx_snprintf(b->pos,bodylen,(char*)output_format.data,&myctx->stock[0],&myctx->stock[1],&myctx->stock[4]);
        b->last = b->pos + bodylen;
        b->last_buf = 1;
        ngx_chain_t out;
        out.buf = b;
        out.next = NULL;
        static ngx_str_t type = ngx_string("text/plain; charset=GBK");
        r->headers_out.content_type = type;
        r->headers_out.status = NGX_HTTP_OK;
        r->connection->buffered |= NGX_HTTP_WRITE_BUFFERED;
        ngx_int_t ret = ngx_http_send_header(r);
        ret = ngx_http_output_filter(r,&out);
        ngx_http_finalize_request(r,ret);
}