一、工具准备

需要先安装 Golang 和 TinyGo 两个程序

1. Golang

(要求 1.18 版本以上)
官方指引链接:https://go.dev/doc/install

Windows

  1. 下载安装文件:https://go.dev/dl/go1.19.windows-amd64.msi
  2. 打开下载好的安装文件直接安装,默认会安装到 Program FilesProgram Files (x86) 目录
  3. 安装完成后,使用键盘上的快捷键“Win+R”打开运行窗口,在运行窗口中输入“cmd”点击确定即可打开命令窗口,输入命令:go version,输出当前安装的版本,表明安装成功

MacOS

  1. 下载安装文件:https://go.dev/dl/go1.19.darwin-amd64.pkg
  2. 打开下载好的安装文件直接安装,默认会安装到/usr/local/go目录
  3. 打开终端命令行工具,输入命令:go version,输出当前安装的版本,表明安装成功

Linux

  1. 下载安装文件:https://go.dev/dl/go1.19.linux-amd64.tar.gz
  2. 执行下列命令进行安装:
  1. rm -rf /usr/local/go && tar -C /usr/local -xzf go1.19.linux-amd64.tar.gz
  2. export PATH=$PATH:/usr/local/go/bin
  1. 执行 go version,输出当前安装的版本,表明安装成功

2. TinyGo

(要求 0.28.1 版本以上)
官方指引链接:https://tinygo.org/getting-started/install/

Windows

  1. 下载安装文件:https://github.com/tinygo-org/tinygo/releases/download/v0.28.1/tinygo0.28.1.windows-amd64.zip
  2. 解压安装文件到指定目录
  3. 如果安装解压后的目录为C:\tinygo,则需要将C:\tinygo\bin添加到环境变量PATH中,例如通过在命令窗口中输入 set 命令设置
  1. set PATH=%PATH%;”C:\tinygo\bin”;
  1. 在命令窗口执行命令 tinygo version,输出当前安装的版本,表明安装成功

MacOS

  1. 下载压缩包并解压
  1. tar -zxf tinygo0.28.1.darwin-amd64.tar.gz
  1. 如果安装解压后的目录为/tmp,则需要将/tmp/tinygo/bin添加到环境变量PATH中:
  1. export PATH=/tmp/tinygo/bin:$PATH
  1. 在终端执行 tinygo version,输出当前安装的版本,表明安装成功

Linux

以 Ubuntu 下 amd64 架构为例,其他系统请参考官方指引链接

  1. 下载 DEB 文件,并安装
  1. sudo dpkg -i tinygo_0.28.1_amd64.deb
  2. export PATH=$PATH:/usr/local/bin
  1. 在终端执行 tinygo version,输出当前安装的版本,表明安装成功

二、编写插件

1. 初始化工程目录

  1. 新建一个工程目录文件,例如wasm-demo-go
  2. 在所建目录下执行以下命令,进行 Go 工程初始化
  1. go mod init wasm-demo-go
  1. 国内环境可能需要设置下载依赖包的代理
  1. 下载构建插件的依赖
  1. go get github.com/higress-group/proxy-wasm-go-sdk
  2. go get github.com/alibaba/higress/plugins/wasm-go@main
  3. go get github.com/tidwall/gjson

2. 编写 main.go 文件

下面是一个简单示例,实现了在插件配置mockEnable: true时直接返回hello world应答;未做插件配置,或者设置mockEnable: false时给原始请求添加 hello: world请求头。更多例子请参考本文第四节。

