diff options
author | Corey Hulen <corey@hulen.com> | 2016-02-04 13:01:14 -0800 |
---|---|---|
committer | Corey Hulen <corey@hulen.com> | 2016-02-04 13:01:14 -0800 |
commit | af6ade338002a215ff7c7771f7fe6bbbb06f0cd7 (patch) | |
tree | a8e696aaaf8fd4e483d52f356e848aafcf2a6c25 /store/sql_license_store.go | |
parent | ec51c31c325b3dc648a261f0e8634b6d70d7ba73 (diff) | |
parent | e45282deaa1d78d7ff3a125e9fd11e3fdc120b07 (diff) | |
download | chat-af6ade338002a215ff7c7771f7fe6bbbb06f0cd7.tar.gz chat-af6ade338002a215ff7c7771f7fe6bbbb06f0cd7.tar.bz2 chat-af6ade338002a215ff7c7771f7fe6bbbb06f0cd7.zip |
Merge pull request #2073 from mattermost/ee-updates
Some general updates to EE
Diffstat (limited to 'store/sql_license_store.go')
-rw-r--r-- | store/sql_license_store.go | 83 |
1 files changed, 83 insertions, 0 deletions
diff --git a/store/sql_license_store.go b/store/sql_license_store.go new file mode 100644 index 000000000..f5d67bc5d --- /dev/null +++ b/store/sql_license_store.go @@ -0,0 +1,83 @@ +// Copyright (c) 2016 Mattermost, Inc. All Rights Reserved. +// See License.txt for license information. + +package store + +import ( + "github.com/mattermost/platform/model" +) + +type SqlLicenseStore struct { + *SqlStore +} + +func NewSqlLicenseStore(sqlStore *SqlStore) LicenseStore { + ls := &SqlLicenseStore{sqlStore} + + for _, db := range sqlStore.GetAllConns() { + table := db.AddTableWithName(model.LicenseRecord{}, "Licenses").SetKeys(false, "Id") + table.ColMap("Id").SetMaxSize(26) + table.ColMap("Bytes").SetMaxSize(10000) + } + + return ls +} + +func (ls SqlLicenseStore) UpgradeSchemaIfNeeded() { +} + +func (ls SqlLicenseStore) CreateIndexesIfNotExists() { +} + +func (ls SqlLicenseStore) Save(license *model.LicenseRecord) StoreChannel { + + storeChannel := make(StoreChannel) + + go func() { + result := StoreResult{} + + license.PreSave() + if result.Err = license.IsValid(); result.Err != nil { + storeChannel <- result + close(storeChannel) + return + } + + // Only insert if not exists + if err := ls.GetReplica().SelectOne(&model.LicenseRecord{}, "SELECT * FROM Licenses WHERE Id = :Id", map[string]interface{}{"Id": license.Id}); err != nil { + if err := ls.GetMaster().Insert(license); err != nil { + result.Err = model.NewLocAppError("SqlLicenseStore.Save", "store.sql_license.save.app_error", nil, "license_id="+license.Id+", "+err.Error()) + } else { + result.Data = license + } + } + + storeChannel <- result + close(storeChannel) + }() + + return storeChannel +} + +func (ls SqlLicenseStore) Get(id string) StoreChannel { + + storeChannel := make(StoreChannel) + + go func() { + result := StoreResult{} + + if obj, err := ls.GetReplica().Get(model.LicenseRecord{}, id); err != nil { + result.Err = model.NewLocAppError("SqlLicenseStore.Get", "store.sql_license.get.app_error", nil, "license_id="+id+", "+err.Error()) + } else if obj == nil { + result.Err = model.NewLocAppError("SqlLicenseStore.Get", "store.sql_license.get.missing.app_error", nil, "license_id="+id) + } else { + result.Data = obj.(*model.LicenseRecord) + } + + storeChannel <- result + close(storeChannel) + + }() + + return storeChannel +} |