From 1ef8ec4073945f8009ceb4779c677299ed2f4645 Mon Sep 17 00:00:00 2001 From: meixiongfeng <766881921@qq.com> Date: Tue, 12 Jul 2022 11:06:07 +0800 Subject: [PATCH 1/2] =?UTF-8?q?uid=20=E5=88=9B=E5=BB=BA=E7=B4=A2=E5=BC=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- modules/hero/model_hero.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/modules/hero/model_hero.go b/modules/hero/model_hero.go index f184a2c72..be52221c1 100644 --- a/modules/hero/model_hero.go +++ b/modules/hero/model_hero.go @@ -11,6 +11,8 @@ import ( mengine "github.com/dengsgo/math-engine/engine" "go.mongodb.org/mongo-driver/bson/primitive" + "go.mongodb.org/mongo-driver/mongo" + "go.mongodb.org/mongo-driver/x/bsonx" ) type ModelHero struct { @@ -22,6 +24,10 @@ func (this *ModelHero) Init(service core.IService, module core.IModule, comp cor err = this.MCompModel.Init(service, module, comp, options) this.moduleHero = module.(*Hero) this.TableName = "hero" + // 通过uid创建索引 + this.DB.CreateIndex(core.SqlTable(this.TableName), mongo.IndexModel{ + Keys: bsonx.Doc{{Key: "uid", Value: bsonx.Int32(1)}}, + }) return } From bce9accc29468ada72a83e9e6f0b17fad953accf Mon Sep 17 00:00:00 2001 From: meixiongfeng <766881921@qq.com> Date: Wed, 13 Jul 2022 09:23:43 +0800 Subject: [PATCH 2/2] =?UTF-8?q?=E4=B8=BB=E7=BA=BF=E5=85=B3=E5=8D=A1?= =?UTF-8?q?=E8=BF=9B=E5=BA=A6=E6=B5=8B=E8=AF=95=E6=96=87=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cmd/robot/robot.go | 3 +++ cmd/robot/story.go | 45 +++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 48 insertions(+) create mode 100644 cmd/robot/story.go diff --git a/cmd/robot/robot.go b/cmd/robot/robot.go index 211274578..b6c410f2c 100644 --- a/cmd/robot/robot.go +++ b/cmd/robot/robot.go @@ -189,6 +189,9 @@ func (r *Robot) onUserLoaded() { //task r.RunTask() + + // story + r.RunStory() } //注册账号 diff --git a/cmd/robot/story.go b/cmd/robot/story.go new file mode 100644 index 000000000..6ed86e1f2 --- /dev/null +++ b/cmd/robot/story.go @@ -0,0 +1,45 @@ +package robot + +import ( + "fmt" + "go_dreamfactory/comm" + "go_dreamfactory/modules/story" + "go_dreamfactory/pb" + + "google.golang.org/protobuf/proto" +) + +var ( + storyBuilders = []*TestCase{ + { + desc: "主线数据", + mainType: string(comm.ModuleStory), + subType: story.StoryGetListResp, + req: &pb.StoryGetListReq{}, + rsp: &pb.StoryGetListResp{}, + print: func(rsp proto.Message) { + out := rsp.(*pb.StoryGetListResp) + for i, v := range out.Data { + fmt.Printf("%d- %v\n", (i + 1), v) + } + }, + enabled: true, + }, { + desc: "主线详情", + mainType: string(comm.ModuleStory), + subType: story.StoryChallengeResp, + req: &pb.StoryChallengeReq{ + ChapterId: 1, + StoryId: 1, + }, + rsp: &pb.StoryChallengeResp{}, + // enabled: true, + }, + } +) + +//声明加入到构建器并发起请求 +func (r *Robot) RunStory() { + r.addBuilders(storyBuilders) + r.batchHandleReq() +}