lua-observer-watcher/Observer.lua at master · zhaolihang/lua-observer-watcher · GitHub

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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
local Dep = import('.Dep',...).Dep;

local function observe(original)

    if type(original) ~= 'table' then
        return original;
    end

    if rawget(original,'__is_proxy__') then
        return original;
    end

    if rawget(original,'__proxy__') then
        return rawget(original,'__proxy__');
    end

    local deps = {};
    local function getDep(k)
        if deps[k] == nil then
            deps[k] = Dep.new();
        end
        return deps[k];
    end
    for k,v in pairs(original) do
        getDep(k);
        original[k] = observe(v);
    end

    local proxy = {};
    local tableDep = Dep.new();
    rawset(proxy, '__is_proxy__', true);
    rawset(proxy, '__dep__', tableDep);
    rawset(original, '__proxy__', proxy);

    function proxy:getn()
        return #original;
    end

    function proxy:pairs(fun)
        for k,v in pairs(original) do
            if v ~= proxy then
                if fun(k,v) then
                    break;
                end
            end
        end
    end

    function proxy:ipairs(fun)
        for k,v in ipairs(original) do
            if fun(k,v) then
                break;
            end
        end
    end

    function proxy:copy()
        local ret = {};
        self:pairs(function(k,v)
            if type(v) == 'table' then
                ret[k] = v:copy();
            else
                ret[k] = v;
            end
        end);
        return ret;
    end

    function proxy:insert(...) -- 数组添加
        local args = {...};
        if #args==1 then
            table.insert(original,observe(args[1])); -- value
        else
            table.insert(original,args[1],observe(args[2]));-- pos,value
        end
        tableDep:notify();
    end
    function proxy:remove(...) -- 数组移除
        local ret = table.remove(original,...);
        tableDep:notify();
        return ret;
    end
    function proxy:sort(...) -- 数组排序
        table.sort(original,...);
        tableDep:notify();
    end

    -- core
    local __index = function(t,k) -- getter
        if Dep.target then
            getDep(k):depend();
            if type(original[k]) == 'table' then
                local childTableDep = rawget(original[k], '__dep__');
                childTableDep:depend();
            end
        end
        return original[k];
    end
    local __newindex = function(t,k,v) -- setter
        local oldValue = original[k];
        local newValue = observe(v);
        if newValue ~= oldValue then
            original[k] = newValue;
            getDep(k):notify();
            tableDep:notify();
        end
    end
    local metatable = { __index = __index; __newindex = __newindex; }
    setmetatable(proxy,metatable);

    return proxy;
end


return {
    observe = observe;
};