diff options
author | Chris <ccbrown112@gmail.com> | 2017-10-06 08:12:10 -0700 |
---|---|---|
committer | Joram Wilander <jwawilander@gmail.com> | 2017-10-06 11:12:10 -0400 |
commit | 363568b4eb3209adb1b88ceb0d8e455e6d4a1073 (patch) | |
tree | 419fb74d8e1a2e91f9f77aa5b873c005bcfbff48 /store/sqlstore/compliance_store.go | |
parent | 12501673d0c70120eebeac633e5072b2e7a2174d (diff) | |
download | chat-363568b4eb3209adb1b88ceb0d8e455e6d4a1073.tar.gz chat-363568b4eb3209adb1b88ceb0d8e455e6d4a1073.tar.bz2 chat-363568b4eb3209adb1b88ceb0d8e455e6d4a1073.zip |
reduce store boiler plate (#7585)
Diffstat (limited to 'store/sqlstore/compliance_store.go')
-rw-r--r-- | store/sqlstore/compliance_store.go | 74 |
1 files changed, 10 insertions, 64 deletions
diff --git a/store/sqlstore/compliance_store.go b/store/sqlstore/compliance_store.go index 95da94673..3d638b1fd 100644 --- a/store/sqlstore/compliance_store.go +++ b/store/sqlstore/compliance_store.go @@ -37,16 +37,9 @@ func (s SqlComplianceStore) CreateIndexesIfNotExists() { } func (s SqlComplianceStore) Save(compliance *model.Compliance) store.StoreChannel { - - storeChannel := make(store.StoreChannel, 1) - - go func() { - result := store.StoreResult{} - + return store.Do(func(result *store.StoreResult) { compliance.PreSave() if result.Err = compliance.IsValid(); result.Err != nil { - storeChannel <- result - close(storeChannel) return } @@ -55,24 +48,12 @@ func (s SqlComplianceStore) Save(compliance *model.Compliance) store.StoreChanne } else { result.Data = compliance } - - storeChannel <- result - close(storeChannel) - }() - - return storeChannel + }) } func (us SqlComplianceStore) Update(compliance *model.Compliance) store.StoreChannel { - - storeChannel := make(store.StoreChannel, 1) - - go func() { - result := store.StoreResult{} - + return store.Do(func(result *store.StoreResult) { if result.Err = compliance.IsValid(); result.Err != nil { - storeChannel <- result - close(storeChannel) return } @@ -81,21 +62,11 @@ func (us SqlComplianceStore) Update(compliance *model.Compliance) store.StoreCha } else { result.Data = compliance } - - storeChannel <- result - close(storeChannel) - }() - - return storeChannel + }) } func (s SqlComplianceStore) GetAll(offset, limit int) store.StoreChannel { - - storeChannel := make(store.StoreChannel, 1) - - go func() { - result := store.StoreResult{} - + return store.Do(func(result *store.StoreResult) { query := "SELECT * FROM Compliances ORDER BY CreateAt DESC LIMIT :Limit OFFSET :Offset" var compliances model.Compliances @@ -104,21 +75,11 @@ func (s SqlComplianceStore) GetAll(offset, limit int) store.StoreChannel { } else { result.Data = compliances } - - storeChannel <- result - close(storeChannel) - }() - - return storeChannel + }) } func (us SqlComplianceStore) Get(id string) store.StoreChannel { - - storeChannel := make(store.StoreChannel, 1) - - go func() { - result := store.StoreResult{} - + return store.Do(func(result *store.StoreResult) { if obj, err := us.GetReplica().Get(model.Compliance{}, id); err != nil { result.Err = model.NewAppError("SqlComplianceStore.Get", "store.sql_compliance.get.finding.app_error", nil, err.Error(), http.StatusInternalServerError) } else if obj == nil { @@ -126,21 +87,11 @@ func (us SqlComplianceStore) Get(id string) store.StoreChannel { } else { result.Data = obj.(*model.Compliance) } - - storeChannel <- result - close(storeChannel) - - }() - - return storeChannel + }) } func (s SqlComplianceStore) ComplianceExport(job *model.Compliance) store.StoreChannel { - storeChannel := make(store.StoreChannel, 1) - - go func() { - result := store.StoreResult{} - + return store.Do(func(result *store.StoreResult) { props := map[string]interface{}{"StartTime": job.StartAt, "EndTime": job.EndAt} keywordQuery := "" @@ -258,10 +209,5 @@ func (s SqlComplianceStore) ComplianceExport(job *model.Compliance) store.StoreC } else { result.Data = cposts } - - storeChannel <- result - close(storeChannel) - }() - - return storeChannel + }) } |