func (this *WorldServer) DoRegister(msg *msgcmd.Cmd, session *libnet.Session) {
resmsg := msgcmd.NewCmd(msgcmd.MCMD_W2M_REGISTER)
resmsg.AddArg(msg.Args[2]) //在返回的消息中，附上传入的临时用户名
//查找是否已经存在该用户名
row := this.database.QueryData(`SELECT * FROM users WHERE name=?`, msg.Args[0])
var userinfo *common.UserInfo = common.NewUserInfo(session)
if err := row.Scan(&userinfo.ID, &userinfo.Name, &userinfo.Pwd, &userinfo.RegDate); err == sql.ErrNoRows {
//可注册该用户名
userinfo.RegDate = time.Now()
err = this.database.InsertData(`INSERT users (name,pwd,regdate) values (?,?,?)`, msg.Args[0], msg.Args[1], userinfo.RegDate)
if err != nil {
//插入数据失败，返回提示信息
println("World Server Error: 写入注册信息 失败. ", err.Error())
logFile.Printf("World Server Error: 写入注册信息 失败. ", err.Error())
resmsg.AddArg("0")
resmsg.AddArg("注册用户时发生异常，请稍后再试。")
if err = session.Send(resmsg); err != nil {
println("World Server Error: 发送注册回馈信息 失败. ", err.Error())
logFile.Printf("World Server Error: 发送注册回馈信息 失败. ", err.Error())
return
}
} else {
resmsg.AddArg("1")
resmsg.AddArg("注册用户成功，试试登陆吧。")
if err = session.Send(resmsg); err != nil {
println("World Server Error: 发送注册回馈信息 失败. ", err.Error())
logFile.Printf("World Server Error: 发送注册回馈信息 失败. ", err.Error())
return
}
}
} else {
//不可注册该用户名
resmsg.AddArg("0")
resmsg.AddArg("该用户名已被抢先1秒注册，换个试试吧。")
if err = session.Send(resmsg); err != nil {
println("World Server Error: 发送注册回馈信息 失败. ", err.Error())
logFile.Printf("World Server Error: 发送注册回馈信息 失败. ", err.Error())
return
}
}
}
func (this *MsgServer) DoClientExit(user *common.UserInfo) {
//通知WS有用户下线
if user.State == 1 {
user.State = 0
println("[Client -> MS] 用户下线:", user.Name)
offlinemsg := msgcmd.NewCmd(msgcmd.MCMD_M2W_USEROFFLINE)
offlinemsg.AddArg(user.Name)
if err := this.wsSession.Send(offlinemsg); err != nil {
println("Msg Server Error: 发送用户下线信息 失败. ", err.Error())
logFile.Printf("Msg Server Error: 发送用户下线信息 失败. ", err.Error())
return
}
}
//MS 清除该用户信息
this.userlist.Delete(user.Name)
}
func (this *WorldServer) DoLogin(msg *msgcmd.Cmd, session *libnet.Session) {
resmsg := msgcmd.NewCmd(msgcmd.MCMD_W2M_LOGIN)
resmsg.AddArg(msg.Args[2])
var userinfo *common.UserInfo
var isInUserMap = false
var result bool
if user, ok := usermap.Find(msg.Args[0]); ok {
if userinfo, result = user.(*common.UserInfo); !result {
println("World Server Error: (*common.UserInfo) 类型断言 失败. ")
logFile.Printf("World Server Error: (*common.UserInfo) 类型断言 失败. ")
return
}
isInUserMap = true
} else {
//查找是否已经存在该用户名
row := this.database.QueryData(`SELECT * FROM users WHERE name=?`, msg.Args[0])
userinfo = common.NewUserInfo(session)
if err := row.Scan(&userinfo.ID, &userinfo.Name, &userinfo.Pwd, &userinfo.RegDate); err == sql.ErrNoRows {
//登陆失败，没有该用户信息
resmsg.AddArg("0")
if err = session.Send(resmsg); err != nil {
println("World Server Error: 发送登陆回馈信息 失败. ", err.Error())
logFile.Printf("World Server Error: 发送登陆回馈信息 失败. ", err.Error())
return
}
}
}
//找到该用户数据，开始验证密码信息
if userinfo.Name == msg.Args[0] && userinfo.Pwd == msg.Args[1] {
//登陆成功
resmsg.AddArg("1")
resmsg.AddArg(userinfo.GetIDbyString())
resmsg.AddArg(userinfo.Name)
if err := session.Send(resmsg); err != nil {
println("World Server Error: 发送登陆回馈信息 失败. ", err.Error())
logFile.Printf("World Server Error: 发送登陆回馈信息 失败. ", err.Error())
return
}
userinfo.State = 1
} else {
//登陆失败
resmsg.AddArg("0")
if err := session.Send(resmsg); err != nil {
println("World Server Error: 发送登陆回馈信息 失败. ", err.Error())
logFile.Printf("World Server Error: 发送登陆回馈信息 失败. ", err.Error())
return
}
userinfo.State = 0
}
if userinfo.State == 1 {
//有用户上线,广播所有用户
bcMsg := msgcmd.NewCmd(msgcmd.MCMD_W2M_BROADCAST)
bcMsg.AddArg(msgcmd.IntToString(msgcmd.ECMD_BROADCAST_USERLOGIN))
bcMsg.AddArg(userinfo.Name)
this.Broadcast(bcMsg)
}
//保存当前用户所在的MS会话地址
userinfo.Session = session
//添加到usermap中
if !isInUserMap {
usermap.Insert(userinfo.Name, userinfo)
}
}