shanshanzhong
2442831cd7
All checks were successful
Build docker and publish / build (20.15.1) (push) Successful in 7m28s
添加FindLatestDeviceOnlineRecord接口用于查询设备最新在线记录 实现GetOnlineDeviceLoginTime方法获取设备登录时间 优化设备列表查询按最后活动时间排序 移除未使用的依赖项
114 lines
3.3 KiB
Go
114 lines
3.3 KiB
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func (m *customUserModel) FindOneDevice(ctx context.Context, id int64) (*Device, error) {
|
|
deviceIdKey := fmt.Sprintf("%s%v", cacheUserDeviceIdPrefix, id)
|
|
var resp Device
|
|
err := m.QueryCtx(ctx, &resp, deviceIdKey, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&Device{}).Where("`id` = ?", id).First(&resp).Error
|
|
})
|
|
switch {
|
|
case err == nil:
|
|
return &resp, nil
|
|
default:
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
func (m *customUserModel) FindOneDeviceByIdentifier(ctx context.Context, id string) (*Device, error) {
|
|
var resp Device
|
|
err := m.QueryNoCacheCtx(ctx, &resp, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&Device{}).Where("`identifier` = ?", id).First(&resp).Error
|
|
})
|
|
switch {
|
|
case err == nil:
|
|
return &resp, nil
|
|
default:
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
// QueryDevicePageList returns a list of records that meet the conditions.
|
|
func (m *customUserModel) QueryDevicePageList(ctx context.Context, userId, subscribeId int64, page, size int) ([]*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` = ? and `subscribe_id` = ?", userId, subscribeId).Count(&total).Order("created_at DESC").Limit(size).Offset((page - 1) * size).Find(&list).Error
|
|
})
|
|
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).Order("created_at DESC").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
|
|
}
|
|
err = m.ExecCtx(ctx, func(conn *gorm.DB) error {
|
|
if len(tx) > 0 {
|
|
conn = tx[0]
|
|
}
|
|
return conn.Save(data).Error
|
|
}, old.GetCacheKeys()...)
|
|
return err
|
|
}
|
|
|
|
func (m *customUserModel) DeleteDevice(ctx context.Context, id int64, tx ...*gorm.DB) error {
|
|
data, err := m.FindOneDevice(ctx, id)
|
|
if err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
err = m.ExecCtx(ctx, func(conn *gorm.DB) error {
|
|
if len(tx) > 0 {
|
|
conn = tx[0]
|
|
}
|
|
return conn.Delete(&Device{}, id).Error
|
|
}, 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
|
|
})
|
|
}
|
|
|
|
func (m *customUserModel) FindLatestDeviceOnlineRecord(ctx context.Context, userId int64, identifier string) (*DeviceOnlineRecord, error) {
|
|
var rec DeviceOnlineRecord
|
|
err := m.QueryNoCacheCtx(ctx, &rec, func(conn *gorm.DB, v interface{}) error {
|
|
return conn.Model(&DeviceOnlineRecord{}).Where("user_id = ? AND identifier = ?", userId, identifier).Order("online_time DESC").First(&rec).Error
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &rec, nil
|
|
}
|