Support multi ontick

plugins/wasm-go/pkg/wrapper/plugin_wrapper.go

@@ -15,6 +15,7 @@
 package wrapper
 
 import (
+	"time"
 	"unsafe"
 
 	"github.com/higress-group/proxy-wasm-go-sdk/proxywasm"
@@ -61,6 +62,29 @@ type CommonVmCtx[PluginConfig any] struct {
 	onHttpResponseHeaders onHttpHeadersFunc[PluginConfig]
 	onHttpResponseBody    onHttpBodyFunc[PluginConfig]
 	onHttpStreamDone      onHttpStreamDoneFunc[PluginConfig]
+	onTickFuncs           []TickFuncEntry
+}
+
+type TickFuncEntry struct {
+	lastExecuted int64
+	tickPeriod   int64
+	tickFunc     func()
+}
+
+var globalOnTickFuncs []TickFuncEntry = []TickFuncEntry{}
+
+// Registe multiple onTick functions. Parameters include:
+// 1) tickPeriod: the execution period of tickFunc;
+// 2) tickFunc: the function to be executed
+// You should call this function in parseConfig phase, for example:
+//
+//	func parseConfig(json gjson.Result, config *HelloWorldConfig, log wrapper.Log) error {
+//		wrapper.RegisteTickFunc(1000, func() { proxywasm.LogInfo("onTick 1s") })
+//		wrapper.RegisteTickFunc(3000, func() { proxywasm.LogInfo("onTick 3s") })
+//		return nil
+//	}
+func RegisteTickFunc(tickPeriod int64, tickFunc func()) {
+	globalOnTickFuncs = append(globalOnTickFuncs, TickFuncEntry{0, tickPeriod, tickFunc})
 }
 
 func SetCtx[PluginConfig any](pluginName string, setFuncs ...SetPluginFunc[PluginConfig]) {
@@ -152,6 +176,8 @@ type CommonPluginCtx[PluginConfig any] struct {
 
 func (ctx *CommonPluginCtx[PluginConfig]) OnPluginStart(int) types.OnPluginStartStatus {
 	data, err := proxywasm.GetPluginConfiguration()
+	globalOnTickFuncs = nil
+	ctx.vm.onTickFuncs = nil
 	if err != nil && err != types.ErrorStatusNotFound {
 		ctx.vm.log.Criticalf("error reading plugin configuration: %v", err)
 		return types.OnPluginStartStatusFailed
@@ -186,9 +212,26 @@ func (ctx *CommonPluginCtx[PluginConfig]) OnPluginStart(int) types.OnPluginStart
 		ctx.vm.log.Warnf("parse rule config failed: %v", err)
 		return types.OnPluginStartStatusFailed
 	}
+	if globalOnTickFuncs != nil {
+		ctx.vm.onTickFuncs = globalOnTickFuncs
+		if err := proxywasm.SetTickPeriodMilliSeconds(100); err != nil {
+			ctx.vm.log.Error("SetTickPeriodMilliSeconds failed, onTick functions will not take effect.")
+			return types.OnPluginStartStatusFailed
+		}
+	}
 	return types.OnPluginStartStatusOK
 }
 
+func (ctx *CommonPluginCtx[PluginConfig]) OnTick() {
+	for i := range ctx.vm.onTickFuncs {
+		currentTimeStamp := time.Now().UnixMilli()
+		if currentTimeStamp-ctx.vm.onTickFuncs[i].lastExecuted >= ctx.vm.onTickFuncs[i].tickPeriod {
+			ctx.vm.onTickFuncs[i].tickFunc()
+			ctx.vm.onTickFuncs[i].lastExecuted = currentTimeStamp
+		}
+	}
+}
+
 func (ctx *CommonPluginCtx[PluginConfig]) NewHttpContext(contextID uint32) types.HttpContext {
 	httpCtx := &CommonHttpCtx[PluginConfig]{
 		plugin:      ctx,