kong/kong/plugins/correlation-id/handler.lua at master · Kong/kong

link管理

链接快照平台

输入网页链接，自动生成快照
标签化管理网页链接

相关文章推荐

重情义的甘蔗 · 再生戒指 (Ring of ...· 1 月前 ·

帅气的枕头 · Why is it so slow? | ...· 3 月前 ·

打酱油的书签 · 《家务优等生》开启第四轮， ...· 3 月前 ·

玩足球的打火机 · Importing a torch ...· 4 月前 ·

重感情的椰子 · 视频去哪了呢？_哔哩哔哩_bilibili· 7 月前 ·

local kong = kong

local worker_uuid

local worker_counter

local generators

local worker_pid = ngx . worker . pid ()

local now = ngx . now

local var = ngx . var

local fmt = string.format

generators = {

[ " uuid " ] = function ()

return uuid ()

end ,

[ " uuid#counter " ] = function ()

worker_counter = worker_counter + 1

return worker_uuid .. " # " .. worker_counter

end ,

[ " tracker " ] = function ()

return fmt ( " %s-%s-%d-%s-%s-%0.3f " ,

var . server_addr ,

var . server_port ,

worker_pid ,

var . connection , -- connection serial number

var . connection_requests , -- current number of requests made through a connection

now () -- the current time stamp from the nginx cached time.

)

end ,

}

end

local CorrelationIdHandler = {}

CorrelationIdHandler . PRIORITY = 1

CorrelationIdHandler . VERSION = kong_meta . version

function CorrelationIdHandler : init_worker ()

worker_uuid = uuid ()

worker_counter = 0

end

function CorrelationIdHandler : access ( conf )

-- Set header for upstream

local correlation_id = kong . request . get_header ( conf . header_name )

if not correlation_id or correlation_id == " " then

-- Generate the header value

correlation_id = generators [ conf . generator ]()

if correlation_id then

kong . service . request . set_header ( conf . header_name , correlation_id )

end

kong . log . set_serialize_value ( " correlation_id " , correlation_id )

if conf . echo_downstream then

-- For later use, to echo it back downstream

kong . ctx . plugin . correlation_id = correlation_id

end

function CorrelationIdHandler : header_filter ( conf )

if not conf . echo_downstream then

return

end

local correlation_id = kong . ctx . plugin . correlation_id or

kong . request . get_header ( conf . header_name )

if not correlation_id or correlation_id == " " then

correlation_id = generators [ conf . generator ]()

end

kong . response . set_header ( conf . header_name , correlation_id )

end

return CorrelationIdHandler