From 0c5a38f4034e009f879029a4f7ef62c60be5350f Mon Sep 17 00:00:00 2001
From: liwei1dao <liwei1dao@gmail.com>
Date: Thu, 13 Oct 2022 12:06:23 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=8A=E4=BC=A0db=E4=BF=9D=E6=8A=A4=E4=BB=A3?=
 =?UTF-8?q?=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/battle/module.go    | 28 ++++++++++++++++++++++------
 modules/forum/modelForum.go |  9 +++++++--
 2 files changed, 29 insertions(+), 8 deletions(-)

diff --git a/modules/battle/module.go b/modules/battle/module.go
index 16226a9cf..9b6008363 100644
--- a/modules/battle/module.go
+++ b/modules/battle/module.go
@@ -68,11 +68,19 @@ func (this *Battle) QueryBattleRecord(oid string) (code pb.ErrorCode, record *pb
 
 //创建pve战斗
 func (this *Battle) CreatePveBattle(session comm.IUserSession, req *pb.BattlePVEReq) (code pb.ErrorCode, record *pb.DBBattleRecord) {
-	var conn *db.DBConn
+	var (
+		conn *db.DBConn
+		err  error
+	)
 	if this.service.GetTag() == session.GetServiecTag() {
-		conn = db.Local()
+		conn, err = db.Local()
 	} else {
-		conn = db.ServerDBConn(session.GetServiecTag())
+		conn, err = db.ServerDBConn(session.GetServiecTag())
+	}
+	if err != nil {
+		code = pb.ErrorCode_DBError
+		this.Errorf("session:%v err:", session, err)
+		return
 	}
 	if req.Ptype < 0 || req.Ptype > 4 ||
 		req.Teamids == nil || len(req.Teamids) != 5 {
@@ -87,11 +95,19 @@ func (this *Battle) CreatePveBattle(session comm.IUserSession, req *pb.BattlePVE
 
 //创建pve战斗
 func (this *Battle) CreatePvbBattle(session comm.IUserSession, req *pb.BattlePVEReq) (code pb.ErrorCode, record *pb.DBBattleRecord) {
-	var conn *db.DBConn
+	var (
+		conn *db.DBConn
+		err  error
+	)
 	if this.service.GetTag() == session.GetServiecTag() {
-		conn = db.Local()
+		conn, err = db.Local()
 	} else {
-		conn = db.ServerDBConn(session.GetServiecTag())
+		conn, err = db.ServerDBConn(session.GetServiecTag())
+	}
+	if err != nil {
+		code = pb.ErrorCode_DBError
+		this.Errorf("session:%v err:", session, err)
+		return
 	}
 	if req.Ptype < 0 || req.Ptype > 4 ||
 		req.Teamids == nil || len(req.Teamids) != 5 {
diff --git a/modules/forum/modelForum.go b/modules/forum/modelForum.go
index 3f8883987..90d507166 100644
--- a/modules/forum/modelForum.go
+++ b/modules/forum/modelForum.go
@@ -125,10 +125,15 @@ func (this *modelForumComp) like(heroid, id string, islike bool) (comment *pb.DB
 
 //查看
 func (this *modelForumComp) watchHero(stage string, uid string, herocid string) (hero *pb.DBHero, err error) {
-	tcoon := db.ServerDBConn(stage)
+
 	var (
-		c *mongo.Cursor
+		tcoon *db.DBConn
+		c     *mongo.Cursor
 	)
+	if tcoon, err = db.ServerDBConn(stage); err != nil {
+		this.module.Errorf("stage:%s err:%v", stage, err)
+		return
+	}
 	if c, err = tcoon.Mgo.Find(comm.TableHero, bson.M{"uid": uid, "heroID": herocid}); err != nil {
 		return
 	} else {