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:
WithoutPants
2022-05-19 17:49:32 +10:00
parent 7b5bd80515
commit 964b559309
244 changed files with 7377 additions and 6699 deletions

View File

@@ -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 PerformerReaderWriter struct {
mock.Mock
}
// All provides a mock function with given fields:
func (_m *PerformerReaderWriter) All() ([]*models.Performer, error) {
ret := _m.Called()
// All provides a mock function with given fields: ctx
func (_m *PerformerReaderWriter) All(ctx context.Context) ([]*models.Performer, error) {
ret := _m.Called(ctx)
var r0 []*models.Performer
if rf, ok := ret.Get(0).(func() []*models.Performer); ok {
r0 = rf()
if rf, ok := ret.Get(0).(func(context.Context) []*models.Performer); ok {
r0 = rf(ctx)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*models.Performer)
@@ -26,8 +28,8 @@ func (_m *PerformerReaderWriter) All() ([]*models.Performer, 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 *PerformerReaderWriter) All() ([]*models.Performer, error) {
return r0, r1
}
// Count provides a mock function with given fields:
func (_m *PerformerReaderWriter) Count() (int, error) {
ret := _m.Called()
// Count provides a mock function with given fields: ctx
func (_m *PerformerReaderWriter) 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,20 +58,20 @@ func (_m *PerformerReaderWriter) Count() (int, error) {
return r0, r1
}
// CountByTagID provides a mock function with given fields: tagID
func (_m *PerformerReaderWriter) CountByTagID(tagID int) (int, error) {
ret := _m.Called(tagID)
// CountByTagID provides a mock function with given fields: ctx, tagID
func (_m *PerformerReaderWriter) CountByTagID(ctx context.Context, tagID int) (int, error) {
ret := _m.Called(ctx, tagID)
var r0 int
if rf, ok := ret.Get(0).(func(int) int); ok {
r0 = rf(tagID)
if rf, ok := ret.Get(0).(func(context.Context, int) int); ok {
r0 = rf(ctx, tagID)
} else {
r0 = ret.Get(0).(int)
}
var r1 error
if rf, ok := ret.Get(1).(func(int) error); ok {
r1 = rf(tagID)
if rf, ok := ret.Get(1).(func(context.Context, int) error); ok {
r1 = rf(ctx, tagID)
} else {
r1 = ret.Error(1)
}
@@ -77,13 +79,13 @@ func (_m *PerformerReaderWriter) CountByTagID(tagID int) (int, error) {
return r0, r1
}
// Create provides a mock function with given fields: newPerformer
func (_m *PerformerReaderWriter) Create(newPerformer models.Performer) (*models.Performer, error) {
ret := _m.Called(newPerformer)
// Create provides a mock function with given fields: ctx, newPerformer
func (_m *PerformerReaderWriter) Create(ctx context.Context, newPerformer models.Performer) (*models.Performer, error) {
ret := _m.Called(ctx, newPerformer)
var r0 *models.Performer
if rf, ok := ret.Get(0).(func(models.Performer) *models.Performer); ok {
r0 = rf(newPerformer)
if rf, ok := ret.Get(0).(func(context.Context, models.Performer) *models.Performer); ok {
r0 = rf(ctx, newPerformer)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*models.Performer)
@@ -91,8 +93,8 @@ func (_m *PerformerReaderWriter) Create(newPerformer models.Performer) (*models.
}
var r1 error
if rf, ok := ret.Get(1).(func(models.Performer) error); ok {
r1 = rf(newPerformer)
if rf, ok := ret.Get(1).(func(context.Context, models.Performer) error); ok {
r1 = rf(ctx, newPerformer)
} else {
r1 = ret.Error(1)
}
@@ -100,13 +102,13 @@ func (_m *PerformerReaderWriter) Create(newPerformer models.Performer) (*models.
return r0, r1
}
// Destroy provides a mock function with given fields: id
func (_m *PerformerReaderWriter) Destroy(id int) error {
ret := _m.Called(id)
// Destroy provides a mock function with given fields: ctx, id
func (_m *PerformerReaderWriter) 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)
}
@@ -114,13 +116,13 @@ func (_m *PerformerReaderWriter) Destroy(id int) error {
return r0
}
// DestroyImage provides a mock function with given fields: performerID
func (_m *PerformerReaderWriter) DestroyImage(performerID int) error {
ret := _m.Called(performerID)
// DestroyImage provides a mock function with given fields: ctx, performerID
func (_m *PerformerReaderWriter) DestroyImage(ctx context.Context, performerID int) error {
ret := _m.Called(ctx, performerID)
var r0 error
if rf, ok := ret.Get(0).(func(int) error); ok {
r0 = rf(performerID)
if rf, ok := ret.Get(0).(func(context.Context, int) error); ok {
r0 = rf(ctx, performerID)
} else {
r0 = ret.Error(0)
}
@@ -128,13 +130,13 @@ func (_m *PerformerReaderWriter) DestroyImage(performerID int) error {
return r0
}
// Find provides a mock function with given fields: id
func (_m *PerformerReaderWriter) Find(id int) (*models.Performer, error) {
ret := _m.Called(id)
// Find provides a mock function with given fields: ctx, id
func (_m *PerformerReaderWriter) Find(ctx context.Context, id int) (*models.Performer, error) {
ret := _m.Called(ctx, id)
var r0 *models.Performer
if rf, ok := ret.Get(0).(func(int) *models.Performer); ok {
r0 = rf(id)
if rf, ok := ret.Get(0).(func(context.Context, int) *models.Performer); ok {
r0 = rf(ctx, id)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*models.Performer)
@@ -142,8 +144,8 @@ func (_m *PerformerReaderWriter) Find(id int) (*models.Performer, 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)
}
@@ -151,13 +153,13 @@ func (_m *PerformerReaderWriter) Find(id int) (*models.Performer, error) {
return r0, r1
}
// FindByGalleryID provides a mock function with given fields: galleryID
func (_m *PerformerReaderWriter) FindByGalleryID(galleryID int) ([]*models.Performer, error) {
ret := _m.Called(galleryID)
// FindByGalleryID provides a mock function with given fields: ctx, galleryID
func (_m *PerformerReaderWriter) FindByGalleryID(ctx context.Context, galleryID int) ([]*models.Performer, error) {
ret := _m.Called(ctx, galleryID)
var r0 []*models.Performer
if rf, ok := ret.Get(0).(func(int) []*models.Performer); ok {
r0 = rf(galleryID)
if rf, ok := ret.Get(0).(func(context.Context, int) []*models.Performer); ok {
r0 = rf(ctx, galleryID)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*models.Performer)
@@ -165,8 +167,8 @@ func (_m *PerformerReaderWriter) FindByGalleryID(galleryID int) ([]*models.Perfo
}
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)
}
@@ -174,13 +176,13 @@ func (_m *PerformerReaderWriter) FindByGalleryID(galleryID int) ([]*models.Perfo
return r0, r1
}
// FindByImageID provides a mock function with given fields: imageID
func (_m *PerformerReaderWriter) FindByImageID(imageID int) ([]*models.Performer, error) {
ret := _m.Called(imageID)
// FindByImageID provides a mock function with given fields: ctx, imageID
func (_m *PerformerReaderWriter) FindByImageID(ctx context.Context, imageID int) ([]*models.Performer, error) {
ret := _m.Called(ctx, imageID)
var r0 []*models.Performer
if rf, ok := ret.Get(0).(func(int) []*models.Performer); ok {
r0 = rf(imageID)
if rf, ok := ret.Get(0).(func(context.Context, int) []*models.Performer); ok {
r0 = rf(ctx, imageID)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*models.Performer)
@@ -188,8 +190,8 @@ func (_m *PerformerReaderWriter) FindByImageID(imageID int) ([]*models.Performer
}
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)
}
@@ -197,13 +199,13 @@ func (_m *PerformerReaderWriter) FindByImageID(imageID int) ([]*models.Performer
return r0, r1
}
// FindByNames provides a mock function with given fields: names, nocase
func (_m *PerformerReaderWriter) FindByNames(names []string, nocase bool) ([]*models.Performer, error) {
ret := _m.Called(names, nocase)
// FindByNames provides a mock function with given fields: ctx, names, nocase
func (_m *PerformerReaderWriter) FindByNames(ctx context.Context, names []string, nocase bool) ([]*models.Performer, error) {
ret := _m.Called(ctx, names, nocase)
var r0 []*models.Performer
if rf, ok := ret.Get(0).(func([]string, bool) []*models.Performer); ok {
r0 = rf(names, nocase)
if rf, ok := ret.Get(0).(func(context.Context, []string, bool) []*models.Performer); ok {
r0 = rf(ctx, names, nocase)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*models.Performer)
@@ -211,8 +213,8 @@ func (_m *PerformerReaderWriter) FindByNames(names []string, nocase bool) ([]*mo
}
var r1 error
if rf, ok := ret.Get(1).(func([]string, bool) error); ok {
r1 = rf(names, nocase)
if rf, ok := ret.Get(1).(func(context.Context, []string, bool) error); ok {
r1 = rf(ctx, names, nocase)
} else {
r1 = ret.Error(1)
}
@@ -220,13 +222,13 @@ func (_m *PerformerReaderWriter) FindByNames(names []string, nocase bool) ([]*mo
return r0, r1
}
// FindBySceneID provides a mock function with given fields: sceneID
func (_m *PerformerReaderWriter) FindBySceneID(sceneID int) ([]*models.Performer, error) {
ret := _m.Called(sceneID)
// FindBySceneID provides a mock function with given fields: ctx, sceneID
func (_m *PerformerReaderWriter) FindBySceneID(ctx context.Context, sceneID int) ([]*models.Performer, error) {
ret := _m.Called(ctx, sceneID)
var r0 []*models.Performer
if rf, ok := ret.Get(0).(func(int) []*models.Performer); ok {
r0 = rf(sceneID)
if rf, ok := ret.Get(0).(func(context.Context, int) []*models.Performer); ok {
r0 = rf(ctx, sceneID)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*models.Performer)
@@ -234,8 +236,8 @@ func (_m *PerformerReaderWriter) FindBySceneID(sceneID int) ([]*models.Performer
}
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)
}
@@ -243,13 +245,13 @@ func (_m *PerformerReaderWriter) FindBySceneID(sceneID int) ([]*models.Performer
return r0, r1
}
// FindByStashID provides a mock function with given fields: stashID
func (_m *PerformerReaderWriter) FindByStashID(stashID models.StashID) ([]*models.Performer, error) {
ret := _m.Called(stashID)
// FindByStashID provides a mock function with given fields: ctx, stashID
func (_m *PerformerReaderWriter) FindByStashID(ctx context.Context, stashID models.StashID) ([]*models.Performer, error) {
ret := _m.Called(ctx, stashID)
var r0 []*models.Performer
if rf, ok := ret.Get(0).(func(models.StashID) []*models.Performer); ok {
r0 = rf(stashID)
if rf, ok := ret.Get(0).(func(context.Context, models.StashID) []*models.Performer); ok {
r0 = rf(ctx, stashID)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*models.Performer)
@@ -257,8 +259,8 @@ func (_m *PerformerReaderWriter) FindByStashID(stashID models.StashID) ([]*model
}
var r1 error
if rf, ok := ret.Get(1).(func(models.StashID) error); ok {
r1 = rf(stashID)
if rf, ok := ret.Get(1).(func(context.Context, models.StashID) error); ok {
r1 = rf(ctx, stashID)
} else {
r1 = ret.Error(1)
}
@@ -266,13 +268,13 @@ func (_m *PerformerReaderWriter) FindByStashID(stashID models.StashID) ([]*model
return r0, r1
}
// FindByStashIDStatus provides a mock function with given fields: hasStashID, stashboxEndpoint
func (_m *PerformerReaderWriter) FindByStashIDStatus(hasStashID bool, stashboxEndpoint string) ([]*models.Performer, error) {
ret := _m.Called(hasStashID, stashboxEndpoint)
// FindByStashIDStatus provides a mock function with given fields: ctx, hasStashID, stashboxEndpoint
func (_m *PerformerReaderWriter) FindByStashIDStatus(ctx context.Context, hasStashID bool, stashboxEndpoint string) ([]*models.Performer, error) {
ret := _m.Called(ctx, hasStashID, stashboxEndpoint)
var r0 []*models.Performer
if rf, ok := ret.Get(0).(func(bool, string) []*models.Performer); ok {
r0 = rf(hasStashID, stashboxEndpoint)
if rf, ok := ret.Get(0).(func(context.Context, bool, string) []*models.Performer); ok {
r0 = rf(ctx, hasStashID, stashboxEndpoint)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*models.Performer)
@@ -280,8 +282,8 @@ func (_m *PerformerReaderWriter) FindByStashIDStatus(hasStashID bool, stashboxEn
}
var r1 error
if rf, ok := ret.Get(1).(func(bool, string) error); ok {
r1 = rf(hasStashID, stashboxEndpoint)
if rf, ok := ret.Get(1).(func(context.Context, bool, string) error); ok {
r1 = rf(ctx, hasStashID, stashboxEndpoint)
} else {
r1 = ret.Error(1)
}
@@ -289,13 +291,13 @@ func (_m *PerformerReaderWriter) FindByStashIDStatus(hasStashID bool, stashboxEn
return r0, r1
}
// FindMany provides a mock function with given fields: ids
func (_m *PerformerReaderWriter) FindMany(ids []int) ([]*models.Performer, error) {
ret := _m.Called(ids)
// FindMany provides a mock function with given fields: ctx, ids
func (_m *PerformerReaderWriter) FindMany(ctx context.Context, ids []int) ([]*models.Performer, error) {
ret := _m.Called(ctx, ids)
var r0 []*models.Performer
if rf, ok := ret.Get(0).(func([]int) []*models.Performer); ok {
r0 = rf(ids)
if rf, ok := ret.Get(0).(func(context.Context, []int) []*models.Performer); ok {
r0 = rf(ctx, ids)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*models.Performer)
@@ -303,8 +305,8 @@ func (_m *PerformerReaderWriter) FindMany(ids []int) ([]*models.Performer, 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)
}
@@ -312,13 +314,13 @@ func (_m *PerformerReaderWriter) FindMany(ids []int) ([]*models.Performer, error
return r0, r1
}
// FindNamesBySceneID provides a mock function with given fields: sceneID
func (_m *PerformerReaderWriter) FindNamesBySceneID(sceneID int) ([]*models.Performer, error) {
ret := _m.Called(sceneID)
// FindNamesBySceneID provides a mock function with given fields: ctx, sceneID
func (_m *PerformerReaderWriter) FindNamesBySceneID(ctx context.Context, sceneID int) ([]*models.Performer, error) {
ret := _m.Called(ctx, sceneID)
var r0 []*models.Performer
if rf, ok := ret.Get(0).(func(int) []*models.Performer); ok {
r0 = rf(sceneID)
if rf, ok := ret.Get(0).(func(context.Context, int) []*models.Performer); ok {
r0 = rf(ctx, sceneID)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*models.Performer)
@@ -326,8 +328,8 @@ func (_m *PerformerReaderWriter) FindNamesBySceneID(sceneID int) ([]*models.Perf
}
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)
}
@@ -335,13 +337,13 @@ func (_m *PerformerReaderWriter) FindNamesBySceneID(sceneID int) ([]*models.Perf
return r0, r1
}
// GetImage provides a mock function with given fields: performerID
func (_m *PerformerReaderWriter) GetImage(performerID int) ([]byte, error) {
ret := _m.Called(performerID)
// GetImage provides a mock function with given fields: ctx, performerID
func (_m *PerformerReaderWriter) GetImage(ctx context.Context, performerID int) ([]byte, error) {
ret := _m.Called(ctx, performerID)
var r0 []byte
if rf, ok := ret.Get(0).(func(int) []byte); ok {
r0 = rf(performerID)
if rf, ok := ret.Get(0).(func(context.Context, int) []byte); ok {
r0 = rf(ctx, performerID)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]byte)
@@ -349,8 +351,8 @@ func (_m *PerformerReaderWriter) GetImage(performerID int) ([]byte, error) {
}
var r1 error
if rf, ok := ret.Get(1).(func(int) error); ok {
r1 = rf(performerID)
if rf, ok := ret.Get(1).(func(context.Context, int) error); ok {
r1 = rf(ctx, performerID)
} else {
r1 = ret.Error(1)
}
@@ -358,13 +360,13 @@ func (_m *PerformerReaderWriter) GetImage(performerID int) ([]byte, error) {
return r0, r1
}
// GetStashIDs provides a mock function with given fields: performerID
func (_m *PerformerReaderWriter) GetStashIDs(performerID int) ([]*models.StashID, error) {
ret := _m.Called(performerID)
// GetStashIDs provides a mock function with given fields: ctx, performerID
func (_m *PerformerReaderWriter) GetStashIDs(ctx context.Context, performerID int) ([]*models.StashID, error) {
ret := _m.Called(ctx, performerID)
var r0 []*models.StashID
if rf, ok := ret.Get(0).(func(int) []*models.StashID); ok {
r0 = rf(performerID)
if rf, ok := ret.Get(0).(func(context.Context, int) []*models.StashID); ok {
r0 = rf(ctx, performerID)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*models.StashID)
@@ -372,8 +374,8 @@ func (_m *PerformerReaderWriter) GetStashIDs(performerID int) ([]*models.StashID
}
var r1 error
if rf, ok := ret.Get(1).(func(int) error); ok {
r1 = rf(performerID)
if rf, ok := ret.Get(1).(func(context.Context, int) error); ok {
r1 = rf(ctx, performerID)
} else {
r1 = ret.Error(1)
}
@@ -381,13 +383,13 @@ func (_m *PerformerReaderWriter) GetStashIDs(performerID int) ([]*models.StashID
return r0, r1
}
// GetTagIDs provides a mock function with given fields: performerID
func (_m *PerformerReaderWriter) GetTagIDs(performerID int) ([]int, error) {
ret := _m.Called(performerID)
// GetTagIDs provides a mock function with given fields: ctx, performerID
func (_m *PerformerReaderWriter) GetTagIDs(ctx context.Context, performerID int) ([]int, error) {
ret := _m.Called(ctx, performerID)
var r0 []int
if rf, ok := ret.Get(0).(func(int) []int); ok {
r0 = rf(performerID)
if rf, ok := ret.Get(0).(func(context.Context, int) []int); ok {
r0 = rf(ctx, performerID)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]int)
@@ -395,8 +397,8 @@ func (_m *PerformerReaderWriter) GetTagIDs(performerID int) ([]int, error) {
}
var r1 error
if rf, ok := ret.Get(1).(func(int) error); ok {
r1 = rf(performerID)
if rf, ok := ret.Get(1).(func(context.Context, int) error); ok {
r1 = rf(ctx, performerID)
} else {
r1 = ret.Error(1)
}
@@ -404,13 +406,13 @@ func (_m *PerformerReaderWriter) GetTagIDs(performerID int) ([]int, error) {
return r0, r1
}
// Query provides a mock function with given fields: performerFilter, findFilter
func (_m *PerformerReaderWriter) Query(performerFilter *models.PerformerFilterType, findFilter *models.FindFilterType) ([]*models.Performer, int, error) {
ret := _m.Called(performerFilter, findFilter)
// Query provides a mock function with given fields: ctx, performerFilter, findFilter
func (_m *PerformerReaderWriter) Query(ctx context.Context, performerFilter *models.PerformerFilterType, findFilter *models.FindFilterType) ([]*models.Performer, int, error) {
ret := _m.Called(ctx, performerFilter, findFilter)
var r0 []*models.Performer
if rf, ok := ret.Get(0).(func(*models.PerformerFilterType, *models.FindFilterType) []*models.Performer); ok {
r0 = rf(performerFilter, findFilter)
if rf, ok := ret.Get(0).(func(context.Context, *models.PerformerFilterType, *models.FindFilterType) []*models.Performer); ok {
r0 = rf(ctx, performerFilter, findFilter)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*models.Performer)
@@ -418,15 +420,15 @@ func (_m *PerformerReaderWriter) Query(performerFilter *models.PerformerFilterTy
}
var r1 int
if rf, ok := ret.Get(1).(func(*models.PerformerFilterType, *models.FindFilterType) int); ok {
r1 = rf(performerFilter, findFilter)
if rf, ok := ret.Get(1).(func(context.Context, *models.PerformerFilterType, *models.FindFilterType) int); ok {
r1 = rf(ctx, performerFilter, findFilter)
} else {
r1 = ret.Get(1).(int)
}
var r2 error
if rf, ok := ret.Get(2).(func(*models.PerformerFilterType, *models.FindFilterType) error); ok {
r2 = rf(performerFilter, findFilter)
if rf, ok := ret.Get(2).(func(context.Context, *models.PerformerFilterType, *models.FindFilterType) error); ok {
r2 = rf(ctx, performerFilter, findFilter)
} else {
r2 = ret.Error(2)
}
@@ -434,13 +436,13 @@ func (_m *PerformerReaderWriter) Query(performerFilter *models.PerformerFilterTy
return r0, r1, r2
}
// QueryForAutoTag provides a mock function with given fields: words
func (_m *PerformerReaderWriter) QueryForAutoTag(words []string) ([]*models.Performer, error) {
ret := _m.Called(words)
// QueryForAutoTag provides a mock function with given fields: ctx, words
func (_m *PerformerReaderWriter) QueryForAutoTag(ctx context.Context, words []string) ([]*models.Performer, error) {
ret := _m.Called(ctx, words)
var r0 []*models.Performer
if rf, ok := ret.Get(0).(func([]string) []*models.Performer); ok {
r0 = rf(words)
if rf, ok := ret.Get(0).(func(context.Context, []string) []*models.Performer); ok {
r0 = rf(ctx, words)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*models.Performer)
@@ -448,8 +450,8 @@ func (_m *PerformerReaderWriter) QueryForAutoTag(words []string) ([]*models.Perf
}
var r1 error
if rf, ok := ret.Get(1).(func([]string) error); ok {
r1 = rf(words)
if rf, ok := ret.Get(1).(func(context.Context, []string) error); ok {
r1 = rf(ctx, words)
} else {
r1 = ret.Error(1)
}
@@ -457,13 +459,13 @@ func (_m *PerformerReaderWriter) QueryForAutoTag(words []string) ([]*models.Perf
return r0, r1
}
// Update provides a mock function with given fields: updatedPerformer
func (_m *PerformerReaderWriter) Update(updatedPerformer models.PerformerPartial) (*models.Performer, error) {
ret := _m.Called(updatedPerformer)
// Update provides a mock function with given fields: ctx, updatedPerformer
func (_m *PerformerReaderWriter) Update(ctx context.Context, updatedPerformer models.PerformerPartial) (*models.Performer, error) {
ret := _m.Called(ctx, updatedPerformer)
var r0 *models.Performer
if rf, ok := ret.Get(0).(func(models.PerformerPartial) *models.Performer); ok {
r0 = rf(updatedPerformer)
if rf, ok := ret.Get(0).(func(context.Context, models.PerformerPartial) *models.Performer); ok {
r0 = rf(ctx, updatedPerformer)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*models.Performer)
@@ -471,8 +473,8 @@ func (_m *PerformerReaderWriter) Update(updatedPerformer models.PerformerPartial
}
var r1 error
if rf, ok := ret.Get(1).(func(models.PerformerPartial) error); ok {
r1 = rf(updatedPerformer)
if rf, ok := ret.Get(1).(func(context.Context, models.PerformerPartial) error); ok {
r1 = rf(ctx, updatedPerformer)
} else {
r1 = ret.Error(1)
}
@@ -480,13 +482,13 @@ func (_m *PerformerReaderWriter) Update(updatedPerformer models.PerformerPartial
return r0, r1
}
// UpdateFull provides a mock function with given fields: updatedPerformer
func (_m *PerformerReaderWriter) UpdateFull(updatedPerformer models.Performer) (*models.Performer, error) {
ret := _m.Called(updatedPerformer)
// UpdateFull provides a mock function with given fields: ctx, updatedPerformer
func (_m *PerformerReaderWriter) UpdateFull(ctx context.Context, updatedPerformer models.Performer) (*models.Performer, error) {
ret := _m.Called(ctx, updatedPerformer)
var r0 *models.Performer
if rf, ok := ret.Get(0).(func(models.Performer) *models.Performer); ok {
r0 = rf(updatedPerformer)
if rf, ok := ret.Get(0).(func(context.Context, models.Performer) *models.Performer); ok {
r0 = rf(ctx, updatedPerformer)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*models.Performer)
@@ -494,8 +496,8 @@ func (_m *PerformerReaderWriter) UpdateFull(updatedPerformer models.Performer) (
}
var r1 error
if rf, ok := ret.Get(1).(func(models.Performer) error); ok {
r1 = rf(updatedPerformer)
if rf, ok := ret.Get(1).(func(context.Context, models.Performer) error); ok {
r1 = rf(ctx, updatedPerformer)
} else {
r1 = ret.Error(1)
}
@@ -503,13 +505,13 @@ func (_m *PerformerReaderWriter) UpdateFull(updatedPerformer models.Performer) (
return r0, r1
}
// UpdateImage provides a mock function with given fields: performerID, image
func (_m *PerformerReaderWriter) UpdateImage(performerID int, image []byte) error {
ret := _m.Called(performerID, image)
// UpdateImage provides a mock function with given fields: ctx, performerID, image
func (_m *PerformerReaderWriter) UpdateImage(ctx context.Context, performerID int, image []byte) error {
ret := _m.Called(ctx, performerID, image)
var r0 error
if rf, ok := ret.Get(0).(func(int, []byte) error); ok {
r0 = rf(performerID, image)
if rf, ok := ret.Get(0).(func(context.Context, int, []byte) error); ok {
r0 = rf(ctx, performerID, image)
} else {
r0 = ret.Error(0)
}
@@ -517,13 +519,13 @@ func (_m *PerformerReaderWriter) UpdateImage(performerID int, image []byte) erro
return r0
}
// UpdateStashIDs provides a mock function with given fields: performerID, stashIDs
func (_m *PerformerReaderWriter) UpdateStashIDs(performerID int, stashIDs []models.StashID) error {
ret := _m.Called(performerID, stashIDs)
// UpdateStashIDs provides a mock function with given fields: ctx, performerID, stashIDs
func (_m *PerformerReaderWriter) UpdateStashIDs(ctx context.Context, performerID int, stashIDs []models.StashID) error {
ret := _m.Called(ctx, performerID, stashIDs)
var r0 error
if rf, ok := ret.Get(0).(func(int, []models.StashID) error); ok {
r0 = rf(performerID, stashIDs)
if rf, ok := ret.Get(0).(func(context.Context, int, []models.StashID) error); ok {
r0 = rf(ctx, performerID, stashIDs)
} else {
r0 = ret.Error(0)
}
@@ -531,13 +533,13 @@ func (_m *PerformerReaderWriter) UpdateStashIDs(performerID int, stashIDs []mode
return r0
}
// UpdateTags provides a mock function with given fields: performerID, tagIDs
func (_m *PerformerReaderWriter) UpdateTags(performerID int, tagIDs []int) error {
ret := _m.Called(performerID, tagIDs)
// UpdateTags provides a mock function with given fields: ctx, performerID, tagIDs
func (_m *PerformerReaderWriter) UpdateTags(ctx context.Context, performerID int, tagIDs []int) error {
ret := _m.Called(ctx, performerID, tagIDs)
var r0 error
if rf, ok := ret.Get(0).(func(int, []int) error); ok {
r0 = rf(performerID, tagIDs)
if rf, ok := ret.Get(0).(func(context.Context, int, []int) error); ok {
r0 = rf(ctx, performerID, tagIDs)
} else {
r0 = ret.Error(0)
}