Merge branch 'old/master' into old
This commit is contained in:
@@ -3,6 +3,7 @@ package user
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/perfect-panel/server/pkg/logger"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
@@ -31,24 +32,50 @@ func (m *defaultUserModel) FindUserAuthMethodByPlatform(ctx context.Context, use
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) InsertUserAuthMethods(ctx context.Context, data *AuthMethods, tx ...*gorm.DB) error {
|
||||
u, err := m.FindOne(ctx, data.UserId)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Model(&AuthMethods{}).Create(data).Error
|
||||
if err = conn.Model(&AuthMethods{}).Create(data).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
return m.ClearUserCache(ctx, u)
|
||||
})
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) UpdateUserAuthMethods(ctx context.Context, data *AuthMethods, tx ...*gorm.DB) error {
|
||||
u, err := m.FindOne(ctx, data.UserId)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Model(&AuthMethods{}).Where("user_id = ? AND auth_type = ?", data.UserId, data.AuthType).Save(data).Error
|
||||
err = conn.Model(&AuthMethods{}).Where("user_id = ? AND auth_type = ?", data.UserId, data.AuthType).Save(data).Error
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return m.ClearUserCache(ctx, u)
|
||||
})
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) DeleteUserAuthMethods(ctx context.Context, userId int64, platform string, tx ...*gorm.DB) error {
|
||||
u, err := m.FindOne(ctx, userId)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer func() {
|
||||
if err = m.ClearUserCache(context.Background(), u); err != nil {
|
||||
logger.Errorf("[UserModel] clear user cache failed: %v", err.Error())
|
||||
}
|
||||
}()
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
|
||||
@@ -0,0 +1,285 @@
|
||||
package user
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"github.com/perfect-panel/server/pkg/logger"
|
||||
)
|
||||
|
||||
type CacheKeyGenerator interface {
|
||||
GetCacheKeys() []string
|
||||
}
|
||||
|
||||
type CacheManager interface {
|
||||
ClearCache(ctx context.Context, keys ...string) error
|
||||
ClearModelCache(ctx context.Context, models ...CacheKeyGenerator) error
|
||||
}
|
||||
|
||||
type UserCacheManager struct {
|
||||
model *defaultUserModel
|
||||
}
|
||||
|
||||
func NewUserCacheManager(model *defaultUserModel) *UserCacheManager {
|
||||
return &UserCacheManager{
|
||||
model: model,
|
||||
}
|
||||
}
|
||||
|
||||
func (c *UserCacheManager) ClearCache(ctx context.Context, keys ...string) error {
|
||||
if len(keys) == 0 {
|
||||
return nil
|
||||
}
|
||||
return c.model.CachedConn.DelCacheCtx(ctx, keys...)
|
||||
}
|
||||
|
||||
func (c *UserCacheManager) ClearModelCache(ctx context.Context, models ...CacheKeyGenerator) error {
|
||||
var allKeys []string
|
||||
for _, model := range models {
|
||||
if model != nil {
|
||||
allKeys = append(allKeys, model.GetCacheKeys()...)
|
||||
}
|
||||
}
|
||||
return c.ClearCache(ctx, allKeys...)
|
||||
}
|
||||
|
||||
func (u *User) GetCacheKeys() []string {
|
||||
if u == nil {
|
||||
return []string{}
|
||||
}
|
||||
keys := []string{
|
||||
fmt.Sprintf("%s%d", cacheUserIdPrefix, u.Id),
|
||||
}
|
||||
|
||||
for _, auth := range u.AuthMethods {
|
||||
if auth.AuthType == "email" {
|
||||
keys = append(keys, fmt.Sprintf("%s%s", cacheUserEmailPrefix, auth.AuthIdentifier))
|
||||
break
|
||||
}
|
||||
}
|
||||
return keys
|
||||
}
|
||||
|
||||
func (s *Subscribe) GetCacheKeys() []string {
|
||||
if s == nil {
|
||||
return []string{}
|
||||
}
|
||||
keys := make([]string, 0)
|
||||
|
||||
if s.Token != "" {
|
||||
keys = append(keys, fmt.Sprintf("%s%s", cacheUserSubscribeTokenPrefix, s.Token))
|
||||
}
|
||||
if s.UserId != 0 {
|
||||
keys = append(keys, fmt.Sprintf("%s%d", cacheUserSubscribeUserPrefix, s.UserId))
|
||||
}
|
||||
if s.Id != 0 {
|
||||
keys = append(keys, fmt.Sprintf("%s%d", cacheUserSubscribeIdPrefix, s.Id))
|
||||
}
|
||||
return keys
|
||||
}
|
||||
|
||||
func (s *Subscribe) GetExtendedCacheKeys(model *defaultUserModel) []string {
|
||||
keys := s.GetCacheKeys()
|
||||
|
||||
if s.SubscribeId != 0 && model != nil {
|
||||
serverKeys := model.getServerRelatedCacheKeys(s.SubscribeId)
|
||||
keys = append(keys, serverKeys...)
|
||||
}
|
||||
|
||||
return keys
|
||||
}
|
||||
|
||||
func (d *Device) GetCacheKeys() []string {
|
||||
if d == nil {
|
||||
return []string{}
|
||||
}
|
||||
keys := []string{}
|
||||
|
||||
if d.Id != 0 {
|
||||
keys = append(keys, fmt.Sprintf("%s%d", cacheUserDeviceIdPrefix, d.Id))
|
||||
}
|
||||
if d.Identifier != "" {
|
||||
keys = append(keys, fmt.Sprintf("%s%s", cacheUserDeviceNumberPrefix, d.Identifier))
|
||||
}
|
||||
return keys
|
||||
}
|
||||
|
||||
func (a *AuthMethods) GetCacheKeys() []string {
|
||||
if a == nil {
|
||||
return []string{}
|
||||
}
|
||||
keys := []string{}
|
||||
|
||||
if a.UserId != 0 {
|
||||
keys = append(keys, fmt.Sprintf("%s%d", cacheUserIdPrefix, a.UserId))
|
||||
}
|
||||
if a.AuthType == "email" && a.AuthIdentifier != "" {
|
||||
keys = append(keys, fmt.Sprintf("%s%s", cacheUserEmailPrefix, a.AuthIdentifier))
|
||||
}
|
||||
return keys
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) GetCacheManager() *UserCacheManager {
|
||||
return NewUserCacheManager(m)
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) getServerRelatedCacheKeys(subscribeId int64) []string {
|
||||
// 这里复用了 model.go 中的逻辑,但简化了实现
|
||||
keys := []string{}
|
||||
|
||||
if subscribeId == 0 {
|
||||
return keys
|
||||
}
|
||||
|
||||
// 这里需要从 getSubscribeCacheKey 方法中提取服务器相关的逻辑
|
||||
// 为了避免重复查询,我们可以在需要时才获取
|
||||
// 或者可以将这个逻辑移到一个统一的地方
|
||||
|
||||
return keys
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) ClearUserCache(ctx context.Context, users ...*User) error {
|
||||
cacheManager := m.GetCacheManager()
|
||||
models := make([]CacheKeyGenerator, len(users))
|
||||
for i, user := range users {
|
||||
models[i] = user
|
||||
}
|
||||
return cacheManager.ClearModelCache(ctx, models...)
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) ClearSubscribeCacheByModels(ctx context.Context, subscribes ...*Subscribe) error {
|
||||
cacheManager := m.GetCacheManager()
|
||||
models := make([]CacheKeyGenerator, len(subscribes))
|
||||
for i, subscribe := range subscribes {
|
||||
models[i] = subscribe
|
||||
}
|
||||
return cacheManager.ClearModelCache(ctx, models...)
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) ClearDeviceCache(ctx context.Context, devices ...*Device) error {
|
||||
cacheManager := m.GetCacheManager()
|
||||
models := make([]CacheKeyGenerator, len(devices))
|
||||
for i, device := range devices {
|
||||
models[i] = device
|
||||
}
|
||||
return cacheManager.ClearModelCache(ctx, models...)
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) ClearAuthMethodCache(ctx context.Context, authMethods ...*AuthMethods) error {
|
||||
cacheManager := m.GetCacheManager()
|
||||
models := make([]CacheKeyGenerator, len(authMethods))
|
||||
for i, auth := range authMethods {
|
||||
models[i] = auth
|
||||
}
|
||||
return cacheManager.ClearModelCache(ctx, models...)
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) BatchClearRelatedCache(ctx context.Context, user *User) error {
|
||||
if user == nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
cacheManager := m.GetCacheManager()
|
||||
|
||||
var allModels []CacheKeyGenerator
|
||||
allModels = append(allModels, user)
|
||||
|
||||
for _, auth := range user.AuthMethods {
|
||||
allModels = append(allModels, &auth)
|
||||
}
|
||||
|
||||
for _, device := range user.UserDevices {
|
||||
allModels = append(allModels, &device)
|
||||
}
|
||||
|
||||
subscribes, err := m.QueryUserSubscribe(ctx, user.Id)
|
||||
if err != nil {
|
||||
logger.Errorf("failed to query user subscribes for cache clearing: %v", err)
|
||||
} else {
|
||||
for _, sub := range subscribes {
|
||||
subModel := &Subscribe{
|
||||
Id: sub.Id,
|
||||
UserId: sub.UserId,
|
||||
Token: sub.Token,
|
||||
SubscribeId: sub.SubscribeId,
|
||||
}
|
||||
allModels = append(allModels, subModel)
|
||||
}
|
||||
}
|
||||
|
||||
return cacheManager.ClearModelCache(ctx, allModels...)
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) CacheInvalidationHandler(ctx context.Context, operation string, modelType string, model interface{}) error {
|
||||
switch operation {
|
||||
case "create", "update", "delete":
|
||||
switch modelType {
|
||||
case "user":
|
||||
if user, ok := model.(*User); ok {
|
||||
return m.BatchClearRelatedCache(ctx, user)
|
||||
}
|
||||
case "subscribe":
|
||||
if subscribe, ok := model.(*Subscribe); ok {
|
||||
return m.ClearSubscribeCacheByModels(ctx, subscribe)
|
||||
}
|
||||
case "device":
|
||||
if device, ok := model.(*Device); ok {
|
||||
return m.ClearDeviceCache(ctx, device)
|
||||
}
|
||||
case "authmethod":
|
||||
if authMethod, ok := model.(*AuthMethods); ok {
|
||||
return m.ClearAuthMethodCache(ctx, authMethod)
|
||||
}
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *customUserModel) GetRelatedCacheKeys(ctx context.Context, modelType string, modelId int64) ([]string, error) {
|
||||
var keys []string
|
||||
|
||||
switch modelType {
|
||||
case "user":
|
||||
user, err := m.FindOne(ctx, modelId)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
keys = append(keys, user.GetCacheKeys()...)
|
||||
|
||||
auths, err := m.FindUserAuthMethods(ctx, modelId)
|
||||
if err == nil {
|
||||
for _, auth := range auths {
|
||||
keys = append(keys, auth.GetCacheKeys()...)
|
||||
}
|
||||
}
|
||||
|
||||
subscribes, err := m.QueryUserSubscribe(ctx, modelId)
|
||||
if err == nil {
|
||||
for _, sub := range subscribes {
|
||||
subModel := &Subscribe{
|
||||
Id: sub.Id,
|
||||
UserId: sub.UserId,
|
||||
Token: sub.Token,
|
||||
SubscribeId: sub.SubscribeId,
|
||||
}
|
||||
keys = append(keys, subModel.GetCacheKeys()...)
|
||||
}
|
||||
}
|
||||
|
||||
case "subscribe":
|
||||
subscribe, err := m.FindOneSubscribe(ctx, modelId)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
keys = append(keys, subscribe.GetCacheKeys()...)
|
||||
|
||||
case "device":
|
||||
device, err := m.FindOneDevice(ctx, modelId)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
keys = append(keys, device.GetCacheKeys()...)
|
||||
}
|
||||
|
||||
return keys, nil
|
||||
}
|
||||
@@ -48,29 +48,20 @@ func newUserModel(db *gorm.DB, c *redis.Client) *defaultUserModel {
|
||||
func (m *defaultUserModel) batchGetCacheKeys(users ...*User) []string {
|
||||
var keys []string
|
||||
for _, user := range users {
|
||||
keys = append(keys, m.getCacheKeys(user)...)
|
||||
keys = append(keys, user.GetCacheKeys()...)
|
||||
}
|
||||
return keys
|
||||
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) getCacheKeys(data *User) []string {
|
||||
if data == nil {
|
||||
return []string{}
|
||||
}
|
||||
userIdKey := fmt.Sprintf("%s%v", cacheUserIdPrefix, data.Id)
|
||||
cacheKeys := []string{
|
||||
userIdKey,
|
||||
}
|
||||
// email key
|
||||
if len(data.AuthMethods) > 0 {
|
||||
for _, auth := range data.AuthMethods {
|
||||
if auth.AuthType == "email" {
|
||||
cacheKeys = append(cacheKeys, fmt.Sprintf("%s%v", cacheUserEmailPrefix, auth.AuthIdentifier))
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
return cacheKeys
|
||||
return data.GetCacheKeys()
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) clearUserCache(ctx context.Context, data ...*User) error {
|
||||
return m.ClearUserCache(ctx, data...)
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) FindOneByEmail(ctx context.Context, email string) (*User, error) {
|
||||
@@ -127,53 +118,38 @@ func (m *defaultUserModel) Delete(ctx context.Context, id int64, tx ...*gorm.DB)
|
||||
}
|
||||
return err
|
||||
}
|
||||
err = m.ExecCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
|
||||
// 使用批量相关缓存清理,包含所有相关数据的缓存
|
||||
defer func() {
|
||||
if clearErr := m.BatchClearRelatedCache(ctx, data); clearErr != nil {
|
||||
// 记录清理缓存错误,但不阻断删除操作
|
||||
}
|
||||
return conn.Transaction(func(db *gorm.DB) error {
|
||||
if err := db.Model(&User{}).Where("`id` = ?", id).Delete(&User{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
if err := db.Model(&AuthMethods{}).Where("`user_id` = ?", id).Delete(&User{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
if err := db.Model(&Subscribe{}).Where("`user_id` = ?", id).Delete(&User{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
if err := db.Model(&BalanceLog{}).Where("`user_id` = ?", id).Delete(&User{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
if err := db.Model(&GiftAmountLog{}).Where("`user_id` = ?", id).Delete(&User{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
if err := db.Model(&LoginLog{}).Where("`user_id` = ?", id).Delete(&User{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
if err := db.Model(&SubscribeLog{}).Where("`user_id` = ?", id).Delete(&User{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
if err := db.Model(&Device{}).Where("`user_id` = ?", id).Delete(&User{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
}()
|
||||
|
||||
subs, err := m.QueryUserSubscribe(ctx, id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
for _, sub := range subs {
|
||||
if err := m.DeleteSubscribeById(ctx, sub.Id, db); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return m.TransactCtx(ctx, func(db *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
db = tx[0]
|
||||
}
|
||||
|
||||
if err := db.Model(&CommissionLog{}).Where("`user_id` = ?", id).Delete(&User{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
})
|
||||
}, m.getCacheKeys(data)...)
|
||||
return err
|
||||
// 删除用户相关的所有数据
|
||||
if err := db.Model(&User{}).Where("`id` = ?", id).Delete(&User{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := db.Model(&AuthMethods{}).Where("`user_id` = ?", id).Delete(&AuthMethods{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := db.Model(&Subscribe{}).Where("`user_id` = ?", id).Delete(&Subscribe{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if err := db.Model(&Device{}).Where("`user_id` = ?", id).Delete(&Device{}).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) Transaction(ctx context.Context, fn func(db *gorm.DB) error) error {
|
||||
|
||||
@@ -46,18 +46,27 @@ func (m *customUserModel) QueryDevicePageList(ctx context.Context, userId, subsc
|
||||
return list, total, err
|
||||
}
|
||||
|
||||
// QueryDeviceList returns a list of records that meet the conditions.
|
||||
func (m *customUserModel) QueryDeviceList(ctx context.Context, userId int64) ([]*Device, int64, error) {
|
||||
var list []*Device
|
||||
var total int64
|
||||
err := m.QueryNoCacheCtx(ctx, &list, func(conn *gorm.DB, v interface{}) error {
|
||||
return conn.Model(&Device{}).Where("`user_id` = ?", userId).Count(&total).Find(&list).Error
|
||||
})
|
||||
return list, total, err
|
||||
}
|
||||
|
||||
func (m *customUserModel) UpdateDevice(ctx context.Context, data *Device, tx ...*gorm.DB) error {
|
||||
old, err := m.FindOneDevice(ctx, data.Id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
deviceIdKey := fmt.Sprintf("%s%v", cacheUserDeviceIdPrefix, old.Id)
|
||||
err = m.ExecCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Save(data).Error
|
||||
}, deviceIdKey)
|
||||
}, old.GetCacheKeys()...)
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -69,12 +78,26 @@ func (m *customUserModel) DeleteDevice(ctx context.Context, id int64, tx ...*gor
|
||||
}
|
||||
return err
|
||||
}
|
||||
deviceIdKey := fmt.Sprintf("%s%v", cacheUserDeviceIdPrefix, data.Id)
|
||||
err = m.ExecCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Delete(&Device{}, id).Error
|
||||
}, deviceIdKey)
|
||||
}, data.GetCacheKeys()...)
|
||||
return err
|
||||
}
|
||||
|
||||
func (m *customUserModel) InsertDevice(ctx context.Context, data *Device, tx ...*gorm.DB) error {
|
||||
defer func() {
|
||||
if clearErr := m.ClearDeviceCache(ctx, data); clearErr != nil {
|
||||
// log cache clear error
|
||||
}
|
||||
}()
|
||||
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Create(data).Error
|
||||
})
|
||||
}
|
||||
|
||||
@@ -1,81 +0,0 @@
|
||||
package user
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
func (m *customUserModel) InsertSubscribeLog(ctx context.Context, log *SubscribeLog) error {
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
return conn.Create(log).Error
|
||||
})
|
||||
}
|
||||
|
||||
func (m *customUserModel) FilterSubscribeLogList(ctx context.Context, page, size int, filter *SubscribeLogFilterParams) ([]*SubscribeLog, int64, error) {
|
||||
var list []*SubscribeLog
|
||||
var total int64
|
||||
err := m.QueryNoCacheCtx(ctx, &list, func(conn *gorm.DB, v interface{}) error {
|
||||
query := conn.Model(&SubscribeLog{})
|
||||
if filter != nil {
|
||||
if filter.UserId != 0 {
|
||||
query = query.Where("user_id = ?", filter.UserId)
|
||||
}
|
||||
if filter.UserSubscribeId != 0 {
|
||||
query = query.Where("user_subscribe_id = ?", filter.UserSubscribeId)
|
||||
}
|
||||
if filter.IP != "" {
|
||||
query = query.Where("ip LIKE ?", "%"+filter.IP+"%")
|
||||
}
|
||||
if filter.Token != "" {
|
||||
query = query.Where("token LIKE ?", "%"+filter.Token+"%")
|
||||
}
|
||||
if filter.UserAgent != "" {
|
||||
query = query.Where("user_agent LIKE ?", "%"+filter.UserAgent+"%")
|
||||
}
|
||||
}
|
||||
return query.Count(&total).Limit(size).Offset((page - 1) * size).Find(v).Error
|
||||
})
|
||||
|
||||
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
return list, total, nil
|
||||
}
|
||||
|
||||
func (m *customUserModel) InsertLoginLog(ctx context.Context, log *LoginLog) error {
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
return conn.Create(log).Error
|
||||
})
|
||||
}
|
||||
|
||||
func (m *customUserModel) FilterLoginLogList(ctx context.Context, page, size int, filter *LoginLogFilterParams) ([]*LoginLog, int64, error) {
|
||||
var list []*LoginLog
|
||||
var total int64
|
||||
err := m.QueryNoCacheCtx(ctx, &list, func(conn *gorm.DB, v interface{}) error {
|
||||
query := conn.Model(&LoginLog{})
|
||||
if filter != nil {
|
||||
if filter.UserId != 0 {
|
||||
query = query.Where("user_id = ?", filter.UserId)
|
||||
}
|
||||
if filter.IP != "" {
|
||||
query = query.Where("ip LIKE ?", "%"+filter.IP+"%")
|
||||
}
|
||||
if filter.UserAgent != "" {
|
||||
query = query.Where("user_agent LIKE ?", "%"+filter.UserAgent+"%")
|
||||
}
|
||||
if filter.Success != nil {
|
||||
query = query.Where("success = ?", *filter.Success)
|
||||
}
|
||||
}
|
||||
return query.Count(&total).Limit(size).Offset((page - 1) * size).Find(v).Error
|
||||
})
|
||||
|
||||
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
return list, total, nil
|
||||
}
|
||||
+97
-171
@@ -2,15 +2,12 @@ package user
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/perfect-panel/server/internal/config"
|
||||
"github.com/perfect-panel/server/internal/model/server"
|
||||
"github.com/perfect-panel/server/internal/model/order"
|
||||
"github.com/perfect-panel/server/internal/model/subscribe"
|
||||
"github.com/perfect-panel/server/pkg/logger"
|
||||
"github.com/perfect-panel/server/pkg/tool"
|
||||
|
||||
"github.com/redis/go-redis/v9"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
@@ -39,6 +36,7 @@ type SubscribeDetails struct {
|
||||
Token string `gorm:"index:idx_token;unique;type:varchar(255);default:'';comment:Token"`
|
||||
UUID string `gorm:"type:varchar(255);unique;index:idx_uuid;default:'';comment:UUID"`
|
||||
Status uint8 `gorm:"type:tinyint(1);default:0;comment:Subscription Status: 0: Pending 1: Active 2: Finished 3: Expired; 4: Cancelled"`
|
||||
Note string `gorm:"type:varchar(500);default:'';comment:User note for subscription"`
|
||||
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
||||
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
||||
}
|
||||
@@ -63,6 +61,7 @@ type UserFilterParams struct {
|
||||
UserId *int64
|
||||
SubscribeId *int64
|
||||
UserSubscribeId *int64
|
||||
Order string // Order by id, e.g., "desc"
|
||||
}
|
||||
|
||||
type customUserLogicModel interface {
|
||||
@@ -79,7 +78,6 @@ type customUserLogicModel interface {
|
||||
QueryUserSubscribe(ctx context.Context, userId int64, status ...int64) ([]*SubscribeDetails, error)
|
||||
FindOneSubscribeDetailsById(ctx context.Context, id int64) (*SubscribeDetails, error)
|
||||
FindOneUserSubscribe(ctx context.Context, id int64) (*SubscribeDetails, error)
|
||||
InsertBalanceLog(ctx context.Context, data *BalanceLog, tx ...*gorm.DB) error
|
||||
FindUsersSubscribeBySubscribeId(ctx context.Context, subscribeId int64) ([]*Subscribe, error)
|
||||
UpdateUserSubscribeWithTraffic(ctx context.Context, id, download, upload int64, tx ...*gorm.DB) error
|
||||
QueryResisterUserTotalByDate(ctx context.Context, date time.Time) (int64, error)
|
||||
@@ -88,7 +86,6 @@ type customUserLogicModel interface {
|
||||
QueryAdminUsers(ctx context.Context) ([]*User, error)
|
||||
UpdateUserCache(ctx context.Context, data *User) error
|
||||
UpdateUserSubscribeCache(ctx context.Context, data *Subscribe) error
|
||||
InsertCommissionLog(ctx context.Context, data *CommissionLog, tx ...*gorm.DB) error
|
||||
QueryActiveSubscriptions(ctx context.Context, subscribeId ...int64) (map[int64]int64, error)
|
||||
FindUserAuthMethods(ctx context.Context, userId int64) ([]*AuthMethods, error)
|
||||
InsertUserAuthMethods(ctx context.Context, data *AuthMethods, tx ...*gorm.DB) error
|
||||
@@ -99,23 +96,25 @@ type customUserLogicModel interface {
|
||||
FindUserAuthMethodByPlatform(ctx context.Context, userId int64, platform string) (*AuthMethods, error)
|
||||
FindOneByEmail(ctx context.Context, email string) (*User, error)
|
||||
FindOneDevice(ctx context.Context, id int64) (*Device, error)
|
||||
QueryDeviceList(ctx context.Context, userid int64) ([]*Device, int64, error)
|
||||
QueryDevicePageList(ctx context.Context, userid, subscribeId int64, page, size int) ([]*Device, int64, error)
|
||||
UpdateDevice(ctx context.Context, data *Device, tx ...*gorm.DB) error
|
||||
FindOneDeviceByIdentifier(ctx context.Context, id string) (*Device, error)
|
||||
DeleteDevice(ctx context.Context, id int64, tx ...*gorm.DB) error
|
||||
|
||||
InsertSubscribeLog(ctx context.Context, log *SubscribeLog) error
|
||||
FilterSubscribeLogList(ctx context.Context, page, size int, filter *SubscribeLogFilterParams) ([]*SubscribeLog, int64, error)
|
||||
InsertLoginLog(ctx context.Context, log *LoginLog) error
|
||||
FilterLoginLogList(ctx context.Context, page, size int, filter *LoginLogFilterParams) ([]*LoginLog, int64, error)
|
||||
InsertDevice(ctx context.Context, data *Device, tx ...*gorm.DB) error
|
||||
|
||||
ClearSubscribeCache(ctx context.Context, data ...*Subscribe) error
|
||||
ClearUserCache(ctx context.Context, data ...*User) error
|
||||
|
||||
InsertResetSubscribeLog(ctx context.Context, log *ResetSubscribeLog, tx ...*gorm.DB) error
|
||||
UpdateResetSubscribeLog(ctx context.Context, log *ResetSubscribeLog, tx ...*gorm.DB) error
|
||||
FindResetSubscribeLog(ctx context.Context, id int64) (*ResetSubscribeLog, error)
|
||||
DeleteResetSubscribeLog(ctx context.Context, id int64, tx ...*gorm.DB) error
|
||||
FilterResetSubscribeLogList(ctx context.Context, filter *FilterResetSubscribeLogParams) ([]*ResetSubscribeLog, int64, error)
|
||||
QueryDailyUserStatisticsList(ctx context.Context, date time.Time) ([]UserStatisticsWithDate, error)
|
||||
QueryMonthlyUserStatisticsList(ctx context.Context, date time.Time) ([]UserStatisticsWithDate, error)
|
||||
}
|
||||
|
||||
type UserStatisticsWithDate struct {
|
||||
Date string
|
||||
Register int64
|
||||
NewOrderUsers int64
|
||||
RenewalOrderUsers int64
|
||||
}
|
||||
|
||||
// NewModel returns a model for the database table.
|
||||
@@ -125,56 +124,6 @@ func NewModel(conn *gorm.DB, c *redis.Client) Model {
|
||||
}
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) getSubscribeCacheKey(data *Subscribe) []string {
|
||||
if data == nil {
|
||||
return []string{}
|
||||
}
|
||||
var keys []string
|
||||
if data.Token != "" {
|
||||
keys = append(keys, fmt.Sprintf("%s%s", cacheUserSubscribeTokenPrefix, data.Token))
|
||||
}
|
||||
if data.UserId != 0 {
|
||||
keys = append(keys, fmt.Sprintf("%s%d", cacheUserSubscribeUserPrefix, data.UserId))
|
||||
}
|
||||
if data.Id != 0 {
|
||||
keys = append(keys, fmt.Sprintf("%s%d", cacheUserSubscribeIdPrefix, data.Id))
|
||||
}
|
||||
|
||||
if data.SubscribeId != 0 {
|
||||
var sub *subscribe.Subscribe
|
||||
err := m.QueryNoCacheCtx(context.Background(), &sub, func(conn *gorm.DB, v interface{}) error {
|
||||
return conn.Model(&subscribe.Subscribe{}).Where("id = ?", data.SubscribeId).First(&sub).Error
|
||||
})
|
||||
if err != nil {
|
||||
logger.Error("getUserSubscribeCacheKey", logger.Field("error", err.Error()), logger.Field("subscribeId", data.SubscribeId))
|
||||
return keys
|
||||
}
|
||||
if sub.Server != "" {
|
||||
ids := tool.StringToInt64Slice(sub.Server)
|
||||
for _, id := range ids {
|
||||
keys = append(keys, fmt.Sprintf("%s%d", config.ServerUserListCacheKey, id))
|
||||
}
|
||||
}
|
||||
if sub.ServerGroup != "" {
|
||||
ids := tool.StringToInt64Slice(sub.ServerGroup)
|
||||
var servers []*server.Server
|
||||
err = m.QueryNoCacheCtx(context.Background(), &servers, func(conn *gorm.DB, v interface{}) error {
|
||||
return conn.Model(&server.Server{}).Where("group_id in ?", ids).Find(v).Error
|
||||
})
|
||||
if err != nil {
|
||||
logger.Error("getUserSubscribeCacheKey", logger.Field("error", err.Error()), logger.Field("subscribeId", data.SubscribeId))
|
||||
return keys
|
||||
}
|
||||
for _, s := range servers {
|
||||
keys = append(keys, fmt.Sprintf("%s%d", config.ServerUserListCacheKey, s.Id))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return keys
|
||||
|
||||
}
|
||||
|
||||
// QueryPageList returns a list of records that meet the conditions.
|
||||
func (m *customUserModel) QueryPageList(ctx context.Context, page, size int, filter *UserFilterParams) ([]*User, int64, error) {
|
||||
var list []*User
|
||||
@@ -196,6 +145,9 @@ func (m *customUserModel) QueryPageList(ctx context.Context, page, size int, fil
|
||||
conn = conn.Joins("LEFT JOIN user_subscribe ON user.id = user_subscribe.user_id").
|
||||
Where("user_subscribe.subscribe_id =? and `status` IN (0,1)", *filter.SubscribeId)
|
||||
}
|
||||
if filter.Order != "" {
|
||||
conn = conn.Order(fmt.Sprintf("user.id %s", filter.Order))
|
||||
}
|
||||
}
|
||||
return conn.Model(&User{}).Group("user.id").Count(&total).Limit(size).Offset((page - 1) * size).Preload("UserDevices").Preload("AuthMethods").Find(&list).Error
|
||||
})
|
||||
@@ -219,33 +171,20 @@ func (m *customUserModel) BatchDeleteUser(ctx context.Context, ids []int64, tx .
|
||||
}, m.batchGetCacheKeys(users...)...)
|
||||
}
|
||||
|
||||
// InsertBalanceLog insert BalanceLog into the database.
|
||||
func (m *customUserModel) InsertBalanceLog(ctx context.Context, data *BalanceLog, tx ...*gorm.DB) error {
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Create(data).Error
|
||||
})
|
||||
}
|
||||
|
||||
// FindUserBalanceLogList returns a list of records that meet the conditions.
|
||||
func (m *customUserModel) FindUserBalanceLogList(ctx context.Context, userId int64, page, size int) ([]*BalanceLog, int64, error) {
|
||||
var list []*BalanceLog
|
||||
var total int64
|
||||
err := m.QueryNoCacheCtx(ctx, &list, func(conn *gorm.DB, v interface{}) error {
|
||||
|
||||
return conn.Model(&BalanceLog{}).Where("`user_id` = ?", userId).Count(&total).Limit(size).Offset((page - 1) * size).Find(&list).Error
|
||||
})
|
||||
return list, total, err
|
||||
}
|
||||
|
||||
func (m *customUserModel) UpdateUserSubscribeWithTraffic(ctx context.Context, id, download, upload int64, tx ...*gorm.DB) error {
|
||||
sub, err := m.FindOneSubscribe(ctx, id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return m.ExecCtx(ctx, func(conn *gorm.DB) error {
|
||||
|
||||
// 使用 defer 确保更新后清理缓存
|
||||
defer func() {
|
||||
if clearErr := m.ClearSubscribeCacheByModels(ctx, sub); clearErr != nil {
|
||||
// 记录清理缓存错误
|
||||
}
|
||||
}()
|
||||
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
@@ -253,7 +192,7 @@ func (m *customUserModel) UpdateUserSubscribeWithTraffic(ctx context.Context, id
|
||||
"download": gorm.Expr("download + ?", download),
|
||||
"upload": gorm.Expr("upload + ?", upload),
|
||||
}).Error
|
||||
}, m.getSubscribeCacheKey(sub)...)
|
||||
})
|
||||
}
|
||||
|
||||
func (m *customUserModel) QueryResisterUserTotalByDate(ctx context.Context, date time.Time) (int64, error) {
|
||||
@@ -293,16 +232,7 @@ func (m *customUserModel) QueryAdminUsers(ctx context.Context) ([]*User, error)
|
||||
}
|
||||
|
||||
func (m *customUserModel) UpdateUserCache(ctx context.Context, data *User) error {
|
||||
return m.CachedConn.DelCacheCtx(ctx, m.getCacheKeys(data)...)
|
||||
}
|
||||
|
||||
func (m *customUserModel) InsertCommissionLog(ctx context.Context, data *CommissionLog, tx ...*gorm.DB) error {
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Model(&CommissionLog{}).Create(data).Error
|
||||
})
|
||||
return m.ClearUserCache(ctx, data)
|
||||
}
|
||||
|
||||
func (m *customUserModel) FindOneByReferCode(ctx context.Context, referCode string) (*User, error) {
|
||||
@@ -321,81 +251,77 @@ func (m *customUserModel) FindOneSubscribeDetailsById(ctx context.Context, id in
|
||||
return &data, err
|
||||
}
|
||||
|
||||
func (m *customUserModel) InsertResetSubscribeLog(ctx context.Context, log *ResetSubscribeLog, tx ...*gorm.DB) error {
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Model(&ResetSubscribeLog{}).Create(log).Error
|
||||
})
|
||||
}
|
||||
// QueryDailyUserStatisticsList Query daily user statistics list for the current month (from 1st to current date)
|
||||
func (m *customUserModel) QueryDailyUserStatisticsList(ctx context.Context, date time.Time) ([]UserStatisticsWithDate, error) {
|
||||
var results []UserStatisticsWithDate
|
||||
|
||||
func (m *customUserModel) UpdateResetSubscribeLog(ctx context.Context, log *ResetSubscribeLog, tx ...*gorm.DB) error {
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Model(&ResetSubscribeLog{}).Where("id = ?", log.Id).Updates(log).Error
|
||||
})
|
||||
}
|
||||
err := m.QueryNoCacheCtx(ctx, &results, func(conn *gorm.DB, v interface{}) error {
|
||||
firstDay := time.Date(date.Year(), date.Month(), 1, 0, 0, 0, 0, date.Location())
|
||||
|
||||
func (m *customUserModel) FindResetSubscribeLog(ctx context.Context, id int64) (*ResetSubscribeLog, error) {
|
||||
var data ResetSubscribeLog
|
||||
err := m.QueryNoCacheCtx(ctx, &data, func(conn *gorm.DB, v interface{}) error {
|
||||
return conn.Model(&ResetSubscribeLog{}).Where("id = ?", id).First(&data).Error
|
||||
})
|
||||
return &data, err
|
||||
}
|
||||
// 子查询:统计每天的新用户订单数量
|
||||
newOrderSub := conn.Model(&order.Order{}).
|
||||
Select("DATE_FORMAT(created_at, '%Y-%m-%d') AS date, COUNT(DISTINCT user_id) AS new_order_users").
|
||||
Where("is_new = 1 AND created_at BETWEEN ? AND ? AND status IN ?", firstDay, date, []int64{2, 5}).
|
||||
Group("DATE_FORMAT(created_at, '%Y-%m-%d')")
|
||||
|
||||
func (m *customUserModel) DeleteResetSubscribeLog(ctx context.Context, id int64, tx ...*gorm.DB) error {
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Model(&ResetSubscribeLog{}).Where("id = ?", id).Delete(&ResetSubscribeLog{}).Error
|
||||
})
|
||||
}
|
||||
// 子查询:统计每天的续费订单数量
|
||||
renewalOrderSub := conn.Model(&order.Order{}).
|
||||
Select("DATE_FORMAT(created_at, '%Y-%m-%d') AS date, COUNT(DISTINCT user_id) AS renewal_order_users").
|
||||
Where("is_new = 0 AND created_at BETWEEN ? AND ? AND status IN ?", firstDay, date, []int64{2, 5}).
|
||||
Group("DATE_FORMAT(created_at, '%Y-%m-%d')")
|
||||
|
||||
func (m *customUserModel) FilterResetSubscribeLogList(ctx context.Context, filter *FilterResetSubscribeLogParams) ([]*ResetSubscribeLog, int64, error) {
|
||||
if filter == nil {
|
||||
return nil, 0, errors.New("filter params is nil")
|
||||
}
|
||||
|
||||
var list []*ResetSubscribeLog
|
||||
var total int64
|
||||
|
||||
err := m.QueryNoCacheCtx(ctx, &list, func(conn *gorm.DB, v interface{}) error {
|
||||
query := conn.Model(&ResetSubscribeLog{})
|
||||
|
||||
// 应用筛选条件
|
||||
if filter.UserId != 0 {
|
||||
query = query.Where("user_id = ?", filter.UserId)
|
||||
}
|
||||
if filter.UserSubscribeId != 0 {
|
||||
query = query.Where("user_subscribe_id = ?", filter.UserSubscribeId)
|
||||
}
|
||||
if filter.Type != 0 {
|
||||
query = query.Where("type = ?", filter.Type)
|
||||
}
|
||||
if filter.OrderNo != "" {
|
||||
query = query.Where("order_no = ?", filter.OrderNo)
|
||||
}
|
||||
|
||||
// 计算总数
|
||||
if err := query.Count(&total).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 应用分页
|
||||
if filter.Page > 0 && filter.Size > 0 {
|
||||
query = query.Offset((filter.Page - 1) * filter.Size)
|
||||
}
|
||||
if filter.Size > 0 {
|
||||
query = query.Limit(filter.Size)
|
||||
}
|
||||
|
||||
return query.Find(&list).Error
|
||||
return conn.Model(&User{}).
|
||||
Select(`
|
||||
DATE_FORMAT(user.created_at, '%Y-%m-%d') AS date,
|
||||
COUNT(*) AS register,
|
||||
IFNULL(MAX(n.new_order_users), 0) AS new_order_users,
|
||||
IFNULL(MAX(r.renewal_order_users), 0) AS renewal_order_users
|
||||
`).
|
||||
Joins("LEFT JOIN (?) AS n ON DATE_FORMAT(user.created_at, '%Y-%m-%d') = n.date", newOrderSub).
|
||||
Joins("LEFT JOIN (?) AS r ON DATE_FORMAT(user.created_at, '%Y-%m-%d') = r.date", renewalOrderSub).
|
||||
Where("user.created_at BETWEEN ? AND ?", firstDay, date).
|
||||
Group("DATE_FORMAT(user.created_at, '%Y-%m-%d')").
|
||||
Order("date ASC").
|
||||
Scan(v).Error
|
||||
})
|
||||
|
||||
return list, total, err
|
||||
return results, err
|
||||
}
|
||||
|
||||
// QueryMonthlyUserStatisticsList Query monthly user statistics list for the past 6 months
|
||||
func (m *customUserModel) QueryMonthlyUserStatisticsList(ctx context.Context, date time.Time) ([]UserStatisticsWithDate, error) {
|
||||
var results []UserStatisticsWithDate
|
||||
|
||||
err := m.QueryNoCacheCtx(ctx, &results, func(conn *gorm.DB, v interface{}) error {
|
||||
// 获取 6 个月前的日期
|
||||
sixMonthsAgo := date.AddDate(0, -5, 0)
|
||||
|
||||
// 子查询:每月新订单用户数量
|
||||
newOrderSub := conn.Model(&order.Order{}).
|
||||
Select("DATE_FORMAT(created_at, '%Y-%m') AS date, COUNT(DISTINCT user_id) AS new_order_users").
|
||||
Where("is_new = 1 AND created_at >= ? AND status IN ?", sixMonthsAgo, []int64{2, 5}).
|
||||
Group("DATE_FORMAT(created_at, '%Y-%m')")
|
||||
|
||||
// 子查询:每月续费订单用户数量
|
||||
renewalOrderSub := conn.Model(&order.Order{}).
|
||||
Select("DATE_FORMAT(created_at, '%Y-%m') AS date, COUNT(DISTINCT user_id) AS renewal_order_users").
|
||||
Where("is_new = 0 AND created_at >= ? AND status IN ?", sixMonthsAgo, []int64{2, 5}).
|
||||
Group("DATE_FORMAT(created_at, '%Y-%m')")
|
||||
|
||||
return conn.Model(&User{}).
|
||||
Select(`
|
||||
DATE_FORMAT(user.created_at, '%Y-%m') AS date,
|
||||
COUNT(*) AS register,
|
||||
IFNULL(MAX(n.new_order_users), 0) AS new_order_users,
|
||||
IFNULL(MAX(r.renewal_order_users), 0) AS renewal_order_users
|
||||
`).
|
||||
Joins("LEFT JOIN (?) AS n ON DATE_FORMAT(user.created_at, '%Y-%m') = n.date", newOrderSub).
|
||||
Joins("LEFT JOIN (?) AS r ON DATE_FORMAT(user.created_at, '%Y-%m') = r.date", renewalOrderSub).
|
||||
Where("user.created_at >= ?", sixMonthsAgo).
|
||||
Group("DATE_FORMAT(user.created_at, '%Y-%m')").
|
||||
Order("date ASC").
|
||||
Scan(v).Error
|
||||
})
|
||||
|
||||
return results, err
|
||||
}
|
||||
|
||||
@@ -9,7 +9,7 @@ import (
|
||||
)
|
||||
|
||||
func (m *defaultUserModel) UpdateUserSubscribeCache(ctx context.Context, data *Subscribe) error {
|
||||
return m.CachedConn.DelCacheCtx(ctx, m.getSubscribeCacheKey(data)...)
|
||||
return m.ClearSubscribeCacheByModels(ctx, data)
|
||||
}
|
||||
|
||||
// QueryActiveSubscriptions returns the number of active subscriptions.
|
||||
@@ -21,7 +21,7 @@ func (m *defaultUserModel) QueryActiveSubscriptions(ctx context.Context, subscri
|
||||
var result []SubscriptionCount
|
||||
err := m.QueryNoCacheCtx(ctx, &result, func(conn *gorm.DB, v interface{}) error {
|
||||
return conn.Model(&Subscribe{}).
|
||||
Where("subscribe_id IN ? AND `status` IN ?", subscribeId, []int64{1, 0, 3}).
|
||||
Where("subscribe_id IN ? AND `status` IN ?", subscribeId, []int64{1, 0}).
|
||||
Select("subscribe_id, COUNT(id) as total").
|
||||
Group("subscribe_id").
|
||||
Scan(&result).
|
||||
@@ -60,7 +60,13 @@ func (m *defaultUserModel) FindOneSubscribe(ctx context.Context, id int64) (*Sub
|
||||
func (m *defaultUserModel) FindUsersSubscribeBySubscribeId(ctx context.Context, subscribeId int64) ([]*Subscribe, error) {
|
||||
var data []*Subscribe
|
||||
err := m.QueryNoCacheCtx(ctx, &data, func(conn *gorm.DB, v interface{}) error {
|
||||
return conn.Model(&Subscribe{}).Where("subscribe_id = ? AND `status` IN ?", subscribeId, []int64{1, 0}).Find(&data).Error
|
||||
err := conn.Model(&Subscribe{}).Where("subscribe_id = ? AND `status` IN ?", subscribeId, []int64{1, 0}).Find(v).Error
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
// update user subscribe status
|
||||
return conn.Model(&Subscribe{}).Where("subscribe_id = ? AND `status` = ?", subscribeId, 0).Update("status", 1).Error
|
||||
})
|
||||
return data, err
|
||||
}
|
||||
@@ -114,12 +120,20 @@ func (m *defaultUserModel) UpdateSubscribe(ctx context.Context, data *Subscribe,
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return m.ExecCtx(ctx, func(conn *gorm.DB) error {
|
||||
|
||||
// 使用 defer 确保更新后清理缓存
|
||||
defer func() {
|
||||
if clearErr := m.ClearSubscribeCacheByModels(ctx, old, data); clearErr != nil {
|
||||
// 记录清理缓存错误
|
||||
}
|
||||
}()
|
||||
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Model(&Subscribe{}).Where("id = ?", data.Id).Save(data).Error
|
||||
}, m.getSubscribeCacheKey(old)...)
|
||||
})
|
||||
}
|
||||
|
||||
// DeleteSubscribe deletes a record.
|
||||
@@ -128,22 +142,37 @@ func (m *defaultUserModel) DeleteSubscribe(ctx context.Context, token string, tx
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return m.ExecCtx(ctx, func(conn *gorm.DB) error {
|
||||
|
||||
// 使用 defer 确保删除后清理缓存
|
||||
defer func() {
|
||||
if clearErr := m.ClearSubscribeCacheByModels(ctx, data); clearErr != nil {
|
||||
// 记录清理缓存错误
|
||||
}
|
||||
}()
|
||||
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Where("token = ?", token).Delete(&Subscribe{}).Error
|
||||
}, m.getSubscribeCacheKey(data)...)
|
||||
})
|
||||
}
|
||||
|
||||
// InsertSubscribe insert Subscribe into the database.
|
||||
func (m *defaultUserModel) InsertSubscribe(ctx context.Context, data *Subscribe, tx ...*gorm.DB) error {
|
||||
return m.ExecCtx(ctx, func(conn *gorm.DB) error {
|
||||
// 使用 defer 确保插入后清理相关缓存
|
||||
defer func() {
|
||||
if clearErr := m.ClearSubscribeCacheByModels(ctx, data); clearErr != nil {
|
||||
// 记录清理缓存错误
|
||||
}
|
||||
}()
|
||||
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Create(data).Error
|
||||
}, m.getSubscribeCacheKey(data)...)
|
||||
})
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) DeleteSubscribeById(ctx context.Context, id int64, tx ...*gorm.DB) error {
|
||||
@@ -151,18 +180,22 @@ func (m *defaultUserModel) DeleteSubscribeById(ctx context.Context, id int64, tx
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return m.ExecCtx(ctx, func(conn *gorm.DB) error {
|
||||
|
||||
// 使用 defer 确保删除后清理缓存
|
||||
defer func() {
|
||||
if clearErr := m.ClearSubscribeCacheByModels(ctx, data); clearErr != nil {
|
||||
// 记录清理缓存错误
|
||||
}
|
||||
}()
|
||||
|
||||
return m.ExecNoCacheCtx(ctx, func(conn *gorm.DB) error {
|
||||
if len(tx) > 0 {
|
||||
conn = tx[0]
|
||||
}
|
||||
return conn.Where("id = ?", id).Delete(&Subscribe{}).Error
|
||||
}, m.getSubscribeCacheKey(data)...)
|
||||
})
|
||||
}
|
||||
|
||||
func (m *defaultUserModel) ClearSubscribeCache(ctx context.Context, data ...*Subscribe) error {
|
||||
var keys []string
|
||||
for _, item := range data {
|
||||
keys = append(keys, m.getSubscribeCacheKey(item)...)
|
||||
}
|
||||
return m.CachedConn.DelCacheCtx(ctx, keys...)
|
||||
return m.ClearSubscribeCacheByModels(ctx, data...)
|
||||
}
|
||||
|
||||
+19
-130
@@ -2,19 +2,20 @@ package user
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/plugin/soft_delete"
|
||||
)
|
||||
|
||||
type User struct {
|
||||
Id int64 `gorm:"primaryKey"`
|
||||
Password string `gorm:"type:varchar(100);not null;comment:User Password"`
|
||||
Algo string `gorm:"type:varchar(20);default:'default';comment:Encryption Algorithm"`
|
||||
Salt string `gorm:"type:varchar(20);default:null;comment:Password Salt"`
|
||||
Avatar string `gorm:"type:MEDIUMTEXT;comment:User Avatar"`
|
||||
Balance int64 `gorm:"default:0;comment:User Balance"` // User Balance Amount
|
||||
ReferCode string `gorm:"type:varchar(20);default:'';comment:Referral Code"`
|
||||
RefererId int64 `gorm:"index:idx_referer;comment:Referrer ID"`
|
||||
Commission int64 `gorm:"default:0;comment:Commission"` // Commission Amount
|
||||
Commission int64 `gorm:"default:0;comment:Commission"` // Commission Amount
|
||||
ReferralPercentage uint8 `gorm:"default:0;comment:Referral"` // Referral Percentage
|
||||
OnlyFirstPurchase *bool `gorm:"default:true;not null;comment:Only First Purchase"` // Only First Purchase Referral
|
||||
GiftAmount int64 `gorm:"default:0;comment:User Gift Amount"`
|
||||
Enable *bool `gorm:"default:true;not null;comment:Is Account Enabled"`
|
||||
IsAdmin *bool `gorm:"default:false;not null;comment:Is Admin"`
|
||||
@@ -24,43 +25,12 @@ type User struct {
|
||||
EnableTradeNotify *bool `gorm:"default:false;not null;comment:Enable Trade Notifications"`
|
||||
AuthMethods []AuthMethods `gorm:"foreignKey:UserId;references:Id"`
|
||||
UserDevices []Device `gorm:"foreignKey:UserId;references:Id"`
|
||||
Rules string `gorm:"type:TEXT;comment:User Rules"`
|
||||
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
||||
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
||||
}
|
||||
|
||||
func (User) TableName() string {
|
||||
return "user"
|
||||
}
|
||||
|
||||
type OldUser struct {
|
||||
Id int64 `gorm:"primaryKey"`
|
||||
Email string `gorm:"index:idx_email;type:varchar(100);comment:Email"`
|
||||
//Telephone string `gorm:"index:idx_telephone;type:varchar(20);default:'';comment:Telephone"`
|
||||
//TelephoneAreaCode string `gorm:"index:idx_telephone;type:varchar(20);default:'';comment:TelephoneAreaCode"`
|
||||
Password string `gorm:"type:varchar(100);not null;comment:User Password"`
|
||||
Avatar string `gorm:"type:varchar(200);default:'';comment:User Avatar"`
|
||||
Balance int64 `gorm:"default:0;comment:User Balance"` // User Balance Amount
|
||||
Telegram int64 `gorm:"default:null;comment:Telegram Account"`
|
||||
ReferCode string `gorm:"type:varchar(20);default:'';comment:Referral Code"`
|
||||
RefererId int64 `gorm:"index:idx_referer;comment:Referrer ID"`
|
||||
Commission int64 `gorm:"default:0;comment:Commission"` // Commission Amount
|
||||
GiftAmount int64 `gorm:"default:0;comment:User Gift Amount"`
|
||||
Enable *bool `gorm:"default:true;not null;comment:Is Account Enabled"`
|
||||
IsAdmin *bool `gorm:"default:false;not null;comment:Is Admin"`
|
||||
ValidEmail *bool `gorm:"default:false;not null;comment:Is Email Verified"`
|
||||
EnableEmailNotify *bool `gorm:"default:false;not null;comment:Enable Email Notifications"`
|
||||
EnableTelegramNotify *bool `gorm:"default:false;not null;comment:Enable Telegram Notifications"`
|
||||
EnableBalanceNotify *bool `gorm:"default:false;not null;comment:Enable Balance Change Notifications"`
|
||||
EnableLoginNotify *bool `gorm:"default:false;not null;comment:Enable Login Notifications"`
|
||||
EnableSubscribeNotify *bool `gorm:"default:false;not null;comment:Enable Subscription Notifications"`
|
||||
EnableTradeNotify *bool `gorm:"default:false;not null;comment:Enable Trade Notifications"`
|
||||
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
||||
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
||||
DeletedAt gorm.DeletedAt `gorm:"default:null;comment:Deletion Time"`
|
||||
IsDel soft_delete.DeletedAt `gorm:"softDelete:flag,DeletedAtField:DeletedAt;comment:1: Normal 0: Deleted"` // Using `1` and `0` to indicate
|
||||
}
|
||||
|
||||
func (OldUser) TableName() string {
|
||||
func (*User) TableName() string {
|
||||
return "user"
|
||||
}
|
||||
|
||||
@@ -79,56 +49,15 @@ type Subscribe struct {
|
||||
Token string `gorm:"index:idx_token;unique;type:varchar(255);default:'';comment:Token"`
|
||||
UUID string `gorm:"type:varchar(255);unique;index:idx_uuid;default:'';comment:UUID"`
|
||||
Status uint8 `gorm:"type:tinyint(1);default:0;comment:Subscription Status: 0: Pending 1: Active 2: Finished 3: Expired 4: Deducted"`
|
||||
Note string `gorm:"type:varchar(500);default:'';comment:User note for subscription"`
|
||||
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
||||
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
||||
}
|
||||
|
||||
func (Subscribe) TableName() string {
|
||||
func (*Subscribe) TableName() string {
|
||||
return "user_subscribe"
|
||||
}
|
||||
|
||||
type BalanceLog struct {
|
||||
Id int64 `gorm:"primaryKey"`
|
||||
UserId int64 `gorm:"index:idx_user_id;not null;comment:User ID"`
|
||||
Amount int64 `gorm:"not null;comment:Amount"`
|
||||
Type uint8 `gorm:"type:tinyint(1);not null;comment:Type: 1: Recharge 2: Withdraw 3: Payment 4: Refund 5: Reward"`
|
||||
OrderId int64 `gorm:"default:null;comment:Order ID"`
|
||||
Balance int64 `gorm:"not null;comment:Balance"`
|
||||
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
||||
}
|
||||
|
||||
func (BalanceLog) TableName() string {
|
||||
return "user_balance_log"
|
||||
}
|
||||
|
||||
type GiftAmountLog struct {
|
||||
Id int64 `gorm:"primaryKey"`
|
||||
UserId int64 `gorm:"index:idx_user_id;not null;comment:User ID"`
|
||||
UserSubscribeId int64 `gorm:"default:null;comment:Deduction User Subscribe ID"`
|
||||
OrderNo string `gorm:"default:null;comment:Order No."`
|
||||
Type uint8 `gorm:"type:tinyint(1);not null;comment:Type: 1: Increase 2: Reduce"`
|
||||
Amount int64 `gorm:"not null;comment:Amount"`
|
||||
Balance int64 `gorm:"not null;comment:Balance"`
|
||||
Remark string `gorm:"type:varchar(255);default:'';comment:Remark"`
|
||||
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
||||
}
|
||||
|
||||
func (GiftAmountLog) TableName() string {
|
||||
return "user_gift_amount_log"
|
||||
}
|
||||
|
||||
type CommissionLog struct {
|
||||
Id int64 `gorm:"primaryKey"`
|
||||
UserId int64 `gorm:"index:idx_user_id;not null;comment:User ID"`
|
||||
OrderNo string `gorm:"default:null;comment:Order No."`
|
||||
Amount int64 `gorm:"not null;comment:Amount"`
|
||||
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
||||
}
|
||||
|
||||
func (CommissionLog) TableName() string {
|
||||
return "user_commission_log"
|
||||
}
|
||||
|
||||
type AuthMethods struct {
|
||||
Id int64 `gorm:"primaryKey"`
|
||||
UserId int64 `gorm:"index:idx_user_id;not null;comment:User ID"`
|
||||
@@ -139,7 +68,7 @@ type AuthMethods struct {
|
||||
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
||||
}
|
||||
|
||||
func (AuthMethods) TableName() string {
|
||||
func (*AuthMethods) TableName() string {
|
||||
return "user_auth_methods"
|
||||
}
|
||||
|
||||
@@ -155,7 +84,7 @@ type Device struct {
|
||||
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
||||
}
|
||||
|
||||
func (Device) TableName() string {
|
||||
func (*Device) TableName() string {
|
||||
return "user_device"
|
||||
}
|
||||
|
||||
@@ -174,57 +103,17 @@ func (DeviceOnlineRecord) TableName() string {
|
||||
return "user_device_online_record"
|
||||
}
|
||||
|
||||
type LoginLog struct {
|
||||
type Withdrawal struct {
|
||||
Id int64 `gorm:"primaryKey"`
|
||||
UserId int64 `gorm:"index:idx_user_id;not null;comment:User ID"`
|
||||
LoginIP string `gorm:"type:varchar(255);not null;comment:Login IP"`
|
||||
UserAgent string `gorm:"type:text;not null;comment:UserAgent"`
|
||||
Success *bool `gorm:"default:false;not null;comment:Login Success"`
|
||||
Amount int64 `gorm:"not null;comment:Withdrawal Amount"`
|
||||
Content string `gorm:"type:text;comment:Withdrawal Content"`
|
||||
Status uint8 `gorm:"type:tinyint(1);default:0;comment:Withdrawal Status: 0: Pending 1: Approved 2: Rejected"`
|
||||
Reason string `gorm:"type:varchar(500);default:'';comment:Rejection Reason"`
|
||||
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
||||
UpdatedAt time.Time `gorm:"comment:Update Time"`
|
||||
}
|
||||
|
||||
func (LoginLog) TableName() string {
|
||||
return "user_login_log"
|
||||
}
|
||||
|
||||
type SubscribeLog struct {
|
||||
Id int64 `gorm:"primaryKey"`
|
||||
UserId int64 `gorm:"index:idx_user_id;not null;comment:User ID"`
|
||||
UserSubscribeId int64 `gorm:"index:idx_user_subscribe_id;not null;comment:User Subscribe ID"`
|
||||
Token string `gorm:"type:varchar(255);not null;comment:Token"`
|
||||
IP string `gorm:"type:varchar(255);not null;comment:IP"`
|
||||
UserAgent string `gorm:"type:text;not null;comment:UserAgent"`
|
||||
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
||||
}
|
||||
|
||||
func (SubscribeLog) TableName() string {
|
||||
return "user_subscribe_log"
|
||||
}
|
||||
|
||||
const (
|
||||
ResetSubscribeTypeAuto uint8 = 1
|
||||
ResetSubscribeTypeAdvance uint8 = 2
|
||||
ResetSubscribeTypePaid uint8 = 3
|
||||
)
|
||||
|
||||
type FilterResetSubscribeLogParams struct {
|
||||
Page int
|
||||
Size int
|
||||
Type uint8
|
||||
UserId int64
|
||||
OrderNo string
|
||||
UserSubscribeId int64
|
||||
}
|
||||
|
||||
type ResetSubscribeLog struct {
|
||||
Id int64 `gorm:"primaryKey"`
|
||||
UserId int64 `gorm:"type:bigint;index:idx_user_id;not null;comment:User ID"`
|
||||
Type uint8 `gorm:"type:tinyint(1);not null;comment:Type: 1: Auto 2: Advance 3: Paid"`
|
||||
OrderNo string `gorm:"type:varchar(255);default:null;comment:Order No."`
|
||||
UserSubscribeId int64 `gorm:"type:bigint;index:idx_user_subscribe_id;not null;comment:User Subscribe ID"`
|
||||
CreatedAt time.Time `gorm:"<-:create;comment:Creation Time"`
|
||||
}
|
||||
|
||||
func (ResetSubscribeLog) TableName() string {
|
||||
return "user_reset_subscribe_log"
|
||||
func (*Withdrawal) TableName() string {
|
||||
return "user_withdrawal"
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user