在這里首先要引入一個概念,go語言的繼承。
在PHP中,我們?nèi)绻胗靡粋€類去繼承父類,直接用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的結(jié)構(gòu)體,并定義一個index,這個index和之前我們創(chuàng)建的索引值是一樣的.
我們?yōu)長oginService創(chuàng)建一個方法,名為Register
func (l *LoginService) Register(umsg common.UserMessage) (result returnData) {
if umsg.FirstTopic == 0 || umsg.SecondTopic == 0 {
logs.Error("必須要帶有一級或二級topic才能通過驗(yàn)證!")
}
if umsg.UserId == 0 {
logs.Error("用戶ID不存在,請重試!")
}
result = l.saveToRedis(umsg)
return
}
在這個方法中傳過來的是之前解析的用戶發(fā)送的消息。
做了判斷之后,通過一個saveToRedis方法來進(jìn)行處理。
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))
//用戶對應(yīng)的fd
client.Do("HSET", services.UserBindRedisKey, strconv.Itoa(l.Index), strconv.Itoa(umsg.UserId))
//fd對應(yīng)的user
client.Do("HSET", services.FdBindUserRedisKey, strconv.Itoa(umsg.UserId), strconv.Itoa(l.Index))
return
}
在我們初始化Redis實(shí)例之后,定義了一個defer.
defer相當(dāng)于PHP的析構(gòu)方法,在邏輯執(zhí)行完成之后,總是會被調(diào)用一次,在這里我們可以做一些自定義的操作。
對于數(shù)據(jù)的操作與之前使用Swoole進(jìn)行的操作也是基本一樣的,值得注意的是,Redis返回的數(shù)據(jù)結(jié)構(gòu)是[]uint8類型的,我們需要將其轉(zhuǎn)換為string類型來進(jìn)行使用。
//將redis返回的數(shù)據(jù)轉(zhuǎn)換為字符串。
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)
}
具體的操作是通過這個結(jié)構(gòu)體來完成的。
完成了這一步,我們也就已經(jīng)完成了整個注冊邏輯。