exception handling

3 years ago · 2b9a3536e7
parent 90cde99f06
commit 2b9a3536e7
1 changed files with 6 additions and 1 deletions
--- a/src/common/db/mongoModel.go
+++ b/src/common/db/mongoModel.go
@ -147,7 +147,12 @@ func (d *DataBases) SaveUserChat(uid string, sendTime int64, m proto.Message) er
 	if session == nil {
 		return errors.New("session == nil")
 	}
-	defer session.Close()
+	defer func() {
+		if session != nil {
+			session.Close()
+		}
+	}()
+
 	log.NewInfo("", "get mgoSession cost time", getCurrentTimestampByMill()-newTime)
 	c := session.DB(config.Config.Mongo.DBDatabase).C(cChat)