IntenseWebs/mattermost
3
0
Fork 0
mirror of https://github.com/mattermost/mattermost.git synced 2025-02-25 18:55:24 -06:00
Code Issues Packages Projects Releases Wiki Activity

Removing unnecesary SqlStore interface (#16057)

Browse Source
This commit is contained in:
Jesús Espino
2020-11-20 10:51:58 +01:00
committed by GitHub
parent 5b1a716e2c
commit d274cf8afa
39 changed files with 538 additions and 1591 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
store/sqlstore/audit_store.go
View File

@@ -12,13 +12,13 @@ import (
)
type SqlAuditStore struct {
SqlStore
*SqlSupplier
}
func newSqlAuditStore(sqlStore SqlStore) store.AuditStore {
s := &SqlAuditStore{sqlStore}
func newSqlAuditStore(sqlSupplier *SqlSupplier) store.AuditStore {
s := &SqlAuditStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Audit{}, "Audits").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("UserId").SetMaxSize(26)

10
store/sqlstore/bot_store.go
View File

@@ -42,18 +42,18 @@ func botFromModel(b *model.Bot) *bot {
// Bots are otherwise normal users with extra metadata record in the Bots table. The primary key
// for a bot matches the primary key value for corresponding User record.
type SqlBotStore struct {
SqlStore
*SqlSupplier
metrics einterfaces.MetricsInterface
}
// newSqlBotStore creates an instance of SqlBotStore, registering the table schema in question.
func newSqlBotStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) store.BotStore {
func newSqlBotStore(sqlSupplier *SqlSupplier, metrics einterfaces.MetricsInterface) store.BotStore {
us := &SqlBotStore{
SqlStore: sqlStore,
metrics: metrics,
SqlSupplier: sqlSupplier,
metrics: metrics,
}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(bot{}, "Bots").SetKeys(false, "UserId")
table.ColMap("UserId").SetMaxSize(26)
table.ColMap("Description").SetMaxSize(1024)

8
store/sqlstore/channel_member_history_store.go
View File

@@ -17,15 +17,15 @@ import (
)
type SqlChannelMemberHistoryStore struct {
SqlStore
*SqlSupplier
}
func newSqlChannelMemberHistoryStore(sqlStore SqlStore) store.ChannelMemberHistoryStore {
func newSqlChannelMemberHistoryStore(sqlSupplier *SqlSupplier) store.ChannelMemberHistoryStore {
s := &SqlChannelMemberHistoryStore{
SqlStore: sqlStore,
SqlSupplier: sqlSupplier,
}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.ChannelMemberHistory{}, "ChannelMemberHistory").SetKeys(false, "ChannelId", "UserId", "JoinTime")
table.ColMap("ChannelId").SetMaxSize(26)
table.ColMap("UserId").SetMaxSize(26)

10
store/sqlstore/channel_store.go
View File

@@ -33,7 +33,7 @@ const (
)
type SqlChannelStore struct {
SqlStore
*SqlSupplier
metrics einterfaces.MetricsInterface
}
@@ -356,13 +356,13 @@ func (s SqlChannelStore) ClearCaches() {
}
}
func newSqlChannelStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) store.ChannelStore {
func newSqlChannelStore(sqlSupplier *SqlSupplier, metrics einterfaces.MetricsInterface) store.ChannelStore {
s := &SqlChannelStore{
SqlStore: sqlStore,
metrics: metrics,
SqlSupplier: sqlSupplier,
metrics: metrics,
}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Channel{}, "Channels").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("TeamId").SetMaxSize(26)

2
store/sqlstore/channel_store_test.go
View File

@@ -28,7 +28,7 @@ func TestChannelSearchQuerySQLInjection(t *testing.T) {
for _, st := range storeTypes {
t.Run(st.Name, func(t *testing.T) {
s := &SqlChannelStore{
SqlStore: st.SqlSupplier,
SqlSupplier: st.SqlSupplier,
}
opts := store.ChannelSearchOpts{}

8
store/sqlstore/cluster_discovery_store.go
View File

@@ -12,13 +12,13 @@ import (
)
type sqlClusterDiscoveryStore struct {
SqlStore
*SqlSupplier
}
func newSqlClusterDiscoveryStore(sqlStore SqlStore) store.ClusterDiscoveryStore {
s := &sqlClusterDiscoveryStore{sqlStore}
func newSqlClusterDiscoveryStore(sqlSupplier *SqlSupplier) store.ClusterDiscoveryStore {
s := &sqlClusterDiscoveryStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.ClusterDiscovery{}, "ClusterDiscovery").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("Type").SetMaxSize(64)

8
store/sqlstore/command_store.go
View File

@@ -14,18 +14,18 @@ import (
)
type SqlCommandStore struct {
SqlStore
*SqlSupplier
commandsQuery sq.SelectBuilder
}
func newSqlCommandStore(sqlStore SqlStore) store.CommandStore {
s := &SqlCommandStore{SqlStore: sqlStore}
func newSqlCommandStore(sqlSupplier *SqlSupplier) store.CommandStore {
s := &SqlCommandStore{SqlSupplier: sqlSupplier}
s.commandsQuery = s.getQueryBuilder().
Select("*").
From("Commands")
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
tableo := db.AddTableWithName(model.Command{}, "Commands").SetKeys(false, "Id")
tableo.ColMap("Id").SetMaxSize(26)
tableo.ColMap("Token").SetMaxSize(26)

8
store/sqlstore/command_webhook_store.go
View File

@@ -16,13 +16,13 @@ import (
)
type SqlCommandWebhookStore struct {
SqlStore
*SqlSupplier
}
func newSqlCommandWebhookStore(sqlStore SqlStore) store.CommandWebhookStore {
s := &SqlCommandWebhookStore{sqlStore}
func newSqlCommandWebhookStore(sqlSupplier *SqlSupplier) store.CommandWebhookStore {
s := &SqlCommandWebhookStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
tablec := db.AddTableWithName(model.CommandWebhook{}, "CommandWebhooks").SetKeys(false, "Id")
tablec.ColMap("Id").SetMaxSize(26)
tablec.ColMap("CommandId").SetMaxSize(26)

8
store/sqlstore/compliance_store.go
View File

@@ -14,13 +14,13 @@ import (
)
type SqlComplianceStore struct {
SqlStore
*SqlSupplier
}
func newSqlComplianceStore(sqlStore SqlStore) store.ComplianceStore {
s := &SqlComplianceStore{sqlStore}
func newSqlComplianceStore(sqlSupplier *SqlSupplier) store.ComplianceStore {
s := &SqlComplianceStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Compliance{}, "Compliances").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("UserId").SetMaxSize(26)

10
store/sqlstore/emoji_store.go
View File

@@ -15,17 +15,17 @@ import (
)
type SqlEmojiStore struct {
SqlStore
*SqlSupplier
metrics einterfaces.MetricsInterface
}
func newSqlEmojiStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) store.EmojiStore {
func newSqlEmojiStore(sqlSupplier *SqlSupplier, metrics einterfaces.MetricsInterface) store.EmojiStore {
s := &SqlEmojiStore{
SqlStore: sqlStore,
metrics: metrics,
SqlSupplier: sqlSupplier,
metrics: metrics,
}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Emoji{}, "Emoji").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("CreatorId").SetMaxSize(26)

10
store/sqlstore/file_info_store.go
View File

@@ -16,7 +16,7 @@ import (
)
type SqlFileInfoStore struct {
SqlStore
*SqlSupplier
metrics einterfaces.MetricsInterface
queryFields []string
}
@@ -24,10 +24,10 @@ type SqlFileInfoStore struct {
func (fs SqlFileInfoStore) ClearCaches() {
}
func newSqlFileInfoStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) store.FileInfoStore {
func newSqlFileInfoStore(sqlSupplier *SqlSupplier, metrics einterfaces.MetricsInterface) store.FileInfoStore {
s := &SqlFileInfoStore{
SqlStore: sqlStore,
metrics: metrics,
SqlSupplier: sqlSupplier,
metrics: metrics,
}
s.queryFields = []string{
@@ -51,7 +51,7 @@ func newSqlFileInfoStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface
"Coalesce(FileInfo.Content, '') AS Content",
}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.FileInfo{}, "FileInfo").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("CreatorId").SetMaxSize(26)

8
store/sqlstore/group_store.go
View File

@@ -48,12 +48,12 @@ type groupChannelJoin struct {
}
type SqlGroupStore struct {
SqlStore
*SqlSupplier
}
func newSqlGroupStore(sqlStore SqlStore) store.GroupStore {
s := &SqlGroupStore{SqlStore: sqlStore}
for _, db := range sqlStore.GetAllConns() {
func newSqlGroupStore(sqlSupplier *SqlSupplier) store.GroupStore {
s := &SqlGroupStore{SqlSupplier: sqlSupplier}
for _, db := range sqlSupplier.GetAllConns() {
groups := db.AddTableWithName(model.Group{}, "UserGroups").SetKeys(false, "Id")
groups.ColMap("Id").SetMaxSize(26)
groups.ColMap("Name").SetMaxSize(model.GroupNameMaxLength).SetUnique(true)

8
store/sqlstore/job_store.go
View File

@@ -17,13 +17,13 @@ import (
)
type SqlJobStore struct {
SqlStore
*SqlSupplier
}
func newSqlJobStore(sqlStore SqlStore) store.JobStore {
s := &SqlJobStore{sqlStore}
func newSqlJobStore(sqlSupplier *SqlSupplier) store.JobStore {
s := &SqlJobStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Job{}, "Jobs").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("Type").SetMaxSize(32)

8
store/sqlstore/license_store.go
View File

@@ -15,13 +15,13 @@ import (
// SqlLicenseStore encapsulates the database writes and reads for
// model.LicenseRecord objects.
type SqlLicenseStore struct {
SqlStore
*SqlSupplier
}
func newSqlLicenseStore(sqlStore SqlStore) store.LicenseStore {
ls := &SqlLicenseStore{sqlStore}
func newSqlLicenseStore(sqlSupplier *SqlSupplier) store.LicenseStore {
ls := &SqlLicenseStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.LicenseRecord{}, "Licenses").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("Bytes").SetMaxSize(10000)

8
store/sqlstore/link_metadata_store.go
View File

@@ -14,13 +14,13 @@ import (
)
type SqlLinkMetadataStore struct {
SqlStore
*SqlSupplier
}
func newSqlLinkMetadataStore(sqlStore SqlStore) store.LinkMetadataStore {
s := &SqlLinkMetadataStore{sqlStore}
func newSqlLinkMetadataStore(sqlSupplier *SqlSupplier) store.LinkMetadataStore {
s := &SqlLinkMetadataStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.LinkMetadata{}, "LinkMetadata").SetKeys(false, "Hash")
table.ColMap("URL").SetMaxSize(2048)
table.ColMap("Type").SetMaxSize(16)

8
store/sqlstore/oauth_store.go
View File

@@ -15,13 +15,13 @@ import (
)
type SqlOAuthStore struct {
SqlStore
*SqlSupplier
}
func newSqlOAuthStore(sqlStore SqlStore) store.OAuthStore {
as := &SqlOAuthStore{sqlStore}
func newSqlOAuthStore(sqlSupplier *SqlSupplier) store.OAuthStore {
as := &SqlOAuthStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.OAuthApp{}, "OAuthApps").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("CreatorId").SetMaxSize(26)

8
store/sqlstore/plugin_store.go
View File

@@ -20,13 +20,13 @@ const (
)
type SqlPluginStore struct {
SqlStore
*SqlSupplier
}
func newSqlPluginStore(sqlStore SqlStore) store.PluginStore {
s := &SqlPluginStore{sqlStore}
func newSqlPluginStore(sqlSupplier *SqlSupplier) store.PluginStore {
s := &SqlPluginStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.PluginKeyValue{}, "PluginKeyValueStore").SetKeys(false, "PluginId", "Key")
table.ColMap("PluginId").SetMaxSize(190)
table.ColMap("Key").SetMaxSize(50)

8
store/sqlstore/post_store.go
View File

@@ -24,7 +24,7 @@ import (
)
type SqlPostStore struct {
SqlStore
*SqlSupplier
metrics einterfaces.MetricsInterface
maxPostSizeOnce sync.Once
maxPostSizeCached int
@@ -60,14 +60,14 @@ func postToSlice(post *model.Post) []interface{} {
}
}
func newSqlPostStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) store.PostStore {
func newSqlPostStore(sqlSupplier *SqlSupplier, metrics einterfaces.MetricsInterface) store.PostStore {
s := &SqlPostStore{
SqlStore: sqlStore,
SqlSupplier: sqlSupplier,
metrics: metrics,
maxPostSizeCached: model.POST_MESSAGE_MAX_RUNES_V1,
}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Post{}, "Posts").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("UserId").SetMaxSize(26)

8
store/sqlstore/preference_store.go
View File

@@ -15,13 +15,13 @@ import (
)
type SqlPreferenceStore struct {
SqlStore
*SqlSupplier
}
func newSqlPreferenceStore(sqlStore SqlStore) store.PreferenceStore {
s := &SqlPreferenceStore{sqlStore}
func newSqlPreferenceStore(sqlSupplier *SqlSupplier) store.PreferenceStore {
s := &SqlPreferenceStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Preference{}, "Preferences").SetKeys(false, "UserId", "Category", "Name")
table.ColMap("UserId").SetMaxSize(26)
table.ColMap("Category").SetMaxSize(32)

11
store/sqlstore/product_notices_store.go
View File

@@ -4,22 +4,23 @@
package sqlstore
import (
"time"
sq "github.com/Masterminds/squirrel"
"github.com/mattermost/mattermost-server/v5/model"
"github.com/mattermost/mattermost-server/v5/store"
"time"
"github.com/pkg/errors"
)
type SqlProductNoticesStore struct {
SqlStore
*SqlSupplier
}
func newSqlProductNoticesStore(sqlStore SqlStore) store.ProductNoticesStore {
s := SqlProductNoticesStore{sqlStore}
func newSqlProductNoticesStore(sqlSupplier *SqlSupplier) store.ProductNoticesStore {
s := SqlProductNoticesStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.ProductNoticeViewState{}, "ProductNoticeViewState").SetKeys(false, "UserId", "NoticeId")
table.ColMap("UserId").SetMaxSize(26)
table.ColMap("NoticeId").SetMaxSize(26)

8
store/sqlstore/role_store.go
View File

@@ -17,7 +17,7 @@ import (
)
type SqlRoleStore struct {
SqlStore
*SqlSupplier
}
type Role struct {
@@ -82,10 +82,10 @@ func (role Role) ToModel() *model.Role {
}
}
func newSqlRoleStore(sqlStore SqlStore) store.RoleStore {
s := &SqlRoleStore{sqlStore}
func newSqlRoleStore(sqlSupplier *SqlSupplier) store.RoleStore {
s := &SqlRoleStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(Role{}, "Roles").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("Name").SetMaxSize(64).SetUnique(true)

22
store/sqlstore/scheme_store.go
View File

@@ -16,13 +16,13 @@ import (
)
type SqlSchemeStore struct {
SqlStore
*SqlSupplier
}
func newSqlSchemeStore(sqlStore SqlStore) store.SchemeStore {
s := &SqlSchemeStore{sqlStore}
func newSqlSchemeStore(sqlSupplier *SqlSupplier) store.SchemeStore {
s := &SqlSchemeStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Scheme{}, "Schemes").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("Name").SetMaxSize(model.SCHEME_NAME_MAX_LENGTH).SetUnique(true)
@@ -85,7 +85,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
// Fetch the default system scheme roles to populate default permissions.
defaultRoleNames := []string{model.TEAM_ADMIN_ROLE_ID, model.TEAM_USER_ROLE_ID, model.TEAM_GUEST_ROLE_ID, model.CHANNEL_ADMIN_ROLE_ID, model.CHANNEL_USER_ROLE_ID, model.CHANNEL_GUEST_ROLE_ID}
defaultRoles := make(map[string]*model.Role)
roles, appErr := s.SqlStore.Role().GetByNames(defaultRoleNames)
roles, appErr := s.SqlSupplier.Role().GetByNames(defaultRoleNames)
if appErr != nil {
return nil, appErr
}
@@ -121,7 +121,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
SchemeManaged: true,
}
savedRole, err := s.SqlStore.Role().(*SqlRoleStore).createRole(teamAdminRole, transaction)
savedRole, err := s.SqlSupplier.Role().(*SqlRoleStore).createRole(teamAdminRole, transaction)
if err != nil {
return nil, err
}
@@ -135,7 +135,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
SchemeManaged: true,
}
savedRole, err = s.SqlStore.Role().(*SqlRoleStore).createRole(teamUserRole, transaction)
savedRole, err = s.SqlSupplier.Role().(*SqlRoleStore).createRole(teamUserRole, transaction)
if err != nil {
return nil, err
}
@@ -149,7 +149,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
SchemeManaged: true,
}
savedRole, err = s.SqlStore.Role().(*SqlRoleStore).createRole(teamGuestRole, transaction)
savedRole, err = s.SqlSupplier.Role().(*SqlRoleStore).createRole(teamGuestRole, transaction)
if err != nil {
return nil, err
}
@@ -169,7 +169,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
channelAdminRole.Permissions = []string{}
}
savedRole, err := s.SqlStore.Role().(*SqlRoleStore).createRole(channelAdminRole, transaction)
savedRole, err := s.SqlSupplier.Role().(*SqlRoleStore).createRole(channelAdminRole, transaction)
if err != nil {
return nil, err
}
@@ -187,7 +187,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
channelUserRole.Permissions = filterModerated(channelUserRole.Permissions)
}
savedRole, err = s.SqlStore.Role().(*SqlRoleStore).createRole(channelUserRole, transaction)
savedRole, err = s.SqlSupplier.Role().(*SqlRoleStore).createRole(channelUserRole, transaction)
if err != nil {
return nil, err
}
@@ -205,7 +205,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
channelGuestRole.Permissions = filterModerated(channelGuestRole.Permissions)
}
savedRole, err = s.SqlStore.Role().(*SqlRoleStore).createRole(channelGuestRole, transaction)
savedRole, err = s.SqlSupplier.Role().(*SqlRoleStore).createRole(channelGuestRole, transaction)
if err != nil {
return nil, err
}

8
store/sqlstore/session_store.go
View File

@@ -21,13 +21,13 @@ const (
)
type SqlSessionStore struct {
SqlStore
*SqlSupplier
}
func newSqlSessionStore(sqlStore SqlStore) store.SessionStore {
us := &SqlSessionStore{sqlStore}
func newSqlSessionStore(sqlSupplier *SqlSupplier) store.SessionStore {
us := &SqlSessionStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Session{}, "Sessions").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("Token").SetMaxSize(26)

8
store/sqlstore/status_store.go
View File

@@ -16,13 +16,13 @@ import (
)
type SqlStatusStore struct {
SqlStore
*SqlSupplier
}
func newSqlStatusStore(sqlStore SqlStore) store.StatusStore {
s := &SqlStatusStore{sqlStore}
func newSqlStatusStore(sqlSupplier *SqlSupplier) store.StatusStore {
s := &SqlStatusStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Status{}, "Status").SetKeys(false, "UserId")
table.ColMap("UserId").SetMaxSize(26)
table.ColMap("Status").SetMaxSize(32)

98
store/sqlstore/store.go
View File

@@ -4,104 +4,6 @@
package sqlstore
import (
sq "github.com/Masterminds/squirrel"
_ "github.com/go-sql-driver/mysql"
_ "github.com/lib/pq"
"github.com/mattermost/gorp"
"github.com/mattermost/mattermost-server/v5/store"
)
/*type SqlStore struct {
master *gorp.DbMap
replicas []*gorp.DbMap
searchReplicas []*gorp.DbMap
team TeamStore
channel ChannelStore
post PostStore
user UserStore
audit AuditStore
compliance ComplianceStore
session SessionStore
oauth OAuthStore
system SystemStore
webhook WebhookStore
command CommandStore
preference PreferenceStore
license LicenseStore
token TokenStore
emoji EmojiStore
status StatusStore
fileInfo FileInfoStore
reaction ReactionStore
jobStatus JobStatusStore
SchemaVersion string
rrCounter int64
srCounter int64
}*/
type SqlStore interface {
DriverName() string
GetCurrentSchemaVersion() string
GetMaster() *gorp.DbMap
GetSearchReplica() *gorp.DbMap
GetReplica() *gorp.DbMap
GetDbVersion() (string, error)
TotalMasterDbConnections() int
TotalReadDbConnections() int
TotalSearchDbConnections() int
MarkSystemRanUnitTests()
DoesTableExist(tablename string) bool
DoesColumnExist(tableName string, columName string) bool
DoesTriggerExist(triggerName string) bool
CreateColumnIfNotExists(tableName string, columnName string, mySqlColType string, postgresColType string, defaultValue string) bool
CreateColumnIfNotExistsNoDefault(tableName string, columnName string, mySqlColType string, postgresColType string) bool
RemoveColumnIfExists(tableName string, columnName string) bool
RemoveTableIfExists(tableName string) bool
RenameColumnIfExists(tableName string, oldColumnName string, newColumnName string, colType string) bool
GetMaxLengthOfColumnIfExists(tableName string, columnName string) string
AlterColumnTypeIfExists(tableName string, columnName string, mySqlColType string, postgresColType string) bool
AlterColumnDefaultIfExists(tableName string, columnName string, mySqlColDefault *string, postgresColDefault *string) bool
AlterPrimaryKey(tableName string, columnNames []string) bool
CreateUniqueIndexIfNotExists(indexName string, tableName string, columnName string) bool
CreateIndexIfNotExists(indexName string, tableName string, columnName string) bool
CreateCompositeIndexIfNotExists(indexName string, tableName string, columnNames []string) bool
CreateUniqueCompositeIndexIfNotExists(indexName string, tableName string, columnNames []string) bool
CreateFullTextIndexIfNotExists(indexName string, tableName string, columnName string) bool
RemoveIndexIfExists(indexName string, tableName string) bool
GetAllConns() []*gorp.DbMap
Close()
LockToMaster()
UnlockFromMaster()
Team() store.TeamStore
Channel() store.ChannelStore
Post() store.PostStore
Thread() store.ThreadStore
User() store.UserStore
Bot() store.BotStore
Audit() store.AuditStore
ClusterDiscovery() store.ClusterDiscoveryStore
Compliance() store.ComplianceStore
Session() store.SessionStore
OAuth() store.OAuthStore
System() store.SystemStore
Webhook() store.WebhookStore
Command() store.CommandStore
CommandWebhook() store.CommandWebhookStore
Preference() store.PreferenceStore
License() store.LicenseStore
Token() store.TokenStore
Emoji() store.EmojiStore
Status() store.StatusStore
FileInfo() store.FileInfoStore
UploadSession() store.UploadSessionStore
Reaction() store.ReactionStore
Job() store.JobStore
Plugin() store.PluginStore
UserAccessToken() store.UserAccessTokenStore
Role() store.RoleStore
Scheme() store.SchemeStore
TermsOfService() store.TermsOfServiceStore
UserTermsOfService() store.UserTermsOfServiceStore
LinkMetadata() store.LinkMetadataStore
getQueryBuilder() sq.StatementBuilderType
}

8
store/sqlstore/supplier_reactions.go
View File

@@ -13,13 +13,13 @@ import (
)
type SqlReactionStore struct {
SqlStore
*SqlSupplier
}
func newSqlReactionStore(sqlStore SqlStore) store.ReactionStore {
s := &SqlReactionStore{sqlStore}
func newSqlReactionStore(sqlSupplier *SqlSupplier) store.ReactionStore {
s := &SqlReactionStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Reaction{}, "Reactions").SetKeys(false, "PostId", "UserId", "EmojiName")
table.ColMap("UserId").SetMaxSize(26)
table.ColMap("PostId").SetMaxSize(26)

8
store/sqlstore/system_store.go
View File

@@ -18,13 +18,13 @@ import (
)
type SqlSystemStore struct {
SqlStore
*SqlSupplier
}
func newSqlSystemStore(sqlStore SqlStore) store.SystemStore {
s := &SqlSystemStore{sqlStore}
func newSqlSystemStore(sqlSupplier *SqlSupplier) store.SystemStore {
s := &SqlSystemStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.System{}, "Systems").SetKeys(false, "Name")
table.ColMap("Name").SetMaxSize(64)
table.ColMap("Value").SetMaxSize(1024)

8
store/sqlstore/team_store.go
View File

@@ -23,7 +23,7 @@ const (
)
type SqlTeamStore struct {
SqlStore
*SqlSupplier
teamsQuery sq.SelectBuilder
}
@@ -202,16 +202,16 @@ func (db teamMemberWithSchemeRolesList) ToModel() []*model.TeamMember {
return tms
}
func newSqlTeamStore(sqlStore SqlStore) store.TeamStore {
func newSqlTeamStore(sqlSupplier *SqlSupplier) store.TeamStore {
s := &SqlTeamStore{
SqlStore: sqlStore,
SqlSupplier: sqlSupplier,
}
s.teamsQuery = s.getQueryBuilder().
Select("Teams.*").
From("Teams")
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Team{}, "Teams").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("DisplayName").SetMaxSize(64)

8
store/sqlstore/terms_of_service_store.go
View File

@@ -14,14 +14,14 @@ import (
)
type SqlTermsOfServiceStore struct {
SqlStore
*SqlSupplier
metrics einterfaces.MetricsInterface
}
func newSqlTermsOfServiceStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) store.TermsOfServiceStore {
s := SqlTermsOfServiceStore{sqlStore, metrics}
func newSqlTermsOfServiceStore(sqlSupplier *SqlSupplier, metrics einterfaces.MetricsInterface) store.TermsOfServiceStore {
s := SqlTermsOfServiceStore{sqlSupplier, metrics}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.TermsOfService{}, "TermsOfService").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("UserId").SetMaxSize(26)

8
store/sqlstore/thread_store.go
View File

@@ -16,18 +16,18 @@ import (
)
type SqlThreadStore struct {
SqlStore
*SqlSupplier
}
func (s *SqlThreadStore) ClearCaches() {
}
func newSqlThreadStore(sqlStore SqlStore) store.ThreadStore {
func newSqlThreadStore(sqlSupplier *SqlSupplier) store.ThreadStore {
s := &SqlThreadStore{
SqlStore: sqlStore,
SqlSupplier: sqlSupplier,
}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
tableThreads := db.AddTableWithName(model.Thread{}, "Threads").SetKeys(false, "PostId")
tableThreads.ColMap("PostId").SetMaxSize(26)
tableThreads.ColMap("ChannelId").SetMaxSize(26)

8
store/sqlstore/tokens_store.go
View File

@@ -15,13 +15,13 @@ import (
)
type SqlTokenStore struct {
SqlStore
*SqlSupplier
}
func newSqlTokenStore(sqlStore SqlStore) store.TokenStore {
s := &SqlTokenStore{sqlStore}
func newSqlTokenStore(sqlSupplier *SqlSupplier) store.TokenStore {
s := &SqlTokenStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.Token{}, "Tokens").SetKeys(false, "Token")
table.ColMap("Token").SetMaxSize(64)
table.ColMap("Type").SetMaxSize(64)

768
store/sqlstore/upgrade.go
View File

File diff suppressed because it is too large Load Diff

4
store/sqlstore/upgrade_test.go
View File

@@ -12,7 +12,7 @@ import (
func TestStoreUpgrade(t *testing.T) {
StoreTest(t, func(t *testing.T, ss store.Store) {
sqlStore := ss.(SqlStore)
sqlStore := ss.(*SqlSupplier)
t.Run("invalid currentModelVersion", func(t *testing.T) {
err := upgradeDatabase(sqlStore, "notaversion")
@@ -81,7 +81,7 @@ func TestStoreUpgrade(t *testing.T) {
func TestSaveSchemaVersion(t *testing.T) {
StoreTest(t, func(t *testing.T, ss store.Store) {
sqlStore := ss.(SqlStore)
sqlStore := ss.(*SqlSupplier)
t.Run("set earliest version", func(t *testing.T) {
saveSchemaVersion(sqlStore, VERSION_3_0_0)

8
store/sqlstore/upload_session_store.go
View File

@@ -15,14 +15,14 @@ import (
)
type SqlUploadSessionStore struct {
SqlStore
*SqlSupplier
}
func newSqlUploadSessionStore(sqlStore SqlStore) store.UploadSessionStore {
func newSqlUploadSessionStore(sqlSupplier *SqlSupplier) store.UploadSessionStore {
s := &SqlUploadSessionStore{
SqlStore: sqlStore,
SqlSupplier: sqlSupplier,
}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.UploadSession{}, "UploadSessions").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("Type").SetMaxSize(32)

8
store/sqlstore/user_access_token_store.go
View File

@@ -15,13 +15,13 @@ import (
)
type SqlUserAccessTokenStore struct {
SqlStore
*SqlSupplier
}
func newSqlUserAccessTokenStore(sqlStore SqlStore) store.UserAccessTokenStore {
s := &SqlUserAccessTokenStore{sqlStore}
func newSqlUserAccessTokenStore(sqlSupplier *SqlSupplier) store.UserAccessTokenStore {
s := &SqlUserAccessTokenStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.UserAccessToken{}, "UserAccessTokens").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("Token").SetMaxSize(26).SetUnique(true)

10
store/sqlstore/user_store.go
View File

@@ -31,7 +31,7 @@ var (
)
type SqlUserStore struct {
SqlStore
*SqlSupplier
metrics einterfaces.MetricsInterface
// usersQuery is a starting point for all queries that return one or more Users.
@@ -42,10 +42,10 @@ func (us SqlUserStore) ClearCaches() {}
func (us SqlUserStore) InvalidateProfileCacheForUser(userId string) {}
func newSqlUserStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) store.UserStore {
func newSqlUserStore(sqlSupplier *SqlSupplier, metrics einterfaces.MetricsInterface) store.UserStore {
us := &SqlUserStore{
SqlStore: sqlStore,
metrics: metrics,
SqlSupplier: sqlSupplier,
metrics: metrics,
}
// note: we are providing field names explicitly here to maintain order of columns (needed when using raw queries)
@@ -55,7 +55,7 @@ func newSqlUserStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) st
From("Users u").
LeftJoin("Bots b ON ( b.UserId = u.Id )")
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.User{}, "Users").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("Username").SetMaxSize(64).SetUnique(true)

8
store/sqlstore/user_terms_of_service.go
View File

@@ -13,13 +13,13 @@ import (
)
type SqlUserTermsOfServiceStore struct {
SqlStore
*SqlSupplier
}
func newSqlUserTermsOfServiceStore(sqlStore SqlStore) store.UserTermsOfServiceStore {
s := SqlUserTermsOfServiceStore{sqlStore}
func newSqlUserTermsOfServiceStore(sqlSupplier *SqlSupplier) store.UserTermsOfServiceStore {
s := SqlUserTermsOfServiceStore{sqlSupplier}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.UserTermsOfService{}, "UserTermsOfService").SetKeys(false, "UserId")
table.ColMap("UserId").SetMaxSize(26)
table.ColMap("TermsOfServiceId").SetMaxSize(26)

10
store/sqlstore/webhook_store.go
View File

@@ -15,20 +15,20 @@ import (
)
type SqlWebhookStore struct {
SqlStore
*SqlSupplier
metrics einterfaces.MetricsInterface
}
func (s SqlWebhookStore) ClearCaches() {
}
func newSqlWebhookStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) store.WebhookStore {
func newSqlWebhookStore(sqlSupplier *SqlSupplier, metrics einterfaces.MetricsInterface) store.WebhookStore {
s := &SqlWebhookStore{
SqlStore: sqlStore,
metrics: metrics,
SqlSupplier: sqlSupplier,
metrics: metrics,
}
for _, db := range sqlStore.GetAllConns() {
for _, db := range sqlSupplier.GetAllConns() {
table := db.AddTableWithName(model.IncomingWebhook{}, "IncomingWebhooks").SetKeys(false, "Id")
table.ColMap("Id").SetMaxSize(26)
table.ColMap("UserId").SetMaxSize(26)

956
store/storetest/mocks/SqlStore.go
View File

@@ -1,956 +0,0 @@
// Code generated by mockery v1.0.0. DO NOT EDIT.
// Regenerate this file using `make store-mocks`.
package mocks
import (
gorp "github.com/mattermost/gorp"
mock "github.com/stretchr/testify/mock"
squirrel "github.com/Masterminds/squirrel"
store "github.com/mattermost/mattermost-server/v5/store"
)
// SqlStore is an autogenerated mock type for the SqlStore type
type SqlStore struct {
mock.Mock
}
// AlterColumnDefaultIfExists provides a mock function with given fields: tableName, columnName, mySqlColDefault, postgresColDefault
func (_m *SqlStore) AlterColumnDefaultIfExists(tableName string, columnName string, mySqlColDefault *string, postgresColDefault *string) bool {
ret := _m.Called(tableName, columnName, mySqlColDefault, postgresColDefault)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string, *string, *string) bool); ok {
r0 = rf(tableName, columnName, mySqlColDefault, postgresColDefault)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// AlterColumnTypeIfExists provides a mock function with given fields: tableName, columnName, mySqlColType, postgresColType
func (_m *SqlStore) AlterColumnTypeIfExists(tableName string, columnName string, mySqlColType string, postgresColType string) bool {
ret := _m.Called(tableName, columnName, mySqlColType, postgresColType)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string, string, string) bool); ok {
r0 = rf(tableName, columnName, mySqlColType, postgresColType)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// AlterPrimaryKey provides a mock function with given fields: tableName, columnNames
func (_m *SqlStore) AlterPrimaryKey(tableName string, columnNames []string) bool {
ret := _m.Called(tableName, columnNames)
var r0 bool
if rf, ok := ret.Get(0).(func(string, []string) bool); ok {
r0 = rf(tableName, columnNames)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// Audit provides a mock function with given fields:
func (_m *SqlStore) Audit() store.AuditStore {
ret := _m.Called()
var r0 store.AuditStore
if rf, ok := ret.Get(0).(func() store.AuditStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.AuditStore)
}
}
return r0
}
// Bot provides a mock function with given fields:
func (_m *SqlStore) Bot() store.BotStore {
ret := _m.Called()
var r0 store.BotStore
if rf, ok := ret.Get(0).(func() store.BotStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.BotStore)
}
}
return r0
}
// Channel provides a mock function with given fields:
func (_m *SqlStore) Channel() store.ChannelStore {
ret := _m.Called()
var r0 store.ChannelStore
if rf, ok := ret.Get(0).(func() store.ChannelStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.ChannelStore)
}
}
return r0
}
// Close provides a mock function with given fields:
func (_m *SqlStore) Close() {
_m.Called()
}
// ClusterDiscovery provides a mock function with given fields:
func (_m *SqlStore) ClusterDiscovery() store.ClusterDiscoveryStore {
ret := _m.Called()
var r0 store.ClusterDiscoveryStore
if rf, ok := ret.Get(0).(func() store.ClusterDiscoveryStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.ClusterDiscoveryStore)
}
}
return r0
}
// Command provides a mock function with given fields:
func (_m *SqlStore) Command() store.CommandStore {
ret := _m.Called()
var r0 store.CommandStore
if rf, ok := ret.Get(0).(func() store.CommandStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.CommandStore)
}
}
return r0
}
// CommandWebhook provides a mock function with given fields:
func (_m *SqlStore) CommandWebhook() store.CommandWebhookStore {
ret := _m.Called()
var r0 store.CommandWebhookStore
if rf, ok := ret.Get(0).(func() store.CommandWebhookStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.CommandWebhookStore)
}
}
return r0
}
// Compliance provides a mock function with given fields:
func (_m *SqlStore) Compliance() store.ComplianceStore {
ret := _m.Called()
var r0 store.ComplianceStore
if rf, ok := ret.Get(0).(func() store.ComplianceStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.ComplianceStore)
}
}
return r0
}
// CreateColumnIfNotExists provides a mock function with given fields: tableName, columnName, mySqlColType, postgresColType, defaultValue
func (_m *SqlStore) CreateColumnIfNotExists(tableName string, columnName string, mySqlColType string, postgresColType string, defaultValue string) bool {
ret := _m.Called(tableName, columnName, mySqlColType, postgresColType, defaultValue)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string, string, string, string) bool); ok {
r0 = rf(tableName, columnName, mySqlColType, postgresColType, defaultValue)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// CreateColumnIfNotExistsNoDefault provides a mock function with given fields: tableName, columnName, mySqlColType, postgresColType
func (_m *SqlStore) CreateColumnIfNotExistsNoDefault(tableName string, columnName string, mySqlColType string, postgresColType string) bool {
ret := _m.Called(tableName, columnName, mySqlColType, postgresColType)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string, string, string) bool); ok {
r0 = rf(tableName, columnName, mySqlColType, postgresColType)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// CreateCompositeIndexIfNotExists provides a mock function with given fields: indexName, tableName, columnNames
func (_m *SqlStore) CreateCompositeIndexIfNotExists(indexName string, tableName string, columnNames []string) bool {
ret := _m.Called(indexName, tableName, columnNames)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string, []string) bool); ok {
r0 = rf(indexName, tableName, columnNames)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// CreateFullTextIndexIfNotExists provides a mock function with given fields: indexName, tableName, columnName
func (_m *SqlStore) CreateFullTextIndexIfNotExists(indexName string, tableName string, columnName string) bool {
ret := _m.Called(indexName, tableName, columnName)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string, string) bool); ok {
r0 = rf(indexName, tableName, columnName)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// CreateIndexIfNotExists provides a mock function with given fields: indexName, tableName, columnName
func (_m *SqlStore) CreateIndexIfNotExists(indexName string, tableName string, columnName string) bool {
ret := _m.Called(indexName, tableName, columnName)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string, string) bool); ok {
r0 = rf(indexName, tableName, columnName)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// CreateUniqueCompositeIndexIfNotExists provides a mock function with given fields: indexName, tableName, columnNames
func (_m *SqlStore) CreateUniqueCompositeIndexIfNotExists(indexName string, tableName string, columnNames []string) bool {
ret := _m.Called(indexName, tableName, columnNames)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string, []string) bool); ok {
r0 = rf(indexName, tableName, columnNames)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// CreateUniqueIndexIfNotExists provides a mock function with given fields: indexName, tableName, columnName
func (_m *SqlStore) CreateUniqueIndexIfNotExists(indexName string, tableName string, columnName string) bool {
ret := _m.Called(indexName, tableName, columnName)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string, string) bool); ok {
r0 = rf(indexName, tableName, columnName)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// DoesColumnExist provides a mock function with given fields: tableName, columName
func (_m *SqlStore) DoesColumnExist(tableName string, columName string) bool {
ret := _m.Called(tableName, columName)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string) bool); ok {
r0 = rf(tableName, columName)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// DoesTableExist provides a mock function with given fields: tablename
func (_m *SqlStore) DoesTableExist(tablename string) bool {
ret := _m.Called(tablename)
var r0 bool
if rf, ok := ret.Get(0).(func(string) bool); ok {
r0 = rf(tablename)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// DoesTriggerExist provides a mock function with given fields: triggerName
func (_m *SqlStore) DoesTriggerExist(triggerName string) bool {
ret := _m.Called(triggerName)
var r0 bool
if rf, ok := ret.Get(0).(func(string) bool); ok {
r0 = rf(triggerName)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// DriverName provides a mock function with given fields:
func (_m *SqlStore) DriverName() string {
ret := _m.Called()
var r0 string
if rf, ok := ret.Get(0).(func() string); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(string)
}
return r0
}
// Emoji provides a mock function with given fields:
func (_m *SqlStore) Emoji() store.EmojiStore {
ret := _m.Called()
var r0 store.EmojiStore
if rf, ok := ret.Get(0).(func() store.EmojiStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.EmojiStore)
}
}
return r0
}
// FileInfo provides a mock function with given fields:
func (_m *SqlStore) FileInfo() store.FileInfoStore {
ret := _m.Called()
var r0 store.FileInfoStore
if rf, ok := ret.Get(0).(func() store.FileInfoStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.FileInfoStore)
}
}
return r0
}
// GetAllConns provides a mock function with given fields:
func (_m *SqlStore) GetAllConns() []*gorp.DbMap {
ret := _m.Called()
var r0 []*gorp.DbMap
if rf, ok := ret.Get(0).(func() []*gorp.DbMap); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).([]*gorp.DbMap)
}
}
return r0
}
// GetCurrentSchemaVersion provides a mock function with given fields:
func (_m *SqlStore) GetCurrentSchemaVersion() string {
ret := _m.Called()
var r0 string
if rf, ok := ret.Get(0).(func() string); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(string)
}
return r0
}
// GetDbVersion provides a mock function with given fields:
func (_m *SqlStore) GetDbVersion() (string, error) {
ret := _m.Called()
var r0 string
if rf, ok := ret.Get(0).(func() string); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(string)
}
var r1 error
if rf, ok := ret.Get(1).(func() error); ok {
r1 = rf()
} else {
r1 = ret.Error(1)
}
return r0, r1
}
// GetMaster provides a mock function with given fields:
func (_m *SqlStore) GetMaster() *gorp.DbMap {
ret := _m.Called()
var r0 *gorp.DbMap
if rf, ok := ret.Get(0).(func() *gorp.DbMap); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*gorp.DbMap)
}
}
return r0
}
// GetMaxLengthOfColumnIfExists provides a mock function with given fields: tableName, columnName
func (_m *SqlStore) GetMaxLengthOfColumnIfExists(tableName string, columnName string) string {
ret := _m.Called(tableName, columnName)
var r0 string
if rf, ok := ret.Get(0).(func(string, string) string); ok {
r0 = rf(tableName, columnName)
} else {
r0 = ret.Get(0).(string)
}
return r0
}
// GetReplica provides a mock function with given fields:
func (_m *SqlStore) GetReplica() *gorp.DbMap {
ret := _m.Called()
var r0 *gorp.DbMap
if rf, ok := ret.Get(0).(func() *gorp.DbMap); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*gorp.DbMap)
}
}
return r0
}
// GetSearchReplica provides a mock function with given fields:
func (_m *SqlStore) GetSearchReplica() *gorp.DbMap {
ret := _m.Called()
var r0 *gorp.DbMap
if rf, ok := ret.Get(0).(func() *gorp.DbMap); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*gorp.DbMap)
}
}
return r0
}
// Job provides a mock function with given fields:
func (_m *SqlStore) Job() store.JobStore {
ret := _m.Called()
var r0 store.JobStore
if rf, ok := ret.Get(0).(func() store.JobStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.JobStore)
}
}
return r0
}
// License provides a mock function with given fields:
func (_m *SqlStore) License() store.LicenseStore {
ret := _m.Called()
var r0 store.LicenseStore
if rf, ok := ret.Get(0).(func() store.LicenseStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.LicenseStore)
}
}
return r0
}
// LinkMetadata provides a mock function with given fields:
func (_m *SqlStore) LinkMetadata() store.LinkMetadataStore {
ret := _m.Called()
var r0 store.LinkMetadataStore
if rf, ok := ret.Get(0).(func() store.LinkMetadataStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.LinkMetadataStore)
}
}
return r0
}
// LockToMaster provides a mock function with given fields:
func (_m *SqlStore) LockToMaster() {
_m.Called()
}
// MarkSystemRanUnitTests provides a mock function with given fields:
func (_m *SqlStore) MarkSystemRanUnitTests() {
_m.Called()
}
// OAuth provides a mock function with given fields:
func (_m *SqlStore) OAuth() store.OAuthStore {
ret := _m.Called()
var r0 store.OAuthStore
if rf, ok := ret.Get(0).(func() store.OAuthStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.OAuthStore)
}
}
return r0
}
// Plugin provides a mock function with given fields:
func (_m *SqlStore) Plugin() store.PluginStore {
ret := _m.Called()
var r0 store.PluginStore
if rf, ok := ret.Get(0).(func() store.PluginStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.PluginStore)
}
}
return r0
}
// Post provides a mock function with given fields:
func (_m *SqlStore) Post() store.PostStore {
ret := _m.Called()
var r0 store.PostStore
if rf, ok := ret.Get(0).(func() store.PostStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.PostStore)
}
}
return r0
}
// Preference provides a mock function with given fields:
func (_m *SqlStore) Preference() store.PreferenceStore {
ret := _m.Called()
var r0 store.PreferenceStore
if rf, ok := ret.Get(0).(func() store.PreferenceStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.PreferenceStore)
}
}
return r0
}
// Reaction provides a mock function with given fields:
func (_m *SqlStore) Reaction() store.ReactionStore {
ret := _m.Called()
var r0 store.ReactionStore
if rf, ok := ret.Get(0).(func() store.ReactionStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.ReactionStore)
}
}
return r0
}
// RemoveColumnIfExists provides a mock function with given fields: tableName, columnName
func (_m *SqlStore) RemoveColumnIfExists(tableName string, columnName string) bool {
ret := _m.Called(tableName, columnName)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string) bool); ok {
r0 = rf(tableName, columnName)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// RemoveIndexIfExists provides a mock function with given fields: indexName, tableName
func (_m *SqlStore) RemoveIndexIfExists(indexName string, tableName string) bool {
ret := _m.Called(indexName, tableName)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string) bool); ok {
r0 = rf(indexName, tableName)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// RemoveTableIfExists provides a mock function with given fields: tableName
func (_m *SqlStore) RemoveTableIfExists(tableName string) bool {
ret := _m.Called(tableName)
var r0 bool
if rf, ok := ret.Get(0).(func(string) bool); ok {
r0 = rf(tableName)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// RenameColumnIfExists provides a mock function with given fields: tableName, oldColumnName, newColumnName, colType
func (_m *SqlStore) RenameColumnIfExists(tableName string, oldColumnName string, newColumnName string, colType string) bool {
ret := _m.Called(tableName, oldColumnName, newColumnName, colType)
var r0 bool
if rf, ok := ret.Get(0).(func(string, string, string, string) bool); ok {
r0 = rf(tableName, oldColumnName, newColumnName, colType)
} else {
r0 = ret.Get(0).(bool)
}
return r0
}
// Role provides a mock function with given fields:
func (_m *SqlStore) Role() store.RoleStore {
ret := _m.Called()
var r0 store.RoleStore
if rf, ok := ret.Get(0).(func() store.RoleStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.RoleStore)
}
}
return r0
}
// Scheme provides a mock function with given fields:
func (_m *SqlStore) Scheme() store.SchemeStore {
ret := _m.Called()
var r0 store.SchemeStore
if rf, ok := ret.Get(0).(func() store.SchemeStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.SchemeStore)
}
}
return r0
}
// Session provides a mock function with given fields:
func (_m *SqlStore) Session() store.SessionStore {
ret := _m.Called()
var r0 store.SessionStore
if rf, ok := ret.Get(0).(func() store.SessionStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.SessionStore)
}
}
return r0
}
// Status provides a mock function with given fields:
func (_m *SqlStore) Status() store.StatusStore {
ret := _m.Called()
var r0 store.StatusStore
if rf, ok := ret.Get(0).(func() store.StatusStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StatusStore)
}
}
return r0
}
// System provides a mock function with given fields:
func (_m *SqlStore) System() store.SystemStore {
ret := _m.Called()
var r0 store.SystemStore
if rf, ok := ret.Get(0).(func() store.SystemStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.SystemStore)
}
}
return r0
}
// Team provides a mock function with given fields:
func (_m *SqlStore) Team() store.TeamStore {
ret := _m.Called()
var r0 store.TeamStore
if rf, ok := ret.Get(0).(func() store.TeamStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.TeamStore)
}
}
return r0
}
// TermsOfService provides a mock function with given fields:
func (_m *SqlStore) TermsOfService() store.TermsOfServiceStore {
ret := _m.Called()
var r0 store.TermsOfServiceStore
if rf, ok := ret.Get(0).(func() store.TermsOfServiceStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.TermsOfServiceStore)
}
}
return r0
}
// Thread provides a mock function with given fields:
func (_m *SqlStore) Thread() store.ThreadStore {
ret := _m.Called()
var r0 store.ThreadStore
if rf, ok := ret.Get(0).(func() store.ThreadStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.ThreadStore)
}
}
return r0
}
// Token provides a mock function with given fields:
func (_m *SqlStore) Token() store.TokenStore {
ret := _m.Called()
var r0 store.TokenStore
if rf, ok := ret.Get(0).(func() store.TokenStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.TokenStore)
}
}
return r0
}
// TotalMasterDbConnections provides a mock function with given fields:
func (_m *SqlStore) TotalMasterDbConnections() int {
ret := _m.Called()
var r0 int
if rf, ok := ret.Get(0).(func() int); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(int)
}
return r0
}
// TotalReadDbConnections provides a mock function with given fields:
func (_m *SqlStore) TotalReadDbConnections() int {
ret := _m.Called()
var r0 int
if rf, ok := ret.Get(0).(func() int); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(int)
}
return r0
}
// TotalSearchDbConnections provides a mock function with given fields:
func (_m *SqlStore) TotalSearchDbConnections() int {
ret := _m.Called()
var r0 int
if rf, ok := ret.Get(0).(func() int); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(int)
}
return r0
}
// UnlockFromMaster provides a mock function with given fields:
func (_m *SqlStore) UnlockFromMaster() {
_m.Called()
}
// UploadSession provides a mock function with given fields:
func (_m *SqlStore) UploadSession() store.UploadSessionStore {
ret := _m.Called()
var r0 store.UploadSessionStore
if rf, ok := ret.Get(0).(func() store.UploadSessionStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.UploadSessionStore)
}
}
return r0
}
// User provides a mock function with given fields:
func (_m *SqlStore) User() store.UserStore {
ret := _m.Called()
var r0 store.UserStore
if rf, ok := ret.Get(0).(func() store.UserStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.UserStore)
}
}
return r0
}
// UserAccessToken provides a mock function with given fields:
func (_m *SqlStore) UserAccessToken() store.UserAccessTokenStore {
ret := _m.Called()
var r0 store.UserAccessTokenStore
if rf, ok := ret.Get(0).(func() store.UserAccessTokenStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.UserAccessTokenStore)
}
}
return r0
}
// UserTermsOfService provides a mock function with given fields:
func (_m *SqlStore) UserTermsOfService() store.UserTermsOfServiceStore {
ret := _m.Called()
var r0 store.UserTermsOfServiceStore
if rf, ok := ret.Get(0).(func() store.UserTermsOfServiceStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.UserTermsOfServiceStore)
}
}
return r0
}
// Webhook provides a mock function with given fields:
func (_m *SqlStore) Webhook() store.WebhookStore {
ret := _m.Called()
var r0 store.WebhookStore
if rf, ok := ret.Get(0).(func() store.WebhookStore); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.WebhookStore)
}
}
return r0
}
// getQueryBuilder provides a mock function with given fields:
func (_m *SqlStore) getQueryBuilder() squirrel.StatementBuilderType {
ret := _m.Called()
var r0 squirrel.StatementBuilderType
if rf, ok := ret.Get(0).(func() squirrel.StatementBuilderType); ok {
r0 = rf()
} else {
r0 = ret.Get(0).(squirrel.StatementBuilderType)
}
return r0
}