相關(guān)依賴

• github.com/higress-group/proxy-wasm-go-sdk
• github.com/alibaba/higress/plugins/wasm-go

標(biāo)準(zhǔn)的redis

下面是一個讀取redis指定key的方法，使用了higress的wasm-go組件實現(xiàn)的

	err := config.Client.SMembers("online", func(response resp.Value) {
		for _, item := range response.Array() {
			str := item.String()
			if strings.HasPrefix(str, "\"") {
				str = str[1:]
			}
			if strings.HasSuffix(str, "\"") {
				str = str[:len(str)-1]
			}
			if str == val {
				msg := fmt.Sprintf("your %v (%v) is illegality.", blackType, val)
				fmt.Println(msg)
				ctx.SetContext("X-Blacklist-Reason", msg)
				proxywasm.SendHttpResponse(ERROR_CODE, nil, []byte(msg), -1)
				break
			}
		}
		proxywasm.ResumeHttpRequest() // 恢復(fù)
	})

擴(kuò)展的Lua方式

在Redis的Lua腳本中，可以使用redis.call來執(zhí)行Redis命令。如果需要在Lua腳本中設(shè)置鍵的過期時間，可以通過調(diào)用EXPIRE命令來實現(xiàn)。

下面是一個示例，演示如何在Lua腳本中使用redis.call執(zhí)行EXPIRE命令來設(shè)置鍵的過期時間：

local key = KEYS[1]
local ttl = ARGV[1]

redis.call('SET', key, 'value')
redis.call('EXPIRE', key, ttl)

return 'Key set with expiration time'

在上面的示例中，首先通過KEYS[1]獲取傳入的鍵名，通過ARGV[1]獲取傳入的過期時間（以秒為單位）。然后使用redis.call('SET', key, 'value')設(shè)置鍵值對，并使用redis.call('EXPIRE', key, ttl)設(shè)置鍵的過期時間為ttl秒。最后返回一個提示信息。

通過這種方式，你可以在Lua腳本中使用redis.call執(zhí)行EXPIRE命令來設(shè)置鍵的過期時間。

github.com/alibaba/higress/plugins/wasm-go中的Eval

阿里封裝的wasm-go插件，在2024-03-18這一天支持了redis，同時也支持使用redis.call，你可以調(diào)用wasm-go中的Eval方法

• 方法原型

Eval(script string, numkeys int, keys, args []interface{}, callback RedisResponseCallback) error

• 開發(fā)人員調(diào)用它

	redisKey := "business:username"
	redisKey2 := "business:session_state"
	var keyArr []interface{}
	keyArr = append(keyArr, redisKey, redisKey2)
	var valueArr []interface{}
	valueArr = append(valueArr, logger.Username, logger.Username+"_"+logger.SessionState, 60*60*24*30)
	err2 := config.Client.Eval("redis.call('ZINCRBY', KEYS[1], 1, ARGV[1]) redis.call('ZINCRBY', KEYS[2], 1, ARGV[2]) redis.call('EXPIRE', KEYS[1], ARGV[3]) redis.call('EXPIRE', KEYS[2], ARGV[3]) return 1", 2, keyArr, valueArr, func(response resp.Value) {
		if response.Integer() == 1 {
			proxywasm.ResumeHttpRequest()
		}
	})
	if err2 != nil {
		return types.ActionContinue
    }
   return types.ActionPause