Hi all,
I'm currently writing an nginx module that sends a number of subrequests to
upstream servers based on application logic, and then aggregates the results
and send them back. That part currently works just fine, thanks to the help I
was given here in the past.
However, I am now trying to add a custom header to the request that is being
proxied and currently failing at it. For various technical reasons, I need to
add a header that contains the IP address and port (from the point of view of
nginx) of the upstream server currently being contacted. In order to get at
that address, I have hooked my module into the peer selection mechanism, much
like the keepalive, ip_hash modules and others do. So far so good, I get at
that address and can log it just fine. However, I can't seem to be able to add
it as a header to my request.
Here's my init_peer callback :
static ngx_int_t
ngx_http_dispatcher_upstream_init_peer(ngx_http_request_t *r,
ngx_http_upstream_srv_conf_t *us)
{
ngx_http_dispatcher_upstream_peer_data_t *dp;
ngx_http_dispatcher_srv_conf_t *dscf;
ngx_int_t rc;
dscf = ngx_http_conf_upstream_srv_conf(us, ngx_http_dispatcher_module);
dp = ngx_palloc(r->pool,
sizeof(ngx_http_dispatcher_upstream_peer_data_t));
if (dp == NULL) {
return NGX_ERROR;
}
rc = dscf->original_init_peer(r, us);
if (rc != NGX_OK) {
return rc;
}
dp->request = r;
dp->data = r->upstream->peer.data;
dp->original_get_peer = r->upstream->peer.get;
dp->original_free_peer = r->upstream->peer.free;
r->upstream->peer.data = dp;
r->upstream->peer.get = ngx_http_dispatcher_upstream_get_peer;
r->upstream->peer.free = ngx_http_dispatcher_upstream_free_peer;
#if (NGX_HTTP_SSL)
dp->original_set_session = r->upstream->peer.set_session;
dp->original_save_session = r->upstream->peer.save_session;
r->upstream->peer.set_session =
ngx_http_dispatcher_upstream_set_session;
r->upstream->peer.save_session =
ngx_http_dispatcher_upstream_save_session;
#endif
return NGX_OK;
}
Note that I save the pointer to the ngx_http_request_t object in my peer data
structure to access it later. Now here's my peer selection hook :
static ngx_int_t
ngx_http_dispatcher_upstream_get_peer(ngx_peer_connection_t *pc, void *data)
{
ngx_http_dispatcher_upstream_peer_data_t *dp = data;
u_char addrbuf[NGX_SOCKADDR_STRLEN];
ngx_http_request_t *r;
ngx_table_elt_t *h;
ngx_str_t peeraddr;
ngx_int_t rc;
rc = dp->original_get_peer(pc, dp->data);
if (rc != NGX_OK) {
return rc;
}
peeraddr.data = addrbuf;
peeraddr.len = ngx_sock_ntop(pc->sockaddr, pc->socklen, addrbuf,
sizeof(addrbuf), 1);
ngx_log_debug1(NGX_LOG_DEBUG_HTTP, pc->log, 0,
"Selected peer address is: %V", &peeraddr);
r = dp->request;
h = ngx_list_push(&r->headers_in.headers);
if (h == NULL) {
return NGX_ERROR;
}
h->hash = 1;
ngx_str_set(&h->key, "X-Shard-Addr");
h->value.data = ngx_pstrdup(r->pool, &peeraddr);
if (h->value.data == NULL) {
return NGX_ERROR;
}
h->value.len = peeraddr.len;
return NGX_OK;
}
So after getting at the actual address we're forwarding our request to, I try
to add a custom header to r->headers_in, on the request object that I saved in
my peer data.
I suspect I'm not actually dealing with the correct request here, because no
matter what I try, I can't get this code to actually alter the headers of the
forwarded request in any way.
Any help would be greatly appreciated!
Thanks,
Maxime
_______________________________________________
nginx-devel mailing list
nginx-devel@nginx.org
http://mailman.nginx.org/mailman/listinfo/nginx-devel
