commit delayed return demo

Signed-off-by: yangyang <yang.yang@daocloud.io>

Author: yangyy93

.github/workflows/ci.yml

@@ -31,4 +31,4 @@ jobs:
         uses: docker/build-push-action@v4
         with:
           push: true
-          tags: release.daocloud.io/skoala/envoy-extproc-payloadlimit-demo-go:${{ github.sha }}
+          tags: release.daocloud.io/skoala/envoy-extproc-delayed-return:${{ github.sha }}

Dockerfile

@@ -20,9 +20,9 @@ FROM busybox
 COPY --from=0 /extproc /bin/extproc
 RUN chmod +x /bin/extproc
 
-ARG EXAMPLE=payload-limit
+ARG EXAMPLE=delayed-return
 
 EXPOSE 50051
 
 ENTRYPOINT [ "/bin/extproc" ]
-CMD [ "payload-limit", "--log-stream", "--log-phases", "payload-limit", "32"  ]
+CMD [ "delayed-return", "--log-stream", "--log-phases" ]

README.md

@@ -1,6 +1,6 @@
-# envoy-extproc-payloadlimit-demo-go
+# envoy-extproc-delayed-return-demo
 
-This repository contains a demo application written in Go that demonstrates the usage of Envoy's External Processor (ExtProc) filter to do `payload limit` for POST request.
+This repository contains a demo application written in Go that demonstrates the usage of Envoy's External Processor (ExtProc) filter to do `delayed return` for POST request.
 
 ## Overview
 
@@ -19,7 +19,7 @@ To get started with the demo application, follow these steps:
 
   1. Clone the repository:
      ```
-     git clone https://github.com/projectsesame/envoy-extproc-payloadlimit-demo-go.git
+     git clone https://github.com/projectsesame/envoy-extproc-delayed-return.git
      ```
 
   2. Build the Go application:
@@ -29,7 +29,7 @@ To get started with the demo application, follow these steps:
 
   3. Run the application:
      ```
-     ./envoy-extproc-payloadlimit-demo-go payload-limit --log-stream --log-phases payload-limit 32
+     ./envoy-extproc-delayed-return delayed-return --log-stream --log-phases
      ```
 
 
@@ -51,4 +51,4 @@ Special thanks to the community for their contributions and support.
 
 ## Contact
 
-For any questions or inquiries, please feel free to reach out to us for any assistance or feedback.
+For any questions or inquiries, please feel free to reach out to us for any assistance or feedback.

delayed-return.go

