Removing unnecesary SqlStore interface (#16057)
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
5b1a716e2c
Коммит
d274cf8afa
@@ -12,13 +12,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlAuditStore struct {
|
type SqlAuditStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlAuditStore(sqlStore SqlStore) store.AuditStore {
|
func newSqlAuditStore(sqlSupplier *SqlSupplier) store.AuditStore {
|
||||||
s := &SqlAuditStore{sqlStore}
|
s := &SqlAuditStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.Audit{}, "Audits").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.Audit{}, "Audits").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("UserId").SetMaxSize(26)
|
table.ColMap("UserId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -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
|
// 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.
|
// for a bot matches the primary key value for corresponding User record.
|
||||||
type SqlBotStore struct {
|
type SqlBotStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
metrics einterfaces.MetricsInterface
|
metrics einterfaces.MetricsInterface
|
||||||
}
|
}
|
||||||
|
|
||||||
// newSqlBotStore creates an instance of SqlBotStore, registering the table schema in question.
|
// 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{
|
us := &SqlBotStore{
|
||||||
SqlStore: sqlStore,
|
SqlSupplier: sqlSupplier,
|
||||||
metrics: metrics,
|
metrics: metrics,
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(bot{}, "Bots").SetKeys(false, "UserId")
|
table := db.AddTableWithName(bot{}, "Bots").SetKeys(false, "UserId")
|
||||||
table.ColMap("UserId").SetMaxSize(26)
|
table.ColMap("UserId").SetMaxSize(26)
|
||||||
table.ColMap("Description").SetMaxSize(1024)
|
table.ColMap("Description").SetMaxSize(1024)
|
||||||
|
|||||||
@@ -17,15 +17,15 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlChannelMemberHistoryStore struct {
|
type SqlChannelMemberHistoryStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlChannelMemberHistoryStore(sqlStore SqlStore) store.ChannelMemberHistoryStore {
|
func newSqlChannelMemberHistoryStore(sqlSupplier *SqlSupplier) store.ChannelMemberHistoryStore {
|
||||||
s := &SqlChannelMemberHistoryStore{
|
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 := db.AddTableWithName(model.ChannelMemberHistory{}, "ChannelMemberHistory").SetKeys(false, "ChannelId", "UserId", "JoinTime")
|
||||||
table.ColMap("ChannelId").SetMaxSize(26)
|
table.ColMap("ChannelId").SetMaxSize(26)
|
||||||
table.ColMap("UserId").SetMaxSize(26)
|
table.ColMap("UserId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -33,7 +33,7 @@ const (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlChannelStore struct {
|
type SqlChannelStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
metrics einterfaces.MetricsInterface
|
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{
|
s := &SqlChannelStore{
|
||||||
SqlStore: sqlStore,
|
SqlSupplier: sqlSupplier,
|
||||||
metrics: metrics,
|
metrics: metrics,
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.Channel{}, "Channels").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.Channel{}, "Channels").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("TeamId").SetMaxSize(26)
|
table.ColMap("TeamId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -28,7 +28,7 @@ func TestChannelSearchQuerySQLInjection(t *testing.T) {
|
|||||||
for _, st := range storeTypes {
|
for _, st := range storeTypes {
|
||||||
t.Run(st.Name, func(t *testing.T) {
|
t.Run(st.Name, func(t *testing.T) {
|
||||||
s := &SqlChannelStore{
|
s := &SqlChannelStore{
|
||||||
SqlStore: st.SqlSupplier,
|
SqlSupplier: st.SqlSupplier,
|
||||||
}
|
}
|
||||||
|
|
||||||
opts := store.ChannelSearchOpts{}
|
opts := store.ChannelSearchOpts{}
|
||||||
|
|||||||
@@ -12,13 +12,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type sqlClusterDiscoveryStore struct {
|
type sqlClusterDiscoveryStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlClusterDiscoveryStore(sqlStore SqlStore) store.ClusterDiscoveryStore {
|
func newSqlClusterDiscoveryStore(sqlSupplier *SqlSupplier) store.ClusterDiscoveryStore {
|
||||||
s := &sqlClusterDiscoveryStore{sqlStore}
|
s := &sqlClusterDiscoveryStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.ClusterDiscovery{}, "ClusterDiscovery").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.ClusterDiscovery{}, "ClusterDiscovery").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("Type").SetMaxSize(64)
|
table.ColMap("Type").SetMaxSize(64)
|
||||||
|
|||||||
@@ -14,18 +14,18 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlCommandStore struct {
|
type SqlCommandStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
|
|
||||||
commandsQuery sq.SelectBuilder
|
commandsQuery sq.SelectBuilder
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlCommandStore(sqlStore SqlStore) store.CommandStore {
|
func newSqlCommandStore(sqlSupplier *SqlSupplier) store.CommandStore {
|
||||||
s := &SqlCommandStore{SqlStore: sqlStore}
|
s := &SqlCommandStore{SqlSupplier: sqlSupplier}
|
||||||
|
|
||||||
s.commandsQuery = s.getQueryBuilder().
|
s.commandsQuery = s.getQueryBuilder().
|
||||||
Select("*").
|
Select("*").
|
||||||
From("Commands")
|
From("Commands")
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
tableo := db.AddTableWithName(model.Command{}, "Commands").SetKeys(false, "Id")
|
tableo := db.AddTableWithName(model.Command{}, "Commands").SetKeys(false, "Id")
|
||||||
tableo.ColMap("Id").SetMaxSize(26)
|
tableo.ColMap("Id").SetMaxSize(26)
|
||||||
tableo.ColMap("Token").SetMaxSize(26)
|
tableo.ColMap("Token").SetMaxSize(26)
|
||||||
|
|||||||
@@ -16,13 +16,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlCommandWebhookStore struct {
|
type SqlCommandWebhookStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlCommandWebhookStore(sqlStore SqlStore) store.CommandWebhookStore {
|
func newSqlCommandWebhookStore(sqlSupplier *SqlSupplier) store.CommandWebhookStore {
|
||||||
s := &SqlCommandWebhookStore{sqlStore}
|
s := &SqlCommandWebhookStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
tablec := db.AddTableWithName(model.CommandWebhook{}, "CommandWebhooks").SetKeys(false, "Id")
|
tablec := db.AddTableWithName(model.CommandWebhook{}, "CommandWebhooks").SetKeys(false, "Id")
|
||||||
tablec.ColMap("Id").SetMaxSize(26)
|
tablec.ColMap("Id").SetMaxSize(26)
|
||||||
tablec.ColMap("CommandId").SetMaxSize(26)
|
tablec.ColMap("CommandId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -14,13 +14,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlComplianceStore struct {
|
type SqlComplianceStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlComplianceStore(sqlStore SqlStore) store.ComplianceStore {
|
func newSqlComplianceStore(sqlSupplier *SqlSupplier) store.ComplianceStore {
|
||||||
s := &SqlComplianceStore{sqlStore}
|
s := &SqlComplianceStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.Compliance{}, "Compliances").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.Compliance{}, "Compliances").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("UserId").SetMaxSize(26)
|
table.ColMap("UserId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -15,17 +15,17 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlEmojiStore struct {
|
type SqlEmojiStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
metrics einterfaces.MetricsInterface
|
metrics einterfaces.MetricsInterface
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlEmojiStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) store.EmojiStore {
|
func newSqlEmojiStore(sqlSupplier *SqlSupplier, metrics einterfaces.MetricsInterface) store.EmojiStore {
|
||||||
s := &SqlEmojiStore{
|
s := &SqlEmojiStore{
|
||||||
SqlStore: sqlStore,
|
SqlSupplier: sqlSupplier,
|
||||||
metrics: metrics,
|
metrics: metrics,
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.Emoji{}, "Emoji").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.Emoji{}, "Emoji").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("CreatorId").SetMaxSize(26)
|
table.ColMap("CreatorId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -16,7 +16,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlFileInfoStore struct {
|
type SqlFileInfoStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
metrics einterfaces.MetricsInterface
|
metrics einterfaces.MetricsInterface
|
||||||
queryFields []string
|
queryFields []string
|
||||||
}
|
}
|
||||||
@@ -24,10 +24,10 @@ type SqlFileInfoStore struct {
|
|||||||
func (fs SqlFileInfoStore) ClearCaches() {
|
func (fs SqlFileInfoStore) ClearCaches() {
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlFileInfoStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) store.FileInfoStore {
|
func newSqlFileInfoStore(sqlSupplier *SqlSupplier, metrics einterfaces.MetricsInterface) store.FileInfoStore {
|
||||||
s := &SqlFileInfoStore{
|
s := &SqlFileInfoStore{
|
||||||
SqlStore: sqlStore,
|
SqlSupplier: sqlSupplier,
|
||||||
metrics: metrics,
|
metrics: metrics,
|
||||||
}
|
}
|
||||||
|
|
||||||
s.queryFields = []string{
|
s.queryFields = []string{
|
||||||
@@ -51,7 +51,7 @@ func newSqlFileInfoStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface
|
|||||||
"Coalesce(FileInfo.Content, '') AS Content",
|
"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 := db.AddTableWithName(model.FileInfo{}, "FileInfo").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("CreatorId").SetMaxSize(26)
|
table.ColMap("CreatorId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -48,12 +48,12 @@ type groupChannelJoin struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type SqlGroupStore struct {
|
type SqlGroupStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlGroupStore(sqlStore SqlStore) store.GroupStore {
|
func newSqlGroupStore(sqlSupplier *SqlSupplier) store.GroupStore {
|
||||||
s := &SqlGroupStore{SqlStore: sqlStore}
|
s := &SqlGroupStore{SqlSupplier: sqlSupplier}
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
groups := db.AddTableWithName(model.Group{}, "UserGroups").SetKeys(false, "Id")
|
groups := db.AddTableWithName(model.Group{}, "UserGroups").SetKeys(false, "Id")
|
||||||
groups.ColMap("Id").SetMaxSize(26)
|
groups.ColMap("Id").SetMaxSize(26)
|
||||||
groups.ColMap("Name").SetMaxSize(model.GroupNameMaxLength).SetUnique(true)
|
groups.ColMap("Name").SetMaxSize(model.GroupNameMaxLength).SetUnique(true)
|
||||||
|
|||||||
@@ -17,13 +17,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlJobStore struct {
|
type SqlJobStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlJobStore(sqlStore SqlStore) store.JobStore {
|
func newSqlJobStore(sqlSupplier *SqlSupplier) store.JobStore {
|
||||||
s := &SqlJobStore{sqlStore}
|
s := &SqlJobStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.Job{}, "Jobs").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.Job{}, "Jobs").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("Type").SetMaxSize(32)
|
table.ColMap("Type").SetMaxSize(32)
|
||||||
|
|||||||
@@ -15,13 +15,13 @@ import (
|
|||||||
// SqlLicenseStore encapsulates the database writes and reads for
|
// SqlLicenseStore encapsulates the database writes and reads for
|
||||||
// model.LicenseRecord objects.
|
// model.LicenseRecord objects.
|
||||||
type SqlLicenseStore struct {
|
type SqlLicenseStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlLicenseStore(sqlStore SqlStore) store.LicenseStore {
|
func newSqlLicenseStore(sqlSupplier *SqlSupplier) store.LicenseStore {
|
||||||
ls := &SqlLicenseStore{sqlStore}
|
ls := &SqlLicenseStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.LicenseRecord{}, "Licenses").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.LicenseRecord{}, "Licenses").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("Bytes").SetMaxSize(10000)
|
table.ColMap("Bytes").SetMaxSize(10000)
|
||||||
|
|||||||
@@ -14,13 +14,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlLinkMetadataStore struct {
|
type SqlLinkMetadataStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlLinkMetadataStore(sqlStore SqlStore) store.LinkMetadataStore {
|
func newSqlLinkMetadataStore(sqlSupplier *SqlSupplier) store.LinkMetadataStore {
|
||||||
s := &SqlLinkMetadataStore{sqlStore}
|
s := &SqlLinkMetadataStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.LinkMetadata{}, "LinkMetadata").SetKeys(false, "Hash")
|
table := db.AddTableWithName(model.LinkMetadata{}, "LinkMetadata").SetKeys(false, "Hash")
|
||||||
table.ColMap("URL").SetMaxSize(2048)
|
table.ColMap("URL").SetMaxSize(2048)
|
||||||
table.ColMap("Type").SetMaxSize(16)
|
table.ColMap("Type").SetMaxSize(16)
|
||||||
|
|||||||
@@ -15,13 +15,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlOAuthStore struct {
|
type SqlOAuthStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlOAuthStore(sqlStore SqlStore) store.OAuthStore {
|
func newSqlOAuthStore(sqlSupplier *SqlSupplier) store.OAuthStore {
|
||||||
as := &SqlOAuthStore{sqlStore}
|
as := &SqlOAuthStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.OAuthApp{}, "OAuthApps").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.OAuthApp{}, "OAuthApps").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("CreatorId").SetMaxSize(26)
|
table.ColMap("CreatorId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -20,13 +20,13 @@ const (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlPluginStore struct {
|
type SqlPluginStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlPluginStore(sqlStore SqlStore) store.PluginStore {
|
func newSqlPluginStore(sqlSupplier *SqlSupplier) store.PluginStore {
|
||||||
s := &SqlPluginStore{sqlStore}
|
s := &SqlPluginStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.PluginKeyValue{}, "PluginKeyValueStore").SetKeys(false, "PluginId", "Key")
|
table := db.AddTableWithName(model.PluginKeyValue{}, "PluginKeyValueStore").SetKeys(false, "PluginId", "Key")
|
||||||
table.ColMap("PluginId").SetMaxSize(190)
|
table.ColMap("PluginId").SetMaxSize(190)
|
||||||
table.ColMap("Key").SetMaxSize(50)
|
table.ColMap("Key").SetMaxSize(50)
|
||||||
|
|||||||
@@ -24,7 +24,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlPostStore struct {
|
type SqlPostStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
metrics einterfaces.MetricsInterface
|
metrics einterfaces.MetricsInterface
|
||||||
maxPostSizeOnce sync.Once
|
maxPostSizeOnce sync.Once
|
||||||
maxPostSizeCached int
|
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{
|
s := &SqlPostStore{
|
||||||
SqlStore: sqlStore,
|
SqlSupplier: sqlSupplier,
|
||||||
metrics: metrics,
|
metrics: metrics,
|
||||||
maxPostSizeCached: model.POST_MESSAGE_MAX_RUNES_V1,
|
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 := db.AddTableWithName(model.Post{}, "Posts").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("UserId").SetMaxSize(26)
|
table.ColMap("UserId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -15,13 +15,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlPreferenceStore struct {
|
type SqlPreferenceStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlPreferenceStore(sqlStore SqlStore) store.PreferenceStore {
|
func newSqlPreferenceStore(sqlSupplier *SqlSupplier) store.PreferenceStore {
|
||||||
s := &SqlPreferenceStore{sqlStore}
|
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 := db.AddTableWithName(model.Preference{}, "Preferences").SetKeys(false, "UserId", "Category", "Name")
|
||||||
table.ColMap("UserId").SetMaxSize(26)
|
table.ColMap("UserId").SetMaxSize(26)
|
||||||
table.ColMap("Category").SetMaxSize(32)
|
table.ColMap("Category").SetMaxSize(32)
|
||||||
|
|||||||
@@ -4,22 +4,23 @@
|
|||||||
package sqlstore
|
package sqlstore
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
sq "github.com/Masterminds/squirrel"
|
sq "github.com/Masterminds/squirrel"
|
||||||
"github.com/mattermost/mattermost-server/v5/model"
|
"github.com/mattermost/mattermost-server/v5/model"
|
||||||
"github.com/mattermost/mattermost-server/v5/store"
|
"github.com/mattermost/mattermost-server/v5/store"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/pkg/errors"
|
"github.com/pkg/errors"
|
||||||
)
|
)
|
||||||
|
|
||||||
type SqlProductNoticesStore struct {
|
type SqlProductNoticesStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlProductNoticesStore(sqlStore SqlStore) store.ProductNoticesStore {
|
func newSqlProductNoticesStore(sqlSupplier *SqlSupplier) store.ProductNoticesStore {
|
||||||
s := SqlProductNoticesStore{sqlStore}
|
s := SqlProductNoticesStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.ProductNoticeViewState{}, "ProductNoticeViewState").SetKeys(false, "UserId", "NoticeId")
|
table := db.AddTableWithName(model.ProductNoticeViewState{}, "ProductNoticeViewState").SetKeys(false, "UserId", "NoticeId")
|
||||||
table.ColMap("UserId").SetMaxSize(26)
|
table.ColMap("UserId").SetMaxSize(26)
|
||||||
table.ColMap("NoticeId").SetMaxSize(26)
|
table.ColMap("NoticeId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -17,7 +17,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlRoleStore struct {
|
type SqlRoleStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
type Role struct {
|
type Role struct {
|
||||||
@@ -82,10 +82,10 @@ func (role Role) ToModel() *model.Role {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlRoleStore(sqlStore SqlStore) store.RoleStore {
|
func newSqlRoleStore(sqlSupplier *SqlSupplier) store.RoleStore {
|
||||||
s := &SqlRoleStore{sqlStore}
|
s := &SqlRoleStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(Role{}, "Roles").SetKeys(false, "Id")
|
table := db.AddTableWithName(Role{}, "Roles").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("Name").SetMaxSize(64).SetUnique(true)
|
table.ColMap("Name").SetMaxSize(64).SetUnique(true)
|
||||||
|
|||||||
@@ -16,13 +16,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlSchemeStore struct {
|
type SqlSchemeStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlSchemeStore(sqlStore SqlStore) store.SchemeStore {
|
func newSqlSchemeStore(sqlSupplier *SqlSupplier) store.SchemeStore {
|
||||||
s := &SqlSchemeStore{sqlStore}
|
s := &SqlSchemeStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.Scheme{}, "Schemes").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.Scheme{}, "Schemes").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("Name").SetMaxSize(model.SCHEME_NAME_MAX_LENGTH).SetUnique(true)
|
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.
|
// 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}
|
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)
|
defaultRoles := make(map[string]*model.Role)
|
||||||
roles, appErr := s.SqlStore.Role().GetByNames(defaultRoleNames)
|
roles, appErr := s.SqlSupplier.Role().GetByNames(defaultRoleNames)
|
||||||
if appErr != nil {
|
if appErr != nil {
|
||||||
return nil, appErr
|
return nil, appErr
|
||||||
}
|
}
|
||||||
@@ -121,7 +121,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
|
|||||||
SchemeManaged: true,
|
SchemeManaged: true,
|
||||||
}
|
}
|
||||||
|
|
||||||
savedRole, err := s.SqlStore.Role().(*SqlRoleStore).createRole(teamAdminRole, transaction)
|
savedRole, err := s.SqlSupplier.Role().(*SqlRoleStore).createRole(teamAdminRole, transaction)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -135,7 +135,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
|
|||||||
SchemeManaged: true,
|
SchemeManaged: true,
|
||||||
}
|
}
|
||||||
|
|
||||||
savedRole, err = s.SqlStore.Role().(*SqlRoleStore).createRole(teamUserRole, transaction)
|
savedRole, err = s.SqlSupplier.Role().(*SqlRoleStore).createRole(teamUserRole, transaction)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -149,7 +149,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
|
|||||||
SchemeManaged: true,
|
SchemeManaged: true,
|
||||||
}
|
}
|
||||||
|
|
||||||
savedRole, err = s.SqlStore.Role().(*SqlRoleStore).createRole(teamGuestRole, transaction)
|
savedRole, err = s.SqlSupplier.Role().(*SqlRoleStore).createRole(teamGuestRole, transaction)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -169,7 +169,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
|
|||||||
channelAdminRole.Permissions = []string{}
|
channelAdminRole.Permissions = []string{}
|
||||||
}
|
}
|
||||||
|
|
||||||
savedRole, err := s.SqlStore.Role().(*SqlRoleStore).createRole(channelAdminRole, transaction)
|
savedRole, err := s.SqlSupplier.Role().(*SqlRoleStore).createRole(channelAdminRole, transaction)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -187,7 +187,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
|
|||||||
channelUserRole.Permissions = filterModerated(channelUserRole.Permissions)
|
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 {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -205,7 +205,7 @@ func (s *SqlSchemeStore) createScheme(scheme *model.Scheme, transaction *gorp.Tr
|
|||||||
channelGuestRole.Permissions = filterModerated(channelGuestRole.Permissions)
|
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 {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -21,13 +21,13 @@ const (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlSessionStore struct {
|
type SqlSessionStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlSessionStore(sqlStore SqlStore) store.SessionStore {
|
func newSqlSessionStore(sqlSupplier *SqlSupplier) store.SessionStore {
|
||||||
us := &SqlSessionStore{sqlStore}
|
us := &SqlSessionStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.Session{}, "Sessions").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.Session{}, "Sessions").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("Token").SetMaxSize(26)
|
table.ColMap("Token").SetMaxSize(26)
|
||||||
|
|||||||
@@ -16,13 +16,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlStatusStore struct {
|
type SqlStatusStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlStatusStore(sqlStore SqlStore) store.StatusStore {
|
func newSqlStatusStore(sqlSupplier *SqlSupplier) store.StatusStore {
|
||||||
s := &SqlStatusStore{sqlStore}
|
s := &SqlStatusStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.Status{}, "Status").SetKeys(false, "UserId")
|
table := db.AddTableWithName(model.Status{}, "Status").SetKeys(false, "UserId")
|
||||||
table.ColMap("UserId").SetMaxSize(26)
|
table.ColMap("UserId").SetMaxSize(26)
|
||||||
table.ColMap("Status").SetMaxSize(32)
|
table.ColMap("Status").SetMaxSize(32)
|
||||||
|
|||||||
@@ -4,104 +4,6 @@
|
|||||||
package sqlstore
|
package sqlstore
|
||||||
|
|
||||||
import (
|
import (
|
||||||
sq "github.com/Masterminds/squirrel"
|
|
||||||
_ "github.com/go-sql-driver/mysql"
|
_ "github.com/go-sql-driver/mysql"
|
||||||
_ "github.com/lib/pq"
|
_ "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
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -13,13 +13,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlReactionStore struct {
|
type SqlReactionStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlReactionStore(sqlStore SqlStore) store.ReactionStore {
|
func newSqlReactionStore(sqlSupplier *SqlSupplier) store.ReactionStore {
|
||||||
s := &SqlReactionStore{sqlStore}
|
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 := db.AddTableWithName(model.Reaction{}, "Reactions").SetKeys(false, "PostId", "UserId", "EmojiName")
|
||||||
table.ColMap("UserId").SetMaxSize(26)
|
table.ColMap("UserId").SetMaxSize(26)
|
||||||
table.ColMap("PostId").SetMaxSize(26)
|
table.ColMap("PostId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -18,13 +18,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlSystemStore struct {
|
type SqlSystemStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlSystemStore(sqlStore SqlStore) store.SystemStore {
|
func newSqlSystemStore(sqlSupplier *SqlSupplier) store.SystemStore {
|
||||||
s := &SqlSystemStore{sqlStore}
|
s := &SqlSystemStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.System{}, "Systems").SetKeys(false, "Name")
|
table := db.AddTableWithName(model.System{}, "Systems").SetKeys(false, "Name")
|
||||||
table.ColMap("Name").SetMaxSize(64)
|
table.ColMap("Name").SetMaxSize(64)
|
||||||
table.ColMap("Value").SetMaxSize(1024)
|
table.ColMap("Value").SetMaxSize(1024)
|
||||||
|
|||||||
@@ -23,7 +23,7 @@ const (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlTeamStore struct {
|
type SqlTeamStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
|
|
||||||
teamsQuery sq.SelectBuilder
|
teamsQuery sq.SelectBuilder
|
||||||
}
|
}
|
||||||
@@ -202,16 +202,16 @@ func (db teamMemberWithSchemeRolesList) ToModel() []*model.TeamMember {
|
|||||||
return tms
|
return tms
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlTeamStore(sqlStore SqlStore) store.TeamStore {
|
func newSqlTeamStore(sqlSupplier *SqlSupplier) store.TeamStore {
|
||||||
s := &SqlTeamStore{
|
s := &SqlTeamStore{
|
||||||
SqlStore: sqlStore,
|
SqlSupplier: sqlSupplier,
|
||||||
}
|
}
|
||||||
|
|
||||||
s.teamsQuery = s.getQueryBuilder().
|
s.teamsQuery = s.getQueryBuilder().
|
||||||
Select("Teams.*").
|
Select("Teams.*").
|
||||||
From("Teams")
|
From("Teams")
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.Team{}, "Teams").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.Team{}, "Teams").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("DisplayName").SetMaxSize(64)
|
table.ColMap("DisplayName").SetMaxSize(64)
|
||||||
|
|||||||
@@ -14,14 +14,14 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlTermsOfServiceStore struct {
|
type SqlTermsOfServiceStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
metrics einterfaces.MetricsInterface
|
metrics einterfaces.MetricsInterface
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlTermsOfServiceStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) store.TermsOfServiceStore {
|
func newSqlTermsOfServiceStore(sqlSupplier *SqlSupplier, metrics einterfaces.MetricsInterface) store.TermsOfServiceStore {
|
||||||
s := SqlTermsOfServiceStore{sqlStore, metrics}
|
s := SqlTermsOfServiceStore{sqlSupplier, metrics}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.TermsOfService{}, "TermsOfService").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.TermsOfService{}, "TermsOfService").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("UserId").SetMaxSize(26)
|
table.ColMap("UserId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -16,18 +16,18 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlThreadStore struct {
|
type SqlThreadStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *SqlThreadStore) ClearCaches() {
|
func (s *SqlThreadStore) ClearCaches() {
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlThreadStore(sqlStore SqlStore) store.ThreadStore {
|
func newSqlThreadStore(sqlSupplier *SqlSupplier) store.ThreadStore {
|
||||||
s := &SqlThreadStore{
|
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 := db.AddTableWithName(model.Thread{}, "Threads").SetKeys(false, "PostId")
|
||||||
tableThreads.ColMap("PostId").SetMaxSize(26)
|
tableThreads.ColMap("PostId").SetMaxSize(26)
|
||||||
tableThreads.ColMap("ChannelId").SetMaxSize(26)
|
tableThreads.ColMap("ChannelId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -15,13 +15,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlTokenStore struct {
|
type SqlTokenStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlTokenStore(sqlStore SqlStore) store.TokenStore {
|
func newSqlTokenStore(sqlSupplier *SqlSupplier) store.TokenStore {
|
||||||
s := &SqlTokenStore{sqlStore}
|
s := &SqlTokenStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.Token{}, "Tokens").SetKeys(false, "Token")
|
table := db.AddTableWithName(model.Token{}, "Tokens").SetKeys(false, "Token")
|
||||||
table.ColMap("Token").SetMaxSize(64)
|
table.ColMap("Token").SetMaxSize(64)
|
||||||
table.ColMap("Type").SetMaxSize(64)
|
table.ColMap("Type").SetMaxSize(64)
|
||||||
|
|||||||
Разница между файлами не показана из-за своего большого размера
Загрузить разницу
@@ -12,7 +12,7 @@ import (
|
|||||||
|
|
||||||
func TestStoreUpgrade(t *testing.T) {
|
func TestStoreUpgrade(t *testing.T) {
|
||||||
StoreTest(t, func(t *testing.T, ss store.Store) {
|
StoreTest(t, func(t *testing.T, ss store.Store) {
|
||||||
sqlStore := ss.(SqlStore)
|
sqlStore := ss.(*SqlSupplier)
|
||||||
|
|
||||||
t.Run("invalid currentModelVersion", func(t *testing.T) {
|
t.Run("invalid currentModelVersion", func(t *testing.T) {
|
||||||
err := upgradeDatabase(sqlStore, "notaversion")
|
err := upgradeDatabase(sqlStore, "notaversion")
|
||||||
@@ -81,7 +81,7 @@ func TestStoreUpgrade(t *testing.T) {
|
|||||||
|
|
||||||
func TestSaveSchemaVersion(t *testing.T) {
|
func TestSaveSchemaVersion(t *testing.T) {
|
||||||
StoreTest(t, func(t *testing.T, ss store.Store) {
|
StoreTest(t, func(t *testing.T, ss store.Store) {
|
||||||
sqlStore := ss.(SqlStore)
|
sqlStore := ss.(*SqlSupplier)
|
||||||
|
|
||||||
t.Run("set earliest version", func(t *testing.T) {
|
t.Run("set earliest version", func(t *testing.T) {
|
||||||
saveSchemaVersion(sqlStore, VERSION_3_0_0)
|
saveSchemaVersion(sqlStore, VERSION_3_0_0)
|
||||||
|
|||||||
@@ -15,14 +15,14 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlUploadSessionStore struct {
|
type SqlUploadSessionStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlUploadSessionStore(sqlStore SqlStore) store.UploadSessionStore {
|
func newSqlUploadSessionStore(sqlSupplier *SqlSupplier) store.UploadSessionStore {
|
||||||
s := &SqlUploadSessionStore{
|
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 := db.AddTableWithName(model.UploadSession{}, "UploadSessions").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("Type").SetMaxSize(32)
|
table.ColMap("Type").SetMaxSize(32)
|
||||||
|
|||||||
@@ -15,13 +15,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlUserAccessTokenStore struct {
|
type SqlUserAccessTokenStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlUserAccessTokenStore(sqlStore SqlStore) store.UserAccessTokenStore {
|
func newSqlUserAccessTokenStore(sqlSupplier *SqlSupplier) store.UserAccessTokenStore {
|
||||||
s := &SqlUserAccessTokenStore{sqlStore}
|
s := &SqlUserAccessTokenStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.UserAccessToken{}, "UserAccessTokens").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.UserAccessToken{}, "UserAccessTokens").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("Token").SetMaxSize(26).SetUnique(true)
|
table.ColMap("Token").SetMaxSize(26).SetUnique(true)
|
||||||
|
|||||||
@@ -31,7 +31,7 @@ var (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlUserStore struct {
|
type SqlUserStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
metrics einterfaces.MetricsInterface
|
metrics einterfaces.MetricsInterface
|
||||||
|
|
||||||
// usersQuery is a starting point for all queries that return one or more Users.
|
// 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 (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{
|
us := &SqlUserStore{
|
||||||
SqlStore: sqlStore,
|
SqlSupplier: sqlSupplier,
|
||||||
metrics: metrics,
|
metrics: metrics,
|
||||||
}
|
}
|
||||||
|
|
||||||
// note: we are providing field names explicitly here to maintain order of columns (needed when using raw queries)
|
// 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").
|
From("Users u").
|
||||||
LeftJoin("Bots b ON ( b.UserId = u.Id )")
|
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 := db.AddTableWithName(model.User{}, "Users").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("Username").SetMaxSize(64).SetUnique(true)
|
table.ColMap("Username").SetMaxSize(64).SetUnique(true)
|
||||||
|
|||||||
@@ -13,13 +13,13 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlUserTermsOfServiceStore struct {
|
type SqlUserTermsOfServiceStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlUserTermsOfServiceStore(sqlStore SqlStore) store.UserTermsOfServiceStore {
|
func newSqlUserTermsOfServiceStore(sqlSupplier *SqlSupplier) store.UserTermsOfServiceStore {
|
||||||
s := SqlUserTermsOfServiceStore{sqlStore}
|
s := SqlUserTermsOfServiceStore{sqlSupplier}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.UserTermsOfService{}, "UserTermsOfService").SetKeys(false, "UserId")
|
table := db.AddTableWithName(model.UserTermsOfService{}, "UserTermsOfService").SetKeys(false, "UserId")
|
||||||
table.ColMap("UserId").SetMaxSize(26)
|
table.ColMap("UserId").SetMaxSize(26)
|
||||||
table.ColMap("TermsOfServiceId").SetMaxSize(26)
|
table.ColMap("TermsOfServiceId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -15,20 +15,20 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type SqlWebhookStore struct {
|
type SqlWebhookStore struct {
|
||||||
SqlStore
|
*SqlSupplier
|
||||||
metrics einterfaces.MetricsInterface
|
metrics einterfaces.MetricsInterface
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s SqlWebhookStore) ClearCaches() {
|
func (s SqlWebhookStore) ClearCaches() {
|
||||||
}
|
}
|
||||||
|
|
||||||
func newSqlWebhookStore(sqlStore SqlStore, metrics einterfaces.MetricsInterface) store.WebhookStore {
|
func newSqlWebhookStore(sqlSupplier *SqlSupplier, metrics einterfaces.MetricsInterface) store.WebhookStore {
|
||||||
s := &SqlWebhookStore{
|
s := &SqlWebhookStore{
|
||||||
SqlStore: sqlStore,
|
SqlSupplier: sqlSupplier,
|
||||||
metrics: metrics,
|
metrics: metrics,
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, db := range sqlStore.GetAllConns() {
|
for _, db := range sqlSupplier.GetAllConns() {
|
||||||
table := db.AddTableWithName(model.IncomingWebhook{}, "IncomingWebhooks").SetKeys(false, "Id")
|
table := db.AddTableWithName(model.IncomingWebhook{}, "IncomingWebhooks").SetKeys(false, "Id")
|
||||||
table.ColMap("Id").SetMaxSize(26)
|
table.ColMap("Id").SetMaxSize(26)
|
||||||
table.ColMap("UserId").SetMaxSize(26)
|
table.ColMap("UserId").SetMaxSize(26)
|
||||||
|
|||||||
@@ -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
|
|
||||||
}
|
|
||||||
Ссылка в новой задаче
Block a user