Blame view

publish/skynet/lualib/skynet.lua 20.3 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
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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
  -- read https://github.com/cloudwu/skynet/wiki/FAQ for the module "skynet.core"
  local c = require "skynet.core"
  local tostring = tostring
  local coroutine = coroutine
  local assert = assert
  local pairs = pairs
  local pcall = pcall
  local table = table
  local tremove = table.remove
  local tinsert = table.insert
  local traceback = debug.traceback
  
  local profile = require "skynet.profile"
  
  local cresume = profile.resume
  local running_thread = nil
  local init_thread = nil
  
  local function coroutine_resume(co, ...)
  	running_thread = co
  	return cresume(co, ...)
  end
  local coroutine_yield = profile.yield
  local coroutine_create = coroutine.create
  
  local proto = {}
  local skynet = {
  	-- read skynet.h
  	PTYPE_TEXT = 0,
  	PTYPE_RESPONSE = 1,
  	PTYPE_MULTICAST = 2,
  	PTYPE_CLIENT = 3,
  	PTYPE_SYSTEM = 4,
  	PTYPE_HARBOR = 5,
  	PTYPE_SOCKET = 6,
  	PTYPE_ERROR = 7,
  	PTYPE_QUEUE = 8,	-- used in deprecated mqueue, use skynet.queue instead
  	PTYPE_DEBUG = 9,
  	PTYPE_LUA = 10,
  	PTYPE_SNAX = 11,
  	PTYPE_TRACE = 12,	-- use for debug trace
  }
  
  -- code cache
  skynet.cache = require "skynet.codecache"
  
  function skynet.register_protocol(class)
  	local name = class.name
  	local id = class.id
  	assert(proto[name] == nil and proto[id] == nil)
  	assert(type(name) == "string" and type(id) == "number" and id >=0 and id <=255)
  	proto[name] = class
  	proto[id] = class
  end
  
  local session_id_coroutine = {}
  local session_coroutine_id = {}
  local session_coroutine_address = {}
  local session_coroutine_tracetag = {}
  local unresponse = {}
  
  local wakeup_queue = {}
  local sleep_session = {}
  
  local watching_session = {}
  local error_queue = {}
  local fork_queue = {}
  
  -- suspend is function
  local suspend
  
  
  ----- monitor exit
  
  local function dispatch_error_queue()
  	local session = tremove(error_queue,1)
  	if session then
  		local co = session_id_coroutine[session]
  		session_id_coroutine[session] = nil
  		return suspend(co, coroutine_resume(co, false))
  	end
  end
  
  local function _error_dispatch(error_session, error_source)
  	skynet.ignoreret()	-- don't return for error
  	if error_session == 0 then
  		-- error_source is down, clear unreponse set
  		for resp, address in pairs(unresponse) do
  			if error_source == address then
  				unresponse[resp] = nil
  			end
  		end
  		for session, srv in pairs(watching_session) do
  			if srv == error_source then
  				tinsert(error_queue, session)
  			end
  		end
  	else
  		-- capture an error for error_session
  		if watching_session[error_session] then
  			tinsert(error_queue, error_session)
  		end
  	end
  end
  
  -- coroutine reuse
  
  local coroutine_pool = setmetatable({}, { __mode = "kv" })
  
  local function co_create(f)
  	local co = tremove(coroutine_pool)
  	if co == nil then
  		co = coroutine_create(function(...)
  			f(...)
  			while true do
  				local session = session_coroutine_id[co]
  				if session and session ~= 0 then
  					local source = debug.getinfo(f,"S")
  					skynet.error(string.format("Maybe forgot response session %s from %s : %s:%d",
  						session,
  						skynet.address(session_coroutine_address[co]),
  						source.source, source.linedefined))
  				end
  				-- coroutine exit
  				local tag = session_coroutine_tracetag[co]
  				if tag ~= nil then
  					if tag then c.trace(tag, "end")	end
  					session_coroutine_tracetag[co] = nil
  				end
  				local address = session_coroutine_address[co]
  				if address then
  					session_coroutine_id[co] = nil
  					session_coroutine_address[co] = nil
  				end
  
  				-- recycle co into pool
  				f = nil
  				coroutine_pool[#coroutine_pool+1] = co
  				-- recv new main function f
  				f = coroutine_yield "SUSPEND"
  				f(coroutine_yield())
  			end
  		end)
  	else
  		-- pass the main function f to coroutine, and restore running thread
  		local running = running_thread
  		coroutine_resume(co, f)
  		running_thread = running
  	end
  	return co
  end
  
  local function dispatch_wakeup()
  	local token = tremove(wakeup_queue,1)
  	if token then
  		local session = sleep_session[token]
  		if session then
  			local co = session_id_coroutine[session]
  			local tag = session_coroutine_tracetag[co]
  			if tag then c.trace(tag, "resume") end
  			session_id_coroutine[session] = "BREAK"
  			return suspend(co, coroutine_resume(co, false, "BREAK"))
  		end
  	end
  end
  
  -- suspend is local function
  function suspend(co, result, command)
  	if not result then
  		local session = session_coroutine_id[co]
  		if session then -- coroutine may fork by others (session is nil)
  			local addr = session_coroutine_address[co]
  			if session ~= 0 then
  				-- only call response error
  				local tag = session_coroutine_tracetag[co]
  				if tag then c.trace(tag, "error") end
  				c.send(addr, skynet.PTYPE_ERROR, session, "")
  			end
  			session_coroutine_id[co] = nil
  		end
  		session_coroutine_address[co] = nil
  		session_coroutine_tracetag[co] = nil
  		skynet.fork(function() end)	-- trigger command "SUSPEND"
  		error(traceback(co,tostring(command)))
  	end
  	if command == "SUSPEND" then
  		dispatch_wakeup()
  		dispatch_error_queue()
  	elseif command == "QUIT" then
  		-- service exit
  		return
  	elseif command == "USER" then
  		-- See skynet.coutine for detail
  		error("Call skynet.coroutine.yield out of skynet.coroutine.resume\n" .. traceback(co))
  	elseif command == nil then
  		-- debug trace
  		return
  	else
  		error("Unknown command : " .. command .. "\n" .. traceback(co))
  	end
  end
  
  local co_create_for_timeout
  local timeout_traceback
  
  function skynet.trace_timeout(on)
  	local function trace_coroutine(func, ti)
  		local co
  		co = co_create(function()
  			timeout_traceback[co] = nil
  			func()
  		end)
  		local info = string.format("TIMER %d+%d : ", skynet.now(), ti)
  		timeout_traceback[co] = traceback(info, 3)
  		return co
  	end
  	if on then
  		timeout_traceback = timeout_traceback or {}
  		co_create_for_timeout = trace_coroutine
  	else
  		timeout_traceback = nil
  		co_create_for_timeout = co_create
  	end
  end
  
  skynet.trace_timeout(false)	-- turn off by default
  
  function skynet.timeout(ti, func)
  	local session = c.intcommand("TIMEOUT",ti)
  	assert(session)
  	local co = co_create_for_timeout(func, ti)
  	assert(session_id_coroutine[session] == nil)
  	session_id_coroutine[session] = co
  	return co	-- for debug
  end
  
  local function suspend_sleep(session, token)
  	local tag = session_coroutine_tracetag[running_thread]
  	if tag then c.trace(tag, "sleep", 2) end
  	session_id_coroutine[session] = running_thread
  	assert(sleep_session[token] == nil, "token duplicative")
  	sleep_session[token] = session
  
  	return coroutine_yield "SUSPEND"
  end
  
  function skynet.sleep(ti, token)
  	local session = c.intcommand("TIMEOUT",ti)
  	assert(session)
  	token = token or coroutine.running()
  	local succ, ret = suspend_sleep(session, token)
  	sleep_session[token] = nil
  	if succ then
  		return
  	end
  	if ret == "BREAK" then
  		return "BREAK"
  	else
  		error(ret)
  	end
  end
  
  function skynet.yield()
  	return skynet.sleep(0)
  end
  
  function skynet.wait(token)
  	local session = c.genid()
  	token = token or coroutine.running()
  	local ret, msg = suspend_sleep(session, token)
  	sleep_session[token] = nil
  	session_id_coroutine[session] = nil
  end
  
  function skynet.self()
  	return c.addresscommand "REG"
  end
  
  function skynet.localname(name)
  	return c.addresscommand("QUERY", name)
  end
  
  skynet.now = c.now
  skynet.hpc = c.hpc	-- high performance counter
  
  local traceid = 0
  function skynet.trace(info)
  	skynet.error("TRACE", session_coroutine_tracetag[running_thread])
  	if session_coroutine_tracetag[running_thread] == false then
  		-- force off trace log
  		return
  	end
  	traceid = traceid + 1
  
  	local tag = string.format(":%08x-%d",skynet.self(), traceid)
  	session_coroutine_tracetag[running_thread] = tag
  	if info then
  		c.trace(tag, "trace " .. info)
  	else
  		c.trace(tag, "trace")
  	end
  end
  
  function skynet.tracetag()
  	return session_coroutine_tracetag[running_thread]
  end
  
  local starttime
  
  function skynet.starttime()
  	if not starttime then
  		starttime = c.intcommand("STARTTIME")
  	end
  	return starttime
  end
  
  function skynet.time()
  	return skynet.now()/100 + (starttime or skynet.starttime())
  end
  
  function skynet.exit()
  	fork_queue = {}	-- no fork coroutine can be execute after skynet.exit
  	skynet.send(".launcher","lua","REMOVE",skynet.self(), false)
  	-- report the sources that call me
  	for co, session in pairs(session_coroutine_id) do
  		local address = session_coroutine_address[co]
  		if session~=0 and address then
  			c.send(address, skynet.PTYPE_ERROR, session, "")
  		end
  	end
  	for resp in pairs(unresponse) do
  		resp(false)
  	end
  	-- report the sources I call but haven't return
  	local tmp = {}
  	for session, address in pairs(watching_session) do
  		tmp[address] = true
  	end
  	for address in pairs(tmp) do
  		c.send(address, skynet.PTYPE_ERROR, 0, "")
  	end
  	c.command("EXIT")
  	-- quit service
  	coroutine_yield "QUIT"
  end
  
  function skynet.getenv(key)
  	return (c.command("GETENV",key))
  end
  
  function skynet.setenv(key, value)
  	assert(c.command("GETENV",key) == nil, "Can't setenv exist key : " .. key)
  	c.command("SETENV",key .. " " ..value)
  end
  
  function skynet.send(addr, typename, ...)
  	local p = proto[typename]
  	return c.send(addr, p.id, 0 , p.pack(...))
  end
  
  function skynet.rawsend(addr, typename, msg, sz)
  	local p = proto[typename]
  	return c.send(addr, p.id, 0 , msg, sz)
  end
  
  skynet.genid = assert(c.genid)
  
  skynet.redirect = function(dest,source,typename,...)
  	return c.redirect(dest, source, proto[typename].id, ...)
  end
  
  skynet.pack = assert(c.pack)
  skynet.packstring = assert(c.packstring)
  skynet.unpack = assert(c.unpack)
  skynet.tostring = assert(c.tostring)
  skynet.trash = assert(c.trash)
  
  local function yield_call(service, session)
  	watching_session[session] = service
  	session_id_coroutine[session] = running_thread
  	local succ, msg, sz = coroutine_yield "SUSPEND"
  	watching_session[session] = nil
  	if not succ then
  		error "call failed"
  	end
  	return msg,sz
  end
  
  function skynet.call(addr, typename, ...)
  	local tag = session_coroutine_tracetag[running_thread]
  	if tag then
  		c.trace(tag, "call", 2)
  		c.send(addr, skynet.PTYPE_TRACE, 0, tag)
  	end
  
  	local p = proto[typename]
  	local session = c.send(addr, p.id , nil , p.pack(...))
  	if session == nil then
  		error("call to invalid address " .. skynet.address(addr))
  	end
  	return p.unpack(yield_call(addr, session))
  end
  
  function skynet.rawcall(addr, typename, msg, sz)
  	local tag = session_coroutine_tracetag[running_thread]
  	if tag then
  		c.trace(tag, "call", 2)
  		c.send(addr, skynet.PTYPE_TRACE, 0, tag)
  	end
  	local p = proto[typename]
  	local session = assert(c.send(addr, p.id , nil , msg, sz), "call to invalid address")
  	return yield_call(addr, session)
  end
  
  function skynet.tracecall(tag, addr, typename, msg, sz)
  	c.trace(tag, "tracecall begin")
  	c.send(addr, skynet.PTYPE_TRACE, 0, tag)
  	local p = proto[typename]
  	local session = assert(c.send(addr, p.id , nil , msg, sz), "call to invalid address")
  	local msg, sz = yield_call(addr, session)
  	c.trace(tag, "tracecall end")
  	return msg, sz
  end
  
  function skynet.ret(msg, sz)
  	msg = msg or ""
  	local tag = session_coroutine_tracetag[running_thread]
  	if tag then c.trace(tag, "response") end
  	local co_session = session_coroutine_id[running_thread]
  	session_coroutine_id[running_thread] = nil
  	if co_session == 0 then
  		if sz ~= nil then
  			c.trash(msg, sz)
  		end
  		return false	-- send don't need ret
  	end
  	local co_address = session_coroutine_address[running_thread]
  	if not co_session then
  		error "No session"
  	end
  	local ret = c.send(co_address, skynet.PTYPE_RESPONSE, co_session, msg, sz)
  	if ret then
  		return true
  	elseif ret == false then
  		-- If the package is too large, returns false. so we should report error back
  		c.send(co_address, skynet.PTYPE_ERROR, co_session, "")
  	end
  	return false
  end
  
  function skynet.context()
  	local co_session = session_coroutine_id[running_thread]
  	local co_address = session_coroutine_address[running_thread]
  	return co_session, co_address
  end
  
  function skynet.ignoreret()
  	-- We use session for other uses
  	session_coroutine_id[running_thread] = nil
  end
  
  function skynet.response(pack)
  	pack = pack or skynet.pack
  
  	local co_session = assert(session_coroutine_id[running_thread], "no session")
  	session_coroutine_id[running_thread] = nil
  	local co_address = session_coroutine_address[running_thread]
  	if co_session == 0 then
  		--  do not response when session == 0 (send)
  		return function() end
  	end
  	local function response(ok, ...)
  		if ok == "TEST" then
  			return unresponse[response] ~= nil
  		end
  		if not pack then
  			error "Can't response more than once"
  		end
  
  		local ret
  		if unresponse[response] then
  			if ok then
  				ret = c.send(co_address, skynet.PTYPE_RESPONSE, co_session, pack(...))
  				if ret == false then
  					-- If the package is too large, returns false. so we should report error back
  					c.send(co_address, skynet.PTYPE_ERROR, co_session, "")
  				end
  			else
  				ret = c.send(co_address, skynet.PTYPE_ERROR, co_session, "")
  			end
  			unresponse[response] = nil
  			ret = ret ~= nil
  		else
  			ret = false
  		end
  		pack = nil
  		return ret
  	end
  	unresponse[response] = co_address
  
  	return response
  end
  
  function skynet.retpack(...)
  	return skynet.ret(skynet.pack(...))
  end
  
  function skynet.wakeup(token)
  	if sleep_session[token] then
  		tinsert(wakeup_queue, token)
  		return true
  	end
  end
  
  function skynet.dispatch(typename, func)
  	local p = proto[typename]
  	if func then
  		local ret = p.dispatch
  		p.dispatch = func
  		return ret
  	else
  		return p and p.dispatch
  	end
  end
  
  local function unknown_request(session, address, msg, sz, prototype)
  	skynet.error(string.format("Unknown request (%s): %s", prototype, c.tostring(msg,sz)))
  	error(string.format("Unknown session : %d from %x", session, address))
  end
  
  function skynet.dispatch_unknown_request(unknown)
  	local prev = unknown_request
  	unknown_request = unknown
  	return prev
  end
  
  local function unknown_response(session, address, msg, sz)
  	skynet.error(string.format("Response message : %s" , c.tostring(msg,sz)))
  	error(string.format("Unknown session : %d from %x", session, address))
  end
  
  function skynet.dispatch_unknown_response(unknown)
  	local prev = unknown_response
  	unknown_response = unknown
  	return prev
  end
  
  function skynet.fork(func,...)
  	local n = select("#", ...)
  	local co
  	if n == 0 then
  		co = co_create(func)
  	else
  		local args = { ... }
  		co = co_create(function() func(table.unpack(args,1,n)) end)
  	end
  	tinsert(fork_queue, co)
  	return co
  end
  
  local trace_source = {}
  
  local function raw_dispatch_message(prototype, msg, sz, session, source)
  	-- skynet.PTYPE_RESPONSE = 1, read skynet.h
  	if prototype == 1 then
  		local co = session_id_coroutine[session]
  		if co == "BREAK" then
  			session_id_coroutine[session] = nil
  		elseif co == nil then
  			unknown_response(session, source, msg, sz)
  		else
  			local tag = session_coroutine_tracetag[co]
  			if tag then c.trace(tag, "resume") end
  			session_id_coroutine[session] = nil
  			suspend(co, coroutine_resume(co, true, msg, sz))
  		end
  	else
  		local p = proto[prototype]
  		if p == nil then
  			if prototype == skynet.PTYPE_TRACE then
  				-- trace next request
  				trace_source[source] = c.tostring(msg,sz)
  			elseif session ~= 0 then
  				c.send(source, skynet.PTYPE_ERROR, session, "")
  			else
  				unknown_request(session, source, msg, sz, prototype)
  			end
  			return
  		end
  
  		local f = p.dispatch
  		if f then
  			local co = co_create(f)
  			session_coroutine_id[co] = session
  			session_coroutine_address[co] = source
  			local traceflag = p.trace
  			if traceflag == false then
  				-- force off
  				trace_source[source] = nil
  				session_coroutine_tracetag[co] = false
  			else
  				local tag = trace_source[source]
  				if tag then
  					trace_source[source] = nil
  					c.trace(tag, "request")
  					session_coroutine_tracetag[co] = tag
  				elseif traceflag then
  					-- set running_thread for trace
  					running_thread = co
  					skynet.trace()
  				end
  			end
  			suspend(co, coroutine_resume(co, session,source, p.unpack(msg,sz)))
  		else
  			trace_source[source] = nil
  			if session ~= 0 then
  				c.send(source, skynet.PTYPE_ERROR, session, "")
  			else
  				unknown_request(session, source, msg, sz, proto[prototype].name)
  			end
  		end
  	end
  end
  
  function skynet.dispatch_message(...)
  	local succ, err = pcall(raw_dispatch_message,...)
  	while true do
  		local co = tremove(fork_queue,1)
  		if co == nil then
  			break
  		end
  		local fork_succ, fork_err = pcall(suspend,co,coroutine_resume(co))
  		if not fork_succ then
  			if succ then
  				succ = false
  				err = tostring(fork_err)
  			else
  				err = tostring(err) .. "\n" .. tostring(fork_err)
  			end
  		end
  	end
  	assert(succ, tostring(err))
  end
  
  function skynet.newservice(name, ...)
  	return skynet.call(".launcher", "lua" , "LAUNCH", "snlua", name, ...)
  end
  
  function skynet.uniqueservice(global, ...)
  	if global == true then
  		return assert(skynet.call(".service", "lua", "GLAUNCH", ...))
  	else
  		return assert(skynet.call(".service", "lua", "LAUNCH", global, ...))
  	end
  end
  
  function skynet.queryservice(global, ...)
  	if global == true then
  		return assert(skynet.call(".service", "lua", "GQUERY", ...))
  	else
  		return assert(skynet.call(".service", "lua", "QUERY", global, ...))
  	end
  end
  
  function skynet.address(addr)
  	if type(addr) == "number" then
  		return string.format(":%08x",addr)
  	else
  		return tostring(addr)
  	end
  end
  
  function skynet.harbor(addr)
  	return c.harbor(addr)
  end
  
  skynet.error = c.error
  skynet.tracelog = c.trace
  
  -- true: force on
  -- false: force off
  -- nil: optional (use skynet.trace() to trace one message)
  function skynet.traceproto(prototype, flag)
  	local p = assert(proto[prototype])
  	p.trace = flag
  end
  
  ----- register protocol
  do
  	local REG = skynet.register_protocol
  
  	REG {
  		name = "lua",
  		id = skynet.PTYPE_LUA,
  		pack = skynet.pack,
  		unpack = skynet.unpack,
  	}
  
  	REG {
  		name = "response",
  		id = skynet.PTYPE_RESPONSE,
  	}
  
  	REG {
  		name = "error",
  		id = skynet.PTYPE_ERROR,
  		unpack = function(...) return ... end,
  		dispatch = _error_dispatch,
  	}
  end
  
  local init_func = {}
  
  function skynet.init(f, name)
  	assert(type(f) == "function")
  	if init_func == nil then
  		f()
  	else
  		tinsert(init_func, f)
  		if name then
  			assert(type(name) == "string")
  			assert(init_func[name] == nil)
  			init_func[name] = f
  		end
  	end
  end
  
  local function init_all()
  	local funcs = init_func
  	init_func = nil
  	if funcs then
  		for _,f in ipairs(funcs) do
  			f()
  		end
  	end
  end
  
  local function ret(f, ...)
  	f()
  	return ...
  end
  
  local function init_template(start, ...)
  	init_all()
  	init_func = {}
  	return ret(init_all, start(...))
  end
  
  function skynet.pcall(start, ...)
  	return xpcall(init_template, traceback, start, ...)
  end
  
  function skynet.init_service(start)
  	local ok, err = skynet.pcall(start)
  	if not ok then
  		skynet.error("init service failed: " .. tostring(err))
  		skynet.send(".launcher","lua", "ERROR")
  		skynet.exit()
  	else
  		skynet.send(".launcher","lua", "LAUNCHOK")
  	end
  end
  
  function skynet.start(start_func)
  	c.callback(skynet.dispatch_message)
  	init_thread = skynet.timeout(0, function()
  		skynet.init_service(start_func)
  		init_thread = nil
  	end)
  end
  
  function skynet.endless()
  	return (c.intcommand("STAT", "endless") == 1)
  end
  
  function skynet.mqlen()
  	return c.intcommand("STAT", "mqlen")
  end
  
  function skynet.stat(what)
  	return c.intcommand("STAT", what)
  end
  
  function skynet.task(ret)
  	if ret == nil then
  		local t = 0
  		for session,co in pairs(session_id_coroutine) do
  			t = t + 1
  		end
  		return t
  	end
  	if ret == "init" then
  		if init_thread then
  			return traceback(init_thread)
  		else
  			return
  		end
  	end
  	local tt = type(ret)
  	if tt == "table" then
  		for session,co in pairs(session_id_coroutine) do
  			if timeout_traceback and timeout_traceback[co] then
  				ret[session] = timeout_traceback[co]
  			else
  				ret[session] = traceback(co)
  			end
  		end
  		return
  	elseif tt == "number" then
  		local co = session_id_coroutine[ret]
  		if co then
  			return traceback(co)
  		else
  			return "No session"
  		end
  	elseif tt == "thread" then
  		for session, co in pairs(session_id_coroutine) do
  			if co == ret then
  				return session
  			end
  		end
  		return
  	end
  end
  
  function skynet.uniqtask()
  	local stacks = {}
  	for session, co in pairs(session_id_coroutine) do
  		local stack = traceback(co)
  		local info = stacks[stack] or {count = 0, sessions = {}}
  		info.count = info.count + 1
  		if info.count < 10 then
  			info.sessions[#info.sessions+1] = session
  		end
  		stacks[stack] = info
  	end
  	local ret = {}
  	for stack, info in pairs(stacks) do
  		local count = info.count
  		local sessions = table.concat(info.sessions, ",")
  		if count > 10 then
  			sessions = sessions .. "..."
  		end
  		local head_line = string.format("%d\tsessions:[%s]\n", count, sessions)
  		ret[head_line] = stack
  	end
  	return ret
  end
  
  function skynet.term(service)
  	return _error_dispatch(0, service)
  end
  
  function skynet.memlimit(bytes)
  	debug.getregistry().memlimit = bytes
  	skynet.memlimit = nil	-- set only once
  end
  
  -- Inject internal debug framework
  local debug = require "skynet.debug"
  debug.init(skynet, {
  	dispatch = skynet.dispatch_message,
  	suspend = suspend,
  })
  
  return skynet