feat: rerun rewrite phase for newly added plugins in consumer

apisix/init.lua

@@ -456,6 +456,8 @@ function _M.http_access_phase()
                 api_ctx.matched_route = route
                 core.table.clear(api_ctx.plugins)
                 api_ctx.plugins = plugin.filter(api_ctx, route, api_ctx.plugins)
+                -- rerun rewrite phase for newly added plugins in consumer
+                plugin.rerun_plugins_of_consumer(api_ctx.plugins, api_ctx)
             end
         end
         plugin.run_plugin("access", plugins, api_ctx)

apisix/plugin.lua

@@ -503,6 +503,11 @@ local function merge_consumer_route(route_conf, consumer_conf)
         end
 
         new_route_conf.value.plugins[name] = conf
+
+        if (route_conf and route_conf.value and route_conf.value.plugins)
+                and not route_conf.value.plugins[name] then
+            new_route_conf.value.plugins[name]["_from_consumer"] = true
+        end
     end
 
     core.log.info("merged conf : ", core.json.delay_encode(new_route_conf))
@@ -805,4 +810,32 @@ function _M.run_global_rules(api_ctx, global_rules, phase_name)
 end
 
 
+function _M.rerun_plugins_of_consumer(plugins, api_ctx)
+    for i = 1, #plugins, 2 do
+        -- no need to rerun the auth plugins
+        if plugins[i + 1]["_from_consumer"] and plugins[i].type ~= "auth" then
+            local phase_func = plugins[i]["rewrite"]
+            if phase_func then
+                local code, body = phase_func(plugins[i + 1], api_ctx)
+                if code or body then
+                    if is_http then
+                        if code >= 400 then
+                            core.log.warn(plugins[i].name, " exits with http status code ", code)
+                        end
+
+                        core.response.exit(code, body)
+                    else
+                        if code >= 400 then
+                            core.log.warn(plugins[i].name, " exits with status code ", code)
+                        end
+
+                        ngx_exit(1)
+                    end
+                end
+            end
+        end
+    end
+    return api_ctx
+end
+
 return _M

t/node/consumer-plugin2.t

