| # |
| # 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); |
| no_long_string(); |
| no_root_location(); |
| no_shuffle(); |
| log_level("info"); |
| |
| |
| add_block_preprocessor(sub { |
| my ($block) = @_; |
| |
| my $http_config = $block->http_config // <<_EOC_; |
| |
| server { |
| listen 30511; |
| |
| location /hello { |
| content_by_lua_block { |
| ngx.say("server 1") |
| } |
| } |
| } |
| _EOC_ |
| |
| $block->set_value("http_config", $http_config); |
| }); |
| |
| our $yaml_config = <<_EOC_; |
| apisix: |
| node_listen: 1984 |
| config_center: yaml |
| enable_admin: false |
| enable_control: true |
| |
| discovery: |
| consul_kv: |
| servers: |
| - "http://127.0.0.1:8500" |
| dump: |
| path: "consul_kv.dump" |
| load_on_init: true |
| _EOC_ |
| |
| |
| run_tests(); |
| |
| __DATA__ |
| |
| === TEST 1: prepare nodes |
| --- config |
| location /v1/kv { |
| proxy_pass http://127.0.0.1:8500; |
| } |
| --- request eval |
| [ |
| "DELETE /v1/kv/upstreams/?recurse=true", |
| "PUT /v1/kv/upstreams/webpages/127.0.0.1:30511\n" . "{\"weight\": 1, \"max_fails\": 1, \"fail_timeout\": 1}", |
| ] |
| --- response_body eval |
| [ |
| 'true', |
| 'true', |
| 'true', |
| ] |
| |
| |
| |
| === TEST 2: show dump services |
| --- yaml_config eval: $::yaml_config |
| --- config |
| location /t { |
| content_by_lua_block { |
| local json = require("toolkit.json") |
| local t = require("lib.test_admin") |
| ngx.sleep(2) |
| |
| local code, body, res = t.test('/v1/discovery/consul_kv/show_dump_file', |
| ngx.HTTP_GET) |
| local entity = json.decode(res) |
| ngx.say(json.encode(entity.services)) |
| } |
| } |
| --- timeout: 3 |
| --- request |
| GET /t |
| --- response_body |
| {"http://127.0.0.1:8500/v1/kv/upstreams/webpages/":[{"host":"127.0.0.1","port":30511,"weight":1}]} |
| |
| |
| |
| === TEST 3: prepare dump file for next test |
| --- yaml_config |
| apisix: |
| node_listen: 1984 |
| config_center: yaml |
| enable_admin: false |
| enable_control: true |
| |
| discovery: |
| consul_kv: |
| servers: |
| - "http://127.0.0.1:8500" |
| dump: |
| path: "/tmp/consul_kv.dump" |
| load_on_init: true |
| #END |
| --- apisix_yaml |
| routes: |
| - |
| uri: /* |
| upstream: |
| service_name: http://127.0.0.1:8500/v1/kv/upstreams/webpages/ |
| discovery_type: consul_kv |
| type: roundrobin |
| #END |
| --- request |
| GET /hello |
| --- response_body |
| server 1 |
| |
| |
| |
| === TEST 4: clean registered nodes |
| --- config |
| location /v1/kv { |
| proxy_pass http://127.0.0.1:8500; |
| } |
| --- request eval |
| [ |
| "DELETE /v1/kv/upstreams/?recurse=true", |
| ] |
| --- response_body eval |
| [ |
| 'true' |
| ] |
| |
| |
| |
| === TEST 5: test load dump on init |
| Configure the invalid consul server addr, and loading the last test 3 generated /tmp/consul_kv.dump file into memory when initializing |
| --- yaml_config |
| apisix: |
| node_listen: 1984 |
| config_center: yaml |
| enable_admin: false |
| enable_control: true |
| |
| discovery: |
| consul_kv: |
| servers: |
| - "http://127.0.0.1:38500" |
| dump: |
| path: "/tmp/consul_kv.dump" |
| load_on_init: true |
| #END |
| --- apisix_yaml |
| routes: |
| - |
| uri: /* |
| upstream: |
| service_name: http://127.0.0.1:8500/v1/kv/upstreams/webpages/ |
| discovery_type: consul_kv |
| type: roundrobin |
| #END |
| --- request |
| GET /hello |
| --- response_body |
| server 1 |
| |
| |
| |
| === TEST 6: delete dump file |
| --- config |
| location /t { |
| content_by_lua_block { |
| local util = require("apisix.cli.util") |
| local succ, err = util.execute_cmd("rm -f /tmp/consul_kv.dump") |
| ngx.say(succ and "success" or err) |
| } |
| } |
| --- request |
| GET /t |
| --- response_body |
| success |
| |
| |
| |
| === TEST 7: miss load dump on init |
| --- yaml_config |
| apisix: |
| node_listen: 1984 |
| config_center: yaml |
| enable_admin: false |
| enable_control: true |
| |
| discovery: |
| consul_kv: |
| servers: |
| - "http://127.0.0.1:38500" |
| dump: |
| path: "/tmp/consul_kv.dump" |
| load_on_init: true |
| #END |
| --- apisix_yaml |
| routes: |
| - |
| uri: /* |
| upstream: |
| service_name: http://127.0.0.1:8500/v1/kv/upstreams/webpages/ |
| discovery_type: consul_kv |
| type: roundrobin |
| #END |
| --- request |
| GET /hello |
| --- error_code: 503 |
| |
| |
| |
| === TEST 8: prepare expired dump file |
| --- config |
| location /t { |
| content_by_lua_block { |
| local util = require("apisix.cli.util") |
| local json = require("toolkit.json") |
| |
| local applications = json.decode('{"http://127.0.0.1:8500/v1/kv/upstreams/webpages/":[{"host":"127.0.0.1","port":30511,"weight":1}]}') |
| local entity = { |
| services = applications, |
| last_update = ngx.time(), |
| expire = 10, |
| } |
| local succ, err = util.write_file("/tmp/consul_kv.dump", json.encode(entity)) |
| |
| ngx.sleep(2) |
| ngx.say(succ and "success" or err) |
| } |
| } |
| --- timeout: 3 |
| --- request |
| GET /t |
| --- response_body |
| success |
| |
| |
| |
| === TEST 9: unexpired dump |
| test load unexpired /tmp/consul_kv.dump file generated by upper test when initializing |
| when initializing |
| --- yaml_config |
| apisix: |
| node_listen: 1984 |
| config_center: yaml |
| enable_admin: false |
| enable_control: true |
| |
| discovery: |
| consul_kv: |
| servers: |
| - "http://127.0.0.1:38500" |
| dump: |
| path: "/tmp/consul_kv.dump" |
| load_on_init: true |
| expire: 5 |
| #END |
| --- apisix_yaml |
| routes: |
| - |
| uri: /* |
| upstream: |
| service_name: http://127.0.0.1:8500/v1/kv/upstreams/webpages/ |
| discovery_type: consul_kv |
| type: roundrobin |
| #END |
| --- request |
| GET /hello |
| --- response_body |
| server 1 |
| |
| |
| |
| === TEST 10: expired dump |
| test load expired ( by check: (dump_file.last_update + dump.expire) < ngx.time ) ) /tmp/consul_kv.dump file generated by upper test when initializing |
| when initializing |
| --- yaml_config |
| apisix: |
| node_listen: 1984 |
| config_center: yaml |
| enable_admin: false |
| enable_control: true |
| |
| discovery: |
| consul_kv: |
| servers: |
| - "http://127.0.0.1:38500" |
| dump: |
| path: "/tmp/consul_kv.dump" |
| load_on_init: true |
| expire: 1 |
| #END |
| --- apisix_yaml |
| routes: |
| - |
| uri: /* |
| upstream: |
| service_name: http://127.0.0.1:8500/v1/kv/upstreams/webpages/ |
| discovery_type: consul_kv |
| type: roundrobin |
| #END |
| --- request |
| GET /hello |
| --- error_code: 503 |
| --- error_log |
| dump file: /tmp/consul_kv.dump had expired, ignored it |
| |
| |
| |
| === TEST 11: delete dump file |
| --- config |
| location /t { |
| content_by_lua_block { |
| local util = require("apisix.cli.util") |
| local succ, err = util.execute_cmd("rm -f /tmp/consul_kv.dump") |
| ngx.say(succ and "success" or err) |
| } |
| } |
| --- request |
| GET /t |
| --- response_body |
| success |
| |
| |
| |
| === TEST 12: dump file inexistence |
| --- yaml_config |
| apisix: |
| node_listen: 1984 |
| config_center: yaml |
| enable_admin: false |
| enable_control: true |
| |
| discovery: |
| consul_kv: |
| servers: |
| - "http://127.0.0.1:38500" |
| dump: |
| path: "/tmp/consul_kv.dump" |
| #END |
| --- request |
| GET /v1/discovery/consul_kv/show_dump_file |
| --- error_code: 503 |
| |
| |
| |
| === TEST 13: no dump config |
| --- yaml_config |
| apisix: |
| node_listen: 1984 |
| config_center: yaml |
| enable_admin: false |
| enable_control: true |
| |
| discovery: |
| consul_kv: |
| servers: |
| - "http://127.0.0.1:38500" |
| #END |
| --- request |
| GET /v1/discovery/consul_kv/show_dump_file |
| --- error_code: 503 |