From 449dd5ed05617abca080cddc9d24058b4d8e169c Mon Sep 17 00:00:00 2001
From: zhaocy <sh_zcy@qq.com>
Date: Thu, 30 Jun 2022 10:55:27 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=B5=8B=E8=AF=95=E8=87=AA?=
 =?UTF-8?q?=E5=AE=9A=E4=B9=89=E6=89=93=E5=8D=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmd/robot/hero.go  | 42 ++++++++++++++++++++++++++++++++++++++----
 cmd/robot/robot.go |  9 ++++++++-
 cmd/robot/user.go  |  2 +-
 3 files changed, 47 insertions(+), 6 deletions(-)

diff --git a/cmd/robot/hero.go b/cmd/robot/hero.go
index 71faaa594..0df8b2e32 100644
--- a/cmd/robot/hero.go
+++ b/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{},
-			rsp:      &pb.HeroChoukaResp{},
-			enabled:  true,
+			req: &pb.HeroChoukaReq{
+				HeroIds: []int32{42911},
+			},
+			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,
 		},
 	}
 )
diff --git a/cmd/robot/robot.go b/cmd/robot/robot.go
index 8f6ca73ce..2cd33221a 100644
--- a/cmd/robot/robot.go
+++ b/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)
diff --git a/cmd/robot/user.go b/cmd/robot/user.go
index 3ede6b8ab..35a00148e 100644
--- a/cmd/robot/user.go
+++ b/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,