快捷導(dǎo)航

PHP利用redis位圖實(shí)現(xiàn)簡(jiǎn)單的簽到功能

更新時(shí)間：2023年06月13日 14:09:08 作者：Masters

在日常開(kāi)發(fā)中, 我們會(huì)遇到需要存儲(chǔ)大量 bool類(lèi)型數(shù)據(jù)的需求, 比如用戶簽到和用戶登陸的記錄等, 本文將為大家介紹如何利用redis位圖輕松實(shí)現(xiàn)簽到功能，感興趣的可以了解一下

前言

不會(huì)吧不會(huì)吧, 都2202年了還有人不會(huì)寫(xiě)簽到? redis位圖實(shí)現(xiàn)簽到功能簡(jiǎn)單方便, 走過(guò)路過(guò)可不要錯(cuò)過(guò)呦!

基礎(chǔ)知識(shí)

位圖源頭

在日常開(kāi)發(fā)中, 我們會(huì)遇到需要存儲(chǔ)大量 bool類(lèi)型數(shù)據(jù)的需求, 比如用戶簽到和用戶登陸的記錄等, 這個(gè)時(shí)候用mysql存儲(chǔ)來(lái)說(shuō)比較占用資源, 所以為了解決這個(gè)問(wèn)題, redis提供了位圖數(shù)據(jù)結(jié)構(gòu)(就是 位數(shù)組), 每個(gè) bool值只占用1個(gè)位, 8個(gè)位組成一個(gè)字節(jié), 這樣存儲(chǔ)空間的節(jié)約率不用我多說(shuō)吧;

Mysql占用對(duì)比

mysql 存一個(gè)thinyint需要占用1個(gè)字節(jié)(bool類(lèi)型默認(rèn)為thinyint(1)), 而且你還需要存一個(gè)主鍵Id(你不存也會(huì)自動(dòng)隱性的幫你存一列), int的話需要占用 4個(gè)字節(jié), 不算其他光是這兩個(gè)字段存儲(chǔ)你就需要5個(gè)字節(jié), 而在位圖里面5個(gè)字節(jié)都?jí)虼?0條記錄了...

使用

需要先知道以下幾點(diǎn):

位圖的內(nèi)容實(shí)際上也就是字符串, 只不過(guò)是更改的個(gè)位的內(nèi)容, 所以分為零存零取和整存零取;
位圖的位數(shù)是會(huì)自動(dòng)補(bǔ)位的, 比如你設(shè)置一個(gè)空鍵第8位為1, 則會(huì)自動(dòng)補(bǔ)充前8位為0 (數(shù)組從0開(kāi)始計(jì)數(shù));

127.0.0.1:6379> setbit zero 0 1     // 設(shè)置第0位為true
(integer) 0
127.0.0.1:6379> getbit zero 0
(integer) 1
127.0.0.1:6379> setbit zero 8 1     // 設(shè)置第8位為true
(integer) 0
127.0.0.1:6379> getbit zero 8
(integer) 1
127.0.0.1:6379> getbit zero 7       // 前面會(huì)自動(dòng)補(bǔ)位
(integer) 0

// 通過(guò)python方法獲取 h 的二進(jìn)制
>>> bin(ord("h"))
'0b1101000'
127.0.0.1:6379> set one h         // 直接存入字符 h = 01101000
OK
127.0.0.1:6379> getbit one 0
(integer) 0
127.0.0.1:6379> getbit one 1
(integer) 1
127.0.0.1:6379> getbit one 2
(integer) 1
127.0.0.1:6379> getbit one 3
(integer) 0
127.0.0.1:6379> getbit one 4
(integer) 1
127.0.0.1:6379> getbit one 5
(integer) 0

功能實(shí)現(xiàn)

看完上面的基礎(chǔ)知識(shí)大家就都基本知道怎么實(shí)現(xiàn)了, 下面是實(shí)戰(zhàn)環(huán)節(jié)

需求

以自然周為周期進(jìn)行簽到;
展示簽到周期;
重復(fù)簽到提示報(bào)錯(cuò);

流程圖

簽到周期獲取

獲取每個(gè)自然周的起始和結(jié)束時(shí)間

// SignStartEndTime 簽到起始時(shí)間&結(jié)束時(shí)間.
func (slf *TaskService) SignStartEndTime() (startTime, endTime int64) {
	now := time.Now()
	weekDay := int(now.Weekday()) 
  // 如果是周日的話weekDay=0
	if weekDay == 0 {  
		weekDay = 7
	}
	startTime = time.Date(now.Year(), now.Month(), now.Day()-weekDay+1, 0, 0, 0, 0, now.Location()).Unix()
	endTime = startTime + 86400*7 - 1
	return
}

簽到存儲(chǔ)key

因?yàn)槭且宰匀恢転閱挝? 所以設(shè)定key的格式為 user:sign:userId:20221107(周一的日期)

// SignKey 簽到key.
func (slf *TaskService) SignKey(userId string) string {
	st, _ := slf.SignStartEndTime()
	return fmt.Sprintf("user:sign:%s:%s", userId, time.Unix(st, 0).Format("20060102"))
}

簽到

const (
	SignTypeNo  = iota // 未簽到
	SignTypeYes        // 已簽到
)
// Sign 簽到.
func (slf *TaskService) Sign(userId string) (code errcode.ErrCode, err error) {
	var (
		client  = redis.GetClient()
		key     = slf.SignKey(userId)
		weekDay = int64(time.Now().Weekday())
	)
	if weekDay == 0 {
		weekDay = 7
	}
	val, err := client.SetBit(key, weekDay, SignTypeYes).Result()
	if err != nil {
		return errcode.ServerErr, fmt.Errorf("sign setBit err: %v", err)
	}
	if val == SignTypeYes {
		return 已簽到錯(cuò)誤碼, fmt.Errorf("sign already, val: %d", val)
	}
  // 如果是第一次簽到需要加個(gè)過(guò)期時(shí)間
	signDay := client.BitCount(key, nil).Val()
	if signDay == 1 {
		client.Expire(key, time.Hour*24*7)
	}
	if err = 簽到成功獲取的獎(jiǎng)勵(lì); err != nil {
		// 如果添加獎(jiǎng)勵(lì)失敗, 重置簽到狀態(tài)
		if err = client.SetBit(key, weekDay, SignTypeNo).Err(); err != nil {
			log.Errorf("sign reset err: %v", err)
		}
		return errcode.ServerErr, fmt.Errorf("sign add reward err: %v", err)
	}
	return errcode.Code200, nil
}