Add low priority list to socket send buffer

Author: cloudwu

Diff for: lualib-src/lua-socket.c

@@ -402,27 +402,43 @@ llisten(lua_State *L) {
 	return 1;
 }
 
-static int
-lsend(lua_State *L) {
-	struct skynet_context * ctx = lua_touserdata(L, lua_upvalueindex(1));
-	int id = luaL_checkinteger(L, 1);
+static void *
+get_buffer(lua_State *L, int *sz) {
 	void *buffer;
-	int sz;
 	if (lua_isuserdata(L,2)) {
 		buffer = lua_touserdata(L,2);
-		sz = luaL_checkinteger(L,3);
+		*sz = luaL_checkinteger(L,3);
 	} else {
 		size_t len = 0;
 		const char * str =  luaL_checklstring(L, 2, &len);
 		buffer = malloc(len);
 		memcpy(buffer, str, len);
-		sz = (int)len;
+		*sz = (int)len;
 	}
+	return buffer;
+}
+
+static int
+lsend(lua_State *L) {
+	struct skynet_context * ctx = lua_touserdata(L, lua_upvalueindex(1));
+	int id = luaL_checkinteger(L, 1);
+	int sz = 0;
+	void *buffer = get_buffer(L, &sz);
 	int err = skynet_socket_send(ctx, id, buffer, sz);
 	lua_pushboolean(L, !err);
 	return 1;
 }
 
+static int
+lsendlow(lua_State *L) {
+	struct skynet_context * ctx = lua_touserdata(L, lua_upvalueindex(1));
+	int id = luaL_checkinteger(L, 1);
+	int sz = 0;
+	void *buffer = get_buffer(L, &sz);
+	skynet_socket_send_lowpriority(ctx, id, buffer, sz);
+	return 0;
+}
+
 static int
 lbind(lua_State *L) {
 	struct skynet_context * ctx = lua_touserdata(L, lua_upvalueindex(1));
@@ -462,6 +478,7 @@ luaopen_socketdriver(lua_State *L) {
 		{ "close", lclose },
 		{ "listen", llisten },
 		{ "send", lsend },
+		{ "lsend", lsendlow },
 		{ "bind", lbind },
 		{ "start", lstart },
 		{ NULL, NULL },

Diff for: lualib/socket.lua

@@ -264,6 +264,7 @@ function socket.block(id)
 end
 
 socket.write = assert(driver.send)
+socket.lwrite = assert(driver.lsend)
 
 function socket.invalid(id)
 	return socket_pool[id] == nil

Diff for: skynet-src/skynet_socket.c

@@ -115,6 +115,11 @@ skynet_socket_send(struct skynet_context *ctx, int id, void *buffer, int sz) {
 	return 0;
 }
 
+void
+skynet_socket_send_lowpriority(struct skynet_context *ctx, int id, void *buffer, int sz) {
+	socket_server_send_lowpriority(SOCKET_SERVER, id, buffer, sz);
+}
+
 int 
 skynet_socket_listen(struct skynet_context *ctx, const char *host, int port, int backlog) {
 	uint32_t source = skynet_context_handle(ctx);

Diff for: skynet-src/skynet_socket.h

@@ -22,6 +22,7 @@ void skynet_socket_free();
 int skynet_socket_poll();
 
 int skynet_socket_send(struct skynet_context *ctx, int id, void *buffer, int sz);
+void skynet_socket_send_lowpriority(struct skynet_context *ctx, int id, void *buffer, int sz);
 int skynet_socket_listen(struct skynet_context *ctx, const char *host, int port, int backlog);
 int skynet_socket_connect(struct skynet_context *ctx, const char *host, int port);
 int skynet_socket_block_connect(struct skynet_context *ctx, const char *host, int port);