Blame view

publish/skynet/lualib/skynet/sharemap.lua 1.47 KB
4d6f285d   zhouhaihai   增加发布功能
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
  local stm = require "skynet.stm"
  local sprotoloader = require "sprotoloader"
  local sproto = require "sproto"
  local setmetatable = setmetatable
  
  local sharemap = {}
  
  function sharemap.register(protofile)
  	-- use global slot 0 for type define
  	sprotoloader.register(protofile, 0)
  end
  
  local sprotoobj
  local function loadsp()
  	if sprotoobj == nil then
  		sprotoobj = sprotoloader.load(0)
  	end
  	return sprotoobj
  end
  
  function sharemap:commit()
  	self.__obj(sprotoobj:encode(self.__typename, self.__data))
  end
  
  function sharemap:copy()
  	return stm.copy(self.__obj)
  end
  
  function sharemap.writer(typename, obj)
  	local sp = loadsp()
  	obj = obj or {}
  	local stmobj = stm.new(sp:encode(typename,obj))
  	local ret = {
  		__typename = typename,
  		__obj = stmobj,
  		__data = obj,
  		commit = sharemap.commit,
  		copy = sharemap.copy,
  	}
  	return setmetatable(ret, { __index = obj, __newindex = obj })
  end
  
  local function decode(msg, sz, self)
  	local data = self.__data
  	for k in pairs(data) do
  		data[k] = nil
  	end
  	return sprotoobj:decode(self.__typename, msg, sz, data)
  end
  
  function sharemap:update()
  	return self.__obj(decode, self)
  end
  
  function sharemap.reader(typename, stmcpy)
  	local sp = loadsp()
  	local stmobj = stm.newcopy(stmcpy)
  	local _, data = stmobj(function(msg, sz)
  		return sp:decode(typename, msg, sz)
  	end)
  
  	local obj = {
  		__typename = typename,
  		__obj = stmobj,
  		__data = data,
  		update = sharemap.update,
  	}
  	return setmetatable(obj, { __index = data, __newindex = error })
  end
  
  return sharemap