@@ -0,0 +1,240 @@
+#
+# 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';
+
+log_level('info');
+repeat_each(1);
+no_long_string();
+no_root_location();
+
+add_block_preprocessor(sub {
+    my ($block) = @_;
+
+    if (!$block->request) {
+        $block->set_value("request", "GET /t");
+    }
+
+    if (!$block->response_body) {
+        $block->set_value("response_body", "passed\n");
+    }
+
+    if (!$block->no_error_log && !$block->error_log) {
+        $block->set_value("no_error_log", "[error]\n[alert]");
+    }
+});
+
+
+our $debug_config = t::APISIX::read_file("conf/debug.yaml");
+$debug_config =~ s/basic:\n  enable: false/basic:\n  enable: true/;
+$debug_config =~ s/hook_conf:\n  enable: false/hook_conf:\n  enable: true/;
+
+run_tests;
+
+__DATA__
+
+=== TEST 1: configure non-auth plugins in the consumer and run it's rewrite phase
+--- config
+    location /t {
+        content_by_lua_block {
+            local t = require("lib.test_admin").test
+            local code, body = t('/apisix/admin/consumers/jack',
+                 ngx.HTTP_PUT,
+                [[{
+                    "username": "jack",
+                    "plugins": {
+                        "key-auth": {
+                            "key": "auth-jack"
+                        },
+                        "proxy-rewrite": {
+                            "uri": "/uri/plugin_proxy_rewrite",
+                            "headers": {
+                                "X-Api-Engine": "APISIX",
+                                "X-CONSUMER-ID": "1"
+                            }
+                        }
+                    }
+                }]]
+                )
+
+            local code, body = t('/apisix/admin/routes/1',
+                 ngx.HTTP_PUT,
+                 [[{
+                        "plugins": {
+                            "key-auth": {}
+                        },
+                        "upstream": {
+                            "nodes": {
+                                "127.0.0.1:1980": 1
+                            },
+                            "type": "roundrobin"
+                        },
+                        "uri": "/hello"
+                }]]
+                )
+
+            if code >= 300 then
+                ngx.status = code
+            end
+            ngx.say(body)
+        }
+    }
+--- response_body
+passed
+
+
+
+=== TEST 2: hit routes
+--- request
+GET /hello
+--- more_headers
+apikey: auth-jack
+--- response_body
+uri: /uri/plugin_proxy_rewrite
+apikey: auth-jack
+host: localhost
+x-api-engine: APISIX
+x-consumer-id: 1
+x-real-ip: 127.0.0.1
+
+
+
+=== TEST 3: trace plugins info for debug
+--- debug_config eval: $::debug_config
+--- config
+    location /t {
+        content_by_lua_block {
+            local json = require("toolkit.json")
+            local ngx_re = require("ngx.re")
+            local http = require "resty.http"
+            local httpc = http.new()
+            local headers = {}
+            headers["apikey"] = "auth-jack"
+            local uri = "http://127.0.0.1:" .. ngx.var.server_port .. "/hello"
+            local res, err = httpc:request_uri(uri, {
+                    method = "GET",
+                    headers = headers,
+                })
+            local debug_header = res.headers["Apisix-Plugins"]
+            local arr = ngx_re.split(debug_header, ", ")
+            local hash = {}
+            for i, v in ipairs(arr) do
+                hash[v] = true
+            end
+            ngx.status = res.status
+            ngx.say(json.encode(hash))
+        }
+    }
+--- response_body
+{"key-auth":true,"proxy-rewrite":true}
+
+
+
+=== TEST 4: configure non-auth plugins in the route and run it's rewrite phase
+--- config
+    location /t {
+        content_by_lua_block {
+            local t = require("lib.test_admin").test
+            local code, body = t('/apisix/admin/consumers/jack',
+                 ngx.HTTP_PUT,
+                [[{
+                    "username": "jack",
+                    "plugins": {
+                        "key-auth": {
+                            "key": "auth-jack"
+                        }
+                    }
+                }]]
+                )
+
+            local code, body = t('/apisix/admin/routes/1',
+                 ngx.HTTP_PUT,
+                 [[{
+                        "plugins": {
+                            "key-auth": {},
+                            "proxy-rewrite": {
+                                "uri": "/uri/plugin_proxy_rewrite",
+                                "headers": {
+                                    "X-Api-Engine": "APISIX",
+                                    "X-CONSUMER-ID": "1"
+                                }
+                            }
+                        },
+                        "upstream": {
+                            "nodes": {
+                                "127.0.0.1:1980": 1
+                            },
+                            "type": "roundrobin"
+                        },
+                        "uri": "/hello"
+                }]]
+                )
+
+            if code >= 300 then
+                ngx.status = code
+            end
+            ngx.say(body)
+        }
+    }
+--- response_body
+passed
+
+
+
+=== TEST 5: hit routes
+--- request
+GET /hello
+--- more_headers
+apikey: auth-jack
+--- response_body
+uri: /uri/plugin_proxy_rewrite
+apikey: auth-jack
+host: localhost
+x-api-engine: APISIX
+x-consumer-id: 1
+x-real-ip: 127.0.0.1
+
+
+
+=== TEST 6: trace plugins info for debug
+--- debug_config eval: $::debug_config
+--- config
+    location /t {
+        content_by_lua_block {
+            local json = require("toolkit.json")
+            local ngx_re = require("ngx.re")
+            local http = require "resty.http"
+            local httpc = http.new()
+            local headers = {}
+            headers["apikey"] = "auth-jack"
+            local uri = "http://127.0.0.1:" .. ngx.var.server_port .. "/hello"
+            local res, err = httpc:request_uri(uri, {
+                    method = "GET",
+                    headers = headers,
+                })
+            local debug_header = res.headers["Apisix-Plugins"]
+            local arr = ngx_re.split(debug_header, ", ")
+            local hash = {}
+            for i, v in ipairs(arr) do
+                hash[v] = true
+            end
+            ngx.status = res.status
+            ngx.say(json.encode(hash))
+        }
+    }
+--- response_body
+{"key-auth":true,"proxy-rewrite":true}