猜颜色

modules/dcolor/api_accept.go

@@ -28,13 +28,14 @@ func (this *apiComp) Accept(session comm.IUserSession, req *pb.DColorAcceptReq)
 		redRecord  *pb.DBDColorQiecuoRecord
 		blueRecord *pb.DBDColorQiecuoRecord
 		room       *Room
+		sessions   []comm.IUserSession
 		keep       bool
 	)
 
 	if errdata = this.AcceptCheck(session, req); errdata != nil {
 		return
 	}
-
+	sessions = append(sessions, session.Clone())
 	//校验切磋请求是否超时
 	if redRecord, err = this.module.modelQiecuo.queryQiecuo(req.Uid); err != nil {
 		errdata = &pb.ErrorData{
@@ -80,6 +81,17 @@ func (this *apiComp) Accept(session comm.IUserSession, req *pb.DColorAcceptReq)
 		return
 	}
 
+	if redsession, ok := this.module.GetUserSession(req.Uid); !ok {
+		errdata = &pb.ErrorData{
+			Code:    pb.ErrorCode_BattleUserOff,
+			Title:   pb.ErrorCode_BattleUserOff.ToString(),
+			Message: "req.uid is Off!",
+		}
+		return
+	} else {
+		sessions = append(sessions, redsession)
+	}
+
 	//发起者 red
 	red, err := this.module.ModuleUser.GetUser(req.Uid)
 	if err != nil {
@@ -123,7 +135,7 @@ func (this *apiComp) Accept(session comm.IUserSession, req *pb.DColorAcceptReq)
 		Blue: &pb.DBDColorRoomPlayer{
 			Info: comm.GetUserBaseInfo(blue),
 		},
-	}); err != nil {
+	}, sessions); err != nil {
 		return
 	}
 

modules/dcolor/rooms.go

@@ -2,6 +2,7 @@ package dcolor
 
 import (
 	"fmt"
+	"go_dreamfactory/comm"
 	"go_dreamfactory/lego/core"
 	"go_dreamfactory/lego/core/cbase"
 	"go_dreamfactory/pb"
@@ -35,7 +36,14 @@ func (this *roomsComp) queryRoom(rid string) (room *Room, err error) {
 	return
 }
 
-func (this *roomsComp) newRoom(data *pb.DBDColorRoom) (room *Room, err error) {
+func (this *roomsComp) newRoom(data *pb.DBDColorRoom, session []comm.IUserSession) (room *Room, err error) {
-
+	room = &Room{
+		module:   this.module,
+		data:     data,
+		sessions: session,
+	}
+	this.lock.Lock()
+	this.rooms[data.Rid] = room
+	this.lock.Unlock()
 	return
 }