2013年10月22日火曜日

nginx plus::statusモジュールを作る#2

ヘッダファイルを追加

ヘッダファイルを作りたくなった場合configに一行追加する。
今回はngx_http_json_status_module.hを追加するのでconfigに以下を追加。
NGX_ADDON_DEPS="$NGX_ADDON_DEPS $ngx_addon_dir/ngx_http_json_status_module.h"

ngx_command_t

ディレクティブの定義をした ngx_command_t 構造体は以下の様になっている。
typedef struct ngx_command_s     ngx_command_t;

struct ngx_command_s {
    ngx_str_t             name;
    ngx_uint_t            type;
    char               *(*set)(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
    ngx_uint_t            conf;
    ngx_uint_t            offset;
    void                 *post;
};
set要素はname要素のディレクティブが設定されたときに呼び出される関数。

ngx_command_tのset要素:ngx_http_json_status

status; が設定されているときに呼び出す関数。
内部でhandlerを設定する。

ngx_http_json_status_module.h

static char *ngx_http_json_status(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);

ngx_http_json_status_module.c

static char *
ngx_http_json_status(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_json_status_handler;

  return NGX_CONF_OK;
}

handler:ngx_http_json_status_handler

status;が設定されたlocationにリクエストが来た時に実行される。

ngx_http_json_status_module.h

static ngx_int_t ngx_http_json_status_handler(ngx_http_request_t *r);

ngx_http_json_status_module.c

jsonで空データを返すところまで。
データを詰めてレスポンスを返すのはどれもこんな感じで実装している。
static ngx_int_t
ngx_http_json_status_handler(ngx_http_request_t *r)
{
  size_t       size;
  ngx_buf_t   *b;
  ngx_int_t    rc;
  ngx_chain_t  out;

  /* GET or HEAD only */
  if (r->method != NGX_HTTP_GET && r->method != NGX_HTTP_HEAD) {
    return NGX_HTTP_NOT_ALLOWED;
  }

  /* request bodyは不要なので破棄する */
  rc = ngx_http_discard_request_body(r);
  if (rc != NGX_OK) {
    return rc;
  }

  size = sizeof("{}");

  b = ngx_create_temp_buf(r->pool, size);
  if (b == NULL) {
    return NGX_HTTP_INTERNAL_SERVER_ERROR;
  }

  out.buf = b;
  out.next = NULL;

  b->last = ngx_sprintf(b->last, "{}");

  b->memory = 1;
  b->flush = 1;
  b->last_buf = 1;
  b->last_in_chain = 1;

  ngx_str_set(&r->headers_out.content_type, "application/json; charset=utf-8");
  r->headers_out.status = NGX_HTTP_OK;
  r->headers_out.content_length_n = b->last - b->pos;
  rc = ngx_http_send_header(r);
  if (rc == NGX_ERROR || rc > NGX_OK || r->header_only) {
    ngx_log_error(NGX_LOG_DEBUG, r->connection->log, 0, "#return. %s:%d", __FUNCTION__, __LINE__);
    return rc;
  }

  return ngx_http_output_filter(r, &out);
}


repository


成果はここに置くことにした。


#3に続く

0 件のコメント:

コメントを投稿