//常用的redis命令

CONFIG SET requirepass "mypass"

//Hashmap

hset [key] [field] value]

hget [key] [field]

hgetall [key]

//List

LPUSH [key] [value]

RPUSH [key] [value]

LPOP [key]

RPOP [key]

//删除表头2个值为value的元素

LREM [key] [count > 0] [value]

//删除表尾2个值为value的元素

LREM [key] [count < 0] [value]

//删除所有值为value的元素

LREM [key] [count = 0] [value]

LREM mylist 2 "hello"

LREM mylist -2 "hello"

//查询0~最后1个元素(不删除)

LRANGE mylist 0 -1

BRPOP [key] [timeout]

BLPOP [key] [timeout]

//例：获取message_1最左的元素，如果没有，等待5秒，超时返回nil

BLPOP message_1 5

采用Redis进行数据存储，主要包括频控、限流、用户表、在线用户表、聊天消息表(redis list实现消息队列)、好友表(TODO)

频控

CheckFrequency(userId uint64) bool

返回true检查通过，false触发频控

visited_{user_id} >3触发

//频控key前缀

const freqPre string = "visited_"

/*

Desc: 检查频控key

Input: userId

Output:

False 触发频控/异常

True 正常请求

*/

func CheckFrequency(userId uint64) bool {

key := freqPre + strconv.FormatUint(userId, 10)

resExists, err := Client.Exists(key).Result()

if err != nil {

return false

}

if resExists == 0 { //该key不存在

Client.Set(key, 0, 100 * time.Second)

return true

}

res, err := Client.Get(key).Result()

if err != nil {

return false

}

resCnt, err := strconv.Atoi(res)

if err != nil {

return false

}

if resCnt > 2 {//10秒同一userid不能访问超过2次

fmt.Println("触发频控")

//...

return false

}

Client.Incr(key)

return true

}

频控测试

redis.CreateClient()

var res bool = true

for res {

time.Sleep(time.Second * 1)

res = redis.CheckFrequency(54508)

fmt.Println(res)

}

限流

/*

令牌桶算法API

*/

const LevelFast int = 1

const LevelMedium int = 2

const LevelSlow int = 3

var tokenBucket *ratelimit.Bucket

func InitToken(speedLevel int) {

var bucketFillDuring time.Duration

if speedLevel == LevelFast {

bucketFillDuring = 20 * time.Millisecond

} else if speedLevel == LevelMedium {

bucketFillDuring = 100 * time.Millisecond

} else if speedLevel == LevelSlow {

bucketFillDuring = 1000 * time.Millisecond

} else {

//日志

fmt.Println("speedLevel只能为Fast, Medium, Slow三个值！")

return

}

var bucketMax int64 = 1//令牌桶最大容量，初始也会有这么多个令牌，此处为1是测试所用，方便看到限流效果，后期需增加大小

tokenBucket = ratelimit.NewBucket(bucketFillDuring, bucketMax)

}

/*

True:得到访问权限，通过

False:目前无令牌，拒绝

*/

func CheckToken() bool {

if tokenBucket == nil {

return false

}

//获取令牌

available := tokenBucket.TakeAvailable(1)

if available <= 0 {

//限流处理

available = tokenBucket.TakeAvailable(1)

fmt.Println("服务器开小差啦")

return false

}

return true

}

限流测试

TokenBucket.InitToken(TokenBucket.LevelFast)

for i := 0; i < 10; i++ {

fmt.Println("尝试连接服务器...")

token := TokenBucket.CheckToken()

if !token {

return

}

fmt.Println("连接成功")

time.Sleep(time.Duration(int64(rand.Int() % 40)) * time.Millisecond)//模拟用户随机访问(间隔0~40秒访问)

}

用户表 users_{user_id}

hashmap

字段

数据类型

描述

userid

Integer

用户uid

type

string

root = 管理员， normal=普通用户，vip=特权用户

username

string

用户名

password

string

密码

password_adv

string

二级密码(密码找回保留信息)

nickname

string

昵称

telephone

string

微信/电话

email

string

电子邮箱

age

Integer

年龄

sex

string

枚举值{boy, gril, unknown}

remark

string

备注

name->uid映射表 username_to_id_{username}

字段

数据类型

描述

userid

Integer

用户uid

在线用户表 users_online_{user_id}

离线用户key为空

数据结构：key-value

取值：

1=在线

2=离开

3=隐身(VIP功能)

聊天消息(redis list实现消息队列) chat_log_{接收者userid}_{发送者userid}

数据结构：list

字段

数据类型

描述

send_id

Integer

发送者uid

recv_id

Integer

接收者uid

send_time

string

发送时间

send_content

string

消息内容

remark

string

保留字段

好友关系表 friendship_{userid}

数据结构：hashmap

字段

数据类型

描述

userid

Integer

用户uid

friendid

Integer

好友uid