注意:在网关控制台中的插件配置为 yaml 格式,下发给插件时将自动转换为 json 格式,所以例子中的 parseConfig 可以直接从 json 中解析配置

  1. package main
  2. import (
  3. github.com/alibaba/higress/plugins/wasm-go/pkg/wrapper
  4. github.com/higress-group/proxy-wasm-go-sdk/proxywasm
  5. github.com/higress-group/proxy-wasm-go-sdk/proxywasm/types
  6. github.com/tidwall/gjson
  7. )
  8. func main() {
  9. wrapper.SetCtx(
  10. // 插件名称
  11. my-plugin”,
  12. // 为解析插件配置,设置自定义函数
  13. wrapper.ParseConfigBy(parseConfig),
  14. // 为处理请求头,设置自定义函数
  15. wrapper.ProcessRequestHeadersBy(onHttpRequestHeaders),
  16. )
  17. }
  18. // 自定义插件配置
  19. type MyConfig struct {
  20. mockEnable bool
  21. }
  22. // 在控制台插件配置中填写的yaml配置会自动转换为json,此处直接从json这个参数里解析配置即可
  23. func parseConfig(json gjson.Result, config *MyConfig, log wrapper.Log) error {
  24. // 解析出配置,更新到config中
  25. config.mockEnable = json.Get(“mockEnable”).Bool()
  26. return nil
  27. }
  28. func onHttpRequestHeaders(ctx wrapper.HttpContext, config MyConfig, log wrapper.Log) types.Action {
  29. proxywasm.AddHttpRequestHeader(“hello”, world”)
  30. if config.mockEnable {
  31. proxywasm.SendHttpResponse(200, nil, []byte(“hello world”), -1)
  32. }
  33. return types.ActionContinue
  34. }

HTTP 处理挂载点

上面示例代码中通过 wrapper.ProcessRequestHeadersBy将自定义函数 onHttpRequestHeaders用于HTTP 请求头处理阶段处理请求。除此之外,还可以通过下面方式,设置其他阶段的自定义处理函数

HTTP 处理阶段触发时机挂载方法
HTTP 请求头处理阶段网关接收到客户端发送来的请求头数据时wrapper.ProcessRequestHeadersBy
HTTP 请求 Body 处理阶段网关接收到客户端发送来的请求 Body 数据时wrapper.ProcessRequestBodyBy
HTTP 应答头处理阶段网关接收到后端服务响应的应答头数据时wrapper.ProcessResponseHeadersBy
HTTP 应答 Body 处理阶段网关接收到后端服务响应的应答 Body 数据时wrapper.ProcessResponseBodyBy

工具方法

上面示例代码中的 proxywasm.AddHttpRequestHeaderproxywasm.SendHttpResponse是插件 SDK 提供的两个工具方法,主要的工具方法见下表:

分类方法名称用途可以生效的
HTTP 处理阶段
请求头处理GetHttpRequestHeaders获取客户端请求的全部请求头HTTP 请求头处理阶段
ReplaceHttpRequestHeaders替换客户端请求的全部请求头HTTP 请求头处理阶段
GetHttpRequestHeader获取客户端请求的指定请求头HTTP 请求头处理阶段
RemoveHttpRequestHeader移除客户端请求的指定请求头HTTP 请求头处理阶段
ReplaceHttpRequestHeader替换客户端请求的指定请求头HTTP 请求头处理阶段
AddHttpRequestHeader新增一个客户端请求头HTTP 请求头处理阶段
请求 Body 处理GetHttpRequestBody获取客户端请求 BodyHTTP 请求 Body 处理阶段
AppendHttpRequestBody将指定的字节串附加到客户端请求 Body 末尾HTTP 请求 Body 处理阶段
PrependHttpRequestBody将指定的字节串附加到客户端请求 Body 的开头HTTP 请求 Body 处理阶段
ReplaceHttpRequestBody替换客户端请求 BodyHTTP 请求 Body 处理阶段
应答头处理GetHttpResponseHeaders获取后端响应的全部应答头HTTP 应答头处理阶段
ReplaceHttpResponseHeaders替换后端响应的全部应答头HTTP 应答头处理阶段
GetHttpResponseHeader获取后端响应的指定应答头HTTP 应答头处理阶段
RemoveHttpResponseHeader移除后端响应的指定应答头HTTP 应答头处理阶段
ReplaceHttpResponseHeader替换后端响应的指定应答头HTTP 应答头处理阶段
AddHttpResponseHeader新增一个后端响应头HTTP 应答头处理阶段
应答 Body 处理GetHttpResponseBody获取客户端请求 BodyHTTP 应答 Body 处理阶段
AppendHttpResponseBody将指定的字节串附加到后端响应 Body 末尾HTTP 应答 Body 处理阶段
PrependHttpResponseBody将指定的字节串附加到后端响应 Body 的开头HTTP 应答 Body 处理阶段
ReplaceHttpResponseBody替换后端响应 BodyHTTP 应答 Body 处理阶段
HTTP 调用DispatchHttpCall发送一个 HTTP 请求-
GetHttpCallResponseHeaders获取 DispatchHttpCall 请求响应的应答头-
GetHttpCallResponseBody获取 DispatchHttpCall 请求响应的应答 Body-
GetHttpCallResponseTrailers获取 DispatchHttpCall 请求响应的应答 Trailer-
直接响应SendHttpResponse直接返回一个特定的 HTTP 应答-
流程恢复ResumeHttpRequest恢复先前被暂停的请求处理流程-
ResumeHttpResponse恢复先前被暂停的应答处理流程-

