skynet的ssdb驱动

以下代码由PHP站长网 52php.cn收集自互联网

现在PHP站长网小编把它分享给大家，仅供参考

local skynet = require "skynet"
local socket = require "socket"
local socketchannel = require "socketchannel"
local int64 = require "int64"

local ssdb = {}
local command = {}
local meta = {
    __index = command,-- DO NOT close channel in __gc
}

local function read_block( fd )
    local result = fd:readline "n"
    -- print('read_block result',result,string.len(result))
    local len = tonumber(result)
    -- print('read_block result len',len)
    if not len  then
        return false
    elseif len == 0 then 
        local result = fd:readline "nn"
        return false
    end

    local value = fd:read(len+1)
    -- print('read_block value',value)
    local data = string.sub(value,1,-2)
    -- print('read_block data',data)
    -- print('data',data)
    return true,data
end


local function read_response(fd)
    local ok,result = read_block(fd)
    -- print('read_response read_block result',ok,result)
    if ok then
        if result~='ok' then
            read_block(fd)
            
            -- print('read_response')
            return true
        end
    end
    
    local bulk = {}
    local ok,d = read_block(fd)
    while ok do

        table.insert(bulk,d)

        ok,d = read_block(fd)
        if not ok then
            break
        end
    end
    
    if #bulk<=1 then
        -- print('read_response')
        return true,bulk[1]
    end
    -- print('read_response')
    return true,bulk
end

local function pack_value(lines,v)
    if v == nil then
        return
    end

    local t = type(v)
    if t == "number" then
        v = tostring(v)
    elseif t == "userdata" then
        v = int64.tostring(int64.new(v),10)
    end
    table.insert(lines,#v)
    table.insert(lines,v)
end

local function compose_message(cmd,msg)
    local t = type(msg)

    local lines = {}
    pack_value(lines,cmd)

    if t == "table" then
        for _,v in ipairs(msg) do
            pack_value(lines,v)
        end
    else
        pack_value(lines,msg)
    end

    local chunk =  table.concat(lines,"n")..'nn'
    -- print("chunk",chunk)
    return chunk
end

setmetatable(command,{ __index = function(t,k)
    local cmd = string.lower(k)
    local f = function (self,v,...)
        if type(v) == "table" then
            return self[1]:request(compose_message(cmd,v),read_response)
        else
            local s = compose_message(cmd,{v,...})
            return self[1]:request(s,read_response)
        end
    end
    t[k] = f
    return f
end})


function ssdb.connect(db_conf)
    local channel = socketchannel.channel {
        host = db_conf.host,port = db_conf.port or 8888,}
    -- try connect first only once
    channel:connect(true)
    return setmetatable( { channel },meta )
end


return ssdb

以上内容由PHP站长网【52php.cn】收集整理供大家参考研究

如果以上内容对您有帮助，欢迎收藏、点赞、推荐、分享。

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!