0000: 2d 2d 20 6d 6f 64 75 6c 65 20 63 6f 75 6e 74 65 -- module counte
0010: 72 20 2d 20 63 6f 75 6e 74 73 20 73 6f 6d 65 20 r - counts some
0020: 6a 75 6e 6b 0a 0a 6c 6f 63 61 6c 20 74 6f 64 61 junk..local toda
0030: 79 20 3d 20 6f 73 2e 64 61 74 65 28 27 2a 74 27 y = os.date('*t'
0040: 29 0a 6c 6f 63 61 6c 20 74 69 6d 65 73 74 61 6d ).local timestam
0050: 70 20 3d 20 6f 73 2e 74 69 6d 65 7b 79 65 61 72 p = os.time{year
0060: 20 3d 20 74 6f 64 61 79 5b 27 79 65 61 72 27 5d = today['year']
0070: 2c 20 6d 6f 6e 74 68 20 3d 20 74 6f 64 61 79 5b , month = today[
0080: 27 6d 6f 6e 74 68 27 5d 2c 20 64 61 79 20 3d 20 'month'], day =
0090: 74 6f 64 61 79 5b 27 64 61 79 27 5d 7d 0a 6c 6f today['day']}.lo
00a0: 63 61 6c 20 72 65 66 65 72 65 72 20 3d 20 27 2d cal referer = '-
00b0: 27 3b 0a 0a 69 66 20 6e 67 78 2e 76 61 72 2e 68 ';..if ngx.var.h
00c0: 74 74 70 5f 72 65 66 65 72 65 72 20 7e 3d 20 6e ttp_referer ~= n
00d0: 69 6c 20 74 68 65 6e 0a 09 5f 2c 20 5f 2c 20 72 il then.._, _, r
00e0: 65 66 65 72 65 72 20 3d 20 73 74 72 69 6e 67 2e eferer = string.
00f0: 66 69 6e 64 28 6e 67 78 2e 76 61 72 2e 68 74 74 find(ngx.var.htt
0100: 70 5f 72 65 66 65 72 65 72 2c 20 27 5e 68 74 74 p_referer, '^htt
0110: 70 73 3f 3a 2f 2f 28 5b 25 77 25 2e 5d 2b 29 27 ps?://([%w%.]+)'
0120: 29 0a 65 6e 64 0a 0a 6c 6f 63 61 6c 20 6b 65 79 ).end..local key
0130: 20 3d 20 27 73 74 61 74 5f 63 6f 75 6e 74 65 72 = 'stat_counter
0140: 5f 27 20 2e 2e 20 74 69 6d 65 73 74 61 6d 70 20 _' .. timestamp
0150: 2e 2e 20 27 5f 27 20 2e 2e 20 72 65 66 65 72 65 .. '_' .. refere
0160: 72 0a 0a 2d 2d 20 63 6f 6e 6e 65 63 74 20 74 6f r..-- connect to
0170: 20 72 65 64 69 73 0a 6c 6f 63 61 6c 20 72 65 64 redis.local red
0180: 69 73 20 3d 20 72 65 71 75 69 72 65 20 27 72 65 is = require 're
0190: 73 74 79 2e 72 65 64 69 73 27 0a 6c 6f 63 61 6c sty.redis'.local
01a0: 20 72 65 64 20 3d 20 72 65 64 69 73 3a 6e 65 77 red = redis:new
01b0: 28 29 0a 6c 6f 63 61 6c 20 6f 6b 2c 20 65 72 72 ().local ok, err
01c0: 20 3d 20 72 65 64 3a 63 6f 6e 6e 65 63 74 28 27 = red:connect('
01d0: 31 32 37 2e 30 2e 30 2e 31 27 2c 20 36 33 37 39 127.0.0.1', 6379
01e0: 29 0a 69 66 20 6e 6f 74 20 6f 6b 20 74 68 65 6e ).if not ok then
01f0: 0a 09 6e 67 78 2e 6c 6f 67 28 6e 67 78 2e 45 52 ..ngx.log(ngx.ER
0200: 52 2c 20 27 72 65 64 69 73 20 63 6f 6e 6e 65 63 R, 'redis connec
0210: 74 69 6f 6e 20 66 61 69 6c 65 64 3a 20 27 2c 20 tion failed: ',
0220: 65 72 72 29 0a 09 6e 67 78 2e 65 78 69 74 28 35 err)..ngx.exit(5
0230: 30 30 29 0a 65 6e 64 0a 0a 6c 6f 63 61 6c 20 63 00).end..local c
0240: 6a 73 6f 6e 20 3d 20 72 65 71 75 69 72 65 20 27 json = require '
0250: 63 6a 73 6f 6e 27 0a 0a 2d 2d 20 64 65 63 6f 64 cjson'..-- decod
0260: 65 73 20 76 61 6c 75 65 73 20 66 72 6f 6d 20 61 es values from a
0270: 72 72 61 79 20 69 6e 74 6f 20 64 69 63 74 0a 66 rray into dict.f
0280: 75 6e 63 74 69 6f 6e 20 64 65 63 6f 64 65 5f 62 unction decode_b
0290: 75 6c 6b 28 72 65 70 6c 79 29 0a 09 6c 6f 63 61 ulk(reply)..loca
02a0: 6c 20 64 61 74 61 20 3d 20 7b 7d 0a 09 66 6f 72 l data = {}..for
02b0: 20 6a 3d 31 2c 20 23 72 65 70 6c 79 2c 20 32 20 j=1, #reply, 2
02c0: 64 6f 0a 09 09 64 61 74 61 5b 72 65 70 6c 79 5b do...data[reply[
02d0: 6a 5d 20 5d 20 3d 20 72 65 70 6c 79 5b 6a 2b 31 j] ] = reply[j+1
02e0: 5d 0a 09 65 6e 64 0a 09 72 65 74 75 72 6e 28 64 ]..end..return(d
02f0: 61 74 61 29 0a 65 6e 64 0a 0a 6c 6f 63 61 6c 20 ata).end..local
0300: 64 61 74 61 2c 20 65 6d 70 74 79 0a 65 6d 70 74 data, empty.empt
0310: 79 2c 20 65 72 72 20 3d 20 72 65 64 3a 68 73 65 y, err = red:hse
0320: 74 6e 78 28 6b 65 79 2c 20 27 74 6f 64 61 79 27 tnx(key, 'today'
0330: 2c 20 30 29 0a 69 66 20 6e 6f 74 20 65 6d 70 74 , 0).if not empt
0340: 79 20 74 68 65 6e 0a 09 6e 67 78 2e 6c 6f 67 28 y then..ngx.log(
0350: 6e 67 78 2e 45 52 52 2c 20 27 72 65 64 69 73 20 ngx.ERR, 'redis
0360: 71 75 65 72 79 20 66 61 69 6c 65 64 3a 20 27 2c query failed: ',
0370: 20 65 72 72 29 0a 09 6e 67 78 2e 65 78 69 74 28 err)..ngx.exit(
0380: 35 30 30 29 0a 65 6e 64 0a 0a 69 66 20 65 6d 70 500).end..if emp
0390: 74 79 20 3d 3d 20 30 20 74 68 65 6e 0a 09 64 61 ty == 0 then..da
03a0: 74 61 2c 20 65 72 72 20 3d 20 72 65 64 3a 68 67 ta, err = red:hg
03b0: 65 74 61 6c 6c 28 6b 65 79 29 0a 09 69 66 20 6e etall(key)..if n
03c0: 6f 74 20 64 61 74 61 20 74 68 65 6e 0a 09 09 6e ot data then...n
03d0: 67 78 2e 6c 6f 67 28 6e 67 78 2e 45 52 52 2c 20 gx.log(ngx.ERR,
03e0: 27 72 65 64 69 73 20 71 75 65 72 79 20 66 61 69 'redis query fai
03f0: 6c 65 64 3a 20 27 2c 20 65 72 72 29 0a 09 09 6e led: ', err)...n
0400: 67 78 2e 65 78 69 74 28 35 30 30 29 0a 09 65 6e gx.exit(500)..en
0410: 64 0a 65 6e 64 0a 0a 69 66 20 65 6d 70 74 79 20 d.end..if empty
0420: 3d 3d 20 31 20 6f 72 20 64 61 74 61 2e 77 68 6f == 1 or data.who
0430: 6c 65 20 3d 3d 20 6e 69 6c 20 74 68 65 6e 0a 09 le == nil then..
0440: 2d 2d 20 70 6f 73 74 67 72 65 73 20 66 61 6c 6c -- postgres fall
0450: 62 61 63 6b 0a 09 72 65 73 75 6c 74 20 3d 20 6e back..result = n
0460: 67 78 2e 6c 6f 63 61 74 69 6f 6e 2e 63 61 70 74 gx.location.capt
0470: 75 72 65 28 27 2f 70 6f 73 74 67 72 65 73 27 2c ure('/postgres',
0480: 20 7b 0a 09 09 6d 65 74 68 6f 64 20 3d 20 6e 67 {...method = ng
0490: 78 2e 48 54 54 50 5f 50 55 54 2c 0a 09 09 62 6f x.HTTP_PUT,...bo
04a0: 64 79 20 3d 20 22 73 65 6c 65 63 74 20 2a 20 66 dy = "select * f
04b0: 72 6f 6d 20 67 65 74 5f 73 74 61 74 73 28 27 22 rom get_stats('"
04c0: 20 2e 2e 20 72 65 66 65 72 65 72 20 2e 2e 20 22 .. referer .. "
04d0: 27 29 20 61 73 20 28 74 6f 64 61 79 20 69 6e 74 ') as (today int
04e0: 2c 20 6c 61 73 74 64 61 79 20 69 6e 74 2c 20 77 , lastday int, w
04f0: 65 65 6b 20 62 69 67 69 6e 74 2c 20 77 68 6f 6c eek bigint, whol
0500: 65 20 62 69 67 69 6e 74 29 3b 22 0a 09 7d 29 0a e bigint);"..}).
0510: 09 69 66 20 72 65 73 75 6c 74 2e 73 74 61 74 75 .if result.statu
0520: 73 20 7e 3d 20 32 30 30 20 6f 72 20 6e 6f 74 20 s ~= 200 or not
0530: 72 65 73 75 6c 74 2e 62 6f 64 79 20 74 68 65 6e result.body then
0540: 0a 09 09 6e 67 78 2e 6c 6f 67 28 6e 67 78 2e 45 ...ngx.log(ngx.E
0550: 52 52 2c 20 27 70 6f 73 74 67 72 65 73 20 61 63 RR, 'postgres ac
0560: 63 65 73 73 20 66 61 69 6c 65 64 27 29 0a 09 09 cess failed')...
0570: 6e 67 78 2e 65 78 69 74 28 35 30 30 29 0a 09 65 ngx.exit(500)..e
0580: 6c 73 65 0a 09 09 6c 6f 63 61 6c 20 75 6e 72 64 lse...local unrd
0590: 73 20 3d 20 72 65 71 75 69 72 65 20 22 72 64 73 s = require "rds
05a0: 2e 70 61 72 73 65 72 22 0a 09 09 6c 6f 63 61 6c .parser"...local
05b0: 20 72 65 73 2c 20 65 72 72 20 3d 20 75 6e 72 64 res, err = unrd
05c0: 73 2e 70 61 72 73 65 28 72 65 73 75 6c 74 2e 62 s.parse(result.b
05d0: 6f 64 79 29 0a 09 09 69 66 20 72 65 73 20 3d 3d ody)...if res ==
05e0: 20 6e 69 6c 20 74 68 65 6e 0a 09 09 09 6e 67 78 nil then....ngx
05f0: 2e 6c 6f 67 28 6e 67 78 2e 45 52 52 2c 20 27 66 .log(ngx.ERR, 'f
0600: 61 69 6c 65 64 20 74 6f 20 6f 62 74 61 69 6e 20 ailed to obtain
0610: 64 61 74 61 3a 20 27 20 2e 2e 20 65 72 72 29 0a data: ' .. err).
0620: 09 09 09 6e 67 78 2e 65 78 69 74 28 35 30 30 29 ...ngx.exit(500)
0630: 0a 09 09 65 6c 73 65 0a 09 09 09 64 61 74 61 20 ...else....data
0640: 3d 20 72 65 73 2e 72 65 73 75 6c 74 73 65 74 5b = res.resultset[
0650: 31 5d 0a 09 09 09 6c 6f 63 61 6c 20 72 65 71 20 1]....local req
0660: 3d 20 7b 6b 65 79 7d 0a 0a 09 09 09 66 6f 72 20 = {key}.....for
0670: 6e 61 6d 65 2c 20 76 61 6c 75 65 20 69 6e 20 70 name, value in p
0680: 61 69 72 73 28 64 61 74 61 29 20 64 6f 0a 09 09 airs(data) do...
0690: 09 09 69 66 20 76 61 6c 75 65 20 3d 3d 20 75 6e ..if value == un
06a0: 72 64 73 2e 6e 75 6c 6c 20 74 68 65 6e 0a 09 09 rds.null then...
06b0: 09 09 09 76 61 6c 75 65 20 3d 20 30 0a 09 09 09 ...value = 0....
06c0: 09 65 6e 64 0a 09 09 09 09 69 66 20 6e 61 6d 65 .end.....if name
06d0: 20 7e 3d 20 27 74 6f 64 61 79 27 20 74 68 65 6e ~= 'today' then
06e0: 0a 09 09 09 09 09 74 61 62 6c 65 2e 69 6e 73 65 ......table.inse
06f0: 72 74 28 72 65 71 2c 20 6e 61 6d 65 29 0a 09 09 rt(req, name)...
0700: 09 09 09 74 61 62 6c 65 2e 69 6e 73 65 72 74 28 ...table.insert(
0710: 72 65 71 2c 20 76 61 6c 75 65 29 0a 09 09 09 09 req, value).....
0720: 65 6e 64 0a 09 09 09 65 6e 64 0a 0a 09 09 09 72 end....end.....r
0730: 65 64 3a 69 6e 69 74 5f 70 69 70 65 6c 69 6e 65 ed:init_pipeline
0740: 28 33 29 0a 09 09 09 72 65 64 3a 68 6d 73 65 74 (3)....red:hmset
0750: 7b 72 65 71 7d 0a 09 09 09 72 65 64 3a 65 78 70 {req}....red:exp
0760: 69 72 65 28 6b 65 79 2c 20 31 32 39 36 30 30 29 ire(key, 129600)
0770: 0a 09 09 09 72 65 64 3a 68 69 6e 63 72 62 79 28 ....red:hincrby(
0780: 6b 65 79 2c 20 27 74 6f 64 61 79 27 2c 20 64 61 key, 'today', da
0790: 74 61 2e 74 6f 64 61 79 29 0a 09 09 09 72 65 73 ta.today)....res
07a0: 2c 20 65 72 72 20 3d 20 72 65 64 3a 63 6f 6d 6d , err = red:comm
07b0: 69 74 5f 70 69 70 65 6c 69 6e 65 28 29 0a 09 09 it_pipeline()...
07c0: 09 69 66 20 6e 6f 74 20 72 65 73 20 74 68 65 6e .if not res then
07d0: 0a 09 09 09 09 6e 67 78 2e 6c 6f 67 28 6e 67 78 .....ngx.log(ngx
07e0: 2e 45 52 52 2c 20 27 72 65 64 69 73 20 70 69 70 .ERR, 'redis pip
07f0: 65 6c 69 6e 65 20 66 61 69 6c 65 64 3a 20 27 2c eline failed: ',
0800: 20 65 72 72 29 0a 09 09 09 09 6e 67 78 2e 65 78 err).....ngx.ex
0810: 69 74 28 35 30 30 29 0a 09 09 09 65 6e 64 0a 09 it(500)....end..
0820: 09 65 6e 64 0a 09 65 6e 64 0a 65 6e 64 0a 0a 64 .end..end.end..d
0830: 61 74 61 2e 74 6f 64 61 79 20 3d 20 64 61 74 61 ata.today = data
0840: 2e 74 6f 64 61 79 20 2b 20 31 0a 6e 67 78 2e 73 .today + 1.ngx.s
0850: 61 79 28 63 6a 73 6f 6e 2e 65 6e 63 6f 64 65 28 ay(cjson.encode(
0860: 64 61 74 61 29 29 0a 6e 67 78 2e 65 6f 66 28 29 data)).ngx.eof()
0870: 0a 0a 72 65 73 2c 20 65 72 72 20 3d 20 72 65 64 ..res, err = red
0880: 3a 68 69 6e 63 72 62 79 28 6b 65 79 2c 20 27 74 :hincrby(key, 't
0890: 6f 64 61 79 27 2c 20 31 29 0a 6c 6f 63 61 6c 20 oday', 1).local
08a0: 75 69 64 20 3d 20 27 27 0a 0a 69 66 20 6e 67 78 uid = ''..if ngx
08b0: 2e 76 61 72 2e 75 69 64 5f 67 6f 74 20 7e 3d 20 .var.uid_got ~=
08c0: 6e 69 6c 20 61 6e 64 20 73 74 72 69 6e 67 2e 66 nil and string.f
08d0: 69 6e 64 28 6e 67 78 2e 76 61 72 2e 75 69 64 5f ind(ngx.var.uid_
08e0: 67 6f 74 2c 20 27 75 69 64 3d 27 29 20 3d 3d 20 got, 'uid=') ==
08f0: 31 20 74 68 65 6e 0a 09 75 69 64 20 3d 20 73 74 1 then..uid = st
0900: 72 69 6e 67 2e 73 75 62 28 6e 67 78 2e 76 61 72 ring.sub(ngx.var
0910: 2e 75 69 64 5f 67 6f 74 2c 20 35 29 0a 65 6c 73 .uid_got, 5).els
0920: 65 69 66 20 6e 67 78 2e 76 61 72 2e 75 69 64 5f eif ngx.var.uid_
0930: 73 65 74 20 7e 3d 20 6e 69 6c 20 61 6e 64 20 73 set ~= nil and s
0940: 74 72 69 6e 67 2e 66 69 6e 64 28 6e 67 78 2e 76 tring.find(ngx.v
0950: 61 72 2e 75 69 64 5f 73 65 74 2c 20 27 75 69 64 ar.uid_set, 'uid
0960: 3d 27 29 20 3d 3d 20 31 20 74 68 65 6e 0a 09 75 =') == 1 then..u
0970: 69 64 20 3d 20 73 74 72 69 6e 67 2e 73 75 62 28 id = string.sub(
0980: 6e 67 78 2e 76 61 72 2e 75 69 64 5f 73 65 74 2c ngx.var.uid_set,
0990: 20 35 29 0a 65 6e 64 0a 0a 6c 6f 63 61 6c 20 68 5).end..local h
09a0: 69 74 5f 6b 65 79 20 3d 20 75 69 64 20 2e 2e 20 it_key = uid ..
09b0: 27 5f 27 20 2e 2e 20 72 65 66 65 72 65 72 20 2e '_' .. referer .
09c0: 2e 20 27 5f 27 20 2e 2e 20 6e 67 78 2e 76 61 72 . '_' .. ngx.var
09d0: 2e 72 65 6d 6f 74 65 5f 61 64 64 72 0a 0a 72 65 .remote_addr..re
09e0: 64 3a 69 6e 69 74 5f 70 69 70 65 6c 69 6e 65 28 d:init_pipeline(
09f0: 35 29 0a 72 65 64 3a 6d 75 6c 74 69 28 29 0a 72 5).red:multi().r
0a00: 65 64 3a 68 69 6e 63 72 62 79 28 27 73 74 61 74 ed:hincrby('stat
0a10: 5f 63 6f 75 6e 74 65 72 5f 70 65 6e 64 69 6e 67 _counter_pending
0a20: 27 2c 20 68 69 74 5f 6b 65 79 2c 20 31 29 0a 72 ', hit_key, 1).r
0a30: 65 64 3a 65 78 70 69 72 65 28 27 73 74 61 74 5f ed:expire('stat_
0a40: 63 6f 75 6e 74 65 72 5f 70 65 6e 64 69 6e 67 27 counter_pending'
0a50: 2c 20 36 30 29 0a 72 65 64 3a 72 65 6e 61 6d 65 , 60).red:rename
0a60: 6e 78 28 27 73 74 61 74 5f 63 6f 75 6e 74 65 72 nx('stat_counter
0a70: 5f 70 65 6e 64 69 6e 67 27 2c 20 27 73 74 61 74 _pending', 'stat
0a80: 5f 63 6f 75 6e 74 65 72 5f 70 65 6e 64 69 6e 67 _counter_pending
0a90: 5f 74 6d 70 27 29 0a 72 65 64 3a 65 78 65 63 28 _tmp').red:exec(
0aa0: 29 0a 72 65 6e 61 6d 65 64 2c 20 65 72 72 20 3d ).renamed, err =
0ab0: 20 72 65 64 3a 63 6f 6d 6d 69 74 5f 70 69 70 65 red:commit_pipe
0ac0: 6c 69 6e 65 28 29 0a 69 66 20 6e 6f 74 20 72 65 line().if not re
0ad0: 6e 61 6d 65 64 20 74 68 65 6e 0a 09 6e 67 78 2e named then..ngx.
0ae0: 6c 6f 67 28 6e 67 78 2e 45 52 52 2c 20 27 72 65 log(ngx.ERR, 're
0af0: 64 69 73 20 6d 75 6c 74 69 20 66 61 69 6c 65 64 dis multi failed
0b00: 3a 20 27 2c 20 65 72 72 29 0a 09 6e 67 78 2e 65 : ', err)..ngx.e
0b10: 78 69 74 28 35 30 30 29 0a 65 6e 64 0a 0a 6c 6f xit(500).end..lo
0b20: 63 61 6c 20 63 6f 6e 74 69 6e 75 65 20 3d 20 66 cal continue = f
0b30: 61 6c 73 65 0a 69 66 20 74 6f 6e 75 6d 62 65 72 alse.if tonumber
0b40: 28 72 65 6e 61 6d 65 64 5b 35 5d 5b 33 5d 29 20 (renamed[5][3])
0b50: 3d 3d 20 31 20 74 68 65 6e 0a 09 63 6f 6e 74 69 == 1 then..conti
0b60: 6e 75 65 20 3d 20 74 72 75 65 0a 65 6e 64 0a 0a nue = true.end..
0b70: 77 68 69 6c 65 20 63 6f 6e 74 69 6e 75 65 20 64 while continue d
0b80: 6f 0a 09 6e 67 78 2e 6c 6f 63 61 74 69 6f 6e 2e o..ngx.location.
0b90: 63 61 70 74 75 72 65 28 27 2f 73 6c 65 65 70 27 capture('/sleep'
0ba0: 29 0a 0a 09 6c 6f 63 61 6c 20 64 61 74 61 0a 09 )...local data..
0bb0: 64 61 74 61 2c 20 65 72 72 20 3d 20 72 65 64 3a data, err = red:
0bc0: 68 67 65 74 61 6c 6c 28 27 73 74 61 74 5f 63 6f hgetall('stat_co
0bd0: 75 6e 74 65 72 5f 70 65 6e 64 69 6e 67 5f 74 6d unter_pending_tm
0be0: 70 27 29 0a 09 69 66 20 6e 6f 74 20 64 61 74 61 p')..if not data
0bf0: 20 74 68 65 6e 0a 09 09 6e 67 78 2e 6c 6f 67 28 then...ngx.log(
0c00: 6e 67 78 2e 45 52 52 2c 20 27 72 65 64 69 73 20 ngx.ERR, 'redis
0c10: 72 65 71 75 65 73 74 20 66 61 69 6c 65 64 3a 20 request failed:
0c20: 27 2c 20 65 72 72 29 0a 09 09 6e 67 78 2e 65 78 ', err)...ngx.ex
0c30: 69 74 28 35 30 30 29 0a 09 65 6e 64 0a 0a 09 66 it(500)..end...f
0c40: 6f 72 20 6e 61 6d 65 2c 20 76 61 6c 75 65 20 69 or name, value i
0c50: 6e 20 70 61 69 72 73 28 64 65 63 6f 64 65 5f 62 n pairs(decode_b
0c60: 75 6c 6b 28 64 61 74 61 29 29 20 64 6f 0a 09 09 ulk(data)) do...
0c70: 6c 6f 63 61 6c 20 66 69 65 6c 64 73 20 3d 20 7b local fields = {
0c80: 7d 0a 09 09 6e 61 6d 65 3a 67 73 75 62 28 22 5b }...name:gsub("[
0c90: 5e 5f 5d 2b 22 2c 20 66 75 6e 63 74 69 6f 6e 28 ^_]+", function(
0ca0: 63 29 20 66 69 65 6c 64 73 5b 23 66 69 65 6c 64 c) fields[#field
0cb0: 73 20 2b 20 31 5d 20 3d 20 63 20 65 6e 64 29 0a s + 1] = c end).
0cc0: 09 09 66 69 65 6c 64 73 5b 23 66 69 65 6c 64 73 ..fields[#fields
0cd0: 20 2b 20 31 5d 20 3d 20 76 61 6c 75 65 0a 09 09 + 1] = value...
0ce0: 6c 6f 63 61 6c 20 72 65 73 75 6c 74 20 3d 20 6e local result = n
0cf0: 67 78 2e 6c 6f 63 61 74 69 6f 6e 2e 63 61 70 74 gx.location.capt
0d00: 75 72 65 28 27 2f 70 6f 73 74 67 72 65 73 27 2c ure('/postgres',
0d10: 20 7b 0a 09 09 09 6d 65 74 68 6f 64 20 3d 20 6e {....method = n
0d20: 67 78 2e 48 54 54 50 5f 50 55 54 2c 0a 09 09 09 gx.HTTP_PUT,....
0d30: 2d 2d 20 70 72 65 70 61 72 65 20 73 65 6c 65 63 -- prepare selec
0d40: 74 3f 20 58 58 58 0a 09 09 09 62 6f 64 79 20 3d t? XXX....body =
0d50: 20 22 73 65 6c 65 63 74 20 6d 65 72 67 65 5f 63 "select merge_c
0d60: 6f 75 6e 74 65 72 28 27 22 20 2e 2e 20 66 69 65 ounter('" .. fie
0d70: 6c 64 73 5b 31 5d 20 2e 2e 20 22 27 3a 3a 74 65 lds[1] .. "'::te
0d80: 78 74 2c 20 27 22 20 2e 2e 20 66 69 65 6c 64 73 xt, '" .. fields
0d90: 5b 32 5d 20 2e 2e 20 22 27 3a 3a 74 65 78 74 2c [2] .. "'::text,
0da0: 20 27 22 20 2e 2e 20 66 69 65 6c 64 73 5b 33 5d '" .. fields[3]
0db0: 20 2e 2e 20 22 27 3a 3a 69 6e 65 74 2c 20 27 22 .. "'::inet, '"
0dc0: 20 2e 2e 20 66 69 65 6c 64 73 5b 34 5d 20 2e 2e .. fields[4] ..
0dd0: 20 22 27 3a 3a 73 6d 61 6c 6c 69 6e 74 29 3b 22 "'::smallint);"
0de0: 0a 09 09 7d 29 0a 09 09 69 66 20 72 65 73 75 6c ...})...if resul
0df0: 74 2e 73 74 61 74 75 73 20 7e 3d 20 32 30 30 20 t.status ~= 200
0e00: 6f 72 20 6e 6f 74 20 72 65 73 75 6c 74 2e 62 6f or not result.bo
0e10: 64 79 20 74 68 65 6e 0a 09 09 09 6e 67 78 2e 6c dy then....ngx.l
0e20: 6f 67 28 6e 67 78 2e 45 52 52 2c 20 27 70 6f 73 og(ngx.ERR, 'pos
0e30: 74 67 72 65 73 20 61 63 63 65 73 73 20 66 61 69 tgres access fai
0e40: 6c 65 64 27 29 0a 09 09 09 6e 67 78 2e 65 78 69 led')....ngx.exi
0e50: 74 28 35 30 30 29 0a 09 09 65 6e 64 0a 09 65 6e t(500)...end..en
0e60: 64 0a 0a 09 72 65 64 3a 69 6e 69 74 5f 70 69 70 d...red:init_pip
0e70: 65 6c 69 6e 65 28 35 29 0a 09 72 65 64 3a 6d 75 eline(5)..red:mu
0e80: 6c 74 69 28 29 0a 09 72 65 64 3a 64 65 6c 28 27 lti()..red:del('
0e90: 73 74 61 74 5f 63 6f 75 6e 74 65 72 5f 70 65 6e stat_counter_pen
0ea0: 64 69 6e 67 5f 74 6d 70 27 29 0a 09 72 65 64 3a ding_tmp')..red:
0eb0: 65 78 69 73 74 73 28 27 73 74 61 74 5f 63 6f 75 exists('stat_cou
0ec0: 6e 74 65 72 5f 70 65 6e 64 69 6e 67 27 29 0a 09 nter_pending')..
0ed0: 72 65 64 3a 72 65 6e 61 6d 65 6e 78 28 27 73 74 red:renamenx('st
0ee0: 61 74 5f 63 6f 75 6e 74 65 72 5f 70 65 6e 64 69 at_counter_pendi
0ef0: 6e 67 27 2c 20 27 73 74 61 74 5f 63 6f 75 6e 74 ng', 'stat_count
0f00: 65 72 5f 70 65 6e 64 69 6e 67 5f 74 6d 70 27 29 er_pending_tmp')
0f10: 0a 09 72 65 64 3a 65 78 65 63 28 29 0a 09 72 65 ..red:exec()..re
0f20: 6e 61 6d 65 64 2c 20 65 72 72 20 3d 20 72 65 64 named, err = red
0f30: 3a 63 6f 6d 6d 69 74 5f 70 69 70 65 6c 69 6e 65 :commit_pipeline
0f40: 28 29 0a 09 69 66 20 6e 6f 74 20 72 65 6e 61 6d ()..if not renam
0f50: 65 64 20 74 68 65 6e 0a 09 09 6e 67 78 2e 6c 6f ed then...ngx.lo
0f60: 67 28 6e 67 78 2e 45 52 52 2c 20 27 72 65 64 69 g(ngx.ERR, 'redi
0f70: 73 20 6d 75 6c 74 69 20 66 61 69 6c 65 64 3a 20 s multi failed:
0f80: 27 2c 20 65 72 72 29 0a 09 09 6e 67 78 2e 65 78 ', err)...ngx.ex
0f90: 69 74 28 35 30 30 29 0a 09 65 6e 64 0a 0a 09 69 it(500)..end...i
0fa0: 66 20 74 6f 6e 75 6d 62 65 72 28 72 65 6e 61 6d f tonumber(renam
0fb0: 65 64 5b 35 5d 5b 31 5d 20 3d 3d 20 30 29 20 6f ed[5][1] == 0) o
0fc0: 72 20 74 6f 6e 75 6d 62 65 72 28 72 65 6e 61 6d r tonumber(renam
0fd0: 65 64 5b 35 5d 5b 32 5d 29 20 3d 3d 20 30 20 74 ed[5][2]) == 0 t
0fe0: 68 65 6e 0a 09 09 63 6f 6e 74 69 6e 75 65 20 3d hen...continue =
0ff0: 20 66 61 6c 73 65 0a 09 65 6e 64 0a 65 6e 64 0a false..end.end.