mirror of
https://github.com/stashapp/stash.git
synced 2025-12-18 04:44: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:
90
pkg/sqlite/scraped_item.go
Normal file
90
pkg/sqlite/scraped_item.go
Normal file
@@ -0,0 +1,90 @@
|
||||
package sqlite
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
|
||||
"github.com/stashapp/stash/pkg/models"
|
||||
)
|
||||
|
||||
const scrapedItemTable = "scraped_items"
|
||||
|
||||
type scrapedItemQueryBuilder struct {
|
||||
repository
|
||||
}
|
||||
|
||||
func NewScrapedItemReaderWriter(tx dbi) *scrapedItemQueryBuilder {
|
||||
return &scrapedItemQueryBuilder{
|
||||
repository{
|
||||
tx: tx,
|
||||
tableName: scrapedItemTable,
|
||||
idColumn: idColumn,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (qb *scrapedItemQueryBuilder) Create(newObject models.ScrapedItem) (*models.ScrapedItem, error) {
|
||||
var ret models.ScrapedItem
|
||||
if err := qb.insertObject(newObject, &ret); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &ret, nil
|
||||
}
|
||||
|
||||
func (qb *scrapedItemQueryBuilder) Update(updatedObject models.ScrapedItem) (*models.ScrapedItem, error) {
|
||||
const partial = false
|
||||
if err := qb.update(updatedObject.ID, updatedObject, partial); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return qb.find(updatedObject.ID)
|
||||
}
|
||||
|
||||
func (qb *scrapedItemQueryBuilder) Find(id int) (*models.ScrapedItem, error) {
|
||||
return qb.find(id)
|
||||
}
|
||||
|
||||
func (qb *scrapedItemQueryBuilder) find(id int) (*models.ScrapedItem, error) {
|
||||
var ret models.ScrapedItem
|
||||
if err := qb.get(id, &ret); err != nil {
|
||||
if err == sql.ErrNoRows {
|
||||
return nil, nil
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return &ret, nil
|
||||
}
|
||||
|
||||
func (qb *scrapedItemQueryBuilder) All() ([]*models.ScrapedItem, error) {
|
||||
return qb.queryScrapedItems(selectAll("scraped_items")+qb.getScrapedItemsSort(nil), nil)
|
||||
}
|
||||
|
||||
func (qb *scrapedItemQueryBuilder) getScrapedItemsSort(findFilter *models.FindFilterType) string {
|
||||
var sort string
|
||||
var direction string
|
||||
if findFilter == nil {
|
||||
sort = "id" // TODO studio_id and title
|
||||
direction = "ASC"
|
||||
} else {
|
||||
sort = findFilter.GetSort("id")
|
||||
direction = findFilter.GetDirection()
|
||||
}
|
||||
return getSort(sort, direction, "scraped_items")
|
||||
}
|
||||
|
||||
func (qb *scrapedItemQueryBuilder) queryScrapedItem(query string, args []interface{}) (*models.ScrapedItem, error) {
|
||||
results, err := qb.queryScrapedItems(query, args)
|
||||
if err != nil || len(results) < 1 {
|
||||
return nil, err
|
||||
}
|
||||
return results[0], nil
|
||||
}
|
||||
|
||||
func (qb *scrapedItemQueryBuilder) queryScrapedItems(query string, args []interface{}) ([]*models.ScrapedItem, error) {
|
||||
var ret models.ScrapedItems
|
||||
if err := qb.query(query, args, &ret); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return []*models.ScrapedItem(ret), nil
|
||||
}
|
||||
Reference in New Issue
Block a user