subscriber_t *longpoll_subscriber_create(ngx_http_request_t *r, nchan_msg_id_t *msg_id) {
DBG("create for req %p", r);
full_subscriber_t *fsub;
//TODO: allocate from pool (but not the request's pool)
if((fsub = ngx_alloc(sizeof(*fsub), ngx_cycle->log)) == NULL) {
ERR("Unable to allocate");
assert(0);
return NULL;
}
nchan_subscriber_init(&fsub->sub, &new_longpoll_sub, r, msg_id);
fsub->privdata = NULL;
fsub->data.cln = NULL;
fsub->data.finalize_request = 1;
fsub->data.holding = 0;
fsub->data.act_as_intervalpoll = 0;
nchan_subscriber_init_timeout_timer(&fsub->sub, &fsub->data.timeout_ev);
fsub->data.dequeue_handler = empty_handler;
fsub->data.dequeue_handler_data = NULL;
fsub->data.already_responded = 0;
fsub->data.awaiting_destruction = 0;
if(fsub->sub.cf->longpoll_multimsg) {
nchan_request_ctx_t *ctx = ngx_http_get_module_ctx(r, ngx_nchan_module);
fsub->sub.dequeue_after_response = 0;
ctx->bcp = ngx_palloc(r->pool, sizeof(nchan_bufchain_pool_t));
nchan_bufchain_pool_init(ctx->bcp, r->pool);
}
fsub->data.multimsg_first = NULL;
fsub->data.multimsg_last = NULL;
#if NCHAN_SUBSCRIBER_LEAK_DEBUG
subscriber_debug_add(&fsub->sub);
//set debug label
fsub->sub.lbl = ngx_calloc(r->uri.len+1, ngx_cycle->log);
ngx_memcpy(fsub->sub.lbl, r->uri.data, r->uri.len);
#endif
//http request sudden close cleanup
if((fsub->data.cln = ngx_http_cleanup_add(r, 0)) == NULL) {
ERR("Unable to add request cleanup for longpoll subscriber");
assert(0);
return NULL;
}
fsub->data.cln->data = fsub;
fsub->data.cln->handler = (ngx_http_cleanup_pt )sudden_abort_handler;
DBG("%p created for request %p", &fsub->sub, r);
return &fsub->sub;
}
subscriber_t *websocket_subscriber_create(ngx_http_request_t *r, nchan_msg_id_t *msg_id) {
nchan_request_ctx_t *ctx = ngx_http_get_module_ctx(r, ngx_nchan_module);
DBG("create for req %p", r);
full_subscriber_t *fsub;
if((fsub = ngx_alloc(sizeof(*fsub), ngx_cycle->log)) == NULL) {
ERR("Unable to allocate");
return NULL;
}
nchan_subscriber_init(&fsub->sub, &new_websocket_sub, r, msg_id);
fsub->cln = NULL;
fsub->ctx = ctx;
fsub->ws_meta_subprotocol = 0;
fsub->finalize_request = 0;
fsub->holding = 0;
fsub->shook_hands = 0;
fsub->connected = 0;
fsub->pinging = 0;
fsub->closing = 0;
ngx_memzero(&fsub->ping_ev, sizeof(fsub->ping_ev));
nchan_subscriber_init_timeout_timer(&fsub->sub, &fsub->timeout_ev);
fsub->dequeue_handler = empty_handler;
fsub->dequeue_handler_data = NULL;
fsub->awaiting_destruction = 0;
ngx_memzero(&fsub->closing_ev, sizeof(fsub->closing_ev));
#if nginx_version >= 1008000
fsub->closing_ev.cancelable = 1;
#endif
//what should the buffers look like?
/*
//mesage buf
b->last_buf = 1;
b->last_in_chain = 1;
b->flush = 1;
b->memory = 1;
b->temporary = 0;
*/
if(fsub->sub.cf->pub.websocket) {
fsub->publish_channel_id = nchan_get_channel_id(r, PUB, 0);
}
fsub->upstream_stuff = NULL;
websocket_init_frame(&fsub->frame);
//http request sudden close cleanup
if((fsub->cln = ngx_http_cleanup_add(r, 0)) == NULL) {
ERR("Unable to add request cleanup for websocket subscriber");
return NULL;
}
fsub->cln->data = fsub;
fsub->cln->handler = (ngx_http_cleanup_pt )sudden_abort_handler;
DBG("%p created for request %p", &fsub->sub, r);
assert(ctx != NULL);
ctx->sub = &fsub->sub; //gonna need this for recv
ctx->subscriber_type = fsub->sub.name;
#if NCHAN_SUBSCRIBER_LEAK_DEBUG
subscriber_debug_add(&fsub->sub);
#endif
//send-frame buffer
ctx->output_str_queue = ngx_palloc(r->pool, sizeof(*ctx->output_str_queue));
nchan_reuse_queue_init(ctx->output_str_queue, offsetof(framebuf_t, prev), offsetof(framebuf_t, next), framebuf_alloc, NULL, r->pool);
//bufchain pool
ctx->bcp = ngx_palloc(r->pool, sizeof(nchan_bufchain_pool_t));
nchan_bufchain_pool_init(ctx->bcp, r->pool);
return &fsub->sub;
}
