skynet內部服務都是由一個一個的消息所驅動,每個服務的上下文結構體struct skynet_context
有個字段struct message_queue *queue
描述其消息隊列,所有服務的消息隊列掛在全局消息對列的列表struct global_queue *Q
中
skynet在啟動時會啟動config->thread
個worker
線程來處理所有服務的消息,worker
線程的入口函數為static void *thread_worker(void *p)
,其處理邏輯如下:
- 如果當前要處理的消息隊列為空,則從全局消息隊列的列表中取下一個消息隊列
- 對消息隊列中的每個消息,調用該消息隊列所屬服務的回調函數,每次至少處理一個消息,之多處理消息隊列長度右移
weight
個消息,其中weight
是事先配置好的
static int weight[] = {
-1, -1, -1, -1, 0, 0, 0, 0,
1, 1, 1, 1, 1, 1, 1, 1,
2, 2, 2, 2, 2, 2, 2, 2,
3, 3, 3, 3, 3, 3, 3, 3, };
比如配置啟動n
個worker
線程,第i
個線程的weight
為:當i小于weight
數組長度時,線程weight
為weight[i-1]
,否則為0
每個服務的消息隊列都會被worker
進程公平的進行處理,但是每個線程一次處理的消息個數由工作線程配置的權重決定。
下面以snlua
為例理解消息回調處理,在dispatch_message
函數中,通過調用服務的回調函數來讓服務處理其收到的消息:
ctx->cb(ctx, ctx->cb_ud, type, msg->session, msg->source, msg->data, sz)
snlua
是執行lua
服務的沙盒環境,啟動一個lua
服務之后,在lua
代碼中會設置回調函數,通常在skynet.lua
文件中的skynet.start
中設置c.callback(skynet.dispatch_message)
,c.callback
調用的是:
83 static int
84 _callback(lua_State *L) {
85 struct skynet_context * context = lua_touserdata(L, lua_upvalueindex(1));
86 int forward = lua_toboolean(L, 2);
87 luaL_checktype(L,1,LUA_TFUNCTION);
88 lua_settop(L,1);
89 lua_rawsetp(L, LUA_REGISTRYINDEX, _cb);
90
91 lua_rawgeti(L, LUA_REGISTRYINDEX, LUA_RIDX_MAINTHREAD);
92 lua_State *gL = lua_tothread(L,-1);
93
94 if (forward) {
95 skynet_callback(context, gL, forward_cb);
96 } else {
97 skynet_callback(context, gL, _cb);
98 }
99
100 return 0;
101 }
- 85行獲取服務的上下文結構,此upvalue是在啟動次服務的時候設置的
- 89行在注冊表中設置
_cb=>skynet.dispatch_message
- 91-92行獲取服務的LUA狀態機結構
- 95或者97行設置服務上下結構體中的回調函數為
_cb
,回調函數私有數據為LUA狀態機gL
下面來分析回調函數_cb
,任何LUA沙盒服務收到的消息的回調函數入口都是_cb
30 static int
31 _cb(struct skynet_context * context, void * ud, int type, int session, uint32_t source, const void * msg, size_t sz) {
32 lua_State *L = ud;
33 int trace = 1;
34 int r;
35 int top = lua_gettop(L);
36 if (top == 0) {
37 lua_pushcfunction(L, traceback);
38 lua_rawgetp(L, LUA_REGISTRYINDEX, _cb);
39 } else {
40 assert(top == 2);
41 }
42 lua_pushvalue(L,2);
43
44 lua_pushinteger(L, type);
45 lua_pushlightuserdata(L, (void *)msg);
46 lua_pushinteger(L,sz);
47 lua_pushinteger(L, session);
48 lua_pushinteger(L, source);
49
50 r = lua_pcall(L, 5, 0 , trace);
51
52 if (r == LUA_OK) {
53 return 0;
54 }
55 const char * self = skynet_command(context, "REG", NULL);
56 switch (r) {
57 case LUA_ERRRUN:
58 skynet_error(context, "lua call [%x to %s : %d msgsz = %d] error : " KRED "%s" KNRM, source , self, session, sz, lua_tostring(L,-1));
59 break;
60 case LUA_ERRMEM:
61 skynet_error(context, "lua memory error : [%x to %s : %d]", source , self, session);
62 break;
63 case LUA_ERRERR:
64 skynet_error(context, "lua error in error : [%x to %s : %d]", source , self, session);
65 break;
66 case LUA_ERRGCMM:
67 skynet_error(context, "lua gc error : [%x to %s : %d]", source , self, session);
68 break;
69 };
70
71 lua_pop(L,1);
72
73 return 0;
74 }
- 32-38行在LUA狀態機的棧中設置即將執行的LUA函數及參數,依次是
traceback
,skynet.dispatch_message
,type
,msg
,sz
,session
,source
- 50行在保護模式執行
skynet.dispatch_message
函數,在此函數進行真正消息處理
以上粗略的分析了skynet框架是如何調度每個服務的消息隊列,以及如何通過回調函數來對服務的消息進行處理