修改测试自定义打印

cmd/robot/hero.go

@@ -1,9 +1,12 @@
 package robot
 
 import (
+	"fmt"
 	"go_dreamfactory/comm"
 	"go_dreamfactory/modules/hero"
 	"go_dreamfactory/pb"
+
+	"google.golang.org/protobuf/proto"
 )
 
 var (
@@ -14,7 +17,13 @@ var (
 			subType:  hero.HeroSubTypeList,
 			req:      &pb.HeroListReq{},
 			rsp:      &pb.HeroListRsp{},
-			// enabled:  true,
+			print: func(rsp proto.Message) {
+				out := rsp.(*pb.HeroListRsp)
+				for i, v := range out.List {
+					fmt.Printf("%d- %v\n", (i + 1), v)
+				}
+			},
+			// enabled: true,
 		}, {
 			mainType: string(comm.ModuleHero),
 			subType:  hero.HeroSubTypeInfo,
@@ -26,9 +35,34 @@ var (
 		}, {
 			mainType: string(comm.ModuleHero),
 			subType:  hero.HeroSubTypeChouka,
-			req:      &pb.HeroChoukaReq{},
+			req: &pb.HeroChoukaReq{
-			rsp:      &pb.HeroChoukaResp{},
+				HeroIds: []int32{42911},
-			enabled:  true,
+			},
+			rsp: &pb.HeroChoukaResp{},
+			// enabled: true,
+		}, {
+			mainType: string(comm.ModuleHero),
+			subType:  hero.StrengthenUpStar,
+			req: &pb.HeroStrengthenUpStarReq{
+				HeroObjID: "62bd0489ff6632434a7d0d1f",
+				Hero: []*pb.CostCardData{
+					{
+						CostCardObj: "",
+					},
+				},
+			},
+			rsp: &pb.HeroStrengthenUpStarResp{},
+			// enabled:  true,
+		}, {
+			mainType: string(comm.ModuleHero),
+			subType:  hero.StrengthenUplv,
+			req: &pb.HeroStrengthenUplvReq{
+				HeroObjID: "62bd0489ff6632434a7d0d1f",
+				ExpCardID: "62bd0b4eca37634b8230d4be",
+				Amount:    1,
+			},
+			rsp:     &pb.HeroStrengthenUplvResp{},
+			enabled: true,
 		},
 	}
 )

cmd/robot/robot.go

@@ -75,6 +75,7 @@ type builder struct {
 	enabled   bool
 	start     time.Time
 	requested bool //请求标识 true已发
+	print     func(rsp proto.Message)
 }
 
 func (r *Robot) addBuilders(builders []*builder) {
@@ -109,7 +110,13 @@ func (r *Robot) handleRsp(msg *pb.UserMessage) {
 			if !comm.ProtoUnmarshal(msg, b.rsp) {
 				return
 			}
-			printReply(msg, b)
+			if b.print == nil {
+				printReply(msg, b)
+			} else {
+				fmt.Printf("===== rsp [%s.%s] =====\n", msg.MainType, msg.SubType)
+				b.print(b.rsp)
+				fmt.Println("==============================")
+			}
 
 			if msg.MainType == "user" && msg.SubType == "login" {
 				r.loginCallback(b.rsp)

cmd/robot/user.go

@@ -13,7 +13,7 @@ var user_builders = []*builder{
 		mainType: string(comm.ModuleUser),
 		subType:  user.UserSubTypeCreate,
 		req: &pb.UserCreateReq{ //设置请求参数
-			NickName: "乐谷62911",
+			NickName: "乐谷6301",
 		},
 		rsp: &pb.UserCreateRsp{},
 		// enabled: true,