callctx 수정

rpc/rpc_test.go

@@ -1,7 +1,42 @@
 package rpc
 
-import "testing"
+import (
+	"context"
+	"testing"
+
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"repositories.action2quare.com/ayo/gocommon"
+	"repositories.action2quare.com/ayo/gocommon/logger"
+)
+
+type testReceiver struct {
+}
+
+func (tr *testReceiver) TargetExists(tid primitive.ObjectID) bool {
+	return tid[0] >= 10
+}
+
+func (tr *testReceiver) TestFunc(a string, b string) {
+	logger.Println("TestFunc :", a, b)
+
+	target := primitive.NewObjectID()
+	target[0] = 0
+	if CallOrGo(tr, target, a, b) != ErrCanExecuteHere {
+		return
+	}
+
+	logger.Println(a, b)
+}
 
 func TestRpc(t *testing.T) {
+	var tr testReceiver
+	RegistReceiver(&tr)
+	myctx, cancel := context.WithCancel(context.Background())
 
+	redisClient, _ := gocommon.NewRedisClient("redis://192.168.8.94:6379", 0)
+	Start(myctx, redisClient)
+
+	tr.TestFunc("aaa", "bb")
+	<-myctx.Done()
+	cancel()
 }