3. 编译生成 WASM 文件

使用 proxy-wasm 社区 0.2.1 版本的 ABI,在 HTTP 请求/响应处理阶段只能使用 types.ActionContinuetypes.ActionPause 两种返回值来控制状态:

  1. types.ActionContinue:继续后续处理,比如继续读取请求 body,或者继续读取响应 body。

  2. types.ActionPause: 暂停后续处理,比如在 onHttpRequestHeaders 通过 Http 或者 Redis 调用外部服务获取认证信息,在调用外部服务回调钩子函数中调用 proxywasm.ResumeHttpRequest() 来恢复请求处理 或者调用 proxywasm.SendHttpResponseWithDetail() 来返回响应。

只需这样简单的状态管理,使用下面的编译方式即可:

  1. go mod tidy
  2. tinygo build -o main.wasm -scheduler=none -target=wasi -gc=custom -tags=”custommalloc nottinygc_finalizer ./

Higress 扩展了 0.2.100 版本的 ABI 来实现更丰富的 Header 状态管理,如果要使用,请用下面的编译方式:

  1. go mod tidy
  2. tinygo build -o main.wasm -scheduler=none -target=wasi -gc=custom -tags=”custommalloc nottinygc_finalizer proxy_wasm_version_0_2_100 ./

Header 的状态管理说明如下:

  1. HeaderContinue:

表示当前 filter 已经处理完毕,可以继续交给下一个 filter 处理。 types.ActionContinue 对应就是这个状态。

  1. HeaderStopIteration:

表示 header 还不能继续交给下一个 filter 来处理。 但是并不停止从连接读数据,继续触发 body data 的处理。 这样可以在 body data 处理阶段可以更新 Http 请求头内容。 如果 body data 要交给下一个 filter 处理, 这时 header 是也会被一起交给下一个 filter 处理。

但注意返回这个状态时,要求必须有 body,如果没有 body,请求/响应将被一直阻塞。

判断是否存在请求 body 可以使用 HasRequestBody()

  1. HeaderContinueAndEndStream:

表示 header 可以继续交给下一个 filter 处理,但是下一个 filter 收到的 end_stream = false,也就是标记请求还未结束。以便当前 filter 再增加 body。

  1. HeaderStopAllIterationAndBuffer:

停止所有迭代,表示 header 不能继续交给下一个 filter,并且当前 filter 也不能收到 body data。 并对当前过滤器及后续过滤器的头部、数据和尾部进行缓冲。如果缓存大小超过了 buffer limit,在请求阶段就直接返回 413,响应阶段就直接返回 500。 同时需要调用 proxywasm.ResumeHttpRequest()、 proxywasm.ResumeHttpResponse() 或者 proxywasm.SendHttpResponseWithDetail() 函数来恢复后续处理。

  1. HeaderStopAllIterationAndWatermark:

