FilesExpand file tree

 master

/

rpc.lua

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

96 lines (87 loc) · 2.2 KB

 master

/

rpc.lua

Copy path

Top

File metadata and controls

• Code
• Blame

96 lines (87 loc) · 2.2 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

--[[

rpc连接只能发送和接收CSocket.rpkdecoder()格式的封包

]]--

local cjson = require "cjson"

local Sche = require "lua.sche"

local CMD_RPC_CALL = 0xABCDDBCA

local CMD_RPC_RESP = 0xDBCAABCD

local function RPC_Process_Call(app,s,rpk)

local str = rpk:Read_string()

local request = cjson.decode(str)

local funname = request.f

local co = request.co

local func = app._RPCService[funname]

local response = {co = co}

if not func then

response.err = funname .. " not found"

else

local ret = table.pack(pcall(func,s,table.unpack(request.arg)))

if ret[1] then

table.remove(ret,1)

response.ret = {table.unpack(ret)}

else

response.err = ret[2]

CLog.SysLog(CLog.LOG_ERROR,string.format("rpc process error:%s",ret[2]))

end

end

local wpk = CPacket.NewWPacket(512)--Packet.WPacket.New(512)

wpk:Write_uint32(CMD_RPC_RESP)

wpk:Write_string(cjson.encode(response))

s:Send(wpk)

end

local function RPC_Process_Response(s,rpk)

local str = rpk:Read_string()

local response = cjson.decode(str)

local co = Sche.GetCoByIdentity(response.co)

if co then

co.response = response

s.pending_rpc[co.identity] = nil

Sche.WakeUp(co)--Schedule(co)

end

end

local rpcCaller = {}

function rpcCaller:new(s,funcname)

local o = {}

self.__index = self

setmetatable(o,self)

o.s = s

o.funcname = funcname

return o

end

function rpcCaller:Call(...)

local request = {}

local co = Sche.Running()

request.f = self.funcname

request.co = co.identity

request.arg = {...}

local wpk = CPacket.NewWPacket(512)

wpk:Write_uint32(CMD_RPC_CALL)

wpk:Write_string(cjson.encode(request))

local ret = self.s:Send(wpk)

if ret then

return "socket error"

else

if not self.s.pending_rpc then

self.s.pending_rpc = {}

end

self.s.pending_rpc[co.identity] = co

Sche.Block()

local response = co.response

co.response = nil

if not response.ret then

return response.err,nil

else

return response.err,table.unpack(response.ret)

end

end

end

local function RPC_MakeCaller(s,funcname)

return rpcCaller:new(s,funcname)

end

return {

ProcessCall = RPC_Process_Call,

ProcessResponse = RPC_Process_Response,

MakeRPC = RPC_MakeCaller,

CMD_RPC_CALL = CMD_RPC_CALL,

CMD_RPC_RESP = CMD_RPC_RESP,

}