lethdat
a0ae2b681a
- Added videoWorkflowRepository with methods to manage video and user interactions. - Introduced catalog_mapper for converting database models to protobuf representations. - Created domain_helpers for normalizing domain and ad format values. - Defined service interfaces for payment, account, notification, domain, ad template, player config, video, and user management. - Implemented OAuth helpers for generating state and caching keys. - Developed payment_proto_helpers for mapping payment-related models to protobuf. - Added service policy helpers to enforce plan requirements and user permissions. - Created user_mapper for converting user payloads to protobuf format. - Implemented value_helpers for handling various value conversions and nil checks. - Developed video_helpers for normalizing video statuses and managing storage types. - Created video_mapper for mapping video models to protobuf format. - Implemented render workflow for managing video creation and job processing.
68 lines
2.0 KiB
Go
68 lines
2.0 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"gorm.io/gorm"
|
|
"stream.api/internal/database/model"
|
|
)
|
|
|
|
type videoWorkflowRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewVideoWorkflowRepository(db *gorm.DB) *videoWorkflowRepository {
|
|
return &videoWorkflowRepository{db: db}
|
|
}
|
|
|
|
func (r *videoWorkflowRepository) GetUserByID(ctx context.Context, userID string) (*model.User, error) {
|
|
var user model.User
|
|
if err := r.db.WithContext(ctx).Where("id = ?", strings.TrimSpace(userID)).First(&user).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func (r *videoWorkflowRepository) CreateVideoWithStorageAndAd(ctx context.Context, video *model.Video, userID string, adTemplateID *string) error {
|
|
return r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
|
if err := tx.Create(video).Error; err != nil {
|
|
return err
|
|
}
|
|
if err := tx.Model(&model.User{}).
|
|
Where("id = ?", strings.TrimSpace(userID)).
|
|
UpdateColumn("storage_used", gorm.Expr("storage_used + ?", video.Size)).Error; err != nil {
|
|
return err
|
|
}
|
|
if video == nil || adTemplateID == nil {
|
|
return nil
|
|
}
|
|
|
|
trimmed := strings.TrimSpace(*adTemplateID)
|
|
if trimmed == "" {
|
|
if err := tx.WithContext(ctx).Model(&model.Video{}).Where("id = ?", video.ID).Update("ad_id", nil).Error; err != nil {
|
|
return err
|
|
}
|
|
video.AdID = nil
|
|
return nil
|
|
}
|
|
|
|
var template model.AdTemplate
|
|
if err := tx.WithContext(ctx).Select("id").Where("id = ? AND user_id = ?", trimmed, strings.TrimSpace(userID)).First(&template).Error; err != nil {
|
|
return err
|
|
}
|
|
if err := tx.WithContext(ctx).Model(&model.Video{}).Where("id = ?", video.ID).Update("ad_id", template.ID).Error; err != nil {
|
|
return err
|
|
}
|
|
video.AdID = &template.ID
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func (r *videoWorkflowRepository) MarkVideoJobFailed(ctx context.Context, videoID string) error {
|
|
return r.db.WithContext(ctx).
|
|
Model(&model.Video{}).
|
|
Where("id = ?", strings.TrimSpace(videoID)).
|
|
Updates(map[string]any{"status": "failed", "processing_status": "FAILED"}).Error
|
|
}
|