fix contractID issue in desc package

clubpay · May 21, 2022 · 1b67aa7 · 1b67aa7
1 parent 2bbeb04
commit 1b67aa7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/desc/desc.go b/desc/desc.go
@@ -1,8 +1,9 @@
 package desc
 
 import (
+	"fmt"
+
 	"github.com/clubpay/ronykit"
-	"github.com/clubpay/ronykit/utils"
 	"github.com/clubpay/ronykit/utils/reflector"
 )
 
@@ -222,7 +223,7 @@ func (s Service) Generate() ronykit.Service {
 
 		contracts := make([]ronykit.Contract, len(c.RouteSelectors))
 		for idx, s := range c.RouteSelectors {
-			ci := (&contractImpl{id: utils.Int64ToStr(index)}).
+			ci := (&contractImpl{id: fmt.Sprintf("%s.%d", svc.name, index)}).
 				addHandler(svc.pre...).
 				addHandler(c.Handlers...).
 				addHandler(svc.post...).

diff --git a/edge_test.go b/edge_test.go
@@ -123,7 +123,7 @@ func (t testBundle) Dispatch(ctx *ronykit.Context, in []byte) (ronykit.ExecuteAr
 			return err
 		},
 		ServiceName: "testService",
-		ContractID:  "1",
+		ContractID:  "testService.1",
 		Route:       "someRoute",
 	}, nil
 }
@@ -144,10 +144,8 @@ func TestServer(t *testing.T) {
 						AddSelector(testSelector{}).
 						AddHandler(
 							func(ctx *ronykit.Context) {
-								m := ctx.In().GetMsg()
-
 								ctx.Out().
-									SetMsg(m).
+									SetMsg(ctx.In().GetMsg()).
 									Send()
 
 								return