summaryrefslogtreecommitdiff
path: root/src/daq/nginx-lua/s5-nginx-fetch.lua
blob: 882109431aa38926d67b9f1bcbc5a100ff188f7b (plain) (blame)
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
--
-- 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
--
-- location /sfive {
--   allow 127.0.0.1;
--   allow ::1;
--   deny  all;
--
--   content_by_lua_file '/path/to/s5-nginx-fetch.lua';
-- }
--

local lock = locks:new("sfive_locks", { exptime = nil, timeout = 0.001 })
local ok, err = lock:lock("fetch")
if not ok then
   if err == "timeout" then
      ngx.exit(ngx.HTTP_SERVICE_UNAVAILABLE)
   else
      ngx.log(ngx.ERR, "SFive(fetch): failed to acquire fetch lock: " .. err)
      ngx.exit(ngx.HTTP_INTERNAL_SERVER_ERROR)
   end
else
   ngx.log(ngx.INFO, "SFive(fetch): client connected")
   local sfive_data = ngx.shared.sfive_data
   while true do
      logs = sfive_data:get_keys()
      for i, k in ipairs(logs) do
         local ok, err = ngx.say(sfive_data:get(k))
         if not ok then
            ngx.log(ngx.ERR, "SFive(fetch): failed to send data set: ", err)
            break
         end
         sfive_data:delete(k)
      end
      ngx.flush()
      ngx.sleep(0.1)
   end
   ok, err = lock:unlock("fetch")
   if not ok then
      ngx.log(ngx.ERR, "SFive(fetch): unlock error: " .. err)
   end
end