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

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

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

前言

不會吧不會吧, 都2202年了還有人不會寫簽到? redis位圖實現(xiàn)簽到功能簡單方便, 走過路過可不要錯過呦!

基礎(chǔ)知識

位圖源頭

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

Mysql占用對比

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

使用

需要先知道以下幾點:

位圖的內(nèi)容實際上也就是字符串, 只不過是更改的個位的內(nèi)容, 所以分為零存零取和整存零取;
位圖的位數(shù)是會自動補位的, 比如你設(shè)置一個空鍵第8位為1, 則會自動補充前8位為0 (數(shù)組從0開始計數(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       // 前面會自動補位
(integer) 0

// 通過python方法獲取 h 的二進制
>>> 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

功能實現(xiàn)

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

需求

以自然周為周期進行簽到;
展示簽到周期;
重復(fù)簽到提示報錯;

流程圖

簽到周期獲取

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

// SignStartEndTime 簽到起始時間&結(jié)束時間.
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
}

簽到存儲key

因為是以自然周為單位, 所以設(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 已簽到錯誤碼, fmt.Errorf("sign already, val: %d", val)
	}
  // 如果是第一次簽到需要加個過期時間
	signDay := client.BitCount(key, nil).Val()
	if signDay == 1 {
		client.Expire(key, time.Hour*24*7)
	}
	if err = 簽到成功獲取的獎勵; err != nil {
		// 如果添加獎勵失敗, 重置簽到狀態(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
}