切磋判断

modules/friend/api_cross_qiecuo.go

@@ -24,9 +24,13 @@ func (this *apiComp) Qiecuo(session comm.IUserSession, req *pb.FriendQiecuoReq)
 		code = pb.ErrorCode_UserOffline
 		return
 	}
+	qr, err := this.moduleFriend.ModelFriendQiecuo.createQiecuoRecord(session.GetUserId(), req.TargetUid)
+	if err != nil {
+		code = pb.ErrorCode_DBError
+		return
+	}
 
 	//判断是否已发送切磋请求
-	qr := this.moduleFriend.ModelFriendQiecuo.getQiecuo(session.GetUserId())
 	if qr == nil {
 		code = pb.ErrorCode_FriendQiecuoTimeout
 		return
@@ -44,9 +48,6 @@ func (this *apiComp) Qiecuo(session comm.IUserSession, req *pb.FriendQiecuoReq)
 		return
 	}
 
-	//保存切磋请求10s
-	this.moduleFriend.ModelFriendQiecuo.createQiecuoRecord(session.GetUserId(), req.TargetUid)
-
 	resp := &pb.FriendQiecuoResp{
 		TargetUid: req.TargetUid,
 		Uid:       session.GetUserId(),

modules/friend/model_qiecuo.go

@@ -37,20 +37,21 @@ func (this *ModelFriendQiecuo) getQiecuo(uid string) *pb.QiecuoRecord {
 }
 
 // 保存切磋记录
-func (this *ModelFriendQiecuo) createQiecuoRecord(uid, targetUid string) error {
+func (this *ModelFriendQiecuo) createQiecuoRecord(uid, targetUid string) (*pb.QiecuoRecord, error) {
 	qr := this.getQiecuo(uid)
 	if qr == nil {
-		if err := this.Add(uid, &pb.QiecuoRecord{
+		r := &pb.QiecuoRecord{
 			Uid:       uid,
 			TargetId:  targetUid,
 			Timestamp: configure.Now().Unix(),
-		}); err != nil {
+		}
+		if err := this.Add(uid, r); err != nil {
 			this.moduleFriend.Errorln(err)
-			return err
+			return nil, err
 		}
 	}
 
-	return nil
+	return qr, nil
 }
 
 // 更新切磋记录