openresty-ext/src/assembly/resources/openresty/nginx/luaext/loadbalance/baseupstream.lua

   1 --[[\r
   2 \r
   3     Copyright (C) 2016 ZTE, Inc. and others. All rights reserved. (ZTE)\r
   4 \r
   5     Licensed under the Apache License, Version 2.0 (the "License");\r
   6     you may not use this file except in compliance with the License.\r
   7     You may obtain a copy of the License at\r
   8 \r
   9             http://www.apache.org/licenses/LICENSE-2.0\r
  10 \r
  11     Unless required by applicable law or agreed to in writing, software\r
  12     distributed under the License is distributed on an "AS IS" BASIS,\r
  13     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  14     See the License for the specific language governing permissions and\r
  15     limitations under the License.\r
  16 \r
  17 --]]\r
  18 \r
  19 local _M = {\r
  20         _VERSION = '1.0.0'\r
  21 }\r
  22 local policymodule = require "loadbalance.policy.roundrobin"\r
  23 local tbl_util  = require('lib.utils.table_util')\r
  24 local peerwatcher = require "loadbalance.peerwatcher"\r
  25 local tbl_isempty = tbl_util.isempty\r
  26 \r
  27 function _M.get_backserver(svc_key,servers)\r
  28         if tbl_isempty(servers) then return nil,"server list is empty" end\r
  29 \r
  30         local servers_num = #servers\r
  31         if not ngx.ctx.tried_num then\r
  32                 ngx.ctx.tried_num = 0\r
  33         end\r
  34         local server\r
  35         if servers_num==1 then\r
  36                 ngx.ctx.tried_num = ngx.ctx.tried_num+1\r
  37                 -- return it directly if there is only one server\r
  38                 server = servers[1]\r
  39                 if peerwatcher.is_server_ok(svc_key,server) then\r
  40                         return server,"" \r
  41                 else \r
  42                         return nil,"only one server but is not available"\r
  43                 end\r
  44         end\r
  45         for i=ngx.ctx.tried_num+1,servers_num do\r
  46                 ngx.ctx.tried_num = ngx.ctx.tried_num+1\r
  47                 server = policymodule.select_backserver(servers,svc_key)\r
  48                 if peerwatcher.is_server_ok(svc_key,server) then\r
  49                         return server,""\r
  50                 end\r
  51         end\r
  52         return nil,"serveral server but no one is available"\r
  53 end\r
  54 \r
  55 function _M.can_retry(svc_key,servers)\r
  56         return ngx.ctx.tried_num < #servers\r
  57 end\r
  58 \r
  59 function _M.mark_srv_failed(svc_key, srv)\r
  60         peerwatcher.set_srv_status(svc_key, srv, true)\r
  61 end\r
  62 \r
  63 function _M.check_and_reset_srv_status_ifneed(svc_key, servers)\r
  64         peerwatcher.check_and_reset_srv_status_ifneed(svc_key,servers)\r
  65 end\r
  66 return _M