optimize(hz): hz client behavior (#1038)

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,
@@ -573,13 +583,24 @@ func (r *request) setHeader(header, value string) *request {
 	return r
 }
 
+func (r *request) addHeader(header, value string) *request {
+	r.header.Add(header, value)
+	return r
+}
+
 func (r *request) setQueryParam(param string, value interface{}) *request {
 	v := reflect.ValueOf(value)
 	switch v.Kind() {
 	case reflect.Slice, reflect.Array:
 		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))
 	}
@@ -911,7 +932,7 @@ func (s *{{$.ServiceName}}Client) {{$MethodInfo.Name}}(context context.Context,
 		setPathParams(map[string]string{
 			{{$MethodInfo.PathParamsCode}}
 		}).
-		setHeaders(map[string]string{
+		addHeaders(map[string]string{
 			{{$MethodInfo.HeaderParamsCode}}
 		}).
 		setFormParams(map[string]string{