go语言聊天室实现（八）注册逻辑的具体操作

在这里首先要引入一个概念，go语言的继承。
在PHP中，我们如果想用一个类去继承父类，直接用extend方法就可以继承父类中的方法和属性。
但是在go中是没有类的概念的，但是我们可以看到struct和类在很大程度上都是相同的。
如果想在一个struct中继承另一个struct,直接将父struct放到子struct中就可以完成。

type animal struct{
//这是一个动物类
}
type cat struct{
  //这是一个子类
  //继承的时候直接将animal放在这里即可
  animal
}

通过上述的操作我们就可以在子类中直接使用父类的属性和方法了。

type LoginService struct {
    services.CommonService
    common.CommonFunction
    Index int
}

首先我们新建一个LoginService的结构体，并定义一个index,这个index和之前我们创建的索引值是一样的.
我们为LoginService创建一个方法，名为Register

func (l *LoginService) Register(umsg common.UserMessage) (result returnData) {
    if umsg.FirstTopic == 0 || umsg.SecondTopic == 0 {
        logs.Error("必须要带有一级或二级topic才能通过验证！")
    }
    if umsg.UserId == 0 {
        logs.Error("用户ID不存在，请重试！")
    }
    result = l.saveToRedis(umsg)
    return
}

在这个方法中传过来的是之前解析的用户发送的消息。
做了判断之后，通过一个saveToRedis方法来进行处理。

func (l *LoginService) saveToRedis(umsg common.UserMessage) (result returnData) {
    rk := services.RedisKeyGroupUser
    rk = strings.Replace(rk, "first_topic", strconv.Itoa(umsg.FirstTopic), -1)
    rk = strings.Replace(rk, "second_topic", strconv.Itoa(umsg.SecondTopic), -1)
    fmt.Println("rk==============", rk)
    client, _ := new(pool.Pool).GetRedisInstance()
    defer client.Close()
    userList, err := client.Do("GET", rk)
    if err != nil {
        logs.Error("get user list error:", err)
    }
    userListArr:=strings.Split(l.B2S(userList),",")
    userListArr = append(userListArr, strconv.Itoa(umsg.UserId))
    userListArr = l.UniqueArr(userListArr)
    client.Do("SET", rk, strings.Replace(strings.Trim(fmt.Sprint(userListArr), "[]"), " ", ",", -1))

    //存入用户-组
    rk = services.RedisKeyUserGroup
    rk = strings.Replace(rk, "fd", strconv.Itoa(l.Index), -1)
    fmt.Println("rk===============", rk)
    client.Do("SET", rk, strconv.Itoa(umsg.FirstTopic)+"_"+strconv.Itoa(umsg.SecondTopic))
    //用户对应的fd
    client.Do("HSET", services.UserBindRedisKey, strconv.Itoa(l.Index), strconv.Itoa(umsg.UserId))
    //fd对应的user
    client.Do("HSET", services.FdBindUserRedisKey, strconv.Itoa(umsg.UserId), strconv.Itoa(l.Index))

    return

}

在我们初始化Redis实例之后，定义了一个defer.
defer相当于PHP的析构方法，在逻辑执行完成之后，总是会被调用一次，在这里我们可以做一些自定义的操作。
对于数据的操作与之前使用Swoole进行的操作也是基本一样的，值得注意的是，Redis返回的数据结构是[]uint8类型的，我们需要将其转换为string类型来进行使用。

//将redis返回的数据转换为字符串。
func (c *CommonFunction) B2S(bs interface{}) string {
    ba := []byte{}
    if s,ok:=bs.([]uint8);ok{
        for _, b := range s {
            ba = append(ba, byte(b))
        }
    }
    return string(ba)
}

具体的操作是通过这个结构体来完成的。
完成了这一步，我们也就已经完成了整个注册逻辑。