src/daq/nginx-lua/s5-nginx-init.lua


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

--
-- sfive
--
-- sfive - spreadspace streaming statistics suite is a generic
-- statistic collection tool for streaming server infrastuctures.
-- The system collects and stores meta data like number of views
-- and throughput from a number of streaming servers and stores
-- it in a global data store.
-- The data acquisition is designed to be generic and extensible in
-- order to support different streaming software.
-- sfive also contains tools and applications to filter and visualize
-- live and recorded data.
--
--
-- Copyright (C) 2014 Christian Pointner <equinox@spreadspace.org>
--                    Markus Grueneis <gimpf@gimpf.org>
--
-- This file is part of sfive.
--
-- sfive is free software: you can redistribute it and/or modify
-- it under the terms of the GNU General Public License version 3
-- as published by the Free Software Foundation.
--
-- sfive is distributed in the hope that it will be useful,
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-- GNU General Public License for more details.
--
-- You should have received a copy of the GNU General Public License
-- along with sfive. If not, see <http://www.gnu.org/licenses/>.
--
--
-- Install this by adding the following to your nginx.conf
--
-- http {
--   lua_shared_dict sfive 64k;
--   lua_shared_dict sfive_locks 64k;
--   lua_shared_dict sfive_data 64m;
--   init_by_lua_file '/path/to/s5-nginx-init.lua';
-- }
--

ngx.log(ngx.DEBUG, "SFive: loaded")

locks = require "resty.lock"

local sfive = ngx.shared.sfive
sfive:flush_all()
sfive:flush_expired()
local sfive_data = ngx.shared.sfive_data
sfive_data:flush_all()
sfive_data:flush_expired()

-- try to create locks so we get an error on init and not on first usage
local lock = locks:new("sfive_locks")
local elapsed, err = lock:lock("test")
if not elapsed then
   ngx.log(ngx.ERR, "SFive(init): lock error: " .. err)
else
   local ok, err, force = sfive:set("log:idx", 0)
   if not ok then
      ngx.log(ngx.ERR, "SFive(init): creating log index counter failed: " .. err)
   else
      ngx.log(ngx.INFO, "SFive(init): initialized successfully!")
   end
   ok, err = lock:unlock("test")
   if not ok then
      ngx.log(ngx.ERR, "SFive(init): unlock error: " .. err)
   end
end