同上,区别是,当缓存超过了 buffer limit 会触发流控,也就是暂停从连接上读数据。 0.2.1 ABI 中的 types.ActionPause 实际上对应就是这个状态。

关于 types.HeaderStopIteration 和 HeaderStopAllIterationAndWatermark 的使用场景可以参考 Higress 官方提供 ai-transformer 插件ai-quota 插件

如果windows下编译出现error: could not find wasm-opt, set the WASMOPT environment variable to override 则需要下载https://github.com/WebAssembly/binaryen/ 里面包含了bin\wasm-opt.exe将这个文件拷贝到tinygo的bin目录下重新编译即可。
编译成功会在当前目录下创建文件 main.wasm。这个文件在下面本地调试的例子中也会被用到。

如果linux/mac下编译出现该错误,使用apt/brew等系统自带包管理工具安装下binaryen即可,例如brew install binaryen

要在Higress中配合Wasmplugin CRD或者Console的UI交互配置该插件,需要将该wasm文件打包成oci或者docker镜像,可以参考这篇文档:《自定义插件》

三、本地调试

工具准备

安装Docker

使用 docker compose 启动验证

  1. 进入在编写插件时创建的目录,例如wasm-demo目录,确认该目录下已经编译生成了main.wasm文件。
  2. 在目录下创建文件docker-compose.yaml,内容如下:
  1. version: 3.7
  2. services:
  3. envoy:
  4. image: higress-registry.cn-hangzhou.cr.aliyuncs.com/higress/gateway:v1.4.0-rc.1
  5. entrypoint: /usr/local/bin/envoy
  6. # 注意这里对wasm开启了debug级别日志,正式部署时则默认info级别
  7. command: -c /etc/envoy/envoy.yaml component-log-level wasm:debug
  8. depends_on:
  9. - httpbin
  10. networks:
  11. - wasmtest
  12. ports:
  13. - 10000:10000
  14. volumes:
  15. - ./envoy.yaml:/etc/envoy/envoy.yaml
  16. - ./main.wasm:/etc/envoy/main.wasm
  17. httpbin:
  18. image: kennethreitz/httpbin:latest
  19. networks:
  20. - wasmtest
  21. ports:
  22. - 12345:80
  23. networks:
  24. wasmtest: {}
  1. 继续在该目录下创建文件envoy.yaml,内容如下:
  1. admin:
  2. address:
  3. socket_address:
  4. protocol: TCP
  5. address: 0.0.0.0
  6. port_value: 9901
  7. static_resources:
  8. listeners:
  9. - name: listener_0
  10. address:
  11. socket_address:
  12. protocol: TCP
  13. address: 0.0.0.0
  14. port_value: 10000
  15. filter_chains:
  16. - filters:
  17. - name: envoy.filters.network.http_connection_manager
  18. typed_config:
  19. @type”: type.googleapis.com/envoy.extensions.filters.network.http_connection_manager.v3.HttpConnectionManager
  20. scheme_header_transformation:
  21. scheme_to_overwrite: https
  22. stat_prefix: ingress_http
  23. route_config:
  24. name: local_route
  25. virtual_hosts:
  26. - name: local_service
  27. domains: [“*”]
  28. routes:
  29. - match:
  30. prefix: “/“
  31. route:
  32. cluster: httpbin
  33. http_filters:
  34. - name: wasmdemo
  35. typed_config:
  36. @type”: type.googleapis.com/udpa.type.v1.TypedStruct
  37. type_url: type.googleapis.com/envoy.extensions.filters.http.wasm.v3.Wasm
  38. value:
  39. config:
  40. name: wasmdemo
  41. vm_config:
  42. runtime: envoy.wasm.runtime.v8
  43. code:
  44. local:
  45. filename: /etc/envoy/main.wasm
  46. configuration:
  47. @type”: type.googleapis.com/google.protobuf.StringValue
  48. value: |
  49. {
  50. mockEnable”: false
  51. }
  52. - name: envoy.filters.http.router
  53. typed_config:
  54. @type”: type.googleapis.com/envoy.extensions.filters.http.router.v3.Router
  55. clusters:
  56. - name: httpbin
  57. connect_timeout: 30s
  58. type: LOGICAL_DNS
  59. # Comment out the following line to test on v6 networks
  60. dns_lookup_family: V4_ONLY
  61. lb_policy: ROUND_ROBIN
  62. load_assignment:
  63. cluster_name: httpbin
  64. endpoints:
  65. - lb_endpoints:
  66. - endpoint:
  67. address:
  68. socket_address:
  69. address: httpbin
  70. port_value: 80
  1. 执行以下命令启动docker compose。
  1. docker compose up

