feat: add quiet flag

kitex-contrib · Oct 23, 2024 · 8c590db · 8c590db
1 parent e6b232e
commit 8c590db
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
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.BoolVar(&a.Quiet, "quiet", false, "Enable only print rpc response.")
+
 	return f
 }
 
@@ -279,6 +281,7 @@ func (a *Argument) BuildConfig() *config.Config {
 		MetaPersistent: a.MetaPersistent,
 		MetaBackward:   a.MetaBackward,
 		BizError:       a.BizError,
+		Quiet:          a.Quiet,
 	}
 }
 

diff --git a/pkg/client/generic_client.go b/pkg/client/generic_client.go
@@ -18,6 +18,7 @@ package client
 
 import (
 	"context"
+	"fmt"
 	"os"
 
 	"github.com/bytedance/gopkg/cloud/metainfo"
@@ -83,8 +84,10 @@ func (c *GenericClientBase) Output() error {
 		return err
 	}
 
-	log.Success()
-	log.Println(result)
+	if !c.Conf.Quiet {
+		log.Success()
+	}
+	fmt.Print(result)
 
 	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
+	Quiet          bool
 }
 
 type ConfigBuilder interface {