整理代码

Author: actboy168

binding/lua_socket.cpp

@@ -29,6 +29,18 @@ namespace bee::lua_socket {
         char* buf;
         size_t len;
     };
+#else
+    struct luabuf {
+        luabuf(lua_State* L, size_t len) {
+            luaL_buffinit(L, &buffer);
+            buf = luaL_prepbuffsize(&buffer, (size_t)len);
+        }
+        void pushresultsize(lua_State* L, int rc) {
+            luaL_pushresultsize(&buffer, rc);
+        }
+        luaL_Buffer buffer;
+        char* buf;
+    };
 #endif
 
     namespace endpoint {
@@ -189,14 +201,8 @@ namespace bee::lua_socket {
         }
         static int recv(lua_State* L, net::fd_t fd) {
             auto len = lua::optinteger<int, LUAL_BUFFERSIZE>(L, 2);
-#if LUA_VERSION_NUM >= 505
             luabuf b(L, (size_t)len);
             char* buf = b.buf;
-#else
-            luaL_Buffer b;
-            luaL_buffinit(L, &b);
-            char* buf = luaL_prepbuffsize(&b, (size_t)len);
-#endif
             int rc;
             switch (net::socket::recv(fd, rc, buf, len)) {
             case net::socket::recv_status::close:
@@ -206,11 +212,7 @@ namespace bee::lua_socket {
                 lua_pushboolean(L, 0);
                 return 1;
             case net::socket::recv_status::success:
-#if LUA_VERSION_NUM >= 505
                 b.pushresultsize(L, rc);
-#else
-                luaL_pushresultsize(&b, rc);
-#endif
                 return 1;
             case net::socket::recv_status::failed:
                 return lua::return_net_error(L, "recv");
@@ -237,22 +239,12 @@ namespace bee::lua_socket {
         static int recvfrom(lua_State* L, net::fd_t fd) {
             auto len = lua::optinteger<int, LUAL_BUFFERSIZE>(L, 2);
             auto& ep = lua::newudata<net::endpoint>(L);
-#if LUA_VERSION_NUM >= 505
             luabuf b(L, (size_t)len);
             char* buf = b.buf;
-#else
-            luaL_Buffer b;
-            luaL_buffinit(L, &b);
-            char* buf = luaL_prepbuffsize(&b, (size_t)len);
-#endif
             int rc;
             switch (net::socket::recvfrom(fd, rc, ep, buf, len)) {
             case net::socket::status::success:
-#if LUA_VERSION_NUM >= 505
                 b.pushresultsize(L, rc);
-#else
-                luaL_pushresultsize(&b, rc);
-#endif
                 lua_insert(L, -2);
                 return 2;
             case net::socket::status::wait: