mirror of
https://github.com/stashapp/stash.git
synced 2025-12-17 12:24:38 +03:00
File storage rewrite (#2676)
* Restructure data layer part 2 (#2599) * Refactor and separate image model * Refactor image query builder * Handle relationships in image query builder * Remove relationship management methods * Refactor gallery model/query builder * Add scenes to gallery model * Convert scene model * Refactor scene models * Remove unused methods * Add unit tests for gallery * Add image tests * Add scene tests * Convert unnecessary scene value pointers to values * Convert unnecessary pointer values to values * Refactor scene partial * Add scene partial tests * Refactor ImagePartial * Add image partial tests * Refactor gallery partial update * Add partial gallery update tests * Use zero/null package for null values * Add files and scan system * Add sqlite implementation for files/folders * Add unit tests for files/folders * Image refactors * Update image data layer * Refactor gallery model and creation * Refactor scene model * Refactor scenes * Don't set title from filename * Allow galleries to freely add/remove images * Add multiple scene file support to graphql and UI * Add multiple file support for images in graphql/UI * Add multiple file for galleries in graphql/UI * Remove use of some deprecated fields * Remove scene path usage * Remove gallery path usage * Remove path from image * Move funscript to video file * Refactor caption detection * Migrate existing data * Add post commit/rollback hook system * Lint. Comment out import/export tests * Add WithDatabase read only wrapper * Prepend tasks to list * Add 32 pre-migration * Add warnings in release and migration notes
This commit is contained in:
75
pkg/sqlite/common.go
Normal file
75
pkg/sqlite/common.go
Normal file
@@ -0,0 +1,75 @@
|
||||
package sqlite
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"github.com/doug-martin/goqu/v9"
|
||||
"github.com/jmoiron/sqlx"
|
||||
)
|
||||
|
||||
type oCounterManager struct {
|
||||
tableMgr *table
|
||||
}
|
||||
|
||||
func (qb *oCounterManager) getOCounter(ctx context.Context, id int) (int, error) {
|
||||
q := dialect.From(qb.tableMgr.table).Select("o_counter").Where(goqu.Ex{"id": id})
|
||||
|
||||
const single = true
|
||||
var ret int
|
||||
if err := queryFunc(ctx, q, single, func(rows *sqlx.Rows) error {
|
||||
if err := rows.Scan(&ret); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}); err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return ret, nil
|
||||
}
|
||||
|
||||
func (qb *oCounterManager) IncrementOCounter(ctx context.Context, id int) (int, error) {
|
||||
if err := qb.tableMgr.checkIDExists(ctx, id); err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
if err := qb.tableMgr.updateByID(ctx, id, goqu.Record{
|
||||
"o_counter": goqu.L("o_counter + 1"),
|
||||
}); err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return qb.getOCounter(ctx, id)
|
||||
}
|
||||
|
||||
func (qb *oCounterManager) DecrementOCounter(ctx context.Context, id int) (int, error) {
|
||||
if err := qb.tableMgr.checkIDExists(ctx, id); err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
table := qb.tableMgr.table
|
||||
q := dialect.Update(table).Set(goqu.Record{
|
||||
"o_counter": goqu.L("o_counter - 1"),
|
||||
}).Where(qb.tableMgr.byID(id), goqu.L("o_counter > 0"))
|
||||
|
||||
if _, err := exec(ctx, q); err != nil {
|
||||
return 0, fmt.Errorf("updating %s: %w", table.GetTable(), err)
|
||||
}
|
||||
|
||||
return qb.getOCounter(ctx, id)
|
||||
}
|
||||
|
||||
func (qb *oCounterManager) ResetOCounter(ctx context.Context, id int) (int, error) {
|
||||
if err := qb.tableMgr.checkIDExists(ctx, id); err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
if err := qb.tableMgr.updateByID(ctx, id, goqu.Record{
|
||||
"o_counter": 0,
|
||||
}); err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
return qb.getOCounter(ctx, id)
|
||||
}
|
||||
Reference in New Issue
Block a user