Mirrors/stash
1
0
Fork 0
mirror of https://github.com/stashapp/stash.git synced 2025-12-17 20:34:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

[Files Refactor] Set primary flag when cleaning (#2880)

Browse Source 
* Ensure single primary per object
* Set primary file during clean
* Only show migration notes for actual migrations
This commit is contained in:
WithoutPants
2022-09-05 11:46:18 +10:00
parent 7159ab69a3
commit 6b0bcdea88
12 changed files with 117 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
pkg/sqlite/table.go
View File

@@ -522,6 +522,28 @@ func (t *relatedFilesTable) replaceJoins(ctx context.Context, id int, fileIDs []
return t.insertJoins(ctx, id, firstPrimary, fileIDs)
}
func (t *relatedFilesTable) setPrimary(ctx context.Context, id int, fileID file.ID) error {
table := t.table.table
q := dialect.Update(table).Prepared(true).Set(goqu.Record{
"primary": 0,
}).Where(t.idColumn.Eq(id), table.Col(fileIDColumn).Neq(fileID))
if _, err := exec(ctx, q); err != nil {
return fmt.Errorf("unsetting primary flags in %s: %w", t.table.table.GetTable(), err)
}
q = dialect.Update(table).Prepared(true).Set(goqu.Record{
"primary": 1,
}).Where(t.idColumn.Eq(id), table.Col(fileIDColumn).Eq(fileID))
if _, err := exec(ctx, q); err != nil {
return fmt.Errorf("setting primary flag in %s: %w", t.table.table.GetTable(), err)
}
return nil
}
type sqler interface {
ToSQL() (sql string, params []interface{}, err error)
}