mirror of
https://gitee.com/iresty/apisix.git
synced 2024-12-15 17:31:45 +08:00
e58a55fbc6
Previously the default number of retry is equal to the number of node, but the same node will be tried again according to its weight. Also ensure the same picker will be used in the whole request, especially during the retry.
559 lines
14 KiB
Perl
Vendored
559 lines
14 KiB
Perl
Vendored
#
|
|
# Licensed to the Apache Software Foundation (ASF) under one or more
|
|
# contributor license agreements. See the NOTICE file distributed with
|
|
# this work for additional information regarding copyright ownership.
|
|
# The ASF licenses this file to You under the Apache License, Version 2.0
|
|
# (the "License"); you may not use this file except in compliance with
|
|
# the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
use t::APISIX 'no_plan';
|
|
|
|
repeat_each(1);
|
|
log_level('info');
|
|
no_root_location();
|
|
worker_connections(1024);
|
|
no_shuffle();
|
|
|
|
run_tests();
|
|
|
|
__DATA__
|
|
|
|
=== TEST 1: set route(two upstream node)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local code, body = t('/apisix/admin/routes/1',
|
|
ngx.HTTP_PUT,
|
|
[[{
|
|
"uri": "/server_port",
|
|
"upstream": {
|
|
"key": "remote_addr",
|
|
"type": "chash",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1,
|
|
"127.0.0.1:1981": 1
|
|
}
|
|
}
|
|
}]]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 2: hit routes
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local http = require "resty.http"
|
|
local uri = "http://127.0.0.1:" .. ngx.var.server_port
|
|
.. "/server_port"
|
|
|
|
local ports_count = {}
|
|
for i = 1, 12 do
|
|
local httpc = http.new()
|
|
local res, err = httpc:request_uri(uri, {method = "GET"})
|
|
if not res then
|
|
ngx.say(err)
|
|
return
|
|
end
|
|
ports_count[res.body] = (ports_count[res.body] or 0) + 1
|
|
end
|
|
|
|
local ports_arr = {}
|
|
for port, count in pairs(ports_count) do
|
|
table.insert(ports_arr, {port = port, count = count})
|
|
end
|
|
|
|
local function cmd(a, b)
|
|
return a.port > b.port
|
|
end
|
|
table.sort(ports_arr, cmd)
|
|
|
|
ngx.say(require("toolkit.json").encode(ports_arr))
|
|
ngx.exit(200)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
[{"count":12,"port":"1980"}]
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 3: set route(three upstream node)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local code, body = t('/apisix/admin/routes/1',
|
|
ngx.HTTP_PUT,
|
|
[[{
|
|
"uri": "/server_port",
|
|
"upstream": {
|
|
"key": "remote_addr",
|
|
"type": "chash",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1,
|
|
"127.0.0.1:1981": 1,
|
|
"127.0.0.1:1982": 1
|
|
}
|
|
}
|
|
}]]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 4: hit routes
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local http = require "resty.http"
|
|
local uri = "http://127.0.0.1:" .. ngx.var.server_port
|
|
.. "/server_port"
|
|
|
|
local ports_count = {}
|
|
for i = 1, 12 do
|
|
local httpc = http.new()
|
|
local res, err = httpc:request_uri(uri, {method = "GET"})
|
|
if not res then
|
|
ngx.say(err)
|
|
return
|
|
end
|
|
ports_count[res.body] = (ports_count[res.body] or 0) + 1
|
|
end
|
|
|
|
local ports_arr = {}
|
|
for port, count in pairs(ports_count) do
|
|
table.insert(ports_arr, {port = port, count = count})
|
|
end
|
|
|
|
local function cmd(a, b)
|
|
return a.port > b.port
|
|
end
|
|
table.sort(ports_arr, cmd)
|
|
|
|
ngx.say(require("toolkit.json").encode(ports_arr))
|
|
ngx.exit(200)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
[{"count":12,"port":"1982"}]
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 5: set route(three upstream node)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local code, body = t('/apisix/admin/routes/1',
|
|
ngx.HTTP_PUT,
|
|
[[{
|
|
"uri": "/server_port",
|
|
"upstream": {
|
|
"key": "remote_addr",
|
|
"type": "chash",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1,
|
|
"127.0.0.1:1981": 0,
|
|
"127.0.0.1:1982": 0
|
|
}
|
|
}
|
|
}]]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 6: hit routes
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local http = require "resty.http"
|
|
local uri = "http://127.0.0.1:" .. ngx.var.server_port
|
|
.. "/server_port"
|
|
|
|
local ports_count = {}
|
|
for i = 1, 12 do
|
|
local httpc = http.new()
|
|
local res, err = httpc:request_uri(uri, {method = "GET"})
|
|
if not res then
|
|
ngx.say(err)
|
|
return
|
|
end
|
|
ports_count[res.body] = (ports_count[res.body] or 0) + 1
|
|
end
|
|
|
|
local ports_arr = {}
|
|
for port, count in pairs(ports_count) do
|
|
table.insert(ports_arr, {port = port, count = count})
|
|
end
|
|
|
|
local function cmd(a, b)
|
|
return a.port > b.port
|
|
end
|
|
table.sort(ports_arr, cmd)
|
|
|
|
ngx.say(require("toolkit.json").encode(ports_arr))
|
|
ngx.exit(200)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
[{"count":12,"port":"1980"}]
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 7 set route(three upstream node with querystring)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local code, body = t('/apisix/admin/routes/1',
|
|
ngx.HTTP_PUT,
|
|
[[{
|
|
"uri": "/server_port",
|
|
"upstream": {
|
|
"key": "query_string",
|
|
"type": "chash",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1,
|
|
"127.0.0.1:1981": 1,
|
|
"127.0.0.1:1982": 1
|
|
}
|
|
}
|
|
}]]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 8: hit routes with querystring
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local http = require "resty.http"
|
|
local uri = "http://127.0.0.1:" .. ngx.var.server_port
|
|
.. "/server_port?var=2&var2="
|
|
|
|
local t = {}
|
|
local ports_count = {}
|
|
for i = 1, 180 do
|
|
local th = assert(ngx.thread.spawn(function(i)
|
|
local httpc = http.new()
|
|
local res, err = httpc:request_uri(uri..i, {method = "GET"})
|
|
if not res then
|
|
ngx.log(ngx.ERR, err)
|
|
return
|
|
end
|
|
ports_count[res.body] = (ports_count[res.body] or 0) + 1
|
|
end, i))
|
|
table.insert(t, th)
|
|
end
|
|
for i, th in ipairs(t) do
|
|
ngx.thread.wait(th)
|
|
end
|
|
|
|
local ports_arr = {}
|
|
for port, count in pairs(ports_count) do
|
|
table.insert(ports_arr, {port = port, count = count})
|
|
end
|
|
|
|
local function cmd(a, b)
|
|
return a.count > b.count
|
|
end
|
|
table.sort(ports_arr, cmd)
|
|
|
|
if (ports_arr[1].count - ports_arr[3].count) / ports_arr[2].count > 0.2 then
|
|
ngx.say(require("toolkit.json").encode(ports_arr))
|
|
else
|
|
ngx.say('ok')
|
|
end
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- wait: 5
|
|
--- response_body
|
|
ok
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 9: set route(three upstream node with arg_xxx)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local code, body = t('/apisix/admin/routes/1',
|
|
ngx.HTTP_PUT,
|
|
[[{
|
|
"uri": "/server_port",
|
|
"upstream": {
|
|
"key": "arg_device_id",
|
|
"type": "chash",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 1,
|
|
"127.0.0.1:1981": 1,
|
|
"127.0.0.1:1982": 1
|
|
}
|
|
}
|
|
}]]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 10: hit routes with args
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local http = require "resty.http"
|
|
local uri = "http://127.0.0.1:" .. ngx.var.server_port
|
|
.. "/server_port?device_id="
|
|
|
|
local t = {}
|
|
local ports_count = {}
|
|
for i = 1, 180 do
|
|
local th = assert(ngx.thread.spawn(function(i)
|
|
local httpc = http.new()
|
|
local res, err = httpc:request_uri(uri..i, {method = "GET"})
|
|
if not res then
|
|
ngx.log(ngx.ERR, err)
|
|
return
|
|
end
|
|
ports_count[res.body] = (ports_count[res.body] or 0) + 1
|
|
end, i))
|
|
table.insert(t, th)
|
|
end
|
|
for i, th in ipairs(t) do
|
|
ngx.thread.wait(th)
|
|
end
|
|
|
|
local ports_arr = {}
|
|
for port, count in pairs(ports_count) do
|
|
table.insert(ports_arr, {port = port, count = count})
|
|
end
|
|
|
|
local function cmd(a, b)
|
|
return a.count > b.count
|
|
end
|
|
table.sort(ports_arr, cmd)
|
|
|
|
if (ports_arr[1].count - ports_arr[3].count) / ports_arr[2].count > 0.2 then
|
|
ngx.say(require("toolkit.json").encode(ports_arr))
|
|
else
|
|
ngx.say('ok')
|
|
end
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- wait: 5
|
|
--- response_body
|
|
ok
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 11: set route(weight 0)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local code, body = t('/apisix/admin/routes/1',
|
|
ngx.HTTP_PUT,
|
|
[[{
|
|
"uri": "/server_port",
|
|
"upstream": {
|
|
"key": "arg_device_id",
|
|
"type": "chash",
|
|
"nodes": {
|
|
"127.0.0.1:1980": 0,
|
|
"127.0.0.1:1981": 0
|
|
}
|
|
}
|
|
}]]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 12: hit routes
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local http = require "resty.http"
|
|
local uri = "http://127.0.0.1:" .. ngx.var.server_port
|
|
.. "/server_port?device_id=1"
|
|
|
|
local httpc = http.new()
|
|
local res, err = httpc:request_uri(uri, {method = "GET"})
|
|
if not res then
|
|
ngx.say(err)
|
|
return
|
|
end
|
|
|
|
ngx.status = res.status
|
|
ngx.say(res.body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- error_code_like: ^(?:50\d)$
|
|
--- error_log
|
|
failed to find valid upstream server, no valid upstream node
|
|
|
|
|
|
|
|
=== TEST 13: set route(ensure retry can try every node)
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local t = require("lib.test_admin").test
|
|
local code, body = t('/apisix/admin/routes/1',
|
|
ngx.HTTP_PUT,
|
|
[[{
|
|
"uri": "/server_port",
|
|
"upstream": {
|
|
"key": "arg_device_id",
|
|
"type": "chash",
|
|
"nodes": {
|
|
"127.0.0.1:1979": 1000,
|
|
"127.0.0.1:1980": 1
|
|
}
|
|
}
|
|
}]]
|
|
)
|
|
|
|
if code >= 300 then
|
|
ngx.status = code
|
|
end
|
|
ngx.say(body)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
passed
|
|
--- no_error_log
|
|
[error]
|
|
|
|
|
|
|
|
=== TEST 14: hit routes
|
|
--- config
|
|
location /t {
|
|
content_by_lua_block {
|
|
local http = require "resty.http"
|
|
local uri = "http://127.0.0.1:" .. ngx.var.server_port
|
|
.. "/server_port?device_id=1"
|
|
|
|
local httpc = http.new()
|
|
local res, err = httpc:request_uri(uri, {method = "GET"})
|
|
if not res then
|
|
ngx.say(err)
|
|
return
|
|
end
|
|
|
|
ngx.say(res.status)
|
|
}
|
|
}
|
|
--- request
|
|
GET /t
|
|
--- response_body
|
|
200
|