123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158 |
- local skynet = require "skynet"
- local core = require "skynet.core"
- require "skynet.manager" -- import manager apis
- local string = string
- local services = {}
- local command = {}
- local instance = {} -- for confirm (function command.LAUNCH / command.ERROR / command.LAUNCHOK)
- local function handle_to_address(handle)
- return tonumber("0x" .. string.sub(handle , 2))
- end
- local NORET = {}
- function command.LIST()
- local list = {}
- for k,v in pairs(services) do
- list[skynet.address(k)] = v
- end
- return list
- end
- function command.STAT()
- local list = {}
- for k,v in pairs(services) do
- local ok, stat = pcall(skynet.call,k,"debug","STAT")
- if not ok then
- stat = string.format("ERROR (%s)",v)
- end
- list[skynet.address(k)] = stat
- end
- return list
- end
- function command.KILL(_, handle)
- handle = handle_to_address(handle)
- skynet.kill(handle)
- local ret = { [skynet.address(handle)] = tostring(services[handle]) }
- services[handle] = nil
- return ret
- end
- function command.MEM()
- local list = {}
- for k,v in pairs(services) do
- local ok, kb, bytes = pcall(skynet.call,k,"debug","MEM")
- if not ok then
- list[skynet.address(k)] = string.format("ERROR (%s)",v)
- else
- list[skynet.address(k)] = string.format("%.2f Kb (%s)",kb,v)
- end
- end
- return list
- end
- function command.GC()
- for k,v in pairs(services) do
- skynet.send(k,"debug","GC")
- end
- return command.MEM()
- end
- function command.REMOVE(_, handle, kill)
- services[handle] = nil
- local response = instance[handle]
- if response then
- -- instance is dead
- response(not kill) -- return nil to caller of newservice, when kill == false
- instance[handle] = nil
- end
- -- don't return (skynet.ret) because the handle may exit
- return NORET
- end
- local function launch_service(service, ...)
- local param = table.concat({...}, " ")
- local inst = skynet.launch(service, param)
- local response = skynet.response()
- if inst then
- services[inst] = service .. " " .. param
- instance[inst] = response
- else
- response(false)
- return
- end
- return inst
- end
- function command.LAUNCH(_, service, ...)
- launch_service(service, ...)
- return NORET
- end
- function command.LOGLAUNCH(_, service, ...)
- local inst = launch_service(service, ...)
- if inst then
- core.command("LOGON", skynet.address(inst))
- end
- return NORET
- end
- function command.ERROR(address)
- -- see serivce-src/service_lua.c
- -- init failed
- local response = instance[address]
- if response then
- response(false)
- instance[address] = nil
- end
- services[address] = nil
- return NORET
- end
- function command.LAUNCHOK(address)
- -- init notice
- local response = instance[address]
- if response then
- response(true, address)
- instance[address] = nil
- end
- return NORET
- end
- -- for historical reasons, launcher support text command (for C service)
- skynet.register_protocol {
- name = "text",
- id = skynet.PTYPE_TEXT,
- unpack = skynet.tostring,
- dispatch = function(session, address , cmd)
- if cmd == "" then
- command.LAUNCHOK(address)
- elseif cmd == "ERROR" then
- command.ERROR(address)
- else
- error ("Invalid text command " .. cmd)
- end
- end,
- }
- skynet.dispatch("lua", function(session, address, cmd , ...)
- cmd = string.upper(cmd)
- local f = command[cmd]
- if f then
- local ret = f(address, ...)
- if ret ~= NORET then
- skynet.ret(skynet.pack(ret))
- end
- else
- skynet.ret(skynet.pack {"Unknown command"} )
- end
- end)
- skynet.start(function() end)
|