mirror of
https://github.com/stashapp/stash.git
synced 2025-12-06 08:26:00 +01:00
33f2ebf2a3
* Remove manager.Repository * Refactor other repositories * Fix tests and add database mock * Add AssertExpectations method * Refactor routes * Move default movie image to internal/static and add convenience methods * Refactor default performer image boxes
41 lines
1,000 B
Go
41 lines
1,000 B
Go
package models
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/stashapp/stash/pkg/txn"
|
|
)
|
|
|
|
type TxnManager interface {
|
|
txn.Manager
|
|
txn.DatabaseProvider
|
|
}
|
|
|
|
type Repository struct {
|
|
TxnManager TxnManager
|
|
|
|
File FileReaderWriter
|
|
Folder FolderReaderWriter
|
|
Gallery GalleryReaderWriter
|
|
GalleryChapter GalleryChapterReaderWriter
|
|
Image ImageReaderWriter
|
|
Movie MovieReaderWriter
|
|
Performer PerformerReaderWriter
|
|
Scene SceneReaderWriter
|
|
SceneMarker SceneMarkerReaderWriter
|
|
Studio StudioReaderWriter
|
|
Tag TagReaderWriter
|
|
SavedFilter SavedFilterReaderWriter
|
|
}
|
|
|
|
func (r *Repository) WithTxn(ctx context.Context, fn txn.TxnFunc) error {
|
|
return txn.WithTxn(ctx, r.TxnManager, fn)
|
|
}
|
|
|
|
func (r *Repository) WithReadTxn(ctx context.Context, fn txn.TxnFunc) error {
|
|
return txn.WithReadTxn(ctx, r.TxnManager, fn)
|
|
}
|
|
|
|
func (r *Repository) WithDB(ctx context.Context, fn txn.TxnFunc) error {
|
|
return txn.WithDatabase(ctx, r.TxnManager, fn)
|
|
}
|