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_system_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_system_store.go')
-rw-r--r-- | store/sql_system_store.go | 26 |
1 files changed, 25 insertions, 1 deletions
diff --git a/store/sql_system_store.go b/store/sql_system_store.go index cfd4a670f..f8da06cec 100644 --- a/store/sql_system_store.go +++ b/store/sql_system_store.go @@ -1,4 +1,4 @@ -// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved. +// Copyright (c) 2016 Mattermost, Inc. All Rights Reserved. // See License.txt for license information. package store @@ -47,6 +47,30 @@ func (s SqlSystemStore) Save(system *model.System) StoreChannel { return storeChannel } +func (s SqlSystemStore) SaveOrUpdate(system *model.System) StoreChannel { + + storeChannel := make(StoreChannel) + + go func() { + result := StoreResult{} + + if err := s.GetReplica().SelectOne(&model.System{}, "SELECT * FROM Systems WHERE Name = :Name", map[string]interface{}{"Name": system.Name}); err == nil { + if _, err := s.GetMaster().Update(system); err != nil { + result.Err = model.NewLocAppError("SqlSystemStore.SaveOrUpdate", "store.sql_system.update.app_error", nil, "") + } + } else { + if err := s.GetMaster().Insert(system); err != nil { + result.Err = model.NewLocAppError("SqlSystemStore.SaveOrUpdate", "store.sql_system.save.app_error", nil, "") + } + } + + storeChannel <- result + close(storeChannel) + }() + + return storeChannel +} + func (s SqlSystemStore) Update(system *model.System) StoreChannel { storeChannel := make(StoreChannel) |