example for sproto

examples/agent.lua

@@ -4,7 +4,8 @@ local socket = require "socket"
 local sproto = require "sproto"
 local bit32 = require "bit32"
 
-local rpc
+local host
+local send_request
 
 local CMD = {}
 local REQUEST = {}
@@ -19,11 +20,10 @@ end
 function REQUEST:set()
 	print("set", self.what, self.value)
 	local r = skynet.call("SIMPLEDB", "lua", "set", self.what, self.value)
-	return { ok = true }
 end
 
 function REQUEST:handshake()
-	return { msg = "Welcome to skynet" }
+	return { msg = "Welcome to skynet, I will send heartbeat every 5 sec." }
 end
 
 local function request(name, args, response)
@@ -47,7 +47,7 @@ skynet.register_protocol {
 	name = "client",
 	id = skynet.PTYPE_CLIENT,
 	unpack = function (msg, sz)
-		return rpc:dispatch(msg, sz)
+		return host:dispatch(msg, sz)
 	end,
 	dispatch = function (_, _, type, ...)
 		if type == "REQUEST" then
@@ -67,7 +67,15 @@ skynet.register_protocol {
 }
 
 function CMD.start(gate, fd, proto)
-	rpc = sproto.new(proto):rpc "package"
+	host = sproto.new(proto.c2s):host "package"
+	send_request = host:attach(sproto.new(proto.s2c))
+	skynet.fork(function()
+		while true do
+			send_package(send_request "heartbeat")
+			skynet.sleep(500)
+		end
+	end)
+
 	client_fd = fd
 	skynet.call(gate, "lua", "forward", fd)
 end

examples/client.lua

@@ -6,7 +6,8 @@ local bit32 = require "bit32"
 local proto = require "proto"
 local sproto = require "sproto"
 
-local rpc = sproto.new(proto):rpc "package"
+local host = sproto.new(proto.s2c):host "package"
+local request = host:attach(sproto.new(proto.c2s))
 
 local fd = assert(socket.connect("127.0.0.1", 8888))
 
@@ -52,13 +53,40 @@ local session = 0
 
 local function send_request(name, args)
 	session = session + 1
-	local str = rpc:request(name, args, session)
+	local str = request(name, args, session)
 	send_package(fd, str)
 	print("Request:", session)
 end
 
 local last = ""
 
+local function print_request(name, args)
+	print("REQUEST", name)
+	if args then
+		for k,v in pairs(args) do
+			print(k,v)
+		end
+	end
+end
+
+local function print_response(session, args)
+	print("RESPONSE", session)
+	if args then
+		for k,v in pairs(args) do
+			print(k,v)
+		end
+	end
+end
+
+local function print_package(t, ...)
+	if t == "REQUEST" then
+		print_request(...)
+	else
+		assert(t == "RESPONSE")
+		print_response(...)
+	end
+end
+
 local function dispatch_package()
 	while true do
 		local v
@@ -67,16 +95,12 @@ local function dispatch_package()
 			break
 		end
 
-		local t, session, response = rpc:dispatch(v)
-		assert(t == "RESPONSE" , "This example only support request , so here must be RESPONSE")
-		print("response session", session)
-		for k,v in pairs(response) do
-			print(k,v)
-		end
+		print_package(host:dispatch(v))
 	end
 end
 
 send_request("handshake")
+send_request("set", { what = "hello", value = "world" })
 while true do
 	dispatch_package()
 	local cmd = socket.readstdin()

examples/proto.lua

@@ -1,6 +1,8 @@
 local sprotoparser = require "sprotoparser"
 
-local proto = sprotoparser.parse [[
+local proto = {}
+
+proto.c2s = sprotoparser.parse [[
 .package {
 	type 0 : integer
 	session 1 : integer
@@ -17,7 +19,7 @@ get 2 {
 		what 0 : string
 	}
 	response {
-		result 0 : boolean
+		result 0 : string
 	}
 }
 
@@ -26,11 +28,17 @@ set 3 {
 		what 0 : string
 		value 1 : string
 	}
-	response {
-		ok 0 : boolean
-	}
 }
 
 ]]
 
+proto.s2c = sprotoparser.parse [[
+.package {
+	type 0 : integer
+	session 1 : integer
+}
+
+heartbeat 1 {}
+]]
+
 return proto