功能验证

  1. WASM功能验证

使用curl直接访问httpbin,可以看到不经过网关时的请求头内容,如下:

  1. {
  2. args”: {},
  3. headers”: {
  4. Accept”: /“,
  5. Host”: 127.0.0.1:12345”,
  6. User-Agent”: curl/7.79.1
  7. },
  8. origin”: 172.18.0.1”,
  9. url”: http://127.0.0.1:12345/get
  10. }

使用curl通过网关访问httpbin,可以看到经过网关处理后的请求头的内容,如下:

  1. {
  2. args”: {},
  3. headers”: {
  4. Accept”: /“,
  5. Hello”: world”,
  6. Host”: 127.0.0.1:10000”,
  7. Original-Host”: 127.0.0.1:10000”,
  8. Req-Start-Time”: 1681269273896”,
  9. User-Agent”: curl/7.79.1”,
  10. X-Envoy-Expected-Rq-Timeout-Ms”: 15000
  11. },
  12. origin”: 172.18.0.3”,
  13. url”: https://127.0.0.1:10000/get
  14. }

此时上文编写插件的功能已经生效了,加入了hello: world请求头。

  1. 插件配置修改验证

修改envoy.yaml,将mockEnable配置修改为true。

  1. configuration:
  2. @type”: type.googleapis.com/google.protobuf.StringValue
  3. value: |
  4. {
  5. mockEnable”: true
  6. }

使用curl通过网关访问httpbin,可以看到经过网关处理后的请求头的内容,如下:

  1. hello world

说明插件配置修改生效,开启了mock应答直接返回了hello world。

更多示例

无配置插件

插件无需配置时,直接定义空结构体即可

  1. package main
  2. import (
  3. github.com/alibaba/higress/plugins/wasm-go/pkg/wrapper
  4. github.com/higress-group/proxy-wasm-go-sdk/proxywasm
  5. github.com/higress-group/proxy-wasm-go-sdk/proxywasm/types
  6. )
  7. func main() {
  8. wrapper.SetCtx(
  9. hello-world”,
  10. wrapper.ProcessRequestHeadersBy(onHttpRequestHeaders),
  11. )
  12. }
  13. type MyConfig struct {}
  14. func onHttpRequestHeaders(ctx wrapper.HttpContext, config MyConfig, log wrapper.Log) types.Action {
  15. proxywasm.SendHttpResponse(200, nil, []byte(“hello world”), -1)
  16. return types.ActionContinue
  17. }

在插件中请求外部服务

