improve stub generator code

desc/service.go

@@ -210,6 +210,7 @@ func (s Service) rpcStub(
 		Predicate: rrs.GetPredicate(),
 		Encoding:  getEncoding(rrs),
 	}
+
 	if dto, ok := stub.getDTO(reflect.TypeOf(c.Input)); ok {
 		m.Request = dto
 	}
@@ -245,6 +246,7 @@ func (s Service) restStub(
 		Path:     rrs.GetPath(),
 		Encoding: getEncoding(rrs),
 	}
+
 	if dto, ok := stub.getDTO(reflect.TypeOf(c.Input)); ok {
 		m.Request = dto
 	}

desc/stub.go

@@ -153,3 +153,7 @@ func (d *Stub) getDTO(mTyp reflect.Type) (DTO, bool) {
 
 	return dto, ok
 }
+
+func (d *Stub) Tags() []string {
+	return d.tags
+}

exmples/simple-rest-server/cmd/client/main.go

@@ -4,7 +4,6 @@ import (
 	"context"
 	"fmt"
 	"net/http"
-	"os"
 
 	sampleservicestub "github.com/clubpay/ronykit/exmples/simple-rest-server/stub"
 	"github.com/clubpay/ronykit/stub"
@@ -14,32 +13,33 @@ import (
 func main() {
 	res1 := sampleservicestub.EchoResponse{}
 	s := stub.New("127.0.0.1")
-	err := s.REST().
+	httpCtx := s.REST().
 		SetMethod(http.MethodGet).
 		SetPath("echo/1230").
+		SetQuery("ok", "true").
 		SetResponseHandler(
 			http.StatusOK,
 			func(ctx context.Context, r stub.RESTResponse) *stub.Error {
 				return stub.WrapError(json.Unmarshal(r.GetBody(), &res1))
 			},
 		).
-		Run(context.Background()).
-		Err()
-	if err != nil {
-		panic(err)
+		Run(context.Background())
+	defer httpCtx.Release()
+
+	if httpCtx.Err() != nil {
+		panic(httpCtx.Err())
 	}
 	//nolint:forbidigo
 	fmt.Println("RESPONSE1: ", res1.Ok, res1.RandomID)
 
 	s2 := sampleservicestub.NewSampleServiceStub(
 		"127.0.0.1",
-		stub.DumpTo(os.Stdout),
 	)
 	res2, err := s2.Echo(
 		context.Background(),
 		&sampleservicestub.EchoRequest{
 			RandomID: 1450,
-			Ok:       false,
+			Ok:       true,
 		},
 	)
 	if err != nil {

stub/gen/go/tpl.go

@@ -2,6 +2,7 @@ package stubgengo
 
 import (
 	_ "embed"
+	"fmt"
 	"strings"
 	"text/template"
 
@@ -19,6 +20,16 @@ func init() {
 }
 
 var funcMaps = map[string]interface{}{
+	"strQuote": func(elems []string) []string {
+		out := make([]string, len(elems))
+		for i, e := range elems {
+			out[i] = fmt.Sprintf("%q", e)
+		}
+
+		return out
+	},
+	"strJoin":         strings.Join,
+	"strSplit":        strings.Split,
 	"toUpper":         strings.ToUpper,
 	"toLower":         strings.ToLower,
 	"toTitle":         strings.ToTitle,

stub/gen/go/tpl/stub.gotmpl

@@ -29,9 +29,10 @@ import (
 	"github.com/clubpay/ronykit/utils/reflector"
 )
 
+{{$tags := strQuote .Tags}}
 func init() {
 {{- range $dtoName, $dto := .DTOs }}
-reflector.Register(&{{$dtoName}}{})
+reflector.Register(&{{$dtoName}}{}, {{strJoin $tags ","}})
 {{- end }}
 }
 
@@ -62,7 +63,7 @@ func New{{$serviceName}}Stub(hostPort string, opts ...stub.Option) *{{$serviceNa
 {{- if ne $methodName "" }}
 func (s {{$serviceName}}Stub) {{$methodName}}(ctx context.Context, req *{{.Request.Name}}) (*{{.Response.Name}}, *stub.Error){
 	res := &{{.Response.Name}}{}
-	err := s.s.REST().
+	httpCtx := s.s.REST().
 		SetMethod("{{.Method}}").
 		{{ range $idx, $errDto := .PossibleErrors }}
 		SetResponseHandler(
@@ -83,9 +84,10 @@ func (s {{$serviceName}}Stub) {{$methodName}}(ctx context.Context, req *{{.Reque
 				return stub.WrapError(ronykit.UnmarshalMessage(r.GetBody(), res))
 			},
 		).
-		AutoRun(ctx, "{{.Path}}", {{.Encoding}}, req).
-		Err()
-	if err != nil {
+		AutoRun(ctx, "{{.Path}}", {{.Encoding}}, req)
+	defer httpCtx.Release()
+
+	if err := httpCtx.Err(); err != nil {
 		return nil, err
 	}
 

utils/reflector/obj.go

@@ -49,9 +49,13 @@ func (fields Fields) Get(m ronykit.Message, fieldName string) interface{} {
 	case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
 		return mVal.Int()
 	case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
-		return mVal.Int()
+		return mVal.Uint()
 	case reflect.String:
 		return mVal.String()
+	case reflect.Float64, reflect.Float32:
+		return mVal.Float()
+	case reflect.Bool:
+		return mVal.Bool()
 	}
 
 	if !mVal.CanInterface() {