lua-resty-core/lib/resty/core/ctx.lua at a94680359b1907c6438d03471757941455b5cbcc · openresty/lua-resty-core · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
-- Copyright (C) Yichun Zhang (agentzh)


local base = require "resty.core.base"
base.allows_subsystem('http')


local ffi = require 'ffi'
local debug = require 'debug'
local misc = require "resty.core.misc"


local register_getter = misc.register_ngx_magic_key_getter
local register_setter = misc.register_ngx_magic_key_setter
local registry = debug.getregistry()
local new_tab = base.new_tab
local ref_in_table = base.ref_in_table
local get_request = base.get_request
local C = ffi.C
local FFI_NO_REQ_CTX = base.FFI_NO_REQ_CTX
local FFI_OK = base.FFI_OK
local error = error


ffi.cdef[[
int ngx_http_lua_ffi_get_ctx_ref(ngx_http_request_t *r);
int ngx_http_lua_ffi_set_ctx_ref(ngx_http_request_t *r, int ref);
]]


local _M = {
    _VERSION = base.version
}


local function get_ctx_table()
    local r = get_request()

    if not r then
        error("no request found")
    end

    local ctx_ref = C.ngx_http_lua_ffi_get_ctx_ref(r)
    if ctx_ref == FFI_NO_REQ_CTX then
        error("no request ctx found")
    end

    local ctxs = registry.ngx_lua_ctx_tables
    if ctx_ref < 0 then
        local ctx = new_tab(0, 4)
        ctx_ref = ref_in_table(ctxs, ctx)
        if C.ngx_http_lua_ffi_set_ctx_ref(r, ctx_ref) ~= FFI_OK then
            return nil
        end
        return ctx
    end
    return ctxs[ctx_ref]
end
register_getter("ctx", get_ctx_table)


local function set_ctx_table(ctx)
    local r = get_request()

    if not r then
        error("no request found")
    end

    local ctx_ref = C.ngx_http_lua_ffi_get_ctx_ref(r)
    if ctx_ref == FFI_NO_REQ_CTX then
        error("no request ctx found")
    end

    local ctxs = registry.ngx_lua_ctx_tables
    if ctx_ref < 0 then
        ctx_ref = ref_in_table(ctxs, ctx)
        C.ngx_http_lua_ffi_set_ctx_ref(r, ctx_ref)
        return
    end
    ctxs[ctx_ref] = ctx
end
register_setter("ctx", set_ctx_table)


return _M