├── .gitignore ├── LICENSE ├── README.md └── dynamic_eureka_balancer.lua /.gitignore: -------------------------------------------------------------------------------- 1 | # Created by .ignore support plugin (hsz.mobi) 2 | ### JetBrains template 3 | # Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and WebStorm 4 | # Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839 5 | 6 | # User-specific stuff 7 | .idea/ 8 | 9 | # CMake 10 | cmake-build-debug/ 11 | cmake-build-release/ 12 | 13 | 14 | # File-based project format 15 | *.iws 16 | 17 | # IntelliJ 18 | out/ 19 | 20 | # mpeltonen/sbt-idea plugin 21 | .idea_modules/ 22 | 23 | # JIRA plugin 24 | atlassian-ide-plugin.xml 25 | 26 | # Crashlytics plugin (for Android Studio and IntelliJ) 27 | com_crashlytics_export_strings.xml 28 | crashlytics.properties 29 | crashlytics-build.properties 30 | fabric.properties 31 | 32 | ### Lua template 33 | # Compiled Lua sources 34 | luac.out 35 | 36 | # luarocks build files 37 | *.src.rock 38 | *.zip 39 | *.tar.gz 40 | 41 | # Object files 42 | *.o 43 | *.os 44 | *.ko 45 | *.obj 46 | *.elf 47 | 48 | # Precompiled Headers 49 | *.gch 50 | *.pch 51 | 52 | # Libraries 53 | *.lib 54 | *.a 55 | *.la 56 | *.lo 57 | *.def 58 | *.exp 59 | 60 | # Shared objects (inc. Windows DLLs) 61 | *.dll 62 | *.so 63 | *.so.* 64 | *.dylib 65 | 66 | # Executables 67 | *.exe 68 | *.out 69 | *.app 70 | *.i*86 71 | *.x86_64 72 | *.hex 73 | 74 | 75 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- 1 | Apache License 2 | Version 2.0, January 2004 3 | http://www.apache.org/licenses/ 4 | 5 | TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 6 | 7 | 1. Definitions. 8 | 9 | "License" shall mean the terms and conditions for use, reproduction, 10 | and distribution as defined by Sections 1 through 9 of this document. 11 | 12 | "Licensor" shall mean the copyright owner or entity authorized by 13 | the copyright owner that is granting the License. 14 | 15 | "Legal Entity" shall mean the union of the acting entity and all 16 | other entities that control, are controlled by, or are under common 17 | control with that entity. For the purposes of this definition, 18 | "control" means (i) the power, direct or indirect, to cause the 19 | direction or management of such entity, whether by contract or 20 | otherwise, or (ii) ownership of fifty percent (50%) or more of the 21 | outstanding shares, or (iii) beneficial ownership of such entity. 22 | 23 | "You" (or "Your") shall mean an individual or Legal Entity 24 | exercising permissions granted by this License. 25 | 26 | "Source" form shall mean the preferred form for making modifications, 27 | including but not limited to software source code, documentation 28 | source, and configuration files. 29 | 30 | "Object" form shall mean any form resulting from mechanical 31 | transformation or translation of a Source form, including but 32 | not limited to compiled object code, generated documentation, 33 | and conversions to other media types. 34 | 35 | "Work" shall mean the work of authorship, whether in Source or 36 | Object form, made available under the License, as indicated by a 37 | copyright notice that is included in or attached to the work 38 | (an example is provided in the Appendix below). 39 | 40 | "Derivative Works" shall mean any work, whether in Source or Object 41 | form, that is based on (or derived from) the Work and for which the 42 | editorial revisions, annotations, elaborations, or other modifications 43 | represent, as a whole, an original work of authorship. For the purposes 44 | of this License, Derivative Works shall not include works that remain 45 | separable from, or merely link (or bind by name) to the interfaces of, 46 | the Work and Derivative Works thereof. 47 | 48 | "Contribution" shall mean any work of authorship, including 49 | the original version of the Work and any modifications or additions 50 | to that Work or Derivative Works thereof, that is intentionally 51 | submitted to Licensor for inclusion in the Work by the copyright owner 52 | or by an individual or Legal Entity authorized to submit on behalf of 53 | the copyright owner. For the purposes of this definition, "submitted" 54 | means any form of electronic, verbal, or written communication sent 55 | to the Licensor or its representatives, including but not limited to 56 | communication on electronic mailing lists, source code control systems, 57 | and issue tracking systems that are managed by, or on behalf of, the 58 | Licensor for the purpose of discussing and improving the Work, but 59 | excluding communication that is conspicuously marked or otherwise 60 | designated in writing by the copyright owner as "Not a Contribution." 61 | 62 | "Contributor" shall mean Licensor and any individual or Legal Entity 63 | on behalf of whom a Contribution has been received by Licensor and 64 | subsequently incorporated within the Work. 65 | 66 | 2. Grant of Copyright License. Subject to the terms and conditions of 67 | this License, each Contributor hereby grants to You a perpetual, 68 | worldwide, non-exclusive, no-charge, royalty-free, irrevocable 69 | copyright license to reproduce, prepare Derivative Works of, 70 | publicly display, publicly perform, sublicense, and distribute the 71 | Work and such Derivative Works in Source or Object form. 72 | 73 | 3. Grant of Patent License. Subject to the terms and conditions of 74 | this License, each Contributor hereby grants to You a perpetual, 75 | worldwide, non-exclusive, no-charge, royalty-free, irrevocable 76 | (except as stated in this section) patent license to make, have made, 77 | use, offer to sell, sell, import, and otherwise transfer the Work, 78 | where such license applies only to those patent claims licensable 79 | by such Contributor that are necessarily infringed by their 80 | Contribution(s) alone or by combination of their Contribution(s) 81 | with the Work to which such Contribution(s) was submitted. If You 82 | institute patent litigation against any entity (including a 83 | cross-claim or counterclaim in a lawsuit) alleging that the Work 84 | or a Contribution incorporated within the Work constitutes direct 85 | or contributory patent infringement, then any patent licenses 86 | granted to You under this License for that Work shall terminate 87 | as of the date such litigation is filed. 88 | 89 | 4. Redistribution. You may reproduce and distribute copies of the 90 | Work or Derivative Works thereof in any medium, with or without 91 | modifications, and in Source or Object form, provided that You 92 | meet the following conditions: 93 | 94 | (a) You must give any other recipients of the Work or 95 | Derivative Works a copy of this License; and 96 | 97 | (b) You must cause any modified files to carry prominent notices 98 | stating that You changed the files; and 99 | 100 | (c) You must retain, in the Source form of any Derivative Works 101 | that You distribute, all copyright, patent, trademark, and 102 | attribution notices from the Source form of the Work, 103 | excluding those notices that do not pertain to any part of 104 | the Derivative Works; and 105 | 106 | (d) If the Work includes a "NOTICE" text file as part of its 107 | distribution, then any Derivative Works that You distribute must 108 | include a readable copy of the attribution notices contained 109 | within such NOTICE file, excluding those notices that do not 110 | pertain to any part of the Derivative Works, in at least one 111 | of the following places: within a NOTICE text file distributed 112 | as part of the Derivative Works; within the Source form or 113 | documentation, if provided along with the Derivative Works; or, 114 | within a display generated by the Derivative Works, if and 115 | wherever such third-party notices normally appear. The contents 116 | of the NOTICE file are for informational purposes only and 117 | do not modify the License. You may add Your own attribution 118 | notices within Derivative Works that You distribute, alongside 119 | or as an addendum to the NOTICE text from the Work, provided 120 | that such additional attribution notices cannot be construed 121 | as modifying the License. 122 | 123 | You may add Your own copyright statement to Your modifications and 124 | may provide additional or different license terms and conditions 125 | for use, reproduction, or distribution of Your modifications, or 126 | for any such Derivative Works as a whole, provided Your use, 127 | reproduction, and distribution of the Work otherwise complies with 128 | the conditions stated in this License. 129 | 130 | 5. Submission of Contributions. Unless You explicitly state otherwise, 131 | any Contribution intentionally submitted for inclusion in the Work 132 | by You to the Licensor shall be under the terms and conditions of 133 | this License, without any additional terms or conditions. 134 | Notwithstanding the above, nothing herein shall supersede or modify 135 | the terms of any separate license agreement you may have executed 136 | with Licensor regarding such Contributions. 137 | 138 | 6. Trademarks. This License does not grant permission to use the trade 139 | names, trademarks, service marks, or product names of the Licensor, 140 | except as required for reasonable and customary use in describing the 141 | origin of the Work and reproducing the content of the NOTICE file. 142 | 143 | 7. Disclaimer of Warranty. Unless required by applicable law or 144 | agreed to in writing, Licensor provides the Work (and each 145 | Contributor provides its Contributions) on an "AS IS" BASIS, 146 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or 147 | implied, including, without limitation, any warranties or conditions 148 | of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A 149 | PARTICULAR PURPOSE. You are solely responsible for determining the 150 | appropriateness of using or redistributing the Work and assume any 151 | risks associated with Your exercise of permissions under this License. 152 | 153 | 8. Limitation of Liability. In no event and under no legal theory, 154 | whether in tort (including negligence), contract, or otherwise, 155 | unless required by applicable law (such as deliberate and grossly 156 | negligent acts) or agreed to in writing, shall any Contributor be 157 | liable to You for damages, including any direct, indirect, special, 158 | incidental, or consequential damages of any character arising as a 159 | result of this License or out of the use or inability to use the 160 | Work (including but not limited to damages for loss of goodwill, 161 | work stoppage, computer failure or malfunction, or any and all 162 | other commercial damages or losses), even if such Contributor 163 | has been advised of the possibility of such damages. 164 | 165 | 9. Accepting Warranty or Additional Liability. While redistributing 166 | the Work or Derivative Works thereof, You may choose to offer, 167 | and charge a fee for, acceptance of support, warranty, indemnity, 168 | or other liability obligations and/or rights consistent with this 169 | License. However, in accepting such obligations, You may act only 170 | on Your own behalf and on Your sole responsibility, not on behalf 171 | of any other Contributor, and only if You agree to indemnify, 172 | defend, and hold each Contributor harmless for any liability 173 | incurred by, or claims asserted against, such Contributor by reason 174 | of your accepting any such warranty or additional liability. 175 | 176 | END OF TERMS AND CONDITIONS 177 | 178 | APPENDIX: How to apply the Apache License to your work. 179 | 180 | To apply the Apache License to your work, attach the following 181 | boilerplate notice, with the fields enclosed by brackets "[]" 182 | replaced with your own identifying information. (Don't include 183 | the brackets!) The text should be enclosed in the appropriate 184 | comment syntax for the file format. We also recommend that a 185 | file or class name and description of purpose be included on the 186 | same "printed page" as the copyright notice for easier 187 | identification within third-party archives. 188 | 189 | Copyright [yyyy] [name of copyright owner] 190 | 191 | Licensed under the Apache License, Version 2.0 (the "License"); 192 | you may not use this file except in compliance with the License. 193 | You may obtain a copy of the License at 194 | 195 | http://www.apache.org/licenses/LICENSE-2.0 196 | 197 | Unless required by applicable law or agreed to in writing, software 198 | distributed under the License is distributed on an "AS IS" BASIS, 199 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 200 | See the License for the specific language governing permissions and 201 | limitations under the License. 202 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | ![image](https://img.shields.io/badge/test-passing-green.svg) 2 | 3 | # nginx-zuul-dynamic-lb 4 | :maple_leaf: 基于Lua的Spring Cloud网关高可用通用Ngnix插件 5 | 6 | *** 7 | 8 | ## 场景痛点 9 | 10 | ![image](https://github.com/SpringCloud/eureka-admin/blob/master/eureka-admin-sample/eureka-admin-sample-eureka-server/img/dynamic-eureka-server.png) 11 | 12 | 在Spring Cloud微服务架构体系中，我们往往会部署一个Zuul集群来横向扩展我们的微服务应用，集群的上层是Nginx软负载，在实际情况中，往往会遇到Zuul宕机的尴尬事情，这时候从Nginx到这台机器的请求就会全部失效。此项目针对此痛点，用lua脚本实现定时拉取特定服务地址，动态无感知增减Zuul在Nginx中的负载节点。 13 | 14 | 如果您希望实现从Nginx直接到普通服务的动态节点负载，在下文配置服务名与Eureka注册中心地址即可。 15 | 16 | ## OpenResty安装与配置 17 | ``` 18 | 1、环境 19 | yum -y install readline-devel pcre-devel openssl-devel gcc 20 | 2、下载解压OpenResty包 21 | wget https://openresty.org/download/openresty-1.13.6.1.tar.gz 22 | tar -zxvf openresty-1.13.6.1.tar.gz 23 | 3、下载ngx_cache_purge模块，该模块用于清理nginx缓存 24 | cd openresty-1.13.6.1/bundle 25 | wget http://labs.frickle.com/files/ngx_cache_purge-2.3.tar.gz 26 | tar -zxvf ngx_cache_purge-2.3.tar.gz 27 | 4、下载nginx_upstream_check_module模块，该模块用于upstream健康检查 28 | wget https://github.com/yaoweibin/nginx_upstream_check_module/archive/v0.3.0.tar.gz 29 | tar -zxvf v0.3.0.tar.gz 30 | 5、OpenResty配置增加 31 | cd openresty-1.13.6.1 32 | ./configure --prefix=/usr/servers --with-http_realip_module --with-pcre --with-luajit --add-module=./bundle/ngx_cache_purge-2.3/ --add-module=./bundle/nginx_upstream_check_module-0.3.0/ -j2 33 | 6、编译安装 34 | make 35 | make install 36 | 37 | 7、OpenResty没有http模块，需要单独安装 38 | cd /usr/servers/lualib/resty 39 | wget https://raw.githubusercontent.com/pintsized/lua-resty-http/master/lib/resty/http_headers.lua 40 | wget https://raw.githubusercontent.com/pintsized/lua-resty-http/master/lib/resty/http.lua 41 | 42 | 8、项目脚本拷贝到这里 43 | copy dynamic_eureka_balancer.lua into this dir 44 | 9、Nginx配置文件 45 | vim /usr/servers/nginx/conf/nginx.conf 46 | ``` 47 | ## Nginx配置 48 | ``` 49 | 50 | http { 51 | #sharing cache area 52 | lua_shared_dict dynamic_eureka_balancer 128m; 53 | 54 | init_worker_by_lua_block { 55 | -- init eureka balancer 56 | local file = require "resty.dynamic_eureka_balancer" 57 | local balancer = file:new({dict_name="dynamic_eureka_balancer"}) 58 | 59 | --eureka server list 60 | balancer.set_eureka_service_url({"127.0.0.1:8888", "127.0.0.1:9999"}) 61 | 62 | --eureka basic authentication 63 | --use this setting if eureka has enabled basic authentication. 64 | --note: basic authentication must use BASE64 encryption in `user:password` format 65 | --balancer.set_eureka_service_basic_authentication("") 66 | 67 | --The service name that needs to be monitored 68 | balancer.watch_service({"zuul", "client"}) 69 | } 70 | 71 | upstream springcloud_cn { 72 | server 127.0.0.1:666; # Required, because empty upstream block is rejected by nginx (nginx+ can use 'zone' instead) 73 | 74 | balancer_by_lua_block { 75 | 76 | --The zuul name that needs to be monitored 77 | local service_name = "zuul" 78 | 79 | local file = require "resty.dynamic_eureka_balancer" 80 | local balancer = file:new({dict_name="dynamic_eureka_balancer"}) 81 | 82 | --balancer.ip_hash(service_name) --IP Hash LB 83 | balancer.round_robin(service_name) --Round Robin LB 84 | } 85 | } 86 | 87 | server { 88 | listen 80; 89 | server_name localhost; 90 | 91 | location / { 92 | proxy_pass http://springcloud_cn/; 93 | proxy_set_header Host $http_host; 94 | proxy_set_header X-Real-IP $remote_addr; 95 | proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for; 96 | proxy_set_header X-Forwarded-Proto $scheme; 97 | } 98 | 99 | error_page 500 502 503 504 /50x.html; 100 | location = /50x.html { 101 | root html; 102 | } 103 | } 104 | } 105 | ``` 106 | -------------------------------------------------------------------------------- /dynamic_eureka_balancer.lua: -------------------------------------------------------------------------------- 1 | --- 2 | --- Generated by EmmyLua(https://github.com/EmmyLua) 3 | --- Created by Lovnx. 4 | --- DateTime: 2018/5/11 11:39 5 | --- 6 | 7 | local http = require "resty.http" 8 | local balancer = require "ngx.balancer" 9 | local json = require "cjson" 10 | 11 | local ok, new_tab = pcall(require, "table.new") 12 | if not ok or type(new_tab) ~= "function" then 13 | new_tab = function (narr, nrec) return {} end 14 | end 15 | 16 | local _M = new_tab(0, 100) 17 | 18 | _M.VERSION = "0.0.1" 19 | 20 | local mt = { __index = _M } 21 | 22 | local default_dict_name = "eureka_balancer" 23 | 24 | -- refresh success idle time (s) 25 | local watch_refresh_idle_time = 10 26 | 27 | -- refresh failed retry idle time (s) 28 | local refresh_retry_idle_time = 5 29 | 30 | -- max idle timeout (in ms) when the connection is in the pool 31 | local http_max_idle_timeout = 60000 32 | -- the maximal size of the pool every nginx worker process 33 | local http_pool_size = 10 34 | 35 | local function _timer(...) 36 | local ok, err = ngx.timer.at(...) 37 | if not ok then 38 | ngx.log(ngx.ERR, "eureka.balancer: failed to create timer: ", err) 39 | end 40 | end 41 | 42 | function split(s, p) 43 | local rt= {} 44 | string.gsub(s, '[^'..p..']+', function(w) table.insert(rt, w) end ) 45 | return rt 46 | end 47 | 48 | -- build eureka service uri 49 | local function build_eureka_uri(eureka_url_index) 50 | local uri = _M.eureka_service_urls[eureka_url_index] 51 | local ipAddr = split(uri, ":") 52 | local ip = ipAddr[1] 53 | local port = ipAddr[2] 54 | return ip, port 55 | end 56 | 57 | -- build eureka request params 58 | local function build_eureka_params(service_name) 59 | local uri = "/eureka/apps/" .. service_name 60 | local headers = {["Accept"]="application/json"} 61 | local eureka_service_basic_authentication = _M.eureka_service_basic_authentication 62 | if eureka_service_basic_authentication then 63 | local basic_authentication = "Basic " .. eureka_service_basic_authentication 64 | headers = {["Accept"]="application/json", ["Authorization"]=basic_authentication} 65 | end 66 | local params = {path=uri, method="GET", headers=headers} 67 | return params 68 | end 69 | 70 | local function incr_eureka_url_index(eureka_url_index) 71 | if eureka_url_index >= #_M.eureka_service_urls then 72 | eureka_url_index = 1 73 | else 74 | eureka_url_index = eureka_url_index + 1 75 | end 76 | return eureka_url_index 77 | end 78 | 79 | -- parse eureka result 80 | local function parse_service(body) 81 | local ok, res_json = pcall(function() 82 | return json.decode(body) 83 | end) 84 | if not ok then 85 | return nil, "JSON decode error" 86 | end 87 | 88 | local service = {} 89 | service.upstreams = {} 90 | 91 | for k, v in pairs(res_json) do 92 | local passing = true 93 | local instances = v["instance"] 94 | 95 | for i, instance in pairs(instances) do 96 | local status = instance["status"] 97 | if status == "UP" then 98 | local ipAddr = instance["ipAddr"] 99 | local port = instance["port"]["$"] 100 | table.insert(service.upstreams, {ip=ipAddr, port=port}) 101 | end 102 | end 103 | 104 | --ngx.log(ngx.INFO, "eureka.balancer: instance", json.encode(service.upstreams)) 105 | end 106 | return service 107 | end 108 | 109 | -- cache service to ngx shared dict 110 | local function cache_service(service_name, service) 111 | if not _M.shared_cache then 112 | return nil 113 | end 114 | ngx.log(ngx.INFO, "eureka.balancer: cache service to ngx shared: ", service_name, " ", json.encode(service)) 115 | _M.shared_cache:set(service_name, json.encode(service)) 116 | end 117 | 118 | -- get service from ngx shared dict 119 | local function aquire_service(service_name) 120 | if not _M.shared_cache then 121 | return nil 122 | end 123 | local service_json = _M.shared_cache:get(service_name) 124 | 125 | ngx.log(ngx.INFO, "eureka.balancer: aquire service from ngx shared: service_name: ", service_name, " ", service_json) 126 | 127 | return service_json and json.decode(service_json) or nil 128 | end 129 | 130 | -- only update upstreams if service already exist (round robin cursor) 131 | local function cache_service_upstreams(service_name, service) 132 | if not _M.shared_cache then 133 | return nil 134 | end 135 | ngx.log(ngx.INFO, "eureka.balancer: cache service upstreams to ngx shared: ", service_name, " ", json.encode(service)) 136 | 137 | local cached_service = aquire_service(service_name) 138 | if cached_service ~= nil then 139 | cached_service.upstreams = service.upstreams 140 | cache_service(service_name, cached_service) 141 | else 142 | cache_service(service_name, service) 143 | end 144 | end 145 | 146 | -- refresh service from eureka 147 | local function refresh_service(service_name, eureka_url_index) 148 | local httpc = http:new() 149 | --connect_timeout, send_timeout, read_timeout (in ms) 150 | httpc:set_timeouts(3000, 10000, 10000) 151 | local params = build_eureka_params(service_name) 152 | local s_ip, s_port = build_eureka_uri(eureka_url_index) 153 | 154 | ngx.log(ngx.INFO, "eureka.balancer: refresh_service : ", eureka_url_index, " ", s_ip, ":", s_port) 155 | 156 | local ok, err = httpc:connect(s_ip, s_port) 157 | if err ~= nil then 158 | ngx.log(ngx.ERR, "eureka.balancer: failed to connect eureka server: ", s_ip, ":", s_port, " ", err) 159 | return nil, err 160 | end 161 | 162 | local res, err = httpc:request(params) 163 | if err ~= nil then 164 | ngx.log(ngx.ERR, "eureka.balancer: failed to request eureka server: ", s_ip, ":", s_port, " ", err) 165 | return nil, err 166 | end 167 | 168 | if res.status ~= 200 then 169 | return nil, "bad response code: " .. res.status 170 | end 171 | 172 | local body = res:read_body() 173 | local service, err = parse_service(body) 174 | if err ~= nil then 175 | ngx.log(ngx.ERR, "eureka.balancer: failed to parse eureka service response: ", s_ip, ":", s_port, " ", err) 176 | return nil, err 177 | end 178 | 179 | local ok, err = httpc:set_keepalive(http_max_idle_timeout, http_pool_size) 180 | --local ok, err = httpc:set_keepalive() 181 | if err ~= nil then 182 | ngx.log(ngx.ERR, "eureka.balancer: failed to set keepalive for http client: ", err) 183 | end 184 | return service 185 | end 186 | 187 | 188 | local function watch(premature, service_name, eureka_url_index) 189 | if premature then 190 | return nil 191 | end 192 | 193 | eureka_url_index = eureka_url_index or 1 194 | 195 | local service, err = refresh_service(service_name, eureka_url_index) 196 | if err ~= nil then 197 | ngx.log(ngx.ERR, "eureka.balancer: failed watching service: ", service_name) 198 | 199 | eureka_url_index = incr_eureka_url_index(eureka_url_index) 200 | 201 | ngx.log(ngx.ERR, "eureka.balancer: failed watching service eureka_url_index: ", eureka_url_index) 202 | 203 | _timer(refresh_retry_idle_time, watch, service_name, eureka_url_index) 204 | return nil 205 | end 206 | 207 | service.name = service_name 208 | cache_service_upstreams(service_name, service) 209 | 210 | _timer(watch_refresh_idle_time, watch, service_name, eureka_url_index) 211 | end 212 | 213 | -- watch services 214 | function _M.watch_service(service_list) 215 | if ngx.worker.id() > 0 then 216 | return 217 | end 218 | for k,v in pairs(service_list) do 219 | _timer(0, watch, v, 1) 220 | end 221 | end 222 | 223 | -- round_robin incr service.cursor 224 | local function incr_service_cursor(service) 225 | service = service or {} 226 | if service.cursor == nil then 227 | service.cursor = 1 228 | else 229 | service.cursor = service.cursor + 1 230 | end 231 | if service.cursor > #service.upstreams then 232 | service.cursor = 1 233 | end 234 | return service.cursor 235 | end 236 | 237 | --round_robin 238 | function _M.round_robin(service_name) 239 | local service = aquire_service(service_name) 240 | if service == nil then 241 | ngx.log(ngx.ERR, "eureka.balancer: service not found: ", service_name) 242 | return ngx.exit(500) 243 | end 244 | if service.upstreams == nil or #service.upstreams == 0 then 245 | ngx.log(ngx.ERR, "eureka.balancer: no peers for service: ", service_name) 246 | return ngx.exit(500) 247 | end 248 | 249 | incr_service_cursor(service) 250 | 251 | -- TODO: https://github.com/openresty/lua-resty-core/blob/master/lib/ngx/balancer.md#get_last_failure 252 | if not balancer.get_last_failure() then 253 | balancer.set_more_tries(#service.upstreams - 1) 254 | end 255 | -- Picking next backend server 256 | local backend_server = service.upstreams[service.cursor] 257 | 258 | -- update service cursor 259 | cache_service(service_name, service) 260 | 261 | local ok, err = balancer.set_current_peer(backend_server["ip"], backend_server["port"]) 262 | if not ok then 263 | ngx.log(ngx.ERR, "eureka.balancer: failed to set the current peer: ", err) 264 | return ngx.exit(500) 265 | end 266 | end 267 | 268 | --ip_hash 269 | function _M.ip_hash(service_name) 270 | local service = aquire_service(service_name) 271 | if service == nil then 272 | ngx.log(ngx.ERR, "eureka.balancer: service not found: ", service_name) 273 | return ngx.exit(500) 274 | end 275 | local bs_size = #service.upstreams 276 | if service.upstreams == nil or bs_size == 0 then 277 | ngx.log(ngx.ERR, "eureka.balancer: no peers for service: ", service_name) 278 | return ngx.exit(500) 279 | end 280 | 281 | local remote_ip = ngx.var.remote_addr 282 | --local remote_port = ngx.var.remote_port 283 | local hash_key = remote_ip 284 | local hash = ngx.crc32_long(hash_key) 285 | local index = (hash % bs_size) + 1 286 | 287 | local backend_server = service.upstreams[index] 288 | local ok, err = balancer.set_current_peer(backend_server["ip"], backend_server["port"]) 289 | if not ok then 290 | ngx.log(ngx.ERR, "eureka.balancer: failed to set the current peer: ", err) 291 | return ngx.exit(500) 292 | end 293 | end 294 | 295 | 296 | local function set_shared_dict_name(dict_name) 297 | dict_name = dict_name or default_dict_name 298 | _M.shared_cache = ngx.shared[dict_name] 299 | if not _M.shared_cache then 300 | ngx.log(ngx.ERR, "eureka.balancer: unable to access shared dict ", dict_name) 301 | return ngx.exit(ngx.ERROR) 302 | end 303 | end 304 | 305 | -- set eureka service urls 306 | function _M.set_eureka_service_url(eureka_service_urls) 307 | _M.eureka_service_urls = eureka_service_urls 308 | if not _M.eureka_service_urls then 309 | ngx.log(ngx.ERR, "eureka.balancer: require set eureka_service_urls") 310 | return ngx.exit(ngx.ERROR) 311 | end 312 | end 313 | 314 | -- set eureka basic authentication info 315 | function _M.set_eureka_service_basic_authentication(eureka_service_basic_authentication) 316 | _M.eureka_service_basic_authentication = eureka_service_basic_authentication 317 | end 318 | 319 | function _M.new(self, opts) 320 | opts = opts or {} 321 | if opts.dict_name ~= nil then 322 | set_shared_dict_name(opts.dict_name) 323 | end 324 | return setmetatable({}, mt) 325 | end 326 | 327 | return _M --------------------------------------------------------------------------------