@@ -0,0 +1,108 @@
+package main
+
+import (
+	"log"
+	"strconv"
+	"time"
+
+	typev3 "github.com/envoyproxy/go-control-plane/envoy/type/v3"
+	ep "github.com/wrossmorrow/envoy-extproc-sdk-go"
+)
+
+type delayedReturnRequestProcessor struct {
+	opts *ep.ProcessingOptions
+}
+
+func (s *delayedReturnRequestProcessor) GetName() string {
+	return "delayed-return"
+}
+
+func (s *delayedReturnRequestProcessor) GetOptions() *ep.ProcessingOptions {
+	return s.opts
+}
+
+const kDelayedTime = "delayed-time"
+
+func (s *delayedReturnRequestProcessor) ProcessRequestHeaders(ctx *ep.RequestContext, headers ep.AllHeaders) error {
+	cancel := func(code int32, msg string) error {
+		return ctx.CancelRequest(code, map[string]ep.HeaderValue{}, typev3.StatusCode_name[code])
+	}
+
+	var delayedTime int64
+	var err error
+
+	if headers.RawHeaders[kDelayedTime] == nil {
+		delayedTime = 100
+		log.Printf("[%s], delayed-time header is not set, use default value 100", ctx.RequestID)
+
+	} else {
+		delayedTime, err = strconv.ParseInt(string(headers.RawHeaders[kDelayedTime]), 10, 64)
+		if err != nil {
+			return cancel(400, "delayed-time header is invalid")
+		}
+	}
+
+	if delayedTime < 0 {
+		return cancel(400, "delayed-time header is invalid")
+	}
+
+	ctx.SetValue("delayed-time", delayedTime)
+
+	logDelayedTime(ctx, "ProcessRequestHeaders")
+
+	return ctx.ContinueRequest()
+}
+
+func (s *delayedReturnRequestProcessor) ProcessRequestBody(ctx *ep.RequestContext, body []byte) error {
+
+	logDelayedTime(ctx, "ProcessRequestBody")
+
+	return ctx.ContinueRequest()
+}
+
+func logDelayedTime(ctx *ep.RequestContext, method string) {
+
+	var delayedTime int64
+
+	v, _ := ctx.GetValue(`delayed-time`)
+
+	delayedTime = v.(int64)
+
+	now := time.Now()
+	time.Sleep(time.Duration(delayedTime) * time.Millisecond)
+
+	log.Printf("[%s],process %s, delayed time: %d, elapsed time: %d\n", ctx.RequestID, method, delayedTime, time.Since(now).Milliseconds())
+
+}
+
+func (s *delayedReturnRequestProcessor) ProcessRequestTrailers(ctx *ep.RequestContext, trailers ep.AllHeaders) error {
+
+	logDelayedTime(ctx, "ProcessRequestTrailers")
+
+	return ctx.ContinueRequest()
+}
+
+func (s *delayedReturnRequestProcessor) ProcessResponseHeaders(ctx *ep.RequestContext, headers ep.AllHeaders) error {
+
+	logDelayedTime(ctx, "ProcessResponseHeaders")
+	return ctx.ContinueRequest()
+}
+
+func (s *delayedReturnRequestProcessor) ProcessResponseBody(ctx *ep.RequestContext, body []byte) error {
+
+	logDelayedTime(ctx, "ProcessResponseBody")
+	return ctx.ContinueRequest()
+}
+
+func (s *delayedReturnRequestProcessor) ProcessResponseTrailers(ctx *ep.RequestContext, trailers ep.AllHeaders) error {
+
+	logDelayedTime(ctx, "ProcessResponseTrailers")
+	return ctx.ContinueRequest()
+}
+
+func (s *delayedReturnRequestProcessor) Init(opts *ep.ProcessingOptions, nonFlagArgs []string) error {
+	s.opts = opts
+	return nil
+}
+
+func (s *delayedReturnRequestProcessor) Finish() {}

envoy.yaml

@@ -35,7 +35,7 @@ static_resources:
                       "@type": type.googleapis.com/envoy.extensions.filters.http.ext_proc.v3.ExternalProcessor
                       grpc_service:
                         envoy_grpc:
-                          cluster_name: payload-limit
+                          cluster_name: delayed-return
                         timeout: 30s
                       failure_mode_allow: true
                       message_timeout: 0.2s
@@ -70,19 +70,19 @@ static_resources:
                   socket_address:
                     address: upstream
                     port_value: 8000
-  - name: payload-limit
+  - name: delayed-return
     dns_lookup_family: V4_ONLY
     lb_policy: LEAST_REQUEST
     load_assignment:
-      cluster_name: payload-limit
+      cluster_name: delayed-return
       endpoints:
       - lb_endpoints:
         - endpoint:
             address:
               socket_address:
-                address: payload-limit
+                address: delayed-return
                 port_value: 50051
-            hostname: payload-limit
+            hostname: delayed-return
     type: LOGICAL_DNS
     typed_extension_protocol_options:
       envoy.extensions.upstreams.http.v3.HttpProtocolOptions:

main.go

@@ -16,7 +16,7 @@ type processor interface {
 }
 
 var processors = map[string]processor{
-	"payload-limit": &payloadLimitRequestProcessor{},
+	"delayed-return": &delayedReturnRequestProcessor{},
 }
 
 func parseArgs(args []string) (port *int, opts *ep.ProcessingOptions, nonFlagArgs []string) {