feat: support hz sdk

cloudwego · Mar 9, 2025 · d552c83 · d552c83
1 parent 2256067
commit d552c83
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/cmd/hz/thrift/plugin.go b/cmd/hz/thrift/plugin.go
@@ -190,7 +190,7 @@ func (plugin *Plugin) Run() int {
 		return meta.PluginError
 	}
 
-	args, err = plugin.parseArgs()
+	args, err = plugin.parseArgs(req)
 	if err != nil {
 		logs.Errorf("parse args failed: %s", err.Error())
 		return meta.PluginError
@@ -229,12 +229,12 @@ func (plugin *Plugin) recvVerboseLogger() string {
 	return verboseLog
 }
 
-func (plugin *Plugin) parseArgs() (*config.Argument, error) {
-	if plugin.req == nil {
+func (plugin *Plugin) parseArgs(req *thriftgo_plugin.Request) (*config.Argument, error) {
+	if req == nil {
 		return nil, fmt.Errorf("request is nil")
 	}
 	args := new(config.Argument)
-	err := args.Unpack(plugin.req.PluginParameters)
+	err := args.Unpack(req.PluginParameters)
 	if err != nil {
 		logs.Errorf("unpack args failed: %s", err.Error())
 	}

diff --git a/cmd/hz/thrift/plugin_test.go b/cmd/hz/thrift/plugin_test.go
@@ -41,7 +41,7 @@ func TestRun(t *testing.T) {
 
 	plu.req = req
 
-	_, err = plu.parseArgs()
+	_, err = plu.parseArgs(req)
 	if err != nil {
 		t.Fatal(err)
 	}