refactor: plugin functions can be defined in any format

plugin/plugin.go

@@ -0,0 +1,67 @@
+package plugin
+
+import (
+	"fmt"
+	"os"
+	"reflect"
+
+	hclog "github.com/hashicorp/go-hclog"
+	"github.com/hashicorp/go-plugin"
+
+	pluginShared "github.com/httprunner/hrp/plugin/shared"
+)
+
+// functionsMap stores plugin functions
+type functionsMap map[string]reflect.Value
+
+// functionPlugin implements the FuncCaller interface
+type functionPlugin struct {
+	logger    hclog.Logger
+	functions functionsMap
+}
+
+func (p *functionPlugin) GetNames() ([]string, error) {
+	var names []string
+	for name := range p.functions {
+		names = append(names, name)
+	}
+	return names, nil
+}
+
+func (p *functionPlugin) Call(funcName string, args ...interface{}) (interface{}, error) {
+	p.logger.Info("call function", "funcName", funcName, "args", args)
+
+	fn, ok := p.functions[funcName]
+	if !ok {
+		return nil, fmt.Errorf("function %s not found", funcName)
+	}
+
+	return pluginShared.CallFunc(fn, args...)
+}
+
+var functions = make(functionsMap)
+
+func Register(funcName string, fn interface{}) {
+	if _, ok := functions[funcName]; ok {
+		return
+	}
+	functions[funcName] = reflect.ValueOf(fn)
+}
+
+func Serve() {
+	funcPlugin := &functionPlugin{
+		logger: hclog.New(&hclog.LoggerOptions{
+			Name:   pluginShared.Name,
+			Output: os.Stdout,
+			Level:  hclog.Info,
+		}),
+		functions: functions,
+	}
+	var pluginMap = map[string]plugin.Plugin{
+		pluginShared.Name: &pluginShared.HashicorpPlugin{Impl: funcPlugin},
+	}
+	plugin.Serve(&plugin.ServeConfig{
+		HandshakeConfig: pluginShared.HandshakeConfig,
+		Plugins:         pluginMap,
+	})
+}