目前仅支持 http 调用,支持访问在网关控制台中设置了服务来源的 Nacos、K8s 服务,以及固定地址或 DNS 来源的服务。请注意,无法直接使用net/http库中的 HTTP client,必须使用如下例中封装的 HTTP client。
下面例子中,在配置解析阶段解析服务类型,生成对应的 HTTP client ;在请求头处理阶段根据配置的请求路径访问对应服务,解析应答头,然后再设置在原始的请求头中。

  1. package main
  2. import (
  3. errors
  4. net/http
  5. strings
  6. github.com/alibaba/higress/plugins/wasm-go/pkg/wrapper
  7. github.com/higress-group/proxy-wasm-go-sdk/proxywasm
  8. github.com/higress-group/proxy-wasm-go-sdk/proxywasm/types
  9. github.com/tidwall/gjson
  10. )
  11. func main() {
  12. wrapper.SetCtx(
  13. http-call”,
  14. wrapper.ParseConfigBy(parseConfig),
  15. wrapper.ProcessRequestHeadersBy(onHttpRequestHeaders),
  16. )
  17. }
  18. type MyConfig struct {
  19. // 用于发起HTTP调用client
  20. client wrapper.HttpClient
  21. // 请求url
  22. requestPath string
  23. // 根据这个key取出调用服务的应答头对应字段,再设置到原始请求的请求头,key为此配置项
  24. tokenHeader string
  25. }
  26. func parseConfig(json gjson.Result, config *MyConfig, log wrapper.Log) error {
  27. config.tokenHeader = json.Get(“tokenHeader”).String()
  28. if config.tokenHeader == “” {
  29. return errors.New(“missing tokenHeader in config”)
  30. }
  31. config.requestPath = json.Get(“requestPath”).String()
  32. if config.requestPath == “” {
  33. return errors.New(“missing requestPath in config”)
  34. }
  35. // 带服务类型的完整 FQDN 名称,例如 my-svc.dns, my-svc.static, service-provider.DEFAULT-GROUP.public.nacos, httpbin.my-ns.svc.cluster.local
  36. serviceName := json.Get(“serviceName”).String()
  37. servicePort := json.Get(“servicePort”).Int()
  38. if servicePort == 0 {
  39. if strings.HasSuffix(serviceName, “.static”) {
  40. // 静态IP类型服务的逻辑端口是80
  41. servicePort = 80
  42. }
  43. }
  44. config.client = wrapper.NewClusterClient(wrapper.FQDNCluster{
  45. FQDN: serviceName,
  46. Port: servicePort,
  47. })
  48. }
  49. func onHttpRequestHeaders(ctx wrapper.HttpContext, config MyConfig, log wrapper.Log) types.Action {
  50. // 使用client的Get方法发起HTTP Get调用,此处省略了timeout参数,默认超时时间500毫秒
  51. config.client.Get(config.requestPath, nil,
  52. // 回调函数,将在响应异步返回时被执行
  53. func(statusCode int, responseHeaders http.Header, responseBody []byte) {
  54. // 请求没有返回200状态码,进行处理
  55. if statusCode != http.StatusOK {
  56. log.Errorf(“http call failed, status: %d”, statusCode)
  57. proxywasm.SendHttpResponse(http.StatusInternalServerError, nil,
  58. []byte(“http call failed”), -1)
  59. return
  60. }
  61. // 打印响应的HTTP状态码和应答body
  62. log.Infof(“get status: %d, response body: %s”, statusCode, responseBody)
  63. // 从应答头中解析token字段设置到原始请求头中
  64. token := responseHeaders.Get(config.tokenHeader)
  65. if token != “” {
  66. proxywasm.AddHttpRequestHeader(config.tokenHeader, token)
  67. }
  68. // 恢复原始请求流程,继续往下处理,才能正常转发给后端服务
  69. proxywasm.ResumeHttpRequest()
  70. })
  71. // 需要等待异步回调完成,返回Pause状态,可以被ResumeHttpRequest恢复
  72. return types.ActionPause
  73. }

在插件中调用Redis

使用如下示例代码实现Redis限流插件

  1. package main
  2. import (
  3. strconv
  4. time
  5. github.com/higress-group/proxy-wasm-go-sdk/proxywasm
  6. github.com/higress-group/proxy-wasm-go-sdk/proxywasm/types
  7. github.com/tidwall/gjson
  8. github.com/tidwall/resp
  9. github.com/alibaba/higress/plugins/wasm-go/pkg/wrapper
  10. )
  11. func main() {
  12. wrapper.SetCtx(
  13. redis-demo”,
  14. wrapper.ParseConfigBy(parseConfig),
  15. wrapper.ProcessRequestHeadersBy(onHttpRequestHeaders),
  16. wrapper.ProcessResponseHeadersBy(onHttpResponseHeaders),
  17. )
  18. }
  19. type RedisCallConfig struct {
  20. client wrapper.RedisClient
  21. qpm int
  22. }
  23. func parseConfig(json gjson.Result, config *RedisCallConfig, log wrapper.Log) error {
  24. // 带服务类型的完整 FQDN 名称,例如 my-redis.dns、redis.my-ns.svc.cluster.local
  25. serviceName := json.Get(“serviceName”).String()
  26. servicePort := json.Get(“servicePort”).Int()
  27. if servicePort == 0 {
  28. if strings.HasSuffix(serviceName, “.static”) {
  29. // 静态IP类型服务的逻辑端口是80
  30. servicePort = 80
  31. } else {
  32. servicePort = 6379
  33. }
  34. }
  35. username := json.Get(“username”).String()
  36. password := json.Get(“password”).String()
  37. // 单位是毫秒
  38. timeout := json.Get(“timeout”).Int()
  39. if timeout == 0 {
  40. timeout = 1000
  41. }
  42. qpm := json.Get(“qpm”).Int()
  43. config.qpm = int(qpm)
  44. config.client = wrapper.NewRedisClusterClient(wrapper.FQDNCluster{
  45. FQDN: serviceName,
  46. Port: servicePort,
  47. })
  48. return config.client.Init(username, password, timeout)
  49. }
  50. func onHttpRequestHeaders(ctx wrapper.HttpContext, config RedisCallConfig, log wrapper.Log) types.Action {
  51. now := time.Now()
  52. minuteAligned := now.Truncate(time.Minute)
  53. timeStamp := strconv.FormatInt(minuteAligned.Unix(), 10)
  54. // 如果 redis api 返回的 err != nil,一般是由于网关找不到 redis 后端服务,请检查是否误删除了 redis 后端服务
  55. err := config.client.Incr(timeStamp, func(response resp.Value) {
  56. if response.Error() != nil {
  57. log.Errorf(“call redis error: %v”, response.Error())
  58. proxywasm.ResumeHttpRequest()
  59. } else {
  60. ctx.SetContext(“timeStamp”, timeStamp)
  61. ctx.SetContext(“callTimeLeft”, strconv.Itoa(config.qpm-response.Integer()))
  62. if response.Integer() == 1 {
  63. err := config.client.Expire(timeStamp, 60, func(response resp.Value) {
  64. if response.Error() != nil {
  65. log.Errorf(“call redis error: %v”, response.Error())
  66. }
  67. proxywasm.ResumeHttpRequest()
  68. })
  69. if err != nil {
  70. log.Errorf(“Error occured while calling redis, it seems cannot find the redis cluster.”)
  71. proxywasm.ResumeHttpRequest()
  72. }
  73. } else {
  74. if response.Integer() > config.qpm {
  75. proxywasm.SendHttpResponse(429, [][2]string{{“timeStamp”, timeStamp}, {“callTimeLeft”, 0”}}, []byte(“Too many requests\n”), -1)
  76. } else {
  77. proxywasm.ResumeHttpRequest()
  78. }
  79. }
  80. }
  81. })
  82. if err != nil {
  83. // 由于调用redis失败,放行请求,记录日志
  84. log.Errorf(“Error occured while calling redis, it seems cannot find the redis cluster.”)
  85. return types.ActionContinue
  86. } else {
  87. // 请求hold住,等待redis调用完成
  88. return types.ActionPause
  89. }
  90. }
  91. func onHttpResponseHeaders(ctx wrapper.HttpContext, config RedisCallConfig, log wrapper.Log) types.Action {
  92. if ctx.GetContext(“timeStamp”) != nil {
  93. proxywasm.AddHttpResponseHeader(“timeStamp”, ctx.GetContext(“timeStamp”).(string))
  94. }
  95. if ctx.GetContext(“callTimeLeft”) != nil {
  96. proxywasm.AddHttpResponseHeader(“callTimeLeft”, ctx.GetContext(“callTimeLeft”).(string))
  97. }
  98. return types.ActionContinue
  99. }