mirror of
https://github.com/stashapp/stash.git
synced 2025-12-18 04:44:37 +03:00
Restructure data layer (#2532)
* Add new txn manager interface * Add txn management to sqlite * Rename get to getByID * Add contexts to repository methods * Update query builders * Add context to reader writer interfaces * Use repository in resolver * Tighten interfaces * Tighten interfaces in dlna * Tighten interfaces in match package * Tighten interfaces in scraper package * Tighten interfaces in scan code * Tighten interfaces on autotag package * Remove ReaderWriter usage * Merge database package into sqlite
This commit is contained in:
@@ -3,6 +3,8 @@
|
||||
package mocks
|
||||
|
||||
import (
|
||||
context "context"
|
||||
|
||||
models "github.com/stashapp/stash/pkg/models"
|
||||
mock "github.com/stretchr/testify/mock"
|
||||
)
|
||||
@@ -12,13 +14,13 @@ type GalleryReaderWriter struct {
|
||||
mock.Mock
|
||||
}
|
||||
|
||||
// All provides a mock function with given fields:
|
||||
func (_m *GalleryReaderWriter) All() ([]*models.Gallery, error) {
|
||||
ret := _m.Called()
|
||||
// All provides a mock function with given fields: ctx
|
||||
func (_m *GalleryReaderWriter) All(ctx context.Context) ([]*models.Gallery, error) {
|
||||
ret := _m.Called(ctx)
|
||||
|
||||
var r0 []*models.Gallery
|
||||
if rf, ok := ret.Get(0).(func() []*models.Gallery); ok {
|
||||
r0 = rf()
|
||||
if rf, ok := ret.Get(0).(func(context.Context) []*models.Gallery); ok {
|
||||
r0 = rf(ctx)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*models.Gallery)
|
||||
@@ -26,8 +28,8 @@ func (_m *GalleryReaderWriter) All() ([]*models.Gallery, error) {
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func() error); ok {
|
||||
r1 = rf()
|
||||
if rf, ok := ret.Get(1).(func(context.Context) error); ok {
|
||||
r1 = rf(ctx)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -35,20 +37,20 @@ func (_m *GalleryReaderWriter) All() ([]*models.Gallery, error) {
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// Count provides a mock function with given fields:
|
||||
func (_m *GalleryReaderWriter) Count() (int, error) {
|
||||
ret := _m.Called()
|
||||
// Count provides a mock function with given fields: ctx
|
||||
func (_m *GalleryReaderWriter) Count(ctx context.Context) (int, error) {
|
||||
ret := _m.Called(ctx)
|
||||
|
||||
var r0 int
|
||||
if rf, ok := ret.Get(0).(func() int); ok {
|
||||
r0 = rf()
|
||||
if rf, ok := ret.Get(0).(func(context.Context) int); ok {
|
||||
r0 = rf(ctx)
|
||||
} else {
|
||||
r0 = ret.Get(0).(int)
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func() error); ok {
|
||||
r1 = rf()
|
||||
if rf, ok := ret.Get(1).(func(context.Context) error); ok {
|
||||
r1 = rf(ctx)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -56,13 +58,13 @@ func (_m *GalleryReaderWriter) Count() (int, error) {
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// Create provides a mock function with given fields: newGallery
|
||||
func (_m *GalleryReaderWriter) Create(newGallery models.Gallery) (*models.Gallery, error) {
|
||||
ret := _m.Called(newGallery)
|
||||
// Create provides a mock function with given fields: ctx, newGallery
|
||||
func (_m *GalleryReaderWriter) Create(ctx context.Context, newGallery models.Gallery) (*models.Gallery, error) {
|
||||
ret := _m.Called(ctx, newGallery)
|
||||
|
||||
var r0 *models.Gallery
|
||||
if rf, ok := ret.Get(0).(func(models.Gallery) *models.Gallery); ok {
|
||||
r0 = rf(newGallery)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, models.Gallery) *models.Gallery); ok {
|
||||
r0 = rf(ctx, newGallery)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*models.Gallery)
|
||||
@@ -70,8 +72,8 @@ func (_m *GalleryReaderWriter) Create(newGallery models.Gallery) (*models.Galler
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(models.Gallery) error); ok {
|
||||
r1 = rf(newGallery)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, models.Gallery) error); ok {
|
||||
r1 = rf(ctx, newGallery)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -79,13 +81,13 @@ func (_m *GalleryReaderWriter) Create(newGallery models.Gallery) (*models.Galler
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// Destroy provides a mock function with given fields: id
|
||||
func (_m *GalleryReaderWriter) Destroy(id int) error {
|
||||
ret := _m.Called(id)
|
||||
// Destroy provides a mock function with given fields: ctx, id
|
||||
func (_m *GalleryReaderWriter) Destroy(ctx context.Context, id int) error {
|
||||
ret := _m.Called(ctx, id)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(int) error); ok {
|
||||
r0 = rf(id)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int) error); ok {
|
||||
r0 = rf(ctx, id)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
@@ -93,13 +95,13 @@ func (_m *GalleryReaderWriter) Destroy(id int) error {
|
||||
return r0
|
||||
}
|
||||
|
||||
// Find provides a mock function with given fields: id
|
||||
func (_m *GalleryReaderWriter) Find(id int) (*models.Gallery, error) {
|
||||
ret := _m.Called(id)
|
||||
// Find provides a mock function with given fields: ctx, id
|
||||
func (_m *GalleryReaderWriter) Find(ctx context.Context, id int) (*models.Gallery, error) {
|
||||
ret := _m.Called(ctx, id)
|
||||
|
||||
var r0 *models.Gallery
|
||||
if rf, ok := ret.Get(0).(func(int) *models.Gallery); ok {
|
||||
r0 = rf(id)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int) *models.Gallery); ok {
|
||||
r0 = rf(ctx, id)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*models.Gallery)
|
||||
@@ -107,8 +109,8 @@ func (_m *GalleryReaderWriter) Find(id int) (*models.Gallery, error) {
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(int) error); ok {
|
||||
r1 = rf(id)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, int) error); ok {
|
||||
r1 = rf(ctx, id)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -116,13 +118,13 @@ func (_m *GalleryReaderWriter) Find(id int) (*models.Gallery, error) {
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// FindByChecksum provides a mock function with given fields: checksum
|
||||
func (_m *GalleryReaderWriter) FindByChecksum(checksum string) (*models.Gallery, error) {
|
||||
ret := _m.Called(checksum)
|
||||
// FindByChecksum provides a mock function with given fields: ctx, checksum
|
||||
func (_m *GalleryReaderWriter) FindByChecksum(ctx context.Context, checksum string) (*models.Gallery, error) {
|
||||
ret := _m.Called(ctx, checksum)
|
||||
|
||||
var r0 *models.Gallery
|
||||
if rf, ok := ret.Get(0).(func(string) *models.Gallery); ok {
|
||||
r0 = rf(checksum)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, string) *models.Gallery); ok {
|
||||
r0 = rf(ctx, checksum)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*models.Gallery)
|
||||
@@ -130,8 +132,8 @@ func (_m *GalleryReaderWriter) FindByChecksum(checksum string) (*models.Gallery,
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(string) error); ok {
|
||||
r1 = rf(checksum)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, string) error); ok {
|
||||
r1 = rf(ctx, checksum)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -139,13 +141,13 @@ func (_m *GalleryReaderWriter) FindByChecksum(checksum string) (*models.Gallery,
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// FindByChecksums provides a mock function with given fields: checksums
|
||||
func (_m *GalleryReaderWriter) FindByChecksums(checksums []string) ([]*models.Gallery, error) {
|
||||
ret := _m.Called(checksums)
|
||||
// FindByChecksums provides a mock function with given fields: ctx, checksums
|
||||
func (_m *GalleryReaderWriter) FindByChecksums(ctx context.Context, checksums []string) ([]*models.Gallery, error) {
|
||||
ret := _m.Called(ctx, checksums)
|
||||
|
||||
var r0 []*models.Gallery
|
||||
if rf, ok := ret.Get(0).(func([]string) []*models.Gallery); ok {
|
||||
r0 = rf(checksums)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, []string) []*models.Gallery); ok {
|
||||
r0 = rf(ctx, checksums)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*models.Gallery)
|
||||
@@ -153,8 +155,8 @@ func (_m *GalleryReaderWriter) FindByChecksums(checksums []string) ([]*models.Ga
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func([]string) error); ok {
|
||||
r1 = rf(checksums)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, []string) error); ok {
|
||||
r1 = rf(ctx, checksums)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -162,13 +164,13 @@ func (_m *GalleryReaderWriter) FindByChecksums(checksums []string) ([]*models.Ga
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// FindByImageID provides a mock function with given fields: imageID
|
||||
func (_m *GalleryReaderWriter) FindByImageID(imageID int) ([]*models.Gallery, error) {
|
||||
ret := _m.Called(imageID)
|
||||
// FindByImageID provides a mock function with given fields: ctx, imageID
|
||||
func (_m *GalleryReaderWriter) FindByImageID(ctx context.Context, imageID int) ([]*models.Gallery, error) {
|
||||
ret := _m.Called(ctx, imageID)
|
||||
|
||||
var r0 []*models.Gallery
|
||||
if rf, ok := ret.Get(0).(func(int) []*models.Gallery); ok {
|
||||
r0 = rf(imageID)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int) []*models.Gallery); ok {
|
||||
r0 = rf(ctx, imageID)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*models.Gallery)
|
||||
@@ -176,8 +178,8 @@ func (_m *GalleryReaderWriter) FindByImageID(imageID int) ([]*models.Gallery, er
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(int) error); ok {
|
||||
r1 = rf(imageID)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, int) error); ok {
|
||||
r1 = rf(ctx, imageID)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -185,13 +187,13 @@ func (_m *GalleryReaderWriter) FindByImageID(imageID int) ([]*models.Gallery, er
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// FindByPath provides a mock function with given fields: path
|
||||
func (_m *GalleryReaderWriter) FindByPath(path string) (*models.Gallery, error) {
|
||||
ret := _m.Called(path)
|
||||
// FindByPath provides a mock function with given fields: ctx, path
|
||||
func (_m *GalleryReaderWriter) FindByPath(ctx context.Context, path string) (*models.Gallery, error) {
|
||||
ret := _m.Called(ctx, path)
|
||||
|
||||
var r0 *models.Gallery
|
||||
if rf, ok := ret.Get(0).(func(string) *models.Gallery); ok {
|
||||
r0 = rf(path)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, string) *models.Gallery); ok {
|
||||
r0 = rf(ctx, path)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*models.Gallery)
|
||||
@@ -199,8 +201,8 @@ func (_m *GalleryReaderWriter) FindByPath(path string) (*models.Gallery, error)
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(string) error); ok {
|
||||
r1 = rf(path)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, string) error); ok {
|
||||
r1 = rf(ctx, path)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -208,13 +210,13 @@ func (_m *GalleryReaderWriter) FindByPath(path string) (*models.Gallery, error)
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// FindBySceneID provides a mock function with given fields: sceneID
|
||||
func (_m *GalleryReaderWriter) FindBySceneID(sceneID int) ([]*models.Gallery, error) {
|
||||
ret := _m.Called(sceneID)
|
||||
// FindBySceneID provides a mock function with given fields: ctx, sceneID
|
||||
func (_m *GalleryReaderWriter) FindBySceneID(ctx context.Context, sceneID int) ([]*models.Gallery, error) {
|
||||
ret := _m.Called(ctx, sceneID)
|
||||
|
||||
var r0 []*models.Gallery
|
||||
if rf, ok := ret.Get(0).(func(int) []*models.Gallery); ok {
|
||||
r0 = rf(sceneID)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int) []*models.Gallery); ok {
|
||||
r0 = rf(ctx, sceneID)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*models.Gallery)
|
||||
@@ -222,8 +224,8 @@ func (_m *GalleryReaderWriter) FindBySceneID(sceneID int) ([]*models.Gallery, er
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(int) error); ok {
|
||||
r1 = rf(sceneID)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, int) error); ok {
|
||||
r1 = rf(ctx, sceneID)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -231,13 +233,13 @@ func (_m *GalleryReaderWriter) FindBySceneID(sceneID int) ([]*models.Gallery, er
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// FindMany provides a mock function with given fields: ids
|
||||
func (_m *GalleryReaderWriter) FindMany(ids []int) ([]*models.Gallery, error) {
|
||||
ret := _m.Called(ids)
|
||||
// FindMany provides a mock function with given fields: ctx, ids
|
||||
func (_m *GalleryReaderWriter) FindMany(ctx context.Context, ids []int) ([]*models.Gallery, error) {
|
||||
ret := _m.Called(ctx, ids)
|
||||
|
||||
var r0 []*models.Gallery
|
||||
if rf, ok := ret.Get(0).(func([]int) []*models.Gallery); ok {
|
||||
r0 = rf(ids)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, []int) []*models.Gallery); ok {
|
||||
r0 = rf(ctx, ids)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*models.Gallery)
|
||||
@@ -245,8 +247,8 @@ func (_m *GalleryReaderWriter) FindMany(ids []int) ([]*models.Gallery, error) {
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func([]int) error); ok {
|
||||
r1 = rf(ids)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, []int) error); ok {
|
||||
r1 = rf(ctx, ids)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -254,13 +256,13 @@ func (_m *GalleryReaderWriter) FindMany(ids []int) ([]*models.Gallery, error) {
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// GetImageIDs provides a mock function with given fields: galleryID
|
||||
func (_m *GalleryReaderWriter) GetImageIDs(galleryID int) ([]int, error) {
|
||||
ret := _m.Called(galleryID)
|
||||
// GetImageIDs provides a mock function with given fields: ctx, galleryID
|
||||
func (_m *GalleryReaderWriter) GetImageIDs(ctx context.Context, galleryID int) ([]int, error) {
|
||||
ret := _m.Called(ctx, galleryID)
|
||||
|
||||
var r0 []int
|
||||
if rf, ok := ret.Get(0).(func(int) []int); ok {
|
||||
r0 = rf(galleryID)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int) []int); ok {
|
||||
r0 = rf(ctx, galleryID)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]int)
|
||||
@@ -268,8 +270,8 @@ func (_m *GalleryReaderWriter) GetImageIDs(galleryID int) ([]int, error) {
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(int) error); ok {
|
||||
r1 = rf(galleryID)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, int) error); ok {
|
||||
r1 = rf(ctx, galleryID)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -277,13 +279,13 @@ func (_m *GalleryReaderWriter) GetImageIDs(galleryID int) ([]int, error) {
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// GetPerformerIDs provides a mock function with given fields: galleryID
|
||||
func (_m *GalleryReaderWriter) GetPerformerIDs(galleryID int) ([]int, error) {
|
||||
ret := _m.Called(galleryID)
|
||||
// GetPerformerIDs provides a mock function with given fields: ctx, galleryID
|
||||
func (_m *GalleryReaderWriter) GetPerformerIDs(ctx context.Context, galleryID int) ([]int, error) {
|
||||
ret := _m.Called(ctx, galleryID)
|
||||
|
||||
var r0 []int
|
||||
if rf, ok := ret.Get(0).(func(int) []int); ok {
|
||||
r0 = rf(galleryID)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int) []int); ok {
|
||||
r0 = rf(ctx, galleryID)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]int)
|
||||
@@ -291,8 +293,8 @@ func (_m *GalleryReaderWriter) GetPerformerIDs(galleryID int) ([]int, error) {
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(int) error); ok {
|
||||
r1 = rf(galleryID)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, int) error); ok {
|
||||
r1 = rf(ctx, galleryID)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -300,13 +302,13 @@ func (_m *GalleryReaderWriter) GetPerformerIDs(galleryID int) ([]int, error) {
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// GetSceneIDs provides a mock function with given fields: galleryID
|
||||
func (_m *GalleryReaderWriter) GetSceneIDs(galleryID int) ([]int, error) {
|
||||
ret := _m.Called(galleryID)
|
||||
// GetSceneIDs provides a mock function with given fields: ctx, galleryID
|
||||
func (_m *GalleryReaderWriter) GetSceneIDs(ctx context.Context, galleryID int) ([]int, error) {
|
||||
ret := _m.Called(ctx, galleryID)
|
||||
|
||||
var r0 []int
|
||||
if rf, ok := ret.Get(0).(func(int) []int); ok {
|
||||
r0 = rf(galleryID)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int) []int); ok {
|
||||
r0 = rf(ctx, galleryID)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]int)
|
||||
@@ -314,8 +316,8 @@ func (_m *GalleryReaderWriter) GetSceneIDs(galleryID int) ([]int, error) {
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(int) error); ok {
|
||||
r1 = rf(galleryID)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, int) error); ok {
|
||||
r1 = rf(ctx, galleryID)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -323,13 +325,13 @@ func (_m *GalleryReaderWriter) GetSceneIDs(galleryID int) ([]int, error) {
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// GetTagIDs provides a mock function with given fields: galleryID
|
||||
func (_m *GalleryReaderWriter) GetTagIDs(galleryID int) ([]int, error) {
|
||||
ret := _m.Called(galleryID)
|
||||
// GetTagIDs provides a mock function with given fields: ctx, galleryID
|
||||
func (_m *GalleryReaderWriter) GetTagIDs(ctx context.Context, galleryID int) ([]int, error) {
|
||||
ret := _m.Called(ctx, galleryID)
|
||||
|
||||
var r0 []int
|
||||
if rf, ok := ret.Get(0).(func(int) []int); ok {
|
||||
r0 = rf(galleryID)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int) []int); ok {
|
||||
r0 = rf(ctx, galleryID)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]int)
|
||||
@@ -337,8 +339,8 @@ func (_m *GalleryReaderWriter) GetTagIDs(galleryID int) ([]int, error) {
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(int) error); ok {
|
||||
r1 = rf(galleryID)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, int) error); ok {
|
||||
r1 = rf(ctx, galleryID)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -346,13 +348,13 @@ func (_m *GalleryReaderWriter) GetTagIDs(galleryID int) ([]int, error) {
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// Query provides a mock function with given fields: galleryFilter, findFilter
|
||||
func (_m *GalleryReaderWriter) Query(galleryFilter *models.GalleryFilterType, findFilter *models.FindFilterType) ([]*models.Gallery, int, error) {
|
||||
ret := _m.Called(galleryFilter, findFilter)
|
||||
// Query provides a mock function with given fields: ctx, galleryFilter, findFilter
|
||||
func (_m *GalleryReaderWriter) Query(ctx context.Context, galleryFilter *models.GalleryFilterType, findFilter *models.FindFilterType) ([]*models.Gallery, int, error) {
|
||||
ret := _m.Called(ctx, galleryFilter, findFilter)
|
||||
|
||||
var r0 []*models.Gallery
|
||||
if rf, ok := ret.Get(0).(func(*models.GalleryFilterType, *models.FindFilterType) []*models.Gallery); ok {
|
||||
r0 = rf(galleryFilter, findFilter)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, *models.GalleryFilterType, *models.FindFilterType) []*models.Gallery); ok {
|
||||
r0 = rf(ctx, galleryFilter, findFilter)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).([]*models.Gallery)
|
||||
@@ -360,15 +362,15 @@ func (_m *GalleryReaderWriter) Query(galleryFilter *models.GalleryFilterType, fi
|
||||
}
|
||||
|
||||
var r1 int
|
||||
if rf, ok := ret.Get(1).(func(*models.GalleryFilterType, *models.FindFilterType) int); ok {
|
||||
r1 = rf(galleryFilter, findFilter)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, *models.GalleryFilterType, *models.FindFilterType) int); ok {
|
||||
r1 = rf(ctx, galleryFilter, findFilter)
|
||||
} else {
|
||||
r1 = ret.Get(1).(int)
|
||||
}
|
||||
|
||||
var r2 error
|
||||
if rf, ok := ret.Get(2).(func(*models.GalleryFilterType, *models.FindFilterType) error); ok {
|
||||
r2 = rf(galleryFilter, findFilter)
|
||||
if rf, ok := ret.Get(2).(func(context.Context, *models.GalleryFilterType, *models.FindFilterType) error); ok {
|
||||
r2 = rf(ctx, galleryFilter, findFilter)
|
||||
} else {
|
||||
r2 = ret.Error(2)
|
||||
}
|
||||
@@ -376,20 +378,20 @@ func (_m *GalleryReaderWriter) Query(galleryFilter *models.GalleryFilterType, fi
|
||||
return r0, r1, r2
|
||||
}
|
||||
|
||||
// QueryCount provides a mock function with given fields: galleryFilter, findFilter
|
||||
func (_m *GalleryReaderWriter) QueryCount(galleryFilter *models.GalleryFilterType, findFilter *models.FindFilterType) (int, error) {
|
||||
ret := _m.Called(galleryFilter, findFilter)
|
||||
// QueryCount provides a mock function with given fields: ctx, galleryFilter, findFilter
|
||||
func (_m *GalleryReaderWriter) QueryCount(ctx context.Context, galleryFilter *models.GalleryFilterType, findFilter *models.FindFilterType) (int, error) {
|
||||
ret := _m.Called(ctx, galleryFilter, findFilter)
|
||||
|
||||
var r0 int
|
||||
if rf, ok := ret.Get(0).(func(*models.GalleryFilterType, *models.FindFilterType) int); ok {
|
||||
r0 = rf(galleryFilter, findFilter)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, *models.GalleryFilterType, *models.FindFilterType) int); ok {
|
||||
r0 = rf(ctx, galleryFilter, findFilter)
|
||||
} else {
|
||||
r0 = ret.Get(0).(int)
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(*models.GalleryFilterType, *models.FindFilterType) error); ok {
|
||||
r1 = rf(galleryFilter, findFilter)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, *models.GalleryFilterType, *models.FindFilterType) error); ok {
|
||||
r1 = rf(ctx, galleryFilter, findFilter)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -397,13 +399,13 @@ func (_m *GalleryReaderWriter) QueryCount(galleryFilter *models.GalleryFilterTyp
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// Update provides a mock function with given fields: updatedGallery
|
||||
func (_m *GalleryReaderWriter) Update(updatedGallery models.Gallery) (*models.Gallery, error) {
|
||||
ret := _m.Called(updatedGallery)
|
||||
// Update provides a mock function with given fields: ctx, updatedGallery
|
||||
func (_m *GalleryReaderWriter) Update(ctx context.Context, updatedGallery models.Gallery) (*models.Gallery, error) {
|
||||
ret := _m.Called(ctx, updatedGallery)
|
||||
|
||||
var r0 *models.Gallery
|
||||
if rf, ok := ret.Get(0).(func(models.Gallery) *models.Gallery); ok {
|
||||
r0 = rf(updatedGallery)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, models.Gallery) *models.Gallery); ok {
|
||||
r0 = rf(ctx, updatedGallery)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*models.Gallery)
|
||||
@@ -411,8 +413,8 @@ func (_m *GalleryReaderWriter) Update(updatedGallery models.Gallery) (*models.Ga
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(models.Gallery) error); ok {
|
||||
r1 = rf(updatedGallery)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, models.Gallery) error); ok {
|
||||
r1 = rf(ctx, updatedGallery)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -420,13 +422,13 @@ func (_m *GalleryReaderWriter) Update(updatedGallery models.Gallery) (*models.Ga
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// UpdateFileModTime provides a mock function with given fields: id, modTime
|
||||
func (_m *GalleryReaderWriter) UpdateFileModTime(id int, modTime models.NullSQLiteTimestamp) error {
|
||||
ret := _m.Called(id, modTime)
|
||||
// UpdateFileModTime provides a mock function with given fields: ctx, id, modTime
|
||||
func (_m *GalleryReaderWriter) UpdateFileModTime(ctx context.Context, id int, modTime models.NullSQLiteTimestamp) error {
|
||||
ret := _m.Called(ctx, id, modTime)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(int, models.NullSQLiteTimestamp) error); ok {
|
||||
r0 = rf(id, modTime)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int, models.NullSQLiteTimestamp) error); ok {
|
||||
r0 = rf(ctx, id, modTime)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
@@ -434,13 +436,13 @@ func (_m *GalleryReaderWriter) UpdateFileModTime(id int, modTime models.NullSQLi
|
||||
return r0
|
||||
}
|
||||
|
||||
// UpdateImages provides a mock function with given fields: galleryID, imageIDs
|
||||
func (_m *GalleryReaderWriter) UpdateImages(galleryID int, imageIDs []int) error {
|
||||
ret := _m.Called(galleryID, imageIDs)
|
||||
// UpdateImages provides a mock function with given fields: ctx, galleryID, imageIDs
|
||||
func (_m *GalleryReaderWriter) UpdateImages(ctx context.Context, galleryID int, imageIDs []int) error {
|
||||
ret := _m.Called(ctx, galleryID, imageIDs)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(int, []int) error); ok {
|
||||
r0 = rf(galleryID, imageIDs)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int, []int) error); ok {
|
||||
r0 = rf(ctx, galleryID, imageIDs)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
@@ -448,13 +450,13 @@ func (_m *GalleryReaderWriter) UpdateImages(galleryID int, imageIDs []int) error
|
||||
return r0
|
||||
}
|
||||
|
||||
// UpdatePartial provides a mock function with given fields: updatedGallery
|
||||
func (_m *GalleryReaderWriter) UpdatePartial(updatedGallery models.GalleryPartial) (*models.Gallery, error) {
|
||||
ret := _m.Called(updatedGallery)
|
||||
// UpdatePartial provides a mock function with given fields: ctx, updatedGallery
|
||||
func (_m *GalleryReaderWriter) UpdatePartial(ctx context.Context, updatedGallery models.GalleryPartial) (*models.Gallery, error) {
|
||||
ret := _m.Called(ctx, updatedGallery)
|
||||
|
||||
var r0 *models.Gallery
|
||||
if rf, ok := ret.Get(0).(func(models.GalleryPartial) *models.Gallery); ok {
|
||||
r0 = rf(updatedGallery)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, models.GalleryPartial) *models.Gallery); ok {
|
||||
r0 = rf(ctx, updatedGallery)
|
||||
} else {
|
||||
if ret.Get(0) != nil {
|
||||
r0 = ret.Get(0).(*models.Gallery)
|
||||
@@ -462,8 +464,8 @@ func (_m *GalleryReaderWriter) UpdatePartial(updatedGallery models.GalleryPartia
|
||||
}
|
||||
|
||||
var r1 error
|
||||
if rf, ok := ret.Get(1).(func(models.GalleryPartial) error); ok {
|
||||
r1 = rf(updatedGallery)
|
||||
if rf, ok := ret.Get(1).(func(context.Context, models.GalleryPartial) error); ok {
|
||||
r1 = rf(ctx, updatedGallery)
|
||||
} else {
|
||||
r1 = ret.Error(1)
|
||||
}
|
||||
@@ -471,13 +473,13 @@ func (_m *GalleryReaderWriter) UpdatePartial(updatedGallery models.GalleryPartia
|
||||
return r0, r1
|
||||
}
|
||||
|
||||
// UpdatePerformers provides a mock function with given fields: galleryID, performerIDs
|
||||
func (_m *GalleryReaderWriter) UpdatePerformers(galleryID int, performerIDs []int) error {
|
||||
ret := _m.Called(galleryID, performerIDs)
|
||||
// UpdatePerformers provides a mock function with given fields: ctx, galleryID, performerIDs
|
||||
func (_m *GalleryReaderWriter) UpdatePerformers(ctx context.Context, galleryID int, performerIDs []int) error {
|
||||
ret := _m.Called(ctx, galleryID, performerIDs)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(int, []int) error); ok {
|
||||
r0 = rf(galleryID, performerIDs)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int, []int) error); ok {
|
||||
r0 = rf(ctx, galleryID, performerIDs)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
@@ -485,13 +487,13 @@ func (_m *GalleryReaderWriter) UpdatePerformers(galleryID int, performerIDs []in
|
||||
return r0
|
||||
}
|
||||
|
||||
// UpdateScenes provides a mock function with given fields: galleryID, sceneIDs
|
||||
func (_m *GalleryReaderWriter) UpdateScenes(galleryID int, sceneIDs []int) error {
|
||||
ret := _m.Called(galleryID, sceneIDs)
|
||||
// UpdateScenes provides a mock function with given fields: ctx, galleryID, sceneIDs
|
||||
func (_m *GalleryReaderWriter) UpdateScenes(ctx context.Context, galleryID int, sceneIDs []int) error {
|
||||
ret := _m.Called(ctx, galleryID, sceneIDs)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(int, []int) error); ok {
|
||||
r0 = rf(galleryID, sceneIDs)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int, []int) error); ok {
|
||||
r0 = rf(ctx, galleryID, sceneIDs)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
@@ -499,13 +501,13 @@ func (_m *GalleryReaderWriter) UpdateScenes(galleryID int, sceneIDs []int) error
|
||||
return r0
|
||||
}
|
||||
|
||||
// UpdateTags provides a mock function with given fields: galleryID, tagIDs
|
||||
func (_m *GalleryReaderWriter) UpdateTags(galleryID int, tagIDs []int) error {
|
||||
ret := _m.Called(galleryID, tagIDs)
|
||||
// UpdateTags provides a mock function with given fields: ctx, galleryID, tagIDs
|
||||
func (_m *GalleryReaderWriter) UpdateTags(ctx context.Context, galleryID int, tagIDs []int) error {
|
||||
ret := _m.Called(ctx, galleryID, tagIDs)
|
||||
|
||||
var r0 error
|
||||
if rf, ok := ret.Get(0).(func(int, []int) error); ok {
|
||||
r0 = rf(galleryID, tagIDs)
|
||||
if rf, ok := ret.Get(0).(func(context.Context, int, []int) error); ok {
|
||||
r0 = rf(ctx, galleryID, tagIDs)
|
||||
} else {
|
||||
r0 = ret.Error(0)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user