MT#11005 add dlglist class

Change-Id: I1f3fe3df48c39e2aad3222431c4b518ce16094ff

ngcp/dlglist.lua

@@ -0,0 +1,159 @@
+--
+-- Copyright 2015 SipWise Team <development@sipwise.com>
+--
+-- This program is free software: you can redistribute it and/or modify
+-- it under the terms of the GNU General Public License as published by
+-- the Free Software Foundation, either version 3 of the License, or
+-- (at your option) any later version.
+--
+-- This package 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 this program. If not, see <http://www.gnu.org/licenses/>.
+-- .
+-- On Debian systems, the complete text of the GNU General
+-- Public License version 3 can be found in "/usr/share/common-licenses/GPL-3".
+--
+local NGCPDlgList = {
+     __class__ = 'NGCPDlgList'
+}
+local redis = require 'redis';
+require 'ngcp.utils';
+
+_ENV = NGCPDlgList
+
+-- class NGCPDlgList
+local NGCPDlgList_MT = { __index = NGCPDlgList }
+
+NGCPDlgList_MT.__tostring = function (t)
+    return string.format("config:%s central:%s pair:%s",
+        table.tostring(t.config), table.tostring(t.central),
+        table.tostring(t.pair));
+end
+
+    function NGCPDlgList.new()
+        local t = NGCPDlgList.init();
+        setmetatable( t, NGCPDlgList_MT );
+        return t;
+    end
+
+    function NGCPDlgList.init()
+        local t = {
+            config = {
+                central = {
+                    host = '127.0.0.1',
+                    port = 6379,
+                    db = "3"
+                },
+                pair = {
+                    host = '127.0.0.1',
+                    port = 6379,
+                    db = "4"
+                },
+                check_pair_dup = false
+            },
+            central = {},
+            pair = {}
+        };
+        return t;
+    end
+
+    local function _test_connection(client)
+        if not client then return nil end
+        local ok, err = pcall(client.ping, client);
+        return ok
+    end
+
+    local function _connect(config)
+        local client = redis.connect(config.host,config.port);
+        client:select(config.db);
+        sr.log("dbg", string.format("connected to redis server %s:%d at %s\n",
+            config.host, config.port, config.db));
+        return client;
+    end
+
+    function NGCPDlgList._del(self, key, callid)
+        self.central:lrem(key, 0, callid);
+        local num = self.central:llen(key);
+        if num == 0 then
+            self.central:del(key);
+            sr.log("dbg", string.format("central[%s] is empty. Removed\n", key));
+        else
+            sr.log("dbg", string.format("central:lrem[%s]=>[%s]\n", key, tostring(num)));
+        end
+        return num;
+    end
+
+    function NGCPDlgList:exists(callid)
+        if not _test_connection(self.pair) then
+            self.pair = _connect(self.config.pair);
+        end
+        local res = self.pair:llen("list:"..callid)
+        if res > 0 then
+            return true
+        else
+            return false
+        end
+    end
+
+    function NGCPDlgList:is_in_set(callid, key)
+        if not _test_connection(self.pair) then
+            self.pair = _connect(self.config.pair);
+        end
+        local res = self.pair:lrange("list:"..callid, 0, -1);
+        return table.contains(res, key);
+    end
+
+    function NGCPDlgList:add(callid, key)
+        if not _test_connection(self.central) then
+            self.central = _connect(self.config.central);
+        end
+        local pos = self.central:rpush(key, callid);
+        sr.log("dbg", string.format("central:rpush[%s]=>[%s] %s\n", key, callid, tostring(pos)));
+        if not _test_connection(self.pair) then
+            self.pair = _connect(self.config.pair);
+        end
+        if self.config.check_pair_dup and self:is_in_set(callid, key) then
+            sr.log("warn", string.format("pair:check_pair_dup[%s]=>[%s] already there!\n", callid, key));
+        end
+        pos = self.pair:lpush("list:"..callid, key);
+        sr.log("dbg", string.format("pair:lpush[list:%s]=>[%s] %s\n", callid, key, tostring(pos)));
+    end
+
+    function NGCPDlgList:del(callid, key)
+        if not _test_connection(self.pair) then
+            self.pair = _connect(self.config.pair);
+        end
+        local num = self.pair:lrem(callid, 0, key);
+        sr.log("dbg", string.format("pair:lrem[%s]=>[%s] %d\n", callid, key, num));
+        if not _test_connection(self.central) then
+            self.central = _connect(self.config.central);
+        end
+        self:_del(key, callid);
+    end
+
+    function NGCPDlgList:destroy(callid)
+        if not _test_connection(self.pair) then
+            self.pair = _connect(self.config.pair);
+        end
+        local key = self.pair:lpop("list:"..callid);
+        if not key then
+            self.pair:del("list:"..callid);
+            error(string.format("callid:%s list empty", callid));
+        end
+        if not _test_connection(self.central) then
+            self.central = _connect(self.config.central);
+        end
+        while key do
+            self:_del(key, callid);
+            sr.log("dbg", string.format("pair:lpop[%s]=>[%s]\n", callid, key));
+            key = self.pair:lpop("list:"..callid);
+        end
+    end
+-- class
+
+return NGCPDlgList
+--EOF