3335 物品相关类型重构

client_Wu Xijin

2019-02-20 a87120c155c48fa45b20a97c1a58bdbeb77318b7

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
74
75
76
77
78
79
80
-- Tencent is pleased to support the open source community by making xLua available.
-- Copyright (C) 2016 THL A29 Limited, a Tencent company. All rights reserved.
-- Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at
-- http://opensource.org/licenses/MIT
-- Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.
 
require "libtdrlua"
local m = {}
for k, v in pairs(libtdrlua) do m[k] = v end
local load_metalib, load_metalib_buf, free_metalib, get_meta, table2buf, buf2table, str2table, metamaxbufsize, bufalloc, buffree, buf2str
load_metalib, m.load_metalib = m.load_metalib, nil
load_metalib_buf, m.load_metalib_buf = m.load_metalib_buf, nil
free_metalib, m.free_metalib = m.free_metalib, nil
get_meta, m.get_meta = m.get_meta, nil
table2buf, m.table2buf = m.table2buf, nil
buf2table, m.buf2table = m.buf2table, nil
str2table, m.str2table = m.str2table, nil
buf2str, m.buf2str = m.buf2str, nil
 
metamaxbufsize, m.metamaxbufsize = m.metamaxbufsize, nil
bufalloc, m.bufalloc = m.bufalloc, nil
buffree, m.buffree = m.buffree, nil
 
local function create_msg_pk(meta, buf, buf_size)
    return {
        buff = buf,
        pack = function(obj)
            local ret_code, used_size = table2buf(meta, obj, buf, buf_size, 0)
            if ret_code ~= 0 then
                return ret_code, used_size
            end
            return buf2str(buf, used_size)
        end,
        unpack = function(str)
            return libtdrlua.str2table(meta, str, 0)
        end
    }
end
 
local function create_lib(metalib)
    return setmetatable({}, {
        __index = function(obj, k)
            local ret_code, meta = libtdrlua.get_meta(metalib, k)
            if ret_code ~= 0 then
                error("libtdrlua.get_meta() failed: errno=".. ret_code .. ",msg=" .. meta)
            end
            local ret_code, buf_size = libtdrlua.metamaxbufsize(metalib, k)
            if ret_code ~= 0 then
                error("libtdrlua.metamaxbufsize() failed: errno=".. ret_code .. ",msg=" .. buf_size)
            end
            
            local ret_code, buf = libtdrlua.bufalloc(buf_size)
            if ret_code ~= 0 then
                error("libtdrlua.bufalloc() failed: errno=".. ret_code .. ",msg=" .. buf)
            end
    
            local pk = create_msg_pk(meta, buf, buf_size)
            rawset(obj, k, pk)
            return pk
        end
    })
end
 
function m.from_file(file)
    local ret_code, metalib = libtdrlua.load_metalib(file)
    if ret_code ~= 0 then
        error("libtdrlua.load_metalib() failed: " .. metalib)
    end
    return create_lib(metalib)
end
 
function m.from_memory(str)
    local ret_code, metalib = libtdrlua.load_metalib_buf(str)
    if ret_code ~= 0 then
        error("libtdrlua.load_metalib_buf() failed: errno=".. ret_code .. ",msg=" .. metalib)
    end
    return create_lib(metalib)
end
 
return m