105 lines
2.9 KiB
Go
105 lines
2.9 KiB
Go
|
package repository
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"app-cloudep-permission-server/internal/domain/repository"
|
|||
|
|
"code.30cm.net/digimon/library-go/utils/invited_code"
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/zeromicro/go-zero/core/stores/redis"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type MemberOnlineStatusRepositoryParam struct {
|
|||
|
|
Store *redis.Redis `name:"redis"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type memberOnlineStatusRepository struct {
|
|||
|
|
store *redis.Redis
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 使用 UID 計算 Bitmap 的 offset
|
|||
|
|
func (t *memberOnlineStatusRepository) uidToOffset(uid string) (int64, error) {
|
|||
|
|
converter := invited_code.MustConverter(10, invited_code.DefaultCodeLen, invited_code.ConvertTable)
|
|||
|
|
|
|||
|
|
// 將 UID 轉換為整數型，並減去基礎 UID (1000000)
|
|||
|
|
uidInt, err := converter.DecodeFromCode(uid)
|
|||
|
|
if err != nil {
|
|||
|
|
return 0, fmt.Errorf("invalid UID: %w", err)
|
|||
|
|
}
|
|||
|
|
// 以 1000000 作為基準
|
|||
|
|
baseUID := invited_code.InitAutoId
|
|||
|
|
if uidInt < int64(baseUID) {
|
|||
|
|
return 0, fmt.Errorf("UID smaller than base: %d", baseUID)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return uidInt - int64(baseUID), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *memberOnlineStatusRepository) SetMemberOnline(ctx context.Context, uid string) (bool, error) {
|
|||
|
|
offset, err := t.uidToOffset(uid)
|
|||
|
|
if err != nil {
|
|||
|
|
return false, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 使用 SET BIT 設置對應的位為 1
|
|||
|
|
_, err = t.store.SetBit("member_status", offset, 1)
|
|||
|
|
if err != nil {
|
|||
|
|
return false, fmt.Errorf("failed to set member online: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return true, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *memberOnlineStatusRepository) SetMemberOffline(ctx context.Context, uid string) (bool, error) {
|
|||
|
|
offset, err := t.uidToOffset(uid)
|
|||
|
|
if err != nil {
|
|||
|
|
return false, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 使用 SET BIT 設置對應的位為 1
|
|||
|
|
_, err = t.store.SetBit("member_status", offset, 0)
|
|||
|
|
if err != nil {
|
|||
|
|
return false, fmt.Errorf("failed to set member offline: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return true, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *memberOnlineStatusRepository) IsMemberOnline(ctx context.Context, uid string) (bool, error) {
|
|||
|
|
offset, err := t.uidToOffset(uid)
|
|||
|
|
if err != nil {
|
|||
|
|
return false, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 使用 GET BIT 獲取對應的位，1 表示在線，0 表示離線
|
|||
|
|
status, err := t.store.GetBit("member_status", offset)
|
|||
|
|
if err != nil {
|
|||
|
|
return false, fmt.Errorf("failed to get member status: %w", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return status == 1, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (t *memberOnlineStatusRepository) QueryMemberOnlineList(ctx context.Context, uids []string) ([]repository.MemberOnlineStatusResp, error) {
|
|||
|
|
statusMap := make([]repository.MemberOnlineStatusResp, 0, len(uids))
|
|||
|
|
|
|||
|
|
// 遍歷所有用戶的 UID，查詢他們的在線狀態
|
|||
|
|
for _, uid := range uids {
|
|||
|
|
isOnline, err := t.IsMemberOnline(ctx, uid)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("failed to query member status for UID %s: %w", uid, err)
|
|||
|
|
}
|
|||
|
|
statusMap = append(statusMap, repository.MemberOnlineStatusResp{
|
|||
|
|
UID: uid,
|
|||
|
|
Status: isOnline,
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return statusMap, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewMemberOnlineStatusRepository(param MemberOnlineStatusRepositoryParam) repository.MemberOnlineStatusRepository {
|
|||
|
|
return &memberOnlineStatusRepository{
|
|||
|
|
store: param.Store,
|
|||
|
|
}
|
|||
|
|
}
|