mirror of
https://github.com/stashapp/stash.git
synced 2025-12-17 20:34:37 +03:00
Data layer restructuring (#997)
* Move query builders to sqlite package * Add transaction system * Wrap model resolvers in transaction * Add error return value for StringSliceToIntSlice * Update/refactor mutation resolvers * Convert query builders * Remove unused join types * Add stash id unit tests * Use WAL journal mode
This commit is contained in:
WithoutPants
702
pkg/sqlite/scene.go
Normal file
702
pkg/sqlite/scene.go
Normal file
@@ -0,0 +1,702 @@
|
||||
package sqlite
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"github.com/jmoiron/sqlx"
|
||||
"github.com/stashapp/stash/pkg/models"
|
||||
)
|
||||
|
||||
const sceneTable = "scenes"
|
||||
const sceneIDColumn = "scene_id"
|
||||
const performersScenesTable = "performers_scenes"
|
||||
const scenesTagsTable = "scenes_tags"
|
||||
const moviesScenesTable = "movies_scenes"
|
||||
|
||||
var scenesForPerformerQuery = selectAll(sceneTable) + `
|
||||
LEFT JOIN performers_scenes as performers_join on performers_join.scene_id = scenes.id
|
||||
WHERE performers_join.performer_id = ?
|
||||
GROUP BY scenes.id
|
||||
`
|
||||
|
||||
var countScenesForPerformerQuery = `
|
||||
SELECT performer_id FROM performers_scenes as performers_join
|
||||
WHERE performer_id = ?
|
||||
GROUP BY scene_id
|
||||
`
|
||||
|
||||
var scenesForStudioQuery = selectAll(sceneTable) + `
|
||||
JOIN studios ON studios.id = scenes.studio_id
|
||||
WHERE studios.id = ?
|
||||
GROUP BY scenes.id
|
||||
`
|
||||
var scenesForMovieQuery = selectAll(sceneTable) + `
|
||||
LEFT JOIN movies_scenes as movies_join on movies_join.scene_id = scenes.id
|
||||
WHERE movies_join.movie_id = ?
|
||||
GROUP BY scenes.id
|
||||
`
|
||||
|
||||
var countScenesForTagQuery = `
|
||||
SELECT tag_id AS id FROM scenes_tags
|
||||
WHERE scenes_tags.tag_id = ?
|
||||
GROUP BY scenes_tags.scene_id
|
||||
`
|
||||
|
||||
var countScenesForMissingChecksumQuery = `
|
||||
SELECT id FROM scenes
|
||||
WHERE scenes.checksum is null
|
||||
`
|
||||
|
||||
var countScenesForMissingOSHashQuery = `
|
||||
SELECT id FROM scenes
|
||||
WHERE scenes.oshash is null
|
||||
`
|
||||
|
||||
type sceneQueryBuilder struct {
|
||||
repository
|
||||
}
|
||||
|
||||
func NewSceneReaderWriter(tx dbi) *sceneQueryBuilder {
|
||||
return &sceneQueryBuilder{
|
||||
repository{
|
||||
tx: tx,
|
||||
tableName: sceneTable,
|
||||
idColumn: idColumn,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) Create(newObject models.Scene) (*models.Scene, error) {
|
||||
var ret models.Scene
|
||||
if err := qb.insertObject(newObject, &ret); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &ret, nil
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) Update(updatedObject models.ScenePartial) (*models.Scene, error) {
|
||||
const partial = true
|
||||
if err := qb.update(updatedObject.ID, updatedObject, partial); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return qb.find(updatedObject.ID)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) UpdateFull(updatedObject models.Scene) (*models.Scene, error) {
|
||||
const partial = false
|
||||
if err := qb.update(updatedObject.ID, updatedObject, partial); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return qb.find(updatedObject.ID)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) UpdateFileModTime(id int, modTime models.NullSQLiteTimestamp) error {
|
||||
return qb.updateMap(id, map[string]interface{}{
|
||||
"file_mod_time": modTime,
|
||||
})
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) IncrementOCounter(id int) (int, error) {
|
||||
_, err := qb.tx.Exec(
|
||||
`UPDATE scenes SET o_counter = o_counter + 1 WHERE scenes.id = ?`,
|
||||
id,
|
||||
)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
scene, err := qb.find(id)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return scene.OCounter, nil
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) DecrementOCounter(id int) (int, error) {
|
||||
_, err := qb.tx.Exec(
|
||||
`UPDATE scenes SET o_counter = o_counter - 1 WHERE scenes.id = ? and scenes.o_counter > 0`,
|
||||
id,
|
||||
)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
scene, err := qb.find(id)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return scene.OCounter, nil
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) ResetOCounter(id int) (int, error) {
|
||||
_, err := qb.tx.Exec(
|
||||
`UPDATE scenes SET o_counter = 0 WHERE scenes.id = ?`,
|
||||
id,
|
||||
)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
scene, err := qb.find(id)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return scene.OCounter, nil
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) Destroy(id int) error {
|
||||
// delete all related table rows
|
||||
// TODO - this should be handled by a delete cascade
|
||||
if err := qb.performersRepository().destroy([]int{id}); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// scene markers should be handled prior to calling destroy
|
||||
// galleries should be handled prior to calling destroy
|
||||
|
||||
return qb.destroyExisting([]int{id})
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) Find(id int) (*models.Scene, error) {
|
||||
return qb.find(id)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) FindMany(ids []int) ([]*models.Scene, error) {
|
||||
var scenes []*models.Scene
|
||||
for _, id := range ids {
|
||||
scene, err := qb.Find(id)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if scene == nil {
|
||||
return nil, fmt.Errorf("scene with id %d not found", id)
|
||||
}
|
||||
|
||||
scenes = append(scenes, scene)
|
||||
}
|
||||
|
||||
return scenes, nil
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) find(id int) (*models.Scene, error) {
|
||||
var ret models.Scene
|
||||
if err := qb.get(id, &ret); err != nil {
|
||||
if err == sql.ErrNoRows {
|
||||
return nil, nil
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return &ret, nil
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) FindByChecksum(checksum string) (*models.Scene, error) {
|
||||
query := "SELECT * FROM scenes WHERE checksum = ? LIMIT 1"
|
||||
args := []interface{}{checksum}
|
||||
return qb.queryScene(query, args)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) FindByOSHash(oshash string) (*models.Scene, error) {
|
||||
query := "SELECT * FROM scenes WHERE oshash = ? LIMIT 1"
|
||||
args := []interface{}{oshash}
|
||||
return qb.queryScene(query, args)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) FindByPath(path string) (*models.Scene, error) {
|
||||
query := selectAll(sceneTable) + "WHERE path = ? LIMIT 1"
|
||||
args := []interface{}{path}
|
||||
return qb.queryScene(query, args)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) FindByPerformerID(performerID int) ([]*models.Scene, error) {
|
||||
args := []interface{}{performerID}
|
||||
return qb.queryScenes(scenesForPerformerQuery, args)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) CountByPerformerID(performerID int) (int, error) {
|
||||
args := []interface{}{performerID}
|
||||
return qb.runCountQuery(qb.buildCountQuery(countScenesForPerformerQuery), args)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) FindByMovieID(movieID int) ([]*models.Scene, error) {
|
||||
args := []interface{}{movieID}
|
||||
return qb.queryScenes(scenesForMovieQuery, args)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) CountByMovieID(movieID int) (int, error) {
|
||||
args := []interface{}{movieID}
|
||||
return qb.runCountQuery(qb.buildCountQuery(scenesForMovieQuery), args)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) Count() (int, error) {
|
||||
return qb.runCountQuery(qb.buildCountQuery("SELECT scenes.id FROM scenes"), nil)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) Size() (float64, error) {
|
||||
return qb.runSumQuery("SELECT SUM(cast(size as double)) as sum FROM scenes", nil)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) CountByStudioID(studioID int) (int, error) {
|
||||
args := []interface{}{studioID}
|
||||
return qb.runCountQuery(qb.buildCountQuery(scenesForStudioQuery), args)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) CountByTagID(tagID int) (int, error) {
|
||||
args := []interface{}{tagID}
|
||||
return qb.runCountQuery(qb.buildCountQuery(countScenesForTagQuery), args)
|
||||
}
|
||||
|
||||
// CountMissingChecksum returns the number of scenes missing a checksum value.
|
||||
func (qb *sceneQueryBuilder) CountMissingChecksum() (int, error) {
|
||||
return qb.runCountQuery(qb.buildCountQuery(countScenesForMissingChecksumQuery), []interface{}{})
|
||||
}
|
||||
|
||||
// CountMissingOSHash returns the number of scenes missing an oshash value.
|
||||
func (qb *sceneQueryBuilder) CountMissingOSHash() (int, error) {
|
||||
return qb.runCountQuery(qb.buildCountQuery(countScenesForMissingOSHashQuery), []interface{}{})
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) Wall(q *string) ([]*models.Scene, error) {
|
||||
s := ""
|
||||
if q != nil {
|
||||
s = *q
|
||||
}
|
||||
query := selectAll(sceneTable) + "WHERE scenes.details LIKE '%" + s + "%' ORDER BY RANDOM() LIMIT 80"
|
||||
return qb.queryScenes(query, nil)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) All() ([]*models.Scene, error) {
|
||||
return qb.queryScenes(selectAll(sceneTable)+qb.getSceneSort(nil), nil)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) Query(sceneFilter *models.SceneFilterType, findFilter *models.FindFilterType) ([]*models.Scene, int, error) {
|
||||
if sceneFilter == nil {
|
||||
sceneFilter = &models.SceneFilterType{}
|
||||
}
|
||||
if findFilter == nil {
|
||||
findFilter = &models.FindFilterType{}
|
||||
}
|
||||
|
||||
query := qb.newQuery()
|
||||
|
||||
query.body = selectDistinctIDs(sceneTable)
|
||||
query.body += `
|
||||
left join scene_markers on scene_markers.scene_id = scenes.id
|
||||
left join performers_scenes as performers_join on performers_join.scene_id = scenes.id
|
||||
left join movies_scenes as movies_join on movies_join.scene_id = scenes.id
|
||||
left join studios as studio on studio.id = scenes.studio_id
|
||||
left join galleries as gallery on gallery.scene_id = scenes.id
|
||||
left join scenes_tags as tags_join on tags_join.scene_id = scenes.id
|
||||
left join scene_stash_ids on scene_stash_ids.scene_id = scenes.id
|
||||
`
|
||||
|
||||
if q := findFilter.Q; q != nil && *q != "" {
|
||||
searchColumns := []string{"scenes.title", "scenes.details", "scenes.path", "scenes.oshash", "scenes.checksum", "scene_markers.title"}
|
||||
clause, thisArgs := getSearchBinding(searchColumns, *q, false)
|
||||
query.addWhere(clause)
|
||||
query.addArg(thisArgs...)
|
||||
}
|
||||
|
||||
query.handleStringCriterionInput(sceneFilter.Path, "scenes.path")
|
||||
query.handleIntCriterionInput(sceneFilter.Rating, "scenes.rating")
|
||||
query.handleIntCriterionInput(sceneFilter.OCounter, "scenes.o_counter")
|
||||
|
||||
if Organized := sceneFilter.Organized; Organized != nil {
|
||||
var organized string
|
||||
if *Organized == true {
|
||||
organized = "1"
|
||||
} else {
|
||||
organized = "0"
|
||||
}
|
||||
query.addWhere("scenes.organized = " + organized)
|
||||
}
|
||||
|
||||
if durationFilter := sceneFilter.Duration; durationFilter != nil {
|
||||
clause, thisArgs := getDurationWhereClause(*durationFilter)
|
||||
query.addWhere(clause)
|
||||
query.addArg(thisArgs...)
|
||||
}
|
||||
|
||||
if resolutionFilter := sceneFilter.Resolution; resolutionFilter != nil {
|
||||
if resolution := resolutionFilter.String(); resolutionFilter.IsValid() {
|
||||
switch resolution {
|
||||
case "VERY_LOW":
|
||||
query.addWhere("MIN(scenes.height, scenes.width) < 240")
|
||||
case "LOW":
|
||||
query.addWhere("(MIN(scenes.height, scenes.width) >= 240 AND MIN(scenes.height, scenes.width) < 360)")
|
||||
case "R360P":
|
||||
query.addWhere("(MIN(scenes.height, scenes.width) >= 360 AND MIN(scenes.height, scenes.width) < 480)")
|
||||
case "STANDARD":
|
||||
query.addWhere("(MIN(scenes.height, scenes.width) >= 480 AND MIN(scenes.height, scenes.width) < 540)")
|
||||
case "WEB_HD":
|
||||
query.addWhere("(MIN(scenes.height, scenes.width) >= 540 AND MIN(scenes.height, scenes.width) < 720)")
|
||||
case "STANDARD_HD":
|
||||
query.addWhere("(MIN(scenes.height, scenes.width) >= 720 AND MIN(scenes.height, scenes.width) < 1080)")
|
||||
case "FULL_HD":
|
||||
query.addWhere("(MIN(scenes.height, scenes.width) >= 1080 AND MIN(scenes.height, scenes.width) < 1440)")
|
||||
case "QUAD_HD":
|
||||
query.addWhere("(MIN(scenes.height, scenes.width) >= 1440 AND MIN(scenes.height, scenes.width) < 1920)")
|
||||
case "VR_HD":
|
||||
query.addWhere("(MIN(scenes.height, scenes.width) >= 1920 AND MIN(scenes.height, scenes.width) < 2160)")
|
||||
case "FOUR_K":
|
||||
query.addWhere("(MIN(scenes.height, scenes.width) >= 2160 AND MIN(scenes.height, scenes.width) < 2880)")
|
||||
case "FIVE_K":
|
||||
query.addWhere("(MIN(scenes.height, scenes.width) >= 2880 AND MIN(scenes.height, scenes.width) < 3384)")
|
||||
case "SIX_K":
|
||||
query.addWhere("(MIN(scenes.height, scenes.width) >= 3384 AND MIN(scenes.height, scenes.width) < 4320)")
|
||||
case "EIGHT_K":
|
||||
query.addWhere("(MIN(scenes.height, scenes.width) >= 4320")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if hasMarkersFilter := sceneFilter.HasMarkers; hasMarkersFilter != nil {
|
||||
if strings.Compare(*hasMarkersFilter, "true") == 0 {
|
||||
query.addHaving("count(scene_markers.scene_id) > 0")
|
||||
} else {
|
||||
query.addWhere("scene_markers.id IS NULL")
|
||||
}
|
||||
}
|
||||
|
||||
if isMissingFilter := sceneFilter.IsMissing; isMissingFilter != nil && *isMissingFilter != "" {
|
||||
switch *isMissingFilter {
|
||||
case "gallery":
|
||||
query.addWhere("gallery.scene_id IS NULL")
|
||||
case "studio":
|
||||
query.addWhere("scenes.studio_id IS NULL")
|
||||
case "movie":
|
||||
query.addWhere("movies_join.scene_id IS NULL")
|
||||
case "performers":
|
||||
query.addWhere("performers_join.scene_id IS NULL")
|
||||
case "date":
|
||||
query.addWhere("scenes.date IS \"\" OR scenes.date IS \"0001-01-01\"")
|
||||
case "tags":
|
||||
query.addWhere("tags_join.scene_id IS NULL")
|
||||
case "stash_id":
|
||||
query.addWhere("scene_stash_ids.scene_id IS NULL")
|
||||
default:
|
||||
query.addWhere("(scenes." + *isMissingFilter + " IS NULL OR TRIM(scenes." + *isMissingFilter + ") = '')")
|
||||
}
|
||||
}
|
||||
|
||||
if tagsFilter := sceneFilter.Tags; tagsFilter != nil && len(tagsFilter.Value) > 0 {
|
||||
for _, tagID := range tagsFilter.Value {
|
||||
query.addArg(tagID)
|
||||
}
|
||||
|
||||
query.body += " LEFT JOIN tags on tags_join.tag_id = tags.id"
|
||||
whereClause, havingClause := getMultiCriterionClause("scenes", "tags", "scenes_tags", "scene_id", "tag_id", tagsFilter)
|
||||
query.addWhere(whereClause)
|
||||
query.addHaving(havingClause)
|
||||
}
|
||||
|
||||
if performersFilter := sceneFilter.Performers; performersFilter != nil && len(performersFilter.Value) > 0 {
|
||||
for _, performerID := range performersFilter.Value {
|
||||
query.addArg(performerID)
|
||||
}
|
||||
|
||||
query.body += " LEFT JOIN performers ON performers_join.performer_id = performers.id"
|
||||
whereClause, havingClause := getMultiCriterionClause("scenes", "performers", "performers_scenes", "scene_id", "performer_id", performersFilter)
|
||||
query.addWhere(whereClause)
|
||||
query.addHaving(havingClause)
|
||||
}
|
||||
|
||||
if studiosFilter := sceneFilter.Studios; studiosFilter != nil && len(studiosFilter.Value) > 0 {
|
||||
for _, studioID := range studiosFilter.Value {
|
||||
query.addArg(studioID)
|
||||
}
|
||||
|
||||
whereClause, havingClause := getMultiCriterionClause("scenes", "studio", "", "", "studio_id", studiosFilter)
|
||||
query.addWhere(whereClause)
|
||||
query.addHaving(havingClause)
|
||||
}
|
||||
|
||||
if moviesFilter := sceneFilter.Movies; moviesFilter != nil && len(moviesFilter.Value) > 0 {
|
||||
for _, movieID := range moviesFilter.Value {
|
||||
query.addArg(movieID)
|
||||
}
|
||||
|
||||
query.body += " LEFT JOIN movies ON movies_join.movie_id = movies.id"
|
||||
whereClause, havingClause := getMultiCriterionClause("scenes", "movies", "movies_scenes", "scene_id", "movie_id", moviesFilter)
|
||||
query.addWhere(whereClause)
|
||||
query.addHaving(havingClause)
|
||||
}
|
||||
|
||||
if stashIDFilter := sceneFilter.StashID; stashIDFilter != nil {
|
||||
query.addWhere("scene_stash_ids.stash_id = ?")
|
||||
query.addArg(stashIDFilter)
|
||||
}
|
||||
|
||||
query.sortAndPagination = qb.getSceneSort(findFilter) + getPagination(findFilter)
|
||||
idsResult, countResult, err := query.executeFind()
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
var scenes []*models.Scene
|
||||
for _, id := range idsResult {
|
||||
scene, err := qb.Find(id)
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
scenes = append(scenes, scene)
|
||||
}
|
||||
|
||||
return scenes, countResult, nil
|
||||
}
|
||||
|
||||
func appendClause(clauses []string, clause string) []string {
|
||||
if clause != "" {
|
||||
return append(clauses, clause)
|
||||
}
|
||||
|
||||
return clauses
|
||||
}
|
||||
|
||||
func getDurationWhereClause(durationFilter models.IntCriterionInput) (string, []interface{}) {
|
||||
// special case for duration. We accept duration as seconds as int but the
|
||||
// field is floating point. Change the equals filter to return a range
|
||||
// between x and x + 1
|
||||
// likewise, not equals needs to be duration < x OR duration >= x
|
||||
var clause string
|
||||
args := []interface{}{}
|
||||
|
||||
value := durationFilter.Value
|
||||
if durationFilter.Modifier == models.CriterionModifierEquals {
|
||||
clause = "scenes.duration >= ? AND scenes.duration < ?"
|
||||
args = append(args, value)
|
||||
args = append(args, value+1)
|
||||
} else if durationFilter.Modifier == models.CriterionModifierNotEquals {
|
||||
clause = "(scenes.duration < ? OR scenes.duration >= ?)"
|
||||
args = append(args, value)
|
||||
args = append(args, value+1)
|
||||
} else {
|
||||
var count int
|
||||
clause, count = getIntCriterionWhereClause("scenes.duration", durationFilter)
|
||||
if count == 1 {
|
||||
args = append(args, value)
|
||||
}
|
||||
}
|
||||
|
||||
return clause, args
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) QueryAllByPathRegex(regex string, ignoreOrganized bool) ([]*models.Scene, error) {
|
||||
var args []interface{}
|
||||
body := selectDistinctIDs("scenes") + " WHERE scenes.path regexp ?"
|
||||
|
||||
if ignoreOrganized {
|
||||
body += " AND scenes.organized = 0"
|
||||
}
|
||||
|
||||
args = append(args, "(?i)"+regex)
|
||||
|
||||
idsResult, err := qb.runIdsQuery(body, args)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var scenes []*models.Scene
|
||||
for _, id := range idsResult {
|
||||
scene, err := qb.Find(id)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
scenes = append(scenes, scene)
|
||||
}
|
||||
|
||||
return scenes, nil
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) QueryByPathRegex(findFilter *models.FindFilterType) ([]*models.Scene, int, error) {
|
||||
if findFilter == nil {
|
||||
findFilter = &models.FindFilterType{}
|
||||
}
|
||||
|
||||
var whereClauses []string
|
||||
var havingClauses []string
|
||||
var args []interface{}
|
||||
body := selectDistinctIDs("scenes")
|
||||
|
||||
if q := findFilter.Q; q != nil && *q != "" {
|
||||
whereClauses = append(whereClauses, "scenes.path regexp ?")
|
||||
args = append(args, "(?i)"+*q)
|
||||
}
|
||||
|
||||
sortAndPagination := qb.getSceneSort(findFilter) + getPagination(findFilter)
|
||||
idsResult, countResult, err := qb.executeFindQuery(body, args, sortAndPagination, whereClauses, havingClauses)
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
var scenes []*models.Scene
|
||||
for _, id := range idsResult {
|
||||
scene, err := qb.Find(id)
|
||||
if err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
scenes = append(scenes, scene)
|
||||
}
|
||||
|
||||
return scenes, countResult, nil
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) getSceneSort(findFilter *models.FindFilterType) string {
|
||||
if findFilter == nil {
|
||||
return " ORDER BY scenes.path, scenes.date ASC "
|
||||
}
|
||||
sort := findFilter.GetSort("title")
|
||||
direction := findFilter.GetDirection()
|
||||
return getSort(sort, direction, "scenes")
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) queryScene(query string, args []interface{}) (*models.Scene, error) {
|
||||
results, err := qb.queryScenes(query, args)
|
||||
if err != nil || len(results) < 1 {
|
||||
return nil, err
|
||||
}
|
||||
return results[0], nil
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) queryScenes(query string, args []interface{}) ([]*models.Scene, error) {
|
||||
var ret models.Scenes
|
||||
if err := qb.query(query, args, &ret); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return []*models.Scene(ret), nil
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) imageRepository() *imageRepository {
|
||||
return &imageRepository{
|
||||
repository: repository{
|
||||
tx: qb.tx,
|
||||
tableName: "scenes_cover",
|
||||
idColumn: sceneIDColumn,
|
||||
},
|
||||
imageColumn: "cover",
|
||||
}
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) GetCover(sceneID int) ([]byte, error) {
|
||||
return qb.imageRepository().get(sceneID)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) UpdateCover(sceneID int, image []byte) error {
|
||||
return qb.imageRepository().replace(sceneID, image)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) DestroyCover(sceneID int) error {
|
||||
return qb.imageRepository().destroy([]int{sceneID})
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) moviesRepository() *repository {
|
||||
return &repository{
|
||||
tx: qb.tx,
|
||||
tableName: moviesScenesTable,
|
||||
idColumn: sceneIDColumn,
|
||||
}
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) GetMovies(id int) (ret []models.MoviesScenes, err error) {
|
||||
if err := qb.moviesRepository().getAll(id, func(rows *sqlx.Rows) error {
|
||||
var ms models.MoviesScenes
|
||||
if err := rows.StructScan(&ms); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
ret = append(ret, ms)
|
||||
return nil
|
||||
}); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return ret, nil
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) UpdateMovies(sceneID int, movies []models.MoviesScenes) error {
|
||||
// destroy existing joins
|
||||
r := qb.moviesRepository()
|
||||
if err := r.destroy([]int{sceneID}); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, m := range movies {
|
||||
m.SceneID = sceneID
|
||||
if _, err := r.insert(m); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) performersRepository() *joinRepository {
|
||||
return &joinRepository{
|
||||
repository: repository{
|
||||
tx: qb.tx,
|
||||
tableName: performersScenesTable,
|
||||
idColumn: sceneIDColumn,
|
||||
},
|
||||
fkColumn: performerIDColumn,
|
||||
}
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) GetPerformerIDs(id int) ([]int, error) {
|
||||
return qb.performersRepository().getIDs(id)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) UpdatePerformers(id int, performerIDs []int) error {
|
||||
// Delete the existing joins and then create new ones
|
||||
return qb.performersRepository().replace(id, performerIDs)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) tagsRepository() *joinRepository {
|
||||
return &joinRepository{
|
||||
repository: repository{
|
||||
tx: qb.tx,
|
||||
tableName: scenesTagsTable,
|
||||
idColumn: sceneIDColumn,
|
||||
},
|
||||
fkColumn: tagIDColumn,
|
||||
}
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) GetTagIDs(id int) ([]int, error) {
|
||||
return qb.tagsRepository().getIDs(id)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) UpdateTags(id int, tagIDs []int) error {
|
||||
// Delete the existing joins and then create new ones
|
||||
return qb.tagsRepository().replace(id, tagIDs)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) stashIDRepository() *stashIDRepository {
|
||||
return &stashIDRepository{
|
||||
repository{
|
||||
tx: qb.tx,
|
||||
tableName: "scene_stash_ids",
|
||||
idColumn: sceneIDColumn,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) GetStashIDs(sceneID int) ([]*models.StashID, error) {
|
||||
return qb.stashIDRepository().get(sceneID)
|
||||
}
|
||||
|
||||
func (qb *sceneQueryBuilder) UpdateStashIDs(sceneID int, stashIDs []models.StashID) error {
|
||||
return qb.stashIDRepository().replace(sceneID, stashIDs)
|
||||
}
|
||||
Reference in New Issue
Block a user