perf(platform): 短剧创作工作台 - 拆分长篇项目接口并优化章节与素材分页
This commit is contained in:
ddCat committed
1 parent
40fd0c54a4
commit
d89f170a08
22 files changed
+1497
-222
No files matched your search
@@ -0,0 +1,218 @@
|
||||
package repository
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"infinite-canvas/backend/internal/model"
|
||||
)
|
||||
|
||||
type ProjectOverviewMetrics struct {
|
||||
UnitCount int64 `gorm:"column:unit_count"`
|
||||
CompletedUnitCount int64 `gorm:"column:completed_unit_count"`
|
||||
TotalWordCount int64 `gorm:"column:total_word_count"`
|
||||
UnitsWithoutText int64 `gorm:"column:units_without_text"`
|
||||
UnitsWithoutShots int64 `gorm:"column:units_without_shots"`
|
||||
CanvasCount int64 `gorm:"column:canvas_count"`
|
||||
AssetCount int64 `gorm:"column:asset_count"`
|
||||
ShotCount int64 `gorm:"column:shot_count"`
|
||||
PendingCandidateCount int64 `gorm:"column:pending_candidate_count"`
|
||||
ReadyStoryboardCount int64 `gorm:"column:ready_storyboard_count"`
|
||||
ReadyPrevizCount int64 `gorm:"column:ready_previz_count"`
|
||||
ReadyVideoCount int64 `gorm:"column:ready_video_count"`
|
||||
StaleArtifactCount int64 `gorm:"column:stale_artifact_count"`
|
||||
}
|
||||
|
||||
type ProjectOverviewUnitRow struct {
|
||||
model.ProjectUnit
|
||||
ShotCount int64 `gorm:"column:shot_count"`
|
||||
CandidateCount int64 `gorm:"column:candidate_count"`
|
||||
CanvasCount int64 `gorm:"column:canvas_count"`
|
||||
}
|
||||
|
||||
type ProjectAssetCountRow struct {
|
||||
Key string `gorm:"column:key"`
|
||||
Count int64 `gorm:"column:count"`
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectUnitCanvasCounts(projectID string) (map[string]int64, error) {
|
||||
var rows []ProjectAssetCountRow
|
||||
if err := r.db.Table("canvas_unit_links").Select("unit_id AS key, COUNT(DISTINCT canvas_id) AS count").Where("project_id = ?", projectID).Group("unit_id").Scan(&rows).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
counts := make(map[string]int64, len(rows))
|
||||
for _, row := range rows {
|
||||
counts[row.Key] = row.Count
|
||||
}
|
||||
return counts, nil
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectOverviewMetrics(projectID string) (ProjectOverviewMetrics, error) {
|
||||
var metrics ProjectOverviewMetrics
|
||||
err := r.db.Raw(`
|
||||
SELECT
|
||||
(SELECT COUNT(*) FROM project_units WHERE project_id = ?) AS unit_count,
|
||||
(SELECT COUNT(*) FROM project_units WHERE project_id = ? AND status = ?) AS completed_unit_count,
|
||||
(SELECT COALESCE(SUM(word_count), 0) FROM project_units WHERE project_id = ?) AS total_word_count,
|
||||
(SELECT COUNT(*) FROM project_units WHERE project_id = ? AND word_count = 0) AS units_without_text,
|
||||
(SELECT COUNT(*) FROM project_units pu WHERE pu.project_id = ? AND pu.status <> ? AND NOT EXISTS (SELECT 1 FROM shots s WHERE s.project_id = pu.project_id AND s.unit_id = pu.id)) AS units_without_shots,
|
||||
(SELECT COUNT(*) FROM canvas_projects WHERE project_id = ?) AS canvas_count,
|
||||
(SELECT COUNT(*) FROM project_asset_links WHERE project_id = ?) AS asset_count,
|
||||
(SELECT COUNT(*) FROM shots WHERE project_id = ?) AS shot_count,
|
||||
(SELECT COUNT(*) FROM project_asset_candidates WHERE project_id = ? AND status = 'pending_confirmation') AS pending_candidate_count,
|
||||
(SELECT COUNT(DISTINCT shot_id) FROM shot_artifacts WHERE project_id = ? AND type = 'storyboard' AND selected = ? AND status = 'ready') AS ready_storyboard_count,
|
||||
(SELECT COUNT(DISTINCT shot_id) FROM shot_artifacts WHERE project_id = ? AND type = 'action_board' AND selected = ? AND status = 'ready') AS ready_previz_count,
|
||||
(SELECT COUNT(DISTINCT shot_id) FROM shot_artifacts WHERE project_id = ? AND type = 'video' AND selected = ? AND status = 'ready') AS ready_video_count,
|
||||
(SELECT COUNT(*) FROM shot_artifacts WHERE project_id = ? AND status = 'stale') AS stale_artifact_count
|
||||
`, projectID, projectID, model.ProjectUnitStatusCompleted, projectID, projectID, projectID, model.ProjectUnitStatusDraft, projectID, projectID, projectID, projectID, projectID, true, projectID, true, projectID, true, projectID).Scan(&metrics).Error
|
||||
return metrics, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectOverviewUnits(projectID string, limit int) ([]ProjectOverviewUnitRow, error) {
|
||||
if limit <= 0 || limit > 20 {
|
||||
limit = 8
|
||||
}
|
||||
var rows []ProjectOverviewUnitRow
|
||||
err := r.db.Raw(`
|
||||
SELECT pu.id, pu.project_id, pu.parent_id, pu.kind, pu.title, pu.word_count, pu.status, pu.position, pu.created_at, pu.updated_at,
|
||||
(SELECT COUNT(*) FROM shots s WHERE s.project_id = pu.project_id AND s.unit_id = pu.id) AS shot_count,
|
||||
(SELECT COUNT(*) FROM project_asset_candidates pac WHERE pac.project_id = pu.project_id AND pac.unit_id = pu.id) AS candidate_count,
|
||||
(SELECT COUNT(DISTINCT cul.canvas_id) FROM canvas_unit_links cul WHERE cul.project_id = pu.project_id AND cul.unit_id = pu.id) AS canvas_count
|
||||
FROM project_units pu
|
||||
WHERE pu.project_id = ?
|
||||
ORDER BY pu.position ASC, pu.created_at ASC
|
||||
LIMIT ?
|
||||
`, projectID, limit).Scan(&rows).Error
|
||||
return rows, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectUnitShots(projectID string, unitID string) ([]model.Shot, error) {
|
||||
var shots []model.Shot
|
||||
err := r.db.Where("project_id = ? AND unit_id = ?", projectID, unitID).Order("position asc, created_at asc").Find(&shots).Error
|
||||
return shots, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectUnitShotRevisions(projectID string, unitID string) ([]model.ShotRevision, error) {
|
||||
var revisions []model.ShotRevision
|
||||
err := r.db.Table("shot_revisions").Select("shot_revisions.*").Joins("JOIN shots ON shots.id = shot_revisions.shot_id").Where("shots.project_id = ? AND shots.unit_id = ?", projectID, unitID).Order("shots.position asc, shot_revisions.version asc").Scan(&revisions).Error
|
||||
return revisions, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectUnitShotArtifacts(projectID string, unitID string) ([]model.ShotArtifact, error) {
|
||||
var artifacts []model.ShotArtifact
|
||||
err := r.db.Where("project_id = ? AND unit_id = ?", projectID, unitID).Order("shot_id asc, type asc, version asc").Find(&artifacts).Error
|
||||
return artifacts, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectUnitShotAssetReferences(projectID string, unitID string) ([]model.ShotAssetReference, error) {
|
||||
var references []model.ShotAssetReference
|
||||
err := r.db.Table("shot_asset_references").Select("shot_asset_references.*").Joins("JOIN shots ON shots.id = shot_asset_references.shot_id").Where("shots.project_id = ? AND shots.unit_id = ?", projectID, unitID).Order("shot_asset_references.created_at asc").Scan(&references).Error
|
||||
return references, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectUnitAssetCandidates(projectID string, unitID string) ([]model.ProjectAssetCandidate, error) {
|
||||
var candidates []model.ProjectAssetCandidate
|
||||
err := r.db.Where("project_id = ? AND (unit_id = ? OR unit_id = '')", projectID, unitID).Order("created_at asc").Find(&candidates).Error
|
||||
return candidates, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectUnitAssets(userID string, projectID string, unitID string) ([]model.Asset, error) {
|
||||
var assets []model.Asset
|
||||
err := r.db.Table("assets").Distinct("assets.*").
|
||||
Joins("JOIN project_asset_links pal ON pal.asset_id = assets.id AND pal.project_id = ?", projectID).
|
||||
Joins("JOIN asset_versions av ON av.asset_id = assets.id").
|
||||
Joins("JOIN shot_asset_references sar ON sar.asset_version_id = av.id").
|
||||
Joins("JOIN shots s ON s.id = sar.shot_id AND s.project_id = ? AND s.unit_id = ?", projectID, unitID).
|
||||
Where("assets.user_id = ?", userID).
|
||||
Order("assets.updated_at desc").
|
||||
Scan(&assets).Error
|
||||
return assets, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectWorkflowInstancesForUnit(projectID string, unitID string) ([]model.WorkflowInstance, error) {
|
||||
var instances []model.WorkflowInstance
|
||||
err := r.db.Where("project_id = ? AND unit_id = ?", projectID, unitID).Order("created_at asc").Find(&instances).Error
|
||||
return instances, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectCanvasSummariesPage(userID string, projectID string, page int, pageSize int) ([]model.CanvasProject, int64, error) {
|
||||
var canvases []model.CanvasProject
|
||||
var total int64
|
||||
query := r.db.Model(&model.CanvasProject{}).Where("user_id = ? AND project_id = ?", userID, projectID)
|
||||
if err := query.Count(&total).Error; err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
err := query.Select("id", "user_id", "project_id", "title", "created_at", "updated_at").Order("updated_at desc").Offset((page - 1) * pageSize).Limit(pageSize).Find(&canvases).Error
|
||||
return canvases, total, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectCanvasUnitLinksForCanvases(projectID string, canvasIDs []string) ([]model.CanvasUnitLink, error) {
|
||||
if len(canvasIDs) == 0 {
|
||||
return []model.CanvasUnitLink{}, nil
|
||||
}
|
||||
var links []model.CanvasUnitLink
|
||||
err := r.db.Where("project_id = ? AND canvas_id IN ?", projectID, canvasIDs).Order("created_at asc").Find(&links).Error
|
||||
return links, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectAssetCandidatesPage(projectID string, page int, pageSize int, unitID string, status string, category string) ([]model.ProjectAssetCandidate, int64, error) {
|
||||
var candidates []model.ProjectAssetCandidate
|
||||
var total int64
|
||||
query := r.db.Model(&model.ProjectAssetCandidate{}).Where("project_id = ?", projectID)
|
||||
if value := strings.TrimSpace(unitID); value != "" {
|
||||
query = query.Where("unit_id = ?", value)
|
||||
}
|
||||
if value := strings.TrimSpace(status); value != "" {
|
||||
query = query.Where("status = ?", value)
|
||||
}
|
||||
if value := strings.TrimSpace(category); value != "" {
|
||||
query = query.Where("category = ?", value)
|
||||
}
|
||||
if err := query.Count(&total).Error; err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
err := query.Order("created_at desc").Offset((page - 1) * pageSize).Limit(pageSize).Find(&candidates).Error
|
||||
return candidates, total, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectAssetsPage(userID string, projectID string, page int, pageSize int, category string, mediaType string, status string, folderID *string, queryText string) ([]model.Asset, int64, error) {
|
||||
var assets []model.Asset
|
||||
var total int64
|
||||
query := r.db.Table("assets").Joins("JOIN project_asset_links ON project_asset_links.asset_id = assets.id").Where("assets.user_id = ? AND project_asset_links.project_id = ?", userID, projectID)
|
||||
if value := strings.TrimSpace(category); value != "" {
|
||||
query = query.Where("assets.category = ?", value)
|
||||
}
|
||||
if value := strings.TrimSpace(mediaType); value != "" {
|
||||
query = query.Where("assets.kind = ?", value)
|
||||
}
|
||||
if value := strings.TrimSpace(status); value != "" {
|
||||
query = query.Where("assets.status = ?", value)
|
||||
}
|
||||
if folderID != nil {
|
||||
query = query.Where("project_asset_links.folder_id = ?", strings.TrimSpace(*folderID))
|
||||
}
|
||||
if value := strings.TrimSpace(queryText); value != "" {
|
||||
query = query.Where("LOWER(assets.title) LIKE ?", "%"+strings.ToLower(value)+"%")
|
||||
}
|
||||
if err := query.Count(&total).Error; err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
err := query.Select("assets.*").Order("assets.updated_at desc").Offset((page - 1) * pageSize).Limit(pageSize).Scan(&assets).Error
|
||||
return assets, total, err
|
||||
}
|
||||
|
||||
func (r *Repository) ProjectAssetFacets(projectID string) ([]ProjectAssetCountRow, []ProjectAssetCountRow, error) {
|
||||
var categoryRows []ProjectAssetCountRow
|
||||
if err := r.db.Table("assets").Select("assets.category AS key, COUNT(*) AS count").
|
||||
Joins("JOIN project_asset_links pal ON pal.asset_id = assets.id").
|
||||
Where("pal.project_id = ?", projectID).
|
||||
Group("assets.category").Scan(&categoryRows).Error; err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
var folderRows []ProjectAssetCountRow
|
||||
if err := r.db.Table("project_asset_links").Select("folder_id AS key, COUNT(*) AS count").
|
||||
Where("project_id = ?", projectID).
|
||||
Group("folder_id").Scan(&folderRows).Error; err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
return categoryRows, folderRows, nil
|
||||
}
|
||||
Reference in new issue
Block a user