feature: add balancer.bind_to_local_addr for stream module.

Co-author: ytlm <[email protected]>

Author: zhuizhuhaomeng

lib/ngx/balancer.lua

@@ -13,10 +13,14 @@ local FFI_OK = base.FFI_OK
 local FFI_ERROR = base.FFI_ERROR
 local int_out = ffi.new("int[1]")
 local get_request = base.get_request
+local get_string_buf = base.get_string_buf
+local get_size_ptr = base.get_size_ptr
+
 local error = error
 local type = type
 local tonumber = tonumber
 local max = math.max
+
 local subsystem = ngx.config.subsystem
 local ngx_lua_ffi_balancer_set_current_peer
 local ngx_lua_ffi_balancer_enable_keepalive
@@ -93,6 +97,10 @@ elseif subsystem == 'stream' then
 
     int ngx_stream_lua_ffi_balancer_set_timeouts(ngx_stream_lua_request_t *r,
         long connect_timeout, long timeout, char **err);
+
+    int ngx_stream_lua_ffi_balancer_bind_to_local_addr(
+        ngx_stream_lua_request_t *r, const char *addr, size_t addr_len,
+        char *errbuf, size_t *errbuf_size);
     ]]
 
     ngx_lua_ffi_balancer_set_current_peer =
@@ -115,6 +123,9 @@ elseif subsystem == 'stream' then
                                                             timeout, err)
         end
 
+    ngx_lua_ffi_balancer_bind_to_local_addr =
+        C.ngx_stream_lua_ffi_balancer_bind_to_local_addr
+
 else
     error("unknown subsystem: " .. subsystem)
 end
@@ -362,37 +373,29 @@ if subsystem == 'http' then
     end
 end
 
-if subsystem == "http" then
-    function _M.bind_to_local_addr(addr)
-        local r = get_request()
-        if not r then
-            error("no request found")
-        end
-
-        if type(addr) ~= "string" then
-            error("bad argument #1 to 'bind_to_local_addr' "
-                  .. "(string expected, got " .. type(addr) .. ")")
-        end
-
-        local errbuf_size = 1024
-        local errbuf = get_string_buf(errbuf_size)
-        local sizep = get_size_ptr()
-        sizep[0] = errbuf_size
-        local rc = ngx_lua_ffi_balancer_bind_to_local_addr(r, addr, #addr,
-                                                           errbuf,
-                                                           sizep)
-        if rc == FFI_OK then
-            return true
-        end
+function _M.bind_to_local_addr(addr)
+    local r = get_request()
+    if not r then
+        error("no request found")
+    end
 
-        return nil, ffi_str(errbuf, sizep[0])
+    if type(addr) ~= "string" then
+        error("bad argument #1 to 'bind_to_local_addr' "
+              .. "(string expected, got " .. type(addr) .. ")")
     end
 
-else
-    function _M.bind_to_local_addr(addr)
-        error("'bind_to_local_addr' not yet implemented in " .. subsystem ..
-              " subsystem", 2)
+    local errbuf_size = 1024
+    local errbuf = get_string_buf(errbuf_size)
+    local sizep = get_size_ptr()
+    sizep[0] = errbuf_size
+    local rc = ngx_lua_ffi_balancer_bind_to_local_addr(r, addr, #addr,
+                                                       errbuf,
+                                                       sizep)
+    if rc == FFI_OK then
+        return true
     end
+
+    return nil, ffi_str(errbuf, sizep[0])
 end
 
 

t/stream/balancer-bind.t

@@ -0,0 +1,107 @@
+# vim:set ft= ts=4 sw=4 et fdm=marker:
+use lib '.';
+use t::TestCore::Stream;
+
+repeat_each(2);
+
+plan tests => repeat_each() * (blocks() * 4);
+
+$ENV{TEST_NGINX_LUA_PACKAGE_PATH} = "$t::TestCore::Stream::lua_package_path";
+
+no_long_string();
+run_tests();
+
+__DATA__
+
+=== TEST 1: balancer 
+--- stream_config
+    lua_package_path "$TEST_NGINX_LUA_PACKAGE_PATH";
+    upstream backend {
+        server 0.0.0.1:1234 down;
+        balancer_by_lua_block {
+            local b = require "ngx.balancer"
+            assert(b.set_current_peer("127.0.0.1", 12345))
+        }
+    }
+    server {
+        listen 127.0.0.1:12345;
+        content_by_lua_block {
+            ngx.print(ngx.var.remote_addr, ":", ngx.var.remote_port)
+        }
+    }
+--- stream_server_config
+    proxy_pass backend;
+--- request
+    GET /t
+--- response_body eval
+[
+qr/127.0.0.1/,
+]
+--- error_code: 200
+--- no_error_log
+[error]
+[warn]
+
+
+
+=== TEST 2: balancer with bind_to_local_addr (addr)
+--- stream_config
+    lua_package_path "$TEST_NGINX_LUA_PACKAGE_PATH";
+    upstream backend {
+        server 0.0.0.1:1234 down;
+        balancer_by_lua_block {
+            local b = require "ngx.balancer"
+            assert(b.set_current_peer("127.0.0.1", 12345))
+            assert(b.bind_to_local_addr("127.0.0.4"))
+        }
+    }
+    server {
+        listen 127.0.0.1:12345;
+        content_by_lua_block {
+            ngx.print(ngx.var.remote_addr, ":", ngx.var.remote_port)
+        }
+    }
+--- stream_server_config
+    proxy_pass backend;
+--- request
+    GET /t
+--- response_body eval
+[
+qr/127.0.0.4/,
+]
+--- error_code: 200
+--- no_error_log
+[error]
+[warn]
+
+
+
+=== TEST 3: balancer with bind_to_local_addr (addr and port)
+--- stream_config
+    lua_package_path "$TEST_NGINX_LUA_PACKAGE_PATH";
+    upstream backend {
+        server 0.0.0.1:1234 down;
+        balancer_by_lua_block {
+            local b = require "ngx.balancer"
+            assert(b.set_current_peer("127.0.0.1", 12345))
+            assert(b.bind_to_local_addr("127.0.0.8:23456"))
+        }
+    }
+    server {
+        listen 127.0.0.1:12345;
+        content_by_lua_block {
+            ngx.print(ngx.var.remote_addr, ":", ngx.var.remote_port)
+        }
+    }
+--- stream_server_config
+    proxy_pass backend;
+--- request
+    GET /t
+--- response_body eval
+[
+qr/127.0.0.8/,
+]
+--- error_code: 200
+--- no_error_log
+[error]
+[warn]