claude
e7fdd0e1ab
- Implemented player_configs table to store multiple player configurations per user. - Migrated existing player settings from user_preferences to player_configs. - Removed player-related columns from user_preferences. - Added referral state fields to user for tracking referral rewards. - Created migration scripts for database changes and data migration. - Added test cases for app services and usage helpers. - Introduced video job service interfaces and implementations.
30 lines
744 B
Go
30 lines
744 B
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gorm.io/gorm"
|
|
"stream.api/internal/database/model"
|
|
"stream.api/pkg/logger"
|
|
)
|
|
|
|
type usagePayload struct {
|
|
UserID string `json:"user_id"`
|
|
TotalVideos int64 `json:"total_videos"`
|
|
TotalStorage int64 `json:"total_storage"`
|
|
}
|
|
|
|
func loadUsage(ctx context.Context, db *gorm.DB, l logger.Logger, user *model.User) (*usagePayload, error) {
|
|
var totalVideos int64
|
|
if err := db.WithContext(ctx).Model(&model.Video{}).Where("user_id = ?", user.ID).Count(&totalVideos).Error; err != nil {
|
|
l.Error("Failed to count user videos", "error", err, "user_id", user.ID)
|
|
return nil, err
|
|
}
|
|
|
|
return &usagePayload{
|
|
UserID: user.ID,
|
|
TotalVideos: totalVideos,
|
|
TotalStorage: user.StorageUsed,
|
|
}, nil
|
|
}
|