mirror of
https://github.com/fusionpbx/fusionpbx.git
synced 2026-01-01 17:53:49 +00:00
5afec68fb1
```Lua
local Database = require 'resources.functions.database'
-- Default backend configured via xml_handler.db_backend = 'native'
dbh = Database.new("system")
-- To use other backends you can use
dbh = Database.backend.luasql("system")
```
82 lines
1.9 KiB
Lua
82 lines
1.9 KiB
Lua
--
|
|
-- Lua-ODBC-Pool backend to FusionPBX database class
|
|
--
|
|
|
|
local log = require "resources.functions.log".database
|
|
local odbc = require "odbc.dba"
|
|
local odbcpool = require "odbc.dba.pool"
|
|
|
|
local function remove_null(row, null, null_value)
|
|
local o = {}
|
|
for k, v in pairs(row) do
|
|
if v == null then
|
|
o[k] = null_value
|
|
else
|
|
o[k] = tostring(v)
|
|
end
|
|
end
|
|
return o
|
|
end
|
|
|
|
-----------------------------------------------------------
|
|
local OdbcPoolDatabase = {} do
|
|
OdbcPoolDatabase.__index = OdbcPoolDatabase
|
|
OdbcPoolDatabase._backend_name = 'ODBC Pool'
|
|
|
|
function OdbcPoolDatabase.new(name)
|
|
local self = setmetatable({}, OdbcPoolDatabase)
|
|
self._cli = odbcpool.client(name)
|
|
self._timeout = 1000
|
|
self._rows_affected = nil
|
|
return self
|
|
end
|
|
|
|
function OdbcPoolDatabase:query(sql, fn)
|
|
self._rows_affected = nil
|
|
local cli = self._cli
|
|
|
|
local ok, err
|
|
if fn then
|
|
ok, err = cli:acquire(self._timeout, function(dbh)
|
|
local ok, err = dbh:neach(sql, function(row)
|
|
return fn(remove_null(row, odbc.NULL, ""))
|
|
end)
|
|
if err and not ok then
|
|
log.errf("Can not execute sql: %s\n%s", tostring(err), sql)
|
|
end
|
|
return not not dbh:connected(), true
|
|
end)
|
|
else
|
|
ok, err = cli:acquire(self._timeout, function(dbh)
|
|
local ok, err = dbh:exec(sql)
|
|
if err and not ok then
|
|
log.errf("Can not execute sql: %s\n%s", tostring(err), sql)
|
|
end
|
|
self._rows_affected = ok
|
|
return not not dbh:connected(), ok
|
|
end)
|
|
end
|
|
|
|
if err and not ok then
|
|
log.errf("Can not get database handle: %s", tostring(err))
|
|
end
|
|
|
|
return ok
|
|
end
|
|
|
|
function OdbcPoolDatabase:affected_rows()
|
|
return self._rows_affected;
|
|
end
|
|
|
|
function OdbcPoolDatabase:release()
|
|
self._cli = nil
|
|
end
|
|
|
|
function OdbcPoolDatabase:connected()
|
|
return not not self._cli
|
|
end
|
|
|
|
end
|
|
-----------------------------------------------------------
|
|
|
|
return OdbcPoolDatabase |