add test cases.

conf/config.yaml

@@ -36,3 +36,5 @@ plugins:                          # plugin list
   - zipkin
   - ip-restriction
   - grpc-transcode
+  - serverless-pre-function
+  - serverless-post-function

lua/apisix/plugins/serverless-post-function.lua

@@ -0,0 +1 @@
+return require("apisix.plugins.serverless")("serverless-post-function", -2000)

lua/apisix/plugins/serverless-pre-function.lua

@@ -1 +1 @@
-return require("apisix.plugins.serverless")("serverless-pre-function", math.huge)
+return require("apisix.plugins.serverless")("serverless-pre-function", 10000)

lua/apisix/plugins/serverless.lua

@@ -4,7 +4,17 @@ return function(plugin_name, priority)
     local schema = {
         type = "object",
         properties = {
-            functions = {type = "array", items = {type = "string"}, minItems = 1},
+            phase = {
+                type = "string",
+                -- the default phase is access
+                enum = {"rewrite", "access", "header_filer", "body_filter",
+                        "log", "balancer"}
+            },
+            functions = {
+                type = "array",
+                items = {type = "string"},
+                minItems = 1
+            },
         },
         required = {"functions"}
     }
@@ -15,16 +25,67 @@ return function(plugin_name, priority)
         name = plugin_name,
     }
 
+    local function load_funcs(functions)
+        local funcs = core.table.new(#functions, 0)
+
+        local index = 1
+        for _, func_str in ipairs(functions) do
+            local _, func = pcall(loadstring(func_str))
+            funcs[index] = func
+            index = index + 1
+        end
+
+        return funcs
+    end
+
+    local function call_funcs(phase, conf, ctx)
+        if phase ~= conf.phase then
+            return
+        end
+
+        local functions = core.lrucache.plugin_ctx(plugin_name, ctx,
+                                                   load_funcs, conf.functions)
+
+        for _, func in ipairs(functions) do
+            func()
+        end
+    end
+
     function _M.check_schema(conf)
         local ok, err = core.schema.check(schema, conf)
         if not ok then
             return false, err
         end
+
+        if not conf.phase then
+            conf.phase = 'access'
+        end
+
         return true
     end
 
+    function _M.rewrite(conf, ctx)
+        call_funcs('rewrite', conf, ctx)
+    end
+
     function _M.access(conf, ctx)
-        local functions = conf.functions
+        call_funcs('access', conf, ctx)
+    end
+
+    function _M.balancer(conf, ctx)
+        call_funcs('balancer', conf, ctx)
+    end
+
+    function _M.header_filer(conf, ctx)
+        call_funcs('header_filer', conf, ctx)
+    end
+
+    function _M.body_filter(conf, ctx)
+        call_funcs('body_filter', conf, ctx)
+    end
+
+    function _M.log(conf, ctx)
+        call_funcs('log', conf, ctx)
     end
 
     return _M

lua/apisix/plugins/zipkin.lua

@@ -24,7 +24,7 @@ local schema = {
 
 local _M = {
     version = 0.1,
-    priority = -1000, -- last running plugin
+    priority = -1000, -- last running plugin, but before serverless post func
     name = plugin_name,
     schema = schema,
 }

t/admin/plugins.t

@@ -14,6 +14,6 @@ __DATA__
 --- request
 GET /apisix/admin/plugins/list
 --- response_body_like eval
-qr/\["limit-req","limit-count","limit-conn","key-auth","prometheus","node-status","jwt-auth","zipkin","ip-restriction","grpc-transcode"\]/
+qr/\["limit-req","limit-count","limit-conn","key-auth","prometheus","node-status","jwt-auth","zipkin","ip-restriction","grpc-transcode","serverless-pre-function","serverless-post-function"\]/
 --- no_error_log
 [error]

t/debug-mode.t

@@ -39,6 +39,7 @@ done
 --- grep_error_log eval
 qr/loaded plugin and sort by priority: [-\d]+ name: [\w-]+/
 --- grep_error_log_out
+loaded plugin and sort by priority: 10000 name: serverless-pre-function
 loaded plugin and sort by priority: 3000 name: ip-restriction
 loaded plugin and sort by priority: 2510 name: jwt-auth
 loaded plugin and sort by priority: 2500 name: key-auth
@@ -50,6 +51,7 @@ loaded plugin and sort by priority: 506 name: grpc-transcode
 loaded plugin and sort by priority: 500 name: prometheus
 loaded plugin and sort by priority: 0 name: example-plugin
 loaded plugin and sort by priority: -1000 name: zipkin
+loaded plugin and sort by priority: -2000 name: serverless-post-function