S-VPN/ario_server
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(quota): update time handling in quota logic and correct subscriber ID query

Browse Source
This commit is contained in:
Chang lue Tsen
2025-09-13 14:02:49 -04:00
parent 9b0e1b1e0f
commit cf4792cdc8
3 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
queue/logic/task/quotaLogic.go
+2 -2
View File
@@ -68,7 +68,7 @@ func (l *QuotaTaskLogic) ProcessTask(ctx context.Context, t *asynq.Task) error {
return err
}
subscribes, err := l.getSubscribes(ctx, scope.Subscribers)
subscribes, err := l.getSubscribes(ctx, scope.Objects)
if err != nil {
return err
}
@@ -156,7 +156,7 @@ func (l *QuotaTaskLogic) parseTaskData(ctx context.Context, taskInfo *task.Task)
func (l *QuotaTaskLogic) getSubscribes(ctx context.Context, subscriberIDs []int64) ([]*user.Subscribe, error) {
var subscribes []*user.Subscribe
if err := l.svcCtx.DB.WithContext(ctx).Model(&user.Subscribe{}).Where("subscribe_id IN ?", subscriberIDs).Find(&subscribes).Error; err != nil {
if err := l.svcCtx.DB.WithContext(ctx).Model(&user.Subscribe{}).Where("id IN ?", subscriberIDs).Find(&subscribes).Error; err != nil {
logger.WithContext(ctx).Error("[QuotaTaskLogic.getSubscribes] find subscribes error",
logger.Field("error", err.Error()),
logger.Field("subscribers", subscriberIDs),