feat: add save rpc response

kitex-contrib · Oct 23, 2024 · 05e2b67 · 05e2b67
1 parent e6b232e
commit 05e2b67
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/pkg/argparse/args.go b/pkg/argparse/args.go
@@ -107,6 +107,8 @@ func (a *Argument) buildFlags() *flag.FlagSet {
 
 	f.BoolVar(&a.version, "version", false, "Show the version of kitexcall.")
 
+	f.StringVar(&a.RespFileName, "o", "", "Specify the file name to save the rpc response.")
+
 	return f
 }
 
@@ -279,6 +281,7 @@ func (a *Argument) BuildConfig() *config.Config {
 		MetaPersistent: a.MetaPersistent,
 		MetaBackward:   a.MetaBackward,
 		BizError:       a.BizError,
+		RespFileName:   a.RespFileName,
 	}
 }
 

diff --git a/pkg/client/generic_client.go b/pkg/client/generic_client.go
@@ -84,7 +84,15 @@ func (c *GenericClientBase) Output() error {
 	}
 
 	log.Success()
-	log.Println(result)
+
+	if c.Conf.RespFileName == "" {
+		log.Println(result)
+	} else {
+		err = os.WriteFile(c.Conf.RespFileName, []byte(result), 0644)
+		if err != nil {
+			return err
+		}
+	}
 
 	if c.Conf.MetaBackward {
 		log.Println("\033[32mReceived metainfo from server: \033[0m")

diff --git a/pkg/config/config.go b/pkg/config/config.go
@@ -46,6 +46,7 @@ type Config struct {
 	MetaPersistent map[string]string
 	MetaBackward   bool
 	BizError       bool
+	RespFileName   string
 }
 
 type ConfigBuilder interface {