optimize: hz client

cloudwego · Dec 26, 2023 · 19ffa49 · 19ffa49
1 parent a327795
commit 19ffa49
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/cmd/hz/generator/package_tpl.go b/cmd/hz/generator/package_tpl.go
@@ -324,6 +324,7 @@ type Option struct {
 
 type Options struct {
 	hostUrl               string
+	enumAsInt             bool
 	doer                  client.Doer
 	header                http.Header
 	requestBodyBind       bindRequestBodyFunc
@@ -375,6 +376,13 @@ func WithResponseResultDecider(decider ResponseResultDecider) Option {
 	}}
 }
 
+// WithQueryEnumAsInt is used to set enum as int for query parameters
+func WithQueryEnumAsInt(enable bool) Option {
+	return Option{func(op *Options) {
+		op.enumAsInt = enable
+	}}
+}
+
 func withHostUrl(HostUrl string) Option {
 	return Option{func(op *Options) {
 		op.hostUrl = HostUrl
@@ -384,6 +392,7 @@ func withHostUrl(HostUrl string) Option {
 // underlying client
 type cli struct {
 	hostUrl               string
+	enumAsInt             bool
 	doer                  client.Doer
 	header                http.Header
 	bindRequestBody       bindRequestBodyFunc
@@ -419,6 +428,7 @@ func newClient(opts *Options) (*cli, error) {
 
 	c := &cli{
 		hostUrl:               opts.hostUrl,
+		enumAsInt:             opts.enumAsInt,
 		doer:                  opts.doer,
 		header:                opts.header,
 		bindRequestBody:       opts.requestBodyBind,
@@ -580,6 +590,12 @@ func (r *request) setQueryParam(param string, value interface{}) *request {
 		for index := 0; index < v.Len(); index++ {
 			r.queryParam.Add(param, fmt.Sprint(v.Index(index).Interface()))
 		}
+	case reflect.Int32, reflect.Int64:
+		if r.client.enumAsInt {
+			r.queryParam.Add(param, fmt.Sprintf("%d", v.Interface()))
+		} else {
+			r.queryParam.Add(param, fmt.Sprint(v))
+		}
 	default:
 		r.queryParam.Set(param, fmt.Sprint(v))
 	}