Migrate Jobs store to sync by default (#11183)
* Migrate Jobs store to sync by default * Fixing compilation * Fixing compilation * Fixing govet
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
bbdd6927de
Коммит
f934502a56
@@ -8,7 +8,7 @@ import (
|
|||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/mattermost/mattermost-server/model"
|
"github.com/mattermost/mattermost-server/model"
|
||||||
"github.com/mattermost/mattermost-server/store"
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestCreateJob(t *testing.T) {
|
func TestCreateJob(t *testing.T) {
|
||||||
@@ -46,9 +46,8 @@ func TestGetJob(t *testing.T) {
|
|||||||
Id: model.NewId(),
|
Id: model.NewId(),
|
||||||
Status: model.JOB_STATUS_PENDING,
|
Status: model.JOB_STATUS_PENDING,
|
||||||
}
|
}
|
||||||
if result := <-th.App.Srv.Store.Job().Save(job); result.Err != nil {
|
_, err := th.App.Srv.Store.Job().Save(job)
|
||||||
t.Fatal(result.Err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
|
|
||||||
defer th.App.Srv.Store.Job().Delete(job.Id)
|
defer th.App.Srv.Store.Job().Delete(job.Id)
|
||||||
|
|
||||||
@@ -95,7 +94,8 @@ func TestGetJobs(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, job := range jobs {
|
for _, job := range jobs {
|
||||||
store.Must(th.App.Srv.Store.Job().Save(job))
|
_, err := th.App.Srv.Store.Job().Save(job)
|
||||||
|
require.Nil(t, err)
|
||||||
defer th.App.Srv.Store.Job().Delete(job.Id)
|
defer th.App.Srv.Store.Job().Delete(job.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -151,7 +151,8 @@ func TestGetJobsByType(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, job := range jobs {
|
for _, job := range jobs {
|
||||||
store.Must(th.App.Srv.Store.Job().Save(job))
|
_, err := th.App.Srv.Store.Job().Save(job)
|
||||||
|
require.Nil(t, err)
|
||||||
defer th.App.Srv.Store.Job().Delete(job.Id)
|
defer th.App.Srv.Store.Job().Delete(job.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -208,7 +209,8 @@ func TestCancelJob(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, job := range jobs {
|
for _, job := range jobs {
|
||||||
store.Must(th.App.Srv.Store.Job().Save(job))
|
_, err := th.App.Srv.Store.Job().Save(job)
|
||||||
|
require.Nil(t, err)
|
||||||
defer th.App.Srv.Store.Job().Delete(job.Id)
|
defer th.App.Srv.Store.Job().Delete(job.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
18
app/job.go
18
app/job.go
@@ -8,11 +8,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func (a *App) GetJob(id string) (*model.Job, *model.AppError) {
|
func (a *App) GetJob(id string) (*model.Job, *model.AppError) {
|
||||||
result := <-a.Srv.Store.Job().Get(id)
|
return a.Srv.Store.Job().Get(id)
|
||||||
if result.Err != nil {
|
|
||||||
return nil, result.Err
|
|
||||||
}
|
|
||||||
return result.Data.(*model.Job), nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) GetJobsPage(page int, perPage int) ([]*model.Job, *model.AppError) {
|
func (a *App) GetJobsPage(page int, perPage int) ([]*model.Job, *model.AppError) {
|
||||||
@@ -20,11 +16,7 @@ func (a *App) GetJobsPage(page int, perPage int) ([]*model.Job, *model.AppError)
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) GetJobs(offset int, limit int) ([]*model.Job, *model.AppError) {
|
func (a *App) GetJobs(offset int, limit int) ([]*model.Job, *model.AppError) {
|
||||||
result := <-a.Srv.Store.Job().GetAllPage(offset, limit)
|
return a.Srv.Store.Job().GetAllPage(offset, limit)
|
||||||
if result.Err != nil {
|
|
||||||
return nil, result.Err
|
|
||||||
}
|
|
||||||
return result.Data.([]*model.Job), nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) GetJobsByTypePage(jobType string, page int, perPage int) ([]*model.Job, *model.AppError) {
|
func (a *App) GetJobsByTypePage(jobType string, page int, perPage int) ([]*model.Job, *model.AppError) {
|
||||||
@@ -32,11 +24,7 @@ func (a *App) GetJobsByTypePage(jobType string, page int, perPage int) ([]*model
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) GetJobsByType(jobType string, offset int, limit int) ([]*model.Job, *model.AppError) {
|
func (a *App) GetJobsByType(jobType string, offset int, limit int) ([]*model.Job, *model.AppError) {
|
||||||
result := <-a.Srv.Store.Job().GetAllByTypePage(jobType, offset, limit)
|
return a.Srv.Store.Job().GetAllByTypePage(jobType, offset, limit)
|
||||||
if result.Err != nil {
|
|
||||||
return nil, result.Err
|
|
||||||
}
|
|
||||||
return result.Data.([]*model.Job), nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) CreateJob(job *model.Job) (*model.Job, *model.AppError) {
|
func (a *App) CreateJob(job *model.Job) (*model.Job, *model.AppError) {
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ import (
|
|||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/mattermost/mattermost-server/model"
|
"github.com/mattermost/mattermost-server/model"
|
||||||
"github.com/mattermost/mattermost-server/store"
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestGetJob(t *testing.T) {
|
func TestGetJob(t *testing.T) {
|
||||||
@@ -18,9 +18,8 @@ func TestGetJob(t *testing.T) {
|
|||||||
Id: model.NewId(),
|
Id: model.NewId(),
|
||||||
Status: model.NewId(),
|
Status: model.NewId(),
|
||||||
}
|
}
|
||||||
if result := <-th.App.Srv.Store.Job().Save(status); result.Err != nil {
|
_, err := th.App.Srv.Store.Job().Save(status)
|
||||||
t.Fatal(result.Err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
|
|
||||||
defer th.App.Srv.Store.Job().Delete(status.Id)
|
defer th.App.Srv.Store.Job().Delete(status.Id)
|
||||||
|
|
||||||
@@ -56,7 +55,8 @@ func TestGetJobByType(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, status := range statuses {
|
for _, status := range statuses {
|
||||||
store.Must(th.App.Srv.Store.Job().Save(status))
|
_, err := th.App.Srv.Store.Job().Save(status)
|
||||||
|
require.Nil(t, err)
|
||||||
defer th.App.Srv.Store.Job().Delete(status.Id)
|
defer th.App.Srv.Store.Job().Delete(status.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
103
jobs/jobs.go
103
jobs/jobs.go
@@ -31,50 +31,42 @@ func (srv *JobServer) CreateJob(jobType string, jobData map[string]string) (*mod
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
if result := <-srv.Store.Job().Save(&job); result.Err != nil {
|
if _, err := srv.Store.Job().Save(&job); err != nil {
|
||||||
return nil, result.Err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
return &job, nil
|
return &job, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *JobServer) GetJob(id string) (*model.Job, *model.AppError) {
|
func (srv *JobServer) GetJob(id string) (*model.Job, *model.AppError) {
|
||||||
if result := <-srv.Store.Job().Get(id); result.Err != nil {
|
return srv.Store.Job().Get(id)
|
||||||
return nil, result.Err
|
|
||||||
} else {
|
|
||||||
return result.Data.(*model.Job), nil
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *JobServer) ClaimJob(job *model.Job) (bool, *model.AppError) {
|
func (srv *JobServer) ClaimJob(job *model.Job) (bool, *model.AppError) {
|
||||||
if result := <-srv.Store.Job().UpdateStatusOptimistically(job.Id, model.JOB_STATUS_PENDING, model.JOB_STATUS_IN_PROGRESS); result.Err != nil {
|
return srv.Store.Job().UpdateStatusOptimistically(job.Id, model.JOB_STATUS_PENDING, model.JOB_STATUS_IN_PROGRESS)
|
||||||
return false, result.Err
|
|
||||||
} else {
|
|
||||||
success := result.Data.(bool)
|
|
||||||
return success, nil
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *JobServer) SetJobProgress(job *model.Job, progress int64) *model.AppError {
|
func (srv *JobServer) SetJobProgress(job *model.Job, progress int64) *model.AppError {
|
||||||
job.Status = model.JOB_STATUS_IN_PROGRESS
|
job.Status = model.JOB_STATUS_IN_PROGRESS
|
||||||
job.Progress = progress
|
job.Progress = progress
|
||||||
|
|
||||||
if result := <-srv.Store.Job().UpdateOptimistically(job, model.JOB_STATUS_IN_PROGRESS); result.Err != nil {
|
if _, err := srv.Store.Job().UpdateOptimistically(job, model.JOB_STATUS_IN_PROGRESS); err != nil {
|
||||||
return result.Err
|
return err
|
||||||
} else {
|
|
||||||
return nil
|
|
||||||
}
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *JobServer) SetJobSuccess(job *model.Job) *model.AppError {
|
func (srv *JobServer) SetJobSuccess(job *model.Job) *model.AppError {
|
||||||
result := <-srv.Store.Job().UpdateStatus(job.Id, model.JOB_STATUS_SUCCESS)
|
if _, err := srv.Store.Job().UpdateStatus(job.Id, model.JOB_STATUS_SUCCESS); err != nil {
|
||||||
return result.Err
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *JobServer) SetJobError(job *model.Job, jobError *model.AppError) *model.AppError {
|
func (srv *JobServer) SetJobError(job *model.Job, jobError *model.AppError) *model.AppError {
|
||||||
if jobError == nil {
|
if jobError == nil {
|
||||||
result := <-srv.Store.Job().UpdateStatus(job.Id, model.JOB_STATUS_ERROR)
|
_, err := srv.Store.Job().UpdateStatus(job.Id, model.JOB_STATUS_ERROR)
|
||||||
return result.Err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
job.Status = model.JOB_STATUS_ERROR
|
job.Status = model.JOB_STATUS_ERROR
|
||||||
@@ -84,17 +76,18 @@ func (srv *JobServer) SetJobError(job *model.Job, jobError *model.AppError) *mod
|
|||||||
}
|
}
|
||||||
job.Data["error"] = jobError.Message + " — " + jobError.DetailedError
|
job.Data["error"] = jobError.Message + " — " + jobError.DetailedError
|
||||||
|
|
||||||
if result := <-srv.Store.Job().UpdateOptimistically(job, model.JOB_STATUS_IN_PROGRESS); result.Err != nil {
|
updated, err := srv.Store.Job().UpdateOptimistically(job, model.JOB_STATUS_IN_PROGRESS)
|
||||||
return result.Err
|
if err != nil {
|
||||||
} else {
|
return err
|
||||||
if !result.Data.(bool) {
|
}
|
||||||
if result := <-srv.Store.Job().UpdateOptimistically(job, model.JOB_STATUS_CANCEL_REQUESTED); result.Err != nil {
|
|
||||||
return result.Err
|
if !updated {
|
||||||
} else {
|
updated, err = srv.Store.Job().UpdateOptimistically(job, model.JOB_STATUS_CANCEL_REQUESTED)
|
||||||
if !result.Data.(bool) {
|
if err != nil {
|
||||||
return model.NewAppError("Jobs.SetJobError", "jobs.set_job_error.update.error", nil, "id="+job.Id, http.StatusInternalServerError)
|
return err
|
||||||
}
|
}
|
||||||
}
|
if !updated {
|
||||||
|
return model.NewAppError("Jobs.SetJobError", "jobs.set_job_error.update.error", nil, "id="+job.Id, http.StatusInternalServerError)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -102,27 +95,36 @@ func (srv *JobServer) SetJobError(job *model.Job, jobError *model.AppError) *mod
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (srv *JobServer) SetJobCanceled(job *model.Job) *model.AppError {
|
func (srv *JobServer) SetJobCanceled(job *model.Job) *model.AppError {
|
||||||
result := <-srv.Store.Job().UpdateStatus(job.Id, model.JOB_STATUS_CANCELED)
|
if _, err := srv.Store.Job().UpdateStatus(job.Id, model.JOB_STATUS_CANCELED); err != nil {
|
||||||
return result.Err
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *JobServer) UpdateInProgressJobData(job *model.Job) *model.AppError {
|
func (srv *JobServer) UpdateInProgressJobData(job *model.Job) *model.AppError {
|
||||||
job.Status = model.JOB_STATUS_IN_PROGRESS
|
job.Status = model.JOB_STATUS_IN_PROGRESS
|
||||||
job.LastActivityAt = model.GetMillis()
|
job.LastActivityAt = model.GetMillis()
|
||||||
result := <-srv.Store.Job().UpdateOptimistically(job, model.JOB_STATUS_IN_PROGRESS)
|
if _, err := srv.Store.Job().UpdateOptimistically(job, model.JOB_STATUS_IN_PROGRESS); err != nil {
|
||||||
return result.Err
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *JobServer) RequestCancellation(jobId string) *model.AppError {
|
func (srv *JobServer) RequestCancellation(jobId string) *model.AppError {
|
||||||
if result := <-srv.Store.Job().UpdateStatusOptimistically(jobId, model.JOB_STATUS_PENDING, model.JOB_STATUS_CANCELED); result.Err != nil {
|
updated, err := srv.Store.Job().UpdateStatusOptimistically(jobId, model.JOB_STATUS_PENDING, model.JOB_STATUS_CANCELED)
|
||||||
return result.Err
|
if err != nil {
|
||||||
} else if result.Data.(bool) {
|
return err
|
||||||
|
}
|
||||||
|
if updated {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if result := <-srv.Store.Job().UpdateStatusOptimistically(jobId, model.JOB_STATUS_IN_PROGRESS, model.JOB_STATUS_CANCEL_REQUESTED); result.Err != nil {
|
updated, err = srv.Store.Job().UpdateStatusOptimistically(jobId, model.JOB_STATUS_IN_PROGRESS, model.JOB_STATUS_CANCEL_REQUESTED)
|
||||||
return result.Err
|
if err != nil {
|
||||||
} else if result.Data.(bool) {
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if updated {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -137,8 +139,7 @@ func (srv *JobServer) CancellationWatcher(ctx context.Context, jobId string, can
|
|||||||
return
|
return
|
||||||
case <-time.After(CANCEL_WATCHER_POLLING_INTERVAL * time.Millisecond):
|
case <-time.After(CANCEL_WATCHER_POLLING_INTERVAL * time.Millisecond):
|
||||||
mlog.Debug(fmt.Sprintf("CancellationWatcher for Job: %v polling.", jobId))
|
mlog.Debug(fmt.Sprintf("CancellationWatcher for Job: %v polling.", jobId))
|
||||||
if result := <-srv.Store.Job().Get(jobId); result.Err == nil {
|
if jobStatus, err := srv.Store.Job().Get(jobId); err == nil {
|
||||||
jobStatus := result.Data.(*model.Job)
|
|
||||||
if jobStatus.Status == model.JOB_STATUS_CANCEL_REQUESTED {
|
if jobStatus.Status == model.JOB_STATUS_CANCEL_REQUESTED {
|
||||||
close(cancelChan)
|
close(cancelChan)
|
||||||
return
|
return
|
||||||
@@ -159,17 +160,13 @@ func GenerateNextStartDateTime(now time.Time, nextStartTime time.Time) *time.Tim
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (srv *JobServer) CheckForPendingJobsByType(jobType string) (bool, *model.AppError) {
|
func (srv *JobServer) CheckForPendingJobsByType(jobType string) (bool, *model.AppError) {
|
||||||
if result := <-srv.Store.Job().GetCountByStatusAndType(model.JOB_STATUS_PENDING, jobType); result.Err != nil {
|
count, err := srv.Store.Job().GetCountByStatusAndType(model.JOB_STATUS_PENDING, jobType)
|
||||||
return false, result.Err
|
if err != nil {
|
||||||
} else {
|
return false, err
|
||||||
return result.Data.(int64) > 0, nil
|
|
||||||
}
|
}
|
||||||
|
return count > 0, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (srv *JobServer) GetLastSuccessfulJobByType(jobType string) (*model.Job, *model.AppError) {
|
func (srv *JobServer) GetLastSuccessfulJobByType(jobType string) (*model.Job, *model.AppError) {
|
||||||
if result := <-srv.Store.Job().GetNewestJobByStatusAndType(model.JOB_STATUS_SUCCESS, jobType); result.Err != nil {
|
return srv.Store.Job().GetNewestJobByStatusAndType(model.JOB_STATUS_SUCCESS, jobType)
|
||||||
return nil, result.Err
|
|
||||||
} else {
|
|
||||||
return result.Data.(*model.Job), nil
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -66,60 +66,60 @@ func (watcher *Watcher) Stop() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (watcher *Watcher) PollAndNotify() {
|
func (watcher *Watcher) PollAndNotify() {
|
||||||
if result := <-watcher.srv.Store.Job().GetAllByStatus(model.JOB_STATUS_PENDING); result.Err != nil {
|
jobs, err := watcher.srv.Store.Job().GetAllByStatus(model.JOB_STATUS_PENDING)
|
||||||
mlog.Error(fmt.Sprintf("Error occurred getting all pending statuses: %v", result.Err.Error()))
|
if err != nil {
|
||||||
} else {
|
mlog.Error(fmt.Sprintf("Error occurred getting all pending statuses: %v", err.Error()))
|
||||||
jobs := result.Data.([]*model.Job)
|
return
|
||||||
|
}
|
||||||
|
|
||||||
for _, job := range jobs {
|
for _, job := range jobs {
|
||||||
if job.Type == model.JOB_TYPE_DATA_RETENTION {
|
if job.Type == model.JOB_TYPE_DATA_RETENTION {
|
||||||
if watcher.workers.DataRetention != nil {
|
if watcher.workers.DataRetention != nil {
|
||||||
select {
|
select {
|
||||||
case watcher.workers.DataRetention.JobChannel() <- *job:
|
case watcher.workers.DataRetention.JobChannel() <- *job:
|
||||||
default:
|
default:
|
||||||
}
|
|
||||||
}
|
}
|
||||||
} else if job.Type == model.JOB_TYPE_MESSAGE_EXPORT {
|
}
|
||||||
if watcher.workers.MessageExport != nil {
|
} else if job.Type == model.JOB_TYPE_MESSAGE_EXPORT {
|
||||||
select {
|
if watcher.workers.MessageExport != nil {
|
||||||
case watcher.workers.MessageExport.JobChannel() <- *job:
|
select {
|
||||||
default:
|
case watcher.workers.MessageExport.JobChannel() <- *job:
|
||||||
}
|
default:
|
||||||
}
|
}
|
||||||
} else if job.Type == model.JOB_TYPE_ELASTICSEARCH_POST_INDEXING {
|
}
|
||||||
if watcher.workers.ElasticsearchIndexing != nil {
|
} else if job.Type == model.JOB_TYPE_ELASTICSEARCH_POST_INDEXING {
|
||||||
select {
|
if watcher.workers.ElasticsearchIndexing != nil {
|
||||||
case watcher.workers.ElasticsearchIndexing.JobChannel() <- *job:
|
select {
|
||||||
default:
|
case watcher.workers.ElasticsearchIndexing.JobChannel() <- *job:
|
||||||
}
|
default:
|
||||||
}
|
}
|
||||||
} else if job.Type == model.JOB_TYPE_ELASTICSEARCH_POST_AGGREGATION {
|
}
|
||||||
if watcher.workers.ElasticsearchAggregation != nil {
|
} else if job.Type == model.JOB_TYPE_ELASTICSEARCH_POST_AGGREGATION {
|
||||||
select {
|
if watcher.workers.ElasticsearchAggregation != nil {
|
||||||
case watcher.workers.ElasticsearchAggregation.JobChannel() <- *job:
|
select {
|
||||||
default:
|
case watcher.workers.ElasticsearchAggregation.JobChannel() <- *job:
|
||||||
}
|
default:
|
||||||
}
|
}
|
||||||
} else if job.Type == model.JOB_TYPE_LDAP_SYNC {
|
}
|
||||||
if watcher.workers.LdapSync != nil {
|
} else if job.Type == model.JOB_TYPE_LDAP_SYNC {
|
||||||
select {
|
if watcher.workers.LdapSync != nil {
|
||||||
case watcher.workers.LdapSync.JobChannel() <- *job:
|
select {
|
||||||
default:
|
case watcher.workers.LdapSync.JobChannel() <- *job:
|
||||||
}
|
default:
|
||||||
}
|
}
|
||||||
} else if job.Type == model.JOB_TYPE_MIGRATIONS {
|
}
|
||||||
if watcher.workers.Migrations != nil {
|
} else if job.Type == model.JOB_TYPE_MIGRATIONS {
|
||||||
select {
|
if watcher.workers.Migrations != nil {
|
||||||
case watcher.workers.Migrations.JobChannel() <- *job:
|
select {
|
||||||
default:
|
case watcher.workers.Migrations.JobChannel() <- *job:
|
||||||
}
|
default:
|
||||||
}
|
}
|
||||||
} else if job.Type == model.JOB_TYPE_PLUGINS {
|
}
|
||||||
if watcher.workers.Plugins != nil {
|
} else if job.Type == model.JOB_TYPE_PLUGINS {
|
||||||
select {
|
if watcher.workers.Plugins != nil {
|
||||||
case watcher.workers.Plugins.JobChannel() <- *job:
|
select {
|
||||||
default:
|
case watcher.workers.Plugins.JobChannel() <- *job:
|
||||||
}
|
default:
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -267,16 +267,15 @@ func (me *TestHelper) ResetRoleMigration() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (me *TestHelper) DeleteAllJobsByTypeAndMigrationKey(jobType string, migrationKey string) {
|
func (me *TestHelper) DeleteAllJobsByTypeAndMigrationKey(jobType string, migrationKey string) {
|
||||||
if res := <-me.App.Srv.Store.Job().GetAllByType(model.JOB_TYPE_MIGRATIONS); res.Err != nil {
|
jobs, err := me.App.Srv.Store.Job().GetAllByType(model.JOB_TYPE_MIGRATIONS)
|
||||||
panic(res.Err)
|
if err != nil {
|
||||||
} else {
|
panic(err)
|
||||||
jobs := res.Data.([]*model.Job)
|
}
|
||||||
|
|
||||||
for _, job := range jobs {
|
for _, job := range jobs {
|
||||||
if key, ok := job.Data[JOB_DATA_KEY_MIGRATION]; ok && key == migrationKey {
|
if key, ok := job.Data[JOB_DATA_KEY_MIGRATION]; ok && key == migrationKey {
|
||||||
if res := <-me.App.Srv.Store.Job().Delete(job.Id); res.Err != nil {
|
if _, err = me.App.Srv.Store.Job().Delete(job.Id); err != nil {
|
||||||
panic(res.Err)
|
panic(err)
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -40,21 +40,22 @@ func GetMigrationState(migration string, store store.Store) (string, *model.Job,
|
|||||||
return MIGRATION_STATE_COMPLETED, nil, nil
|
return MIGRATION_STATE_COMPLETED, nil, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if result := <-store.Job().GetAllByType(model.JOB_TYPE_MIGRATIONS); result.Err != nil {
|
jobs, err := store.Job().GetAllByType(model.JOB_TYPE_MIGRATIONS)
|
||||||
return "", nil, result.Err
|
if err != nil {
|
||||||
} else {
|
return "", nil, err
|
||||||
for _, job := range result.Data.([]*model.Job) {
|
}
|
||||||
if key, ok := job.Data[JOB_DATA_KEY_MIGRATION]; ok {
|
|
||||||
if key != migration {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
switch job.Status {
|
for _, job := range jobs {
|
||||||
case model.JOB_STATUS_IN_PROGRESS, model.JOB_STATUS_PENDING:
|
if key, ok := job.Data[JOB_DATA_KEY_MIGRATION]; ok {
|
||||||
return MIGRATION_STATE_IN_PROGRESS, job, nil
|
if key != migration {
|
||||||
default:
|
continue
|
||||||
return MIGRATION_STATE_UNSCHEDULED, job, nil
|
}
|
||||||
}
|
|
||||||
|
switch job.Status {
|
||||||
|
case model.JOB_STATUS_IN_PROGRESS, model.JOB_STATUS_PENDING:
|
||||||
|
return MIGRATION_STATE_IN_PROGRESS, job, nil
|
||||||
|
default:
|
||||||
|
return MIGRATION_STATE_UNSCHEDULED, job, nil
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,6 +7,7 @@ import (
|
|||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
|
||||||
"github.com/mattermost/mattermost-server/model"
|
"github.com/mattermost/mattermost-server/model"
|
||||||
)
|
)
|
||||||
@@ -52,7 +53,8 @@ func TestGetMigrationState(t *testing.T) {
|
|||||||
Type: model.JOB_TYPE_MIGRATIONS,
|
Type: model.JOB_TYPE_MIGRATIONS,
|
||||||
}
|
}
|
||||||
|
|
||||||
j1 = (<-th.App.Srv.Store.Job().Save(j1)).Data.(*model.Job)
|
j1, err = th.App.Srv.Store.Job().Save(j1)
|
||||||
|
require.Nil(t, err)
|
||||||
|
|
||||||
state, job, err = GetMigrationState(migrationKey, th.App.Srv.Store)
|
state, job, err = GetMigrationState(migrationKey, th.App.Srv.Store)
|
||||||
assert.Nil(t, err)
|
assert.Nil(t, err)
|
||||||
@@ -70,7 +72,8 @@ func TestGetMigrationState(t *testing.T) {
|
|||||||
Type: model.JOB_TYPE_MIGRATIONS,
|
Type: model.JOB_TYPE_MIGRATIONS,
|
||||||
}
|
}
|
||||||
|
|
||||||
j2 = (<-th.App.Srv.Store.Job().Save(j2)).Data.(*model.Job)
|
j2, err = th.App.Srv.Store.Job().Save(j2)
|
||||||
|
require.Nil(t, err)
|
||||||
|
|
||||||
state, job, err = GetMigrationState(migrationKey, th.App.Srv.Store)
|
state, job, err = GetMigrationState(migrationKey, th.App.Srv.Store)
|
||||||
assert.Nil(t, err)
|
assert.Nil(t, err)
|
||||||
@@ -88,7 +91,8 @@ func TestGetMigrationState(t *testing.T) {
|
|||||||
Type: model.JOB_TYPE_MIGRATIONS,
|
Type: model.JOB_TYPE_MIGRATIONS,
|
||||||
}
|
}
|
||||||
|
|
||||||
j3 = (<-th.App.Srv.Store.Job().Save(j3)).Data.(*model.Job)
|
j3, err = th.App.Srv.Store.Job().Save(j3)
|
||||||
|
require.Nil(t, err)
|
||||||
|
|
||||||
state, job, err = GetMigrationState(migrationKey, th.App.Srv.Store)
|
state, job, err = GetMigrationState(migrationKey, th.App.Srv.Store)
|
||||||
assert.Nil(t, err)
|
assert.Nil(t, err)
|
||||||
|
|||||||
@@ -34,267 +34,161 @@ func (jss SqlJobStore) CreateIndexesIfNotExists() {
|
|||||||
jss.CreateIndexIfNotExists("idx_jobs_type", "Jobs", "Type")
|
jss.CreateIndexIfNotExists("idx_jobs_type", "Jobs", "Type")
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jss SqlJobStore) Save(job *model.Job) store.StoreChannel {
|
func (jss SqlJobStore) Save(job *model.Job) (*model.Job, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
if err := jss.GetMaster().Insert(job); err != nil {
|
||||||
if err := jss.GetMaster().Insert(job); err != nil {
|
return nil, model.NewAppError("SqlJobStore.Save", "store.sql_job.save.app_error", nil, "id="+job.Id+", "+err.Error(), http.StatusInternalServerError)
|
||||||
result.Err = model.NewAppError("SqlJobStore.Save", "store.sql_job.save.app_error", nil, "id="+job.Id+", "+err.Error(), http.StatusInternalServerError)
|
}
|
||||||
} else {
|
return job, nil
|
||||||
result.Data = job
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jss SqlJobStore) UpdateOptimistically(job *model.Job, currentStatus string) store.StoreChannel {
|
func (jss SqlJobStore) UpdateOptimistically(job *model.Job, currentStatus string) (bool, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
query := "UPDATE Jobs SET LastActivityAt = :LastActivityAt, Status = :Status, Progress = :Progress, Data = :Data WHERE Id = :Id AND Status = :OldStatus"
|
||||||
if sqlResult, err := jss.GetMaster().Exec(
|
params := map[string]interface{}{
|
||||||
`UPDATE
|
"Id": job.Id,
|
||||||
Jobs
|
"OldStatus": currentStatus,
|
||||||
SET
|
"LastActivityAt": model.GetMillis(),
|
||||||
LastActivityAt = :LastActivityAt,
|
"Status": job.Status,
|
||||||
Status = :Status,
|
"Data": job.DataToJson(),
|
||||||
Progress = :Progress,
|
"Progress": job.Progress,
|
||||||
Data = :Data
|
}
|
||||||
WHERE
|
sqlResult, err := jss.GetMaster().Exec(query, params)
|
||||||
Id = :Id
|
if err != nil {
|
||||||
AND
|
return false, model.NewAppError("SqlJobStore.UpdateOptimistically", "store.sql_job.update.app_error", nil, "id="+job.Id+", "+err.Error(), http.StatusInternalServerError)
|
||||||
Status = :OldStatus`,
|
}
|
||||||
map[string]interface{}{
|
|
||||||
"Id": job.Id,
|
|
||||||
"OldStatus": currentStatus,
|
|
||||||
"LastActivityAt": model.GetMillis(),
|
|
||||||
"Status": job.Status,
|
|
||||||
"Data": job.DataToJson(),
|
|
||||||
"Progress": job.Progress,
|
|
||||||
}); err != nil {
|
|
||||||
result.Err = model.NewAppError("SqlJobStore.UpdateOptimistically", "store.sql_job.update.app_error", nil, "id="+job.Id+", "+err.Error(), http.StatusInternalServerError)
|
|
||||||
} else {
|
|
||||||
rows, err := sqlResult.RowsAffected()
|
|
||||||
|
|
||||||
if err != nil {
|
rows, err := sqlResult.RowsAffected()
|
||||||
result.Err = model.NewAppError("SqlJobStore.UpdateStatus", "store.sql_job.update.app_error", nil, "id="+job.Id+", "+err.Error(), http.StatusInternalServerError)
|
|
||||||
} else {
|
if err != nil {
|
||||||
if rows == 1 {
|
return false, model.NewAppError("SqlJobStore.UpdateStatus", "store.sql_job.update.app_error", nil, "id="+job.Id+", "+err.Error(), http.StatusInternalServerError)
|
||||||
result.Data = true
|
}
|
||||||
} else {
|
|
||||||
result.Data = false
|
if rows != 1 {
|
||||||
}
|
return false, nil
|
||||||
}
|
}
|
||||||
}
|
|
||||||
})
|
return true, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jss SqlJobStore) UpdateStatus(id string, status string) store.StoreChannel {
|
func (jss SqlJobStore) UpdateStatus(id string, status string) (*model.Job, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
job := &model.Job{
|
||||||
job := &model.Job{
|
Id: id,
|
||||||
Id: id,
|
Status: status,
|
||||||
Status: status,
|
LastActivityAt: model.GetMillis(),
|
||||||
LastActivityAt: model.GetMillis(),
|
}
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := jss.GetMaster().UpdateColumns(func(col *gorp.ColumnMap) bool {
|
if _, err := jss.GetMaster().UpdateColumns(func(col *gorp.ColumnMap) bool {
|
||||||
return col.ColumnName == "Status" || col.ColumnName == "LastActivityAt"
|
return col.ColumnName == "Status" || col.ColumnName == "LastActivityAt"
|
||||||
}, job); err != nil {
|
}, job); err != nil {
|
||||||
result.Err = model.NewAppError("SqlJobStore.UpdateStatus", "store.sql_job.update.app_error", nil, "id="+id+", "+err.Error(), http.StatusInternalServerError)
|
return nil, model.NewAppError("SqlJobStore.UpdateStatus", "store.sql_job.update.app_error", nil, "id="+id+", "+err.Error(), http.StatusInternalServerError)
|
||||||
}
|
}
|
||||||
|
|
||||||
if result.Err == nil {
|
return job, nil
|
||||||
result.Data = job
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jss SqlJobStore) UpdateStatusOptimistically(id string, currentStatus string, newStatus string) store.StoreChannel {
|
func (jss SqlJobStore) UpdateStatusOptimistically(id string, currentStatus string, newStatus string) (bool, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
var startAtClause string
|
||||||
var startAtClause string
|
if newStatus == model.JOB_STATUS_IN_PROGRESS {
|
||||||
if newStatus == model.JOB_STATUS_IN_PROGRESS {
|
startAtClause = "StartAt = :StartAt,"
|
||||||
startAtClause = `StartAt = :StartAt,`
|
}
|
||||||
}
|
query := "UPDATE Jobs SET " + startAtClause + " Status = :NewStatus, LastActivityAt = :LastActivityAt WHERE Id = :Id AND Status = :OldStatus"
|
||||||
|
params := map[string]interface{}{
|
||||||
|
"Id": id,
|
||||||
|
"OldStatus": currentStatus,
|
||||||
|
"NewStatus": newStatus,
|
||||||
|
"StartAt": model.GetMillis(),
|
||||||
|
"LastActivityAt": model.GetMillis(),
|
||||||
|
}
|
||||||
|
|
||||||
if sqlResult, err := jss.GetMaster().Exec(
|
sqlResult, err := jss.GetMaster().Exec(query, params)
|
||||||
`UPDATE
|
if err != nil {
|
||||||
Jobs
|
return false, model.NewAppError("SqlJobStore.UpdateStatus", "store.sql_job.update.app_error", nil, "id="+id+", "+err.Error(), http.StatusInternalServerError)
|
||||||
SET `+startAtClause+`
|
}
|
||||||
Status = :NewStatus,
|
rows, err := sqlResult.RowsAffected()
|
||||||
LastActivityAt = :LastActivityAt
|
if err != nil {
|
||||||
WHERE
|
return false, model.NewAppError("SqlJobStore.UpdateStatus", "store.sql_job.update.app_error", nil, "id="+id+", "+err.Error(), http.StatusInternalServerError)
|
||||||
Id = :Id
|
}
|
||||||
AND
|
if rows != 1 {
|
||||||
Status = :OldStatus`, map[string]interface{}{"Id": id, "OldStatus": currentStatus, "NewStatus": newStatus, "StartAt": model.GetMillis(), "LastActivityAt": model.GetMillis()}); err != nil {
|
return false, nil
|
||||||
result.Err = model.NewAppError("SqlJobStore.UpdateStatus", "store.sql_job.update.app_error", nil, "id="+id+", "+err.Error(), http.StatusInternalServerError)
|
}
|
||||||
} else {
|
|
||||||
rows, err := sqlResult.RowsAffected()
|
|
||||||
|
|
||||||
if err != nil {
|
return true, nil
|
||||||
result.Err = model.NewAppError("SqlJobStore.UpdateStatus", "store.sql_job.update.app_error", nil, "id="+id+", "+err.Error(), http.StatusInternalServerError)
|
|
||||||
} else {
|
|
||||||
if rows == 1 {
|
|
||||||
result.Data = true
|
|
||||||
} else {
|
|
||||||
result.Data = false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jss SqlJobStore) Get(id string) store.StoreChannel {
|
func (jss SqlJobStore) Get(id string) (*model.Job, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
query := "SELECT * FROM Jobs WHERE Id = :Id"
|
||||||
var status *model.Job
|
|
||||||
|
|
||||||
if err := jss.GetReplica().SelectOne(&status,
|
var status *model.Job
|
||||||
`SELECT
|
if err := jss.GetReplica().SelectOne(&status, query, map[string]interface{}{"Id": id}); err != nil {
|
||||||
*
|
if err == sql.ErrNoRows {
|
||||||
FROM
|
return nil, model.NewAppError("SqlJobStore.Get", "store.sql_job.get.app_error", nil, "Id="+id+", "+err.Error(), http.StatusNotFound)
|
||||||
Jobs
|
|
||||||
WHERE
|
|
||||||
Id = :Id`, map[string]interface{}{"Id": id}); err != nil {
|
|
||||||
if err == sql.ErrNoRows {
|
|
||||||
result.Err = model.NewAppError("SqlJobStore.Get", "store.sql_job.get.app_error", nil, "Id="+id+", "+err.Error(), http.StatusNotFound)
|
|
||||||
} else {
|
|
||||||
result.Err = model.NewAppError("SqlJobStore.Get", "store.sql_job.get.app_error", nil, "Id="+id+", "+err.Error(), http.StatusInternalServerError)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
result.Data = status
|
|
||||||
}
|
}
|
||||||
})
|
return nil, model.NewAppError("SqlJobStore.Get", "store.sql_job.get.app_error", nil, "Id="+id+", "+err.Error(), http.StatusInternalServerError)
|
||||||
|
}
|
||||||
|
return status, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jss SqlJobStore) GetAllPage(offset int, limit int) store.StoreChannel {
|
func (jss SqlJobStore) GetAllPage(offset int, limit int) ([]*model.Job, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
query := "SELECT * FROM Jobs ORDER BY CreateAt DESC LIMIT :Limit OFFSET :Offset"
|
||||||
var statuses []*model.Job
|
|
||||||
|
|
||||||
if _, err := jss.GetReplica().Select(&statuses,
|
var statuses []*model.Job
|
||||||
`SELECT
|
if _, err := jss.GetReplica().Select(&statuses, query, map[string]interface{}{"Limit": limit, "Offset": offset}); err != nil {
|
||||||
*
|
return nil, model.NewAppError("SqlJobStore.GetAllPage", "store.sql_job.get_all.app_error", nil, err.Error(), http.StatusInternalServerError)
|
||||||
FROM
|
}
|
||||||
Jobs
|
return statuses, nil
|
||||||
ORDER BY
|
|
||||||
CreateAt DESC
|
|
||||||
LIMIT
|
|
||||||
:Limit
|
|
||||||
OFFSET
|
|
||||||
:Offset`, map[string]interface{}{"Limit": limit, "Offset": offset}); err != nil {
|
|
||||||
result.Err = model.NewAppError("SqlJobStore.GetAllPage", "store.sql_job.get_all.app_error", nil, err.Error(), http.StatusInternalServerError)
|
|
||||||
} else {
|
|
||||||
result.Data = statuses
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jss SqlJobStore) GetAllByType(jobType string) store.StoreChannel {
|
func (jss SqlJobStore) GetAllByType(jobType string) ([]*model.Job, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
query := "SELECT * FROM Jobs WHERE Type = :Type ORDER BY CreateAt DESC"
|
||||||
var statuses []*model.Job
|
var statuses []*model.Job
|
||||||
|
if _, err := jss.GetReplica().Select(&statuses, query, map[string]interface{}{"Type": jobType}); err != nil {
|
||||||
if _, err := jss.GetReplica().Select(&statuses,
|
return nil, model.NewAppError("SqlJobStore.GetAllByType", "store.sql_job.get_all.app_error", nil, "Type="+jobType+", "+err.Error(), http.StatusInternalServerError)
|
||||||
`SELECT
|
}
|
||||||
*
|
return statuses, nil
|
||||||
FROM
|
|
||||||
Jobs
|
|
||||||
WHERE
|
|
||||||
Type = :Type
|
|
||||||
ORDER BY
|
|
||||||
CreateAt DESC`, map[string]interface{}{"Type": jobType}); err != nil {
|
|
||||||
result.Err = model.NewAppError("SqlJobStore.GetAllByType", "store.sql_job.get_all.app_error", nil, "Type="+jobType+", "+err.Error(), http.StatusInternalServerError)
|
|
||||||
} else {
|
|
||||||
result.Data = statuses
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jss SqlJobStore) GetAllByTypePage(jobType string, offset int, limit int) store.StoreChannel {
|
func (jss SqlJobStore) GetAllByTypePage(jobType string, offset int, limit int) ([]*model.Job, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
query := "SELECT * FROM Jobs WHERE Type = :Type ORDER BY CreateAt DESC LIMIT :Limit OFFSET :Offset"
|
||||||
var statuses []*model.Job
|
|
||||||
|
|
||||||
if _, err := jss.GetReplica().Select(&statuses,
|
var statuses []*model.Job
|
||||||
`SELECT
|
if _, err := jss.GetReplica().Select(&statuses, query, map[string]interface{}{"Type": jobType, "Limit": limit, "Offset": offset}); err != nil {
|
||||||
*
|
return nil, model.NewAppError("SqlJobStore.GetAllByTypePage", "store.sql_job.get_all.app_error", nil, "Type="+jobType+", "+err.Error(), http.StatusInternalServerError)
|
||||||
FROM
|
}
|
||||||
Jobs
|
return statuses, nil
|
||||||
WHERE
|
|
||||||
Type = :Type
|
|
||||||
ORDER BY
|
|
||||||
CreateAt DESC
|
|
||||||
LIMIT
|
|
||||||
:Limit
|
|
||||||
OFFSET
|
|
||||||
:Offset`, map[string]interface{}{"Type": jobType, "Limit": limit, "Offset": offset}); err != nil {
|
|
||||||
result.Err = model.NewAppError("SqlJobStore.GetAllByTypePage", "store.sql_job.get_all.app_error", nil, "Type="+jobType+", "+err.Error(), http.StatusInternalServerError)
|
|
||||||
} else {
|
|
||||||
result.Data = statuses
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jss SqlJobStore) GetAllByStatus(status string) store.StoreChannel {
|
func (jss SqlJobStore) GetAllByStatus(status string) ([]*model.Job, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
var statuses []*model.Job
|
||||||
var statuses []*model.Job
|
query := "SELECT * FROM Jobs WHERE Status = :Status ORDER BY CreateAt ASC"
|
||||||
|
|
||||||
if _, err := jss.GetReplica().Select(&statuses,
|
if _, err := jss.GetReplica().Select(&statuses, query, map[string]interface{}{"Status": status}); err != nil {
|
||||||
`SELECT
|
return nil, model.NewAppError("SqlJobStore.GetAllByStatus", "store.sql_job.get_all.app_error", nil, "Status="+status+", "+err.Error(), http.StatusInternalServerError)
|
||||||
*
|
}
|
||||||
FROM
|
return statuses, nil
|
||||||
Jobs
|
|
||||||
WHERE
|
|
||||||
Status = :Status
|
|
||||||
ORDER BY
|
|
||||||
CreateAt ASC`, map[string]interface{}{"Status": status}); err != nil {
|
|
||||||
result.Err = model.NewAppError("SqlJobStore.GetAllByStatus", "store.sql_job.get_all.app_error", nil, "Status="+status+", "+err.Error(), http.StatusInternalServerError)
|
|
||||||
} else {
|
|
||||||
result.Data = statuses
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jss SqlJobStore) GetNewestJobByStatusAndType(status string, jobType string) store.StoreChannel {
|
func (jss SqlJobStore) GetNewestJobByStatusAndType(status string, jobType string) (*model.Job, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
query := "SELECT * FROM Jobs WHERE Status = :Status AND Type = :Type ORDER BY CreateAt DESC LIMIT 1"
|
||||||
var job *model.Job
|
|
||||||
|
|
||||||
if err := jss.GetReplica().SelectOne(&job,
|
var job *model.Job
|
||||||
`SELECT
|
if err := jss.GetReplica().SelectOne(&job, query, map[string]interface{}{"Status": status, "Type": jobType}); err != nil && err != sql.ErrNoRows {
|
||||||
*
|
return nil, model.NewAppError("SqlJobStore.GetNewestJobByStatusAndType", "store.sql_job.get_newest_job_by_status_and_type.app_error", nil, "Status="+status+", "+err.Error(), http.StatusInternalServerError)
|
||||||
FROM
|
}
|
||||||
Jobs
|
return job, nil
|
||||||
WHERE
|
|
||||||
Status = :Status
|
|
||||||
AND
|
|
||||||
Type = :Type
|
|
||||||
ORDER BY
|
|
||||||
CreateAt DESC
|
|
||||||
LIMIT 1`, map[string]interface{}{"Status": status, "Type": jobType}); err != nil && err != sql.ErrNoRows {
|
|
||||||
result.Err = model.NewAppError("SqlJobStore.GetNewestJobByStatusAndType", "store.sql_job.get_newest_job_by_status_and_type.app_error", nil, "Status="+status+", "+err.Error(), http.StatusInternalServerError)
|
|
||||||
} else {
|
|
||||||
result.Data = job
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jss SqlJobStore) GetCountByStatusAndType(status string, jobType string) store.StoreChannel {
|
func (jss SqlJobStore) GetCountByStatusAndType(status string, jobType string) (int64, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
query := "SELECT COUNT(*) FROM Jobs WHERE Status = :Status AND Type = :Type"
|
||||||
if count, err := jss.GetReplica().SelectInt(`SELECT
|
count, err := jss.GetReplica().SelectInt(query, map[string]interface{}{"Status": status, "Type": jobType})
|
||||||
COUNT(*)
|
if err != nil {
|
||||||
FROM
|
return int64(0), model.NewAppError("SqlJobStore.GetCountByStatusAndType", "store.sql_job.get_count_by_status_and_type.app_error", nil, "Status="+status+", "+err.Error(), http.StatusInternalServerError)
|
||||||
Jobs
|
}
|
||||||
WHERE
|
return count, nil
|
||||||
Status = :Status
|
|
||||||
AND
|
|
||||||
Type = :Type`, map[string]interface{}{"Status": status, "Type": jobType}); err != nil {
|
|
||||||
result.Err = model.NewAppError("SqlJobStore.GetCountByStatusAndType", "store.sql_job.get_count_by_status_and_type.app_error", nil, "Status="+status+", "+err.Error(), http.StatusInternalServerError)
|
|
||||||
} else {
|
|
||||||
result.Data = count
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (jss SqlJobStore) Delete(id string) store.StoreChannel {
|
func (jss SqlJobStore) Delete(id string) (string, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
query := "DELETE FROM Jobs WHERE Id = :Id"
|
||||||
if _, err := jss.GetMaster().Exec(
|
if _, err := jss.GetMaster().Exec(query, map[string]interface{}{"Id": id}); err != nil {
|
||||||
`DELETE FROM
|
return "", model.NewAppError("SqlJobStore.DeleteByType", "store.sql_job.delete.app_error", nil, "id="+id+", "+err.Error(), http.StatusInternalServerError)
|
||||||
Jobs
|
}
|
||||||
WHERE
|
return id, nil
|
||||||
Id = :Id`, map[string]interface{}{"Id": id}); err != nil {
|
|
||||||
result.Err = model.NewAppError("SqlJobStore.DeleteByType", "store.sql_job.delete.app_error", nil, "id="+id+", "+err.Error(), http.StatusInternalServerError)
|
|
||||||
} else {
|
|
||||||
result.Data = id
|
|
||||||
}
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -500,18 +500,18 @@ type ReactionStore interface {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type JobStore interface {
|
type JobStore interface {
|
||||||
Save(job *model.Job) StoreChannel
|
Save(job *model.Job) (*model.Job, *model.AppError)
|
||||||
UpdateOptimistically(job *model.Job, currentStatus string) StoreChannel
|
UpdateOptimistically(job *model.Job, currentStatus string) (bool, *model.AppError)
|
||||||
UpdateStatus(id string, status string) StoreChannel
|
UpdateStatus(id string, status string) (*model.Job, *model.AppError)
|
||||||
UpdateStatusOptimistically(id string, currentStatus string, newStatus string) StoreChannel
|
UpdateStatusOptimistically(id string, currentStatus string, newStatus string) (bool, *model.AppError)
|
||||||
Get(id string) StoreChannel
|
Get(id string) (*model.Job, *model.AppError)
|
||||||
GetAllPage(offset int, limit int) StoreChannel
|
GetAllPage(offset int, limit int) ([]*model.Job, *model.AppError)
|
||||||
GetAllByType(jobType string) StoreChannel
|
GetAllByType(jobType string) ([]*model.Job, *model.AppError)
|
||||||
GetAllByTypePage(jobType string, offset int, limit int) StoreChannel
|
GetAllByTypePage(jobType string, offset int, limit int) ([]*model.Job, *model.AppError)
|
||||||
GetAllByStatus(status string) StoreChannel
|
GetAllByStatus(status string) ([]*model.Job, *model.AppError)
|
||||||
GetNewestJobByStatusAndType(status string, jobType string) StoreChannel
|
GetNewestJobByStatusAndType(status string, jobType string) (*model.Job, *model.AppError)
|
||||||
GetCountByStatusAndType(status string, jobType string) StoreChannel
|
GetCountByStatusAndType(status string, jobType string) (int64, *model.AppError)
|
||||||
Delete(id string) StoreChannel
|
Delete(id string) (string, *model.AppError)
|
||||||
}
|
}
|
||||||
|
|
||||||
type UserAccessTokenStore interface {
|
type UserAccessTokenStore interface {
|
||||||
|
|||||||
@@ -11,6 +11,7 @@ import (
|
|||||||
"github.com/mattermost/mattermost-server/model"
|
"github.com/mattermost/mattermost-server/model"
|
||||||
"github.com/mattermost/mattermost-server/store"
|
"github.com/mattermost/mattermost-server/store"
|
||||||
"github.com/stretchr/testify/assert"
|
"github.com/stretchr/testify/assert"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestJobStore(t *testing.T, ss store.Store) {
|
func TestJobStore(t *testing.T, ss store.Store) {
|
||||||
@@ -38,17 +39,14 @@ func testJobSaveGet(t *testing.T, ss store.Store) {
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
if result := <-ss.Job().Save(job); result.Err != nil {
|
_, err := ss.Job().Save(job)
|
||||||
t.Fatal(result.Err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
|
|
||||||
defer func() {
|
defer ss.Job().Delete(job.Id)
|
||||||
<-ss.Job().Delete(job.Id)
|
|
||||||
}()
|
|
||||||
|
|
||||||
if result := <-ss.Job().Get(job.Id); result.Err != nil {
|
if received, err := ss.Job().Get(job.Id); err != nil {
|
||||||
t.Fatal(result.Err)
|
t.Fatal(err)
|
||||||
} else if received := result.Data.(*model.Job); received.Id != job.Id {
|
} else if received.Id != job.Id {
|
||||||
t.Fatal("received incorrect job after save")
|
t.Fatal("received incorrect job after save")
|
||||||
} else if received.Data["Total"] != "12345" {
|
} else if received.Data["Total"] != "12345" {
|
||||||
t.Fatal("data field was not retrieved successfully:", received.Data)
|
t.Fatal("data field was not retrieved successfully:", received.Data)
|
||||||
@@ -74,13 +72,14 @@ func testJobGetAllByType(t *testing.T, ss store.Store) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, job := range jobs {
|
for _, job := range jobs {
|
||||||
store.Must(ss.Job().Save(job))
|
_, err := ss.Job().Save(job)
|
||||||
|
require.Nil(t, err)
|
||||||
defer ss.Job().Delete(job.Id)
|
defer ss.Job().Delete(job.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
if result := <-ss.Job().GetAllByType(jobType); result.Err != nil {
|
if received, err := ss.Job().GetAllByType(jobType); err != nil {
|
||||||
t.Fatal(result.Err)
|
t.Fatal(err)
|
||||||
} else if received := result.Data.([]*model.Job); len(received) != 2 {
|
} else if len(received) != 2 {
|
||||||
t.Fatal("received wrong number of jobs")
|
t.Fatal("received wrong number of jobs")
|
||||||
} else if received[0].Id != jobs[0].Id && received[1].Id != jobs[0].Id {
|
} else if received[0].Id != jobs[0].Id && received[1].Id != jobs[0].Id {
|
||||||
t.Fatal("should've received first jobs")
|
t.Fatal("should've received first jobs")
|
||||||
@@ -116,13 +115,14 @@ func testJobGetAllByTypePage(t *testing.T, ss store.Store) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, job := range jobs {
|
for _, job := range jobs {
|
||||||
store.Must(ss.Job().Save(job))
|
_, err := ss.Job().Save(job)
|
||||||
|
require.Nil(t, err)
|
||||||
defer ss.Job().Delete(job.Id)
|
defer ss.Job().Delete(job.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
if result := <-ss.Job().GetAllByTypePage(jobType, 0, 2); result.Err != nil {
|
if received, err := ss.Job().GetAllByTypePage(jobType, 0, 2); err != nil {
|
||||||
t.Fatal(result.Err)
|
t.Fatal(err)
|
||||||
} else if received := result.Data.([]*model.Job); len(received) != 2 {
|
} else if len(received) != 2 {
|
||||||
t.Fatal("received wrong number of jobs")
|
t.Fatal("received wrong number of jobs")
|
||||||
} else if received[0].Id != jobs[2].Id {
|
} else if received[0].Id != jobs[2].Id {
|
||||||
t.Fatal("should've received newest job first")
|
t.Fatal("should've received newest job first")
|
||||||
@@ -130,9 +130,9 @@ func testJobGetAllByTypePage(t *testing.T, ss store.Store) {
|
|||||||
t.Fatal("should've received second newest job second")
|
t.Fatal("should've received second newest job second")
|
||||||
}
|
}
|
||||||
|
|
||||||
if result := <-ss.Job().GetAllByTypePage(jobType, 2, 2); result.Err != nil {
|
if received, err := ss.Job().GetAllByTypePage(jobType, 2, 2); err != nil {
|
||||||
t.Fatal(result.Err)
|
t.Fatal(err)
|
||||||
} else if received := result.Data.([]*model.Job); len(received) != 1 {
|
} else if len(received) != 1 {
|
||||||
t.Fatal("received wrong number of jobs")
|
t.Fatal("received wrong number of jobs")
|
||||||
} else if received[0].Id != jobs[1].Id {
|
} else if received[0].Id != jobs[1].Id {
|
||||||
t.Fatal("should've received oldest job last")
|
t.Fatal("should've received oldest job last")
|
||||||
@@ -162,13 +162,14 @@ func testJobGetAllPage(t *testing.T, ss store.Store) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, job := range jobs {
|
for _, job := range jobs {
|
||||||
store.Must(ss.Job().Save(job))
|
_, err := ss.Job().Save(job)
|
||||||
|
require.Nil(t, err)
|
||||||
defer ss.Job().Delete(job.Id)
|
defer ss.Job().Delete(job.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
if result := <-ss.Job().GetAllPage(0, 2); result.Err != nil {
|
if received, err := ss.Job().GetAllPage(0, 2); err != nil {
|
||||||
t.Fatal(result.Err)
|
t.Fatal(err)
|
||||||
} else if received := result.Data.([]*model.Job); len(received) != 2 {
|
} else if len(received) != 2 {
|
||||||
t.Fatal("received wrong number of jobs")
|
t.Fatal("received wrong number of jobs")
|
||||||
} else if received[0].Id != jobs[2].Id {
|
} else if received[0].Id != jobs[2].Id {
|
||||||
t.Fatal("should've received newest job first")
|
t.Fatal("should've received newest job first")
|
||||||
@@ -176,9 +177,9 @@ func testJobGetAllPage(t *testing.T, ss store.Store) {
|
|||||||
t.Fatal("should've received second newest job second")
|
t.Fatal("should've received second newest job second")
|
||||||
}
|
}
|
||||||
|
|
||||||
if result := <-ss.Job().GetAllPage(2, 2); result.Err != nil {
|
if received, err := ss.Job().GetAllPage(2, 2); err != nil {
|
||||||
t.Fatal(result.Err)
|
t.Fatal(err)
|
||||||
} else if received := result.Data.([]*model.Job); len(received) < 1 {
|
} else if len(received) < 1 {
|
||||||
t.Fatal("received wrong number of jobs")
|
t.Fatal("received wrong number of jobs")
|
||||||
} else if received[0].Id != jobs[1].Id {
|
} else if received[0].Id != jobs[1].Id {
|
||||||
t.Fatal("should've received oldest job last")
|
t.Fatal("should've received oldest job last")
|
||||||
@@ -220,13 +221,14 @@ func testJobGetAllByStatus(t *testing.T, ss store.Store) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, job := range jobs {
|
for _, job := range jobs {
|
||||||
store.Must(ss.Job().Save(job))
|
_, err := ss.Job().Save(job)
|
||||||
|
require.Nil(t, err)
|
||||||
defer ss.Job().Delete(job.Id)
|
defer ss.Job().Delete(job.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
if result := <-ss.Job().GetAllByStatus(status); result.Err != nil {
|
if received, err := ss.Job().GetAllByStatus(status); err != nil {
|
||||||
t.Fatal(result.Err)
|
t.Fatal(err)
|
||||||
} else if received := result.Data.([]*model.Job); len(received) != 3 {
|
} else if len(received) != 3 {
|
||||||
t.Fatal("received wrong number of jobs")
|
t.Fatal("received wrong number of jobs")
|
||||||
} else if received[0].Id != jobs[1].Id || received[1].Id != jobs[0].Id || received[2].Id != jobs[2].Id {
|
} else if received[0].Id != jobs[1].Id || received[1].Id != jobs[0].Id || received[2].Id != jobs[2].Id {
|
||||||
t.Fatal("should've received jobs ordered by CreateAt time")
|
t.Fatal("should've received jobs ordered by CreateAt time")
|
||||||
@@ -269,17 +271,18 @@ func testJobStoreGetNewestJobByStatusAndType(t *testing.T, ss store.Store) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, job := range jobs {
|
for _, job := range jobs {
|
||||||
store.Must(ss.Job().Save(job))
|
_, err := ss.Job().Save(job)
|
||||||
|
require.Nil(t, err)
|
||||||
defer ss.Job().Delete(job.Id)
|
defer ss.Job().Delete(job.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
result := <-ss.Job().GetNewestJobByStatusAndType(status1, jobType1)
|
received, err := ss.Job().GetNewestJobByStatusAndType(status1, jobType1)
|
||||||
assert.Nil(t, result.Err)
|
assert.Nil(t, err)
|
||||||
assert.EqualValues(t, jobs[0].Id, result.Data.(*model.Job).Id)
|
assert.EqualValues(t, jobs[0].Id, received.Id)
|
||||||
|
|
||||||
result = <-ss.Job().GetNewestJobByStatusAndType(model.NewId(), model.NewId())
|
received, err = ss.Job().GetNewestJobByStatusAndType(model.NewId(), model.NewId())
|
||||||
assert.Nil(t, result.Err)
|
assert.Nil(t, err)
|
||||||
assert.Nil(t, result.Data.(*model.Job))
|
assert.Nil(t, received)
|
||||||
}
|
}
|
||||||
|
|
||||||
func testJobStoreGetCountByStatusAndType(t *testing.T, ss store.Store) {
|
func testJobStoreGetCountByStatusAndType(t *testing.T, ss store.Store) {
|
||||||
@@ -316,25 +319,26 @@ func testJobStoreGetCountByStatusAndType(t *testing.T, ss store.Store) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, job := range jobs {
|
for _, job := range jobs {
|
||||||
store.Must(ss.Job().Save(job))
|
_, err := ss.Job().Save(job)
|
||||||
|
require.Nil(t, err)
|
||||||
defer ss.Job().Delete(job.Id)
|
defer ss.Job().Delete(job.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
result := <-ss.Job().GetCountByStatusAndType(status1, jobType1)
|
count, err := ss.Job().GetCountByStatusAndType(status1, jobType1)
|
||||||
assert.Nil(t, result.Err)
|
assert.Nil(t, err)
|
||||||
assert.EqualValues(t, 2, result.Data.(int64))
|
assert.EqualValues(t, 2, count)
|
||||||
|
|
||||||
result = <-ss.Job().GetCountByStatusAndType(status2, jobType2)
|
count, err = ss.Job().GetCountByStatusAndType(status2, jobType2)
|
||||||
assert.Nil(t, result.Err)
|
assert.Nil(t, err)
|
||||||
assert.EqualValues(t, 0, result.Data.(int64))
|
assert.EqualValues(t, 0, count)
|
||||||
|
|
||||||
result = <-ss.Job().GetCountByStatusAndType(status1, jobType2)
|
count, err = ss.Job().GetCountByStatusAndType(status1, jobType2)
|
||||||
assert.Nil(t, result.Err)
|
assert.Nil(t, err)
|
||||||
assert.EqualValues(t, 1, result.Data.(int64))
|
assert.EqualValues(t, 1, count)
|
||||||
|
|
||||||
result = <-ss.Job().GetCountByStatusAndType(status2, jobType1)
|
count, err = ss.Job().GetCountByStatusAndType(status2, jobType1)
|
||||||
assert.Nil(t, result.Err)
|
assert.Nil(t, err)
|
||||||
assert.EqualValues(t, 1, result.Data.(int64))
|
assert.EqualValues(t, 1, count)
|
||||||
}
|
}
|
||||||
|
|
||||||
func testJobUpdateOptimistically(t *testing.T, ss store.Store) {
|
func testJobUpdateOptimistically(t *testing.T, ss store.Store) {
|
||||||
@@ -345,9 +349,8 @@ func testJobUpdateOptimistically(t *testing.T, ss store.Store) {
|
|||||||
Status: model.JOB_STATUS_PENDING,
|
Status: model.JOB_STATUS_PENDING,
|
||||||
}
|
}
|
||||||
|
|
||||||
if result := <-ss.Job().Save(job); result.Err != nil {
|
_, err := ss.Job().Save(job)
|
||||||
t.Fatal(result.Err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
defer ss.Job().Delete(job.Id)
|
defer ss.Job().Delete(job.Id)
|
||||||
|
|
||||||
job.LastActivityAt = model.GetMillis()
|
job.LastActivityAt = model.GetMillis()
|
||||||
@@ -357,34 +360,24 @@ func testJobUpdateOptimistically(t *testing.T, ss store.Store) {
|
|||||||
"Foo": "Bar",
|
"Foo": "Bar",
|
||||||
}
|
}
|
||||||
|
|
||||||
if result := <-ss.Job().UpdateOptimistically(job, model.JOB_STATUS_SUCCESS); result.Err != nil {
|
if updated, err2 := ss.Job().UpdateOptimistically(job, model.JOB_STATUS_SUCCESS); err2 != nil {
|
||||||
if result.Data.(bool) {
|
if updated {
|
||||||
t.Fatal("should have failed due to incorrect old status")
|
t.Fatal("should have failed due to incorrect old status")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
time.Sleep(2 * time.Millisecond)
|
time.Sleep(2 * time.Millisecond)
|
||||||
|
|
||||||
if result := <-ss.Job().UpdateOptimistically(job, model.JOB_STATUS_PENDING); result.Err != nil {
|
updated, err := ss.Job().UpdateOptimistically(job, model.JOB_STATUS_PENDING)
|
||||||
t.Fatal(result.Err)
|
require.Nil(t, err)
|
||||||
} else {
|
require.True(t, updated)
|
||||||
if !result.Data.(bool) {
|
|
||||||
t.Fatal("Should have successfully updated")
|
|
||||||
}
|
|
||||||
|
|
||||||
var updatedJob *model.Job
|
updatedJob, err := ss.Job().Get(job.Id)
|
||||||
|
require.Nil(t, err)
|
||||||
|
|
||||||
if result := <-ss.Job().Get(job.Id); result.Err != nil {
|
if updatedJob.Type != job.Type || updatedJob.CreateAt != job.CreateAt || updatedJob.Status != job.Status || updatedJob.LastActivityAt <= job.LastActivityAt || updatedJob.Progress != job.Progress || updatedJob.Data["Foo"] != job.Data["Foo"] {
|
||||||
t.Fatal(result.Err)
|
t.Fatal("Some update property was not as expected")
|
||||||
} else {
|
|
||||||
updatedJob = result.Data.(*model.Job)
|
|
||||||
}
|
|
||||||
|
|
||||||
if updatedJob.Type != job.Type || updatedJob.CreateAt != job.CreateAt || updatedJob.Status != job.Status || updatedJob.LastActivityAt <= job.LastActivityAt || updatedJob.Progress != job.Progress || updatedJob.Data["Foo"] != job.Data["Foo"] {
|
|
||||||
t.Fatal("Some update property was not as expected")
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func testJobUpdateStatusUpdateStatusOptimistically(t *testing.T, ss store.Store) {
|
func testJobUpdateStatusUpdateStatusOptimistically(t *testing.T, ss store.Store) {
|
||||||
@@ -396,111 +389,85 @@ func testJobUpdateStatusUpdateStatusOptimistically(t *testing.T, ss store.Store)
|
|||||||
}
|
}
|
||||||
|
|
||||||
var lastUpdateAt int64
|
var lastUpdateAt int64
|
||||||
if result := <-ss.Job().Save(job); result.Err != nil {
|
received, err := ss.Job().Save(job)
|
||||||
t.Fatal(result.Err)
|
require.Nil(t, err)
|
||||||
} else {
|
lastUpdateAt = received.LastActivityAt
|
||||||
lastUpdateAt = result.Data.(*model.Job).LastActivityAt
|
|
||||||
}
|
|
||||||
|
|
||||||
defer ss.Job().Delete(job.Id)
|
defer ss.Job().Delete(job.Id)
|
||||||
|
|
||||||
time.Sleep(2 * time.Millisecond)
|
time.Sleep(2 * time.Millisecond)
|
||||||
|
|
||||||
if result := <-ss.Job().UpdateStatus(job.Id, model.JOB_STATUS_PENDING); result.Err != nil {
|
received, err = ss.Job().UpdateStatus(job.Id, model.JOB_STATUS_PENDING)
|
||||||
t.Fatal(result.Err)
|
require.Nil(t, err)
|
||||||
} else {
|
|
||||||
received := result.Data.(*model.Job)
|
if received.Status != model.JOB_STATUS_PENDING {
|
||||||
if received.Status != model.JOB_STATUS_PENDING {
|
t.Fatal("status wasn't updated")
|
||||||
t.Fatal("status wasn't updated")
|
}
|
||||||
}
|
if received.LastActivityAt <= lastUpdateAt {
|
||||||
if received.LastActivityAt <= lastUpdateAt {
|
t.Fatal("lastActivityAt wasn't updated")
|
||||||
t.Fatal("lastActivityAt wasn't updated")
|
}
|
||||||
}
|
lastUpdateAt = received.LastActivityAt
|
||||||
lastUpdateAt = received.LastActivityAt
|
|
||||||
|
time.Sleep(2 * time.Millisecond)
|
||||||
|
|
||||||
|
updated, err := ss.Job().UpdateStatusOptimistically(job.Id, model.JOB_STATUS_IN_PROGRESS, model.JOB_STATUS_SUCCESS)
|
||||||
|
require.Nil(t, err)
|
||||||
|
require.False(t, updated)
|
||||||
|
|
||||||
|
received, err = ss.Job().Get(job.Id)
|
||||||
|
require.Nil(t, err)
|
||||||
|
|
||||||
|
if received.Status != model.JOB_STATUS_PENDING {
|
||||||
|
t.Fatal("should still be pending")
|
||||||
|
}
|
||||||
|
if received.LastActivityAt != lastUpdateAt {
|
||||||
|
t.Fatal("last activity at shouldn't have changed")
|
||||||
}
|
}
|
||||||
|
|
||||||
time.Sleep(2 * time.Millisecond)
|
time.Sleep(2 * time.Millisecond)
|
||||||
|
|
||||||
if result := <-ss.Job().UpdateStatusOptimistically(job.Id, model.JOB_STATUS_IN_PROGRESS, model.JOB_STATUS_SUCCESS); result.Err != nil {
|
updated, err = ss.Job().UpdateStatusOptimistically(job.Id, model.JOB_STATUS_PENDING, model.JOB_STATUS_IN_PROGRESS)
|
||||||
t.Fatal(result.Err)
|
require.Nil(t, err)
|
||||||
} else {
|
require.True(t, updated, "should have succeeded")
|
||||||
if result.Data.(bool) {
|
|
||||||
t.Fatal("should be false due to incorrect original status")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if result := <-ss.Job().Get(job.Id); result.Err != nil {
|
|
||||||
t.Fatal(result.Err)
|
|
||||||
} else {
|
|
||||||
received := result.Data.(*model.Job)
|
|
||||||
if received.Status != model.JOB_STATUS_PENDING {
|
|
||||||
t.Fatal("should still be pending")
|
|
||||||
}
|
|
||||||
if received.LastActivityAt != lastUpdateAt {
|
|
||||||
t.Fatal("last activity at shouldn't have changed")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
time.Sleep(2 * time.Millisecond)
|
|
||||||
|
|
||||||
if result := <-ss.Job().UpdateStatusOptimistically(job.Id, model.JOB_STATUS_PENDING, model.JOB_STATUS_IN_PROGRESS); result.Err != nil {
|
|
||||||
t.Fatal(result.Err)
|
|
||||||
} else {
|
|
||||||
if !result.Data.(bool) {
|
|
||||||
t.Fatal("should have succeeded")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
var startAtSet int64
|
var startAtSet int64
|
||||||
if result := <-ss.Job().Get(job.Id); result.Err != nil {
|
received, err = ss.Job().Get(job.Id)
|
||||||
t.Fatal(result.Err)
|
require.Nil(t, err)
|
||||||
} else {
|
if received.Status != model.JOB_STATUS_IN_PROGRESS {
|
||||||
received := result.Data.(*model.Job)
|
t.Fatal("should be in progress")
|
||||||
if received.Status != model.JOB_STATUS_IN_PROGRESS {
|
|
||||||
t.Fatal("should be in progress")
|
|
||||||
}
|
|
||||||
if received.StartAt == 0 {
|
|
||||||
t.Fatal("received should have start at set")
|
|
||||||
}
|
|
||||||
if received.LastActivityAt <= lastUpdateAt {
|
|
||||||
t.Fatal("lastActivityAt wasn't updated")
|
|
||||||
}
|
|
||||||
lastUpdateAt = received.LastActivityAt
|
|
||||||
startAtSet = received.StartAt
|
|
||||||
}
|
}
|
||||||
|
if received.StartAt == 0 {
|
||||||
|
t.Fatal("received should have start at set")
|
||||||
|
}
|
||||||
|
if received.LastActivityAt <= lastUpdateAt {
|
||||||
|
t.Fatal("lastActivityAt wasn't updated")
|
||||||
|
}
|
||||||
|
lastUpdateAt = received.LastActivityAt
|
||||||
|
startAtSet = received.StartAt
|
||||||
|
|
||||||
time.Sleep(2 * time.Millisecond)
|
time.Sleep(2 * time.Millisecond)
|
||||||
|
|
||||||
if result := <-ss.Job().UpdateStatusOptimistically(job.Id, model.JOB_STATUS_IN_PROGRESS, model.JOB_STATUS_SUCCESS); result.Err != nil {
|
updated, err = ss.Job().UpdateStatusOptimistically(job.Id, model.JOB_STATUS_IN_PROGRESS, model.JOB_STATUS_SUCCESS)
|
||||||
t.Fatal(result.Err)
|
require.Nil(t, err)
|
||||||
} else {
|
require.True(t, updated, "should have succeeded")
|
||||||
if !result.Data.(bool) {
|
|
||||||
t.Fatal("should have succeeded")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if result := <-ss.Job().Get(job.Id); result.Err != nil {
|
received, err = ss.Job().Get(job.Id)
|
||||||
t.Fatal(result.Err)
|
require.Nil(t, err)
|
||||||
} else {
|
if received.Status != model.JOB_STATUS_SUCCESS {
|
||||||
received := result.Data.(*model.Job)
|
t.Fatal("should be success status")
|
||||||
if received.Status != model.JOB_STATUS_SUCCESS {
|
}
|
||||||
t.Fatal("should be success status")
|
if received.StartAt != startAtSet {
|
||||||
}
|
t.Fatal("startAt should not have changed")
|
||||||
if received.StartAt != startAtSet {
|
}
|
||||||
t.Fatal("startAt should not have changed")
|
if received.LastActivityAt <= lastUpdateAt {
|
||||||
}
|
t.Fatal("lastActivityAt wasn't updated")
|
||||||
if received.LastActivityAt <= lastUpdateAt {
|
|
||||||
t.Fatal("lastActivityAt wasn't updated")
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func testJobDelete(t *testing.T, ss store.Store) {
|
func testJobDelete(t *testing.T, ss store.Store) {
|
||||||
job := store.Must(ss.Job().Save(&model.Job{
|
job, err := ss.Job().Save(&model.Job{Id: model.NewId()})
|
||||||
Id: model.NewId(),
|
require.Nil(t, err)
|
||||||
})).(*model.Job)
|
|
||||||
|
|
||||||
if result := <-ss.Job().Delete(job.Id); result.Err != nil {
|
_, err = ss.Job().Delete(job.Id)
|
||||||
t.Fatal(result.Err)
|
assert.Nil(t, err)
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -6,7 +6,6 @@ package mocks
|
|||||||
|
|
||||||
import mock "github.com/stretchr/testify/mock"
|
import mock "github.com/stretchr/testify/mock"
|
||||||
import model "github.com/mattermost/mattermost-server/model"
|
import model "github.com/mattermost/mattermost-server/model"
|
||||||
import store "github.com/mattermost/mattermost-server/store"
|
|
||||||
|
|
||||||
// JobStore is an autogenerated mock type for the JobStore type
|
// JobStore is an autogenerated mock type for the JobStore type
|
||||||
type JobStore struct {
|
type JobStore struct {
|
||||||
@@ -14,193 +13,293 @@ type JobStore struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Delete provides a mock function with given fields: id
|
// Delete provides a mock function with given fields: id
|
||||||
func (_m *JobStore) Delete(id string) store.StoreChannel {
|
func (_m *JobStore) Delete(id string) (string, *model.AppError) {
|
||||||
ret := _m.Called(id)
|
ret := _m.Called(id)
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 string
|
||||||
if rf, ok := ret.Get(0).(func(string) store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func(string) string); ok {
|
||||||
r0 = rf(id)
|
r0 = rf(id)
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
r0 = ret.Get(0).(string)
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
}
|
||||||
|
|
||||||
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func(string) *model.AppError); ok {
|
||||||
|
r1 = rf(id)
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|
||||||
// Get provides a mock function with given fields: id
|
// Get provides a mock function with given fields: id
|
||||||
func (_m *JobStore) Get(id string) store.StoreChannel {
|
func (_m *JobStore) Get(id string) (*model.Job, *model.AppError) {
|
||||||
ret := _m.Called(id)
|
ret := _m.Called(id)
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 *model.Job
|
||||||
if rf, ok := ret.Get(0).(func(string) store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func(string) *model.Job); ok {
|
||||||
r0 = rf(id)
|
r0 = rf(id)
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
if ret.Get(0) != nil {
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
r0 = ret.Get(0).(*model.Job)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func(string) *model.AppError); ok {
|
||||||
|
r1 = rf(id)
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetAllByStatus provides a mock function with given fields: status
|
// GetAllByStatus provides a mock function with given fields: status
|
||||||
func (_m *JobStore) GetAllByStatus(status string) store.StoreChannel {
|
func (_m *JobStore) GetAllByStatus(status string) ([]*model.Job, *model.AppError) {
|
||||||
ret := _m.Called(status)
|
ret := _m.Called(status)
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 []*model.Job
|
||||||
if rf, ok := ret.Get(0).(func(string) store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func(string) []*model.Job); ok {
|
||||||
r0 = rf(status)
|
r0 = rf(status)
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
if ret.Get(0) != nil {
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
r0 = ret.Get(0).([]*model.Job)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func(string) *model.AppError); ok {
|
||||||
|
r1 = rf(status)
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetAllByType provides a mock function with given fields: jobType
|
// GetAllByType provides a mock function with given fields: jobType
|
||||||
func (_m *JobStore) GetAllByType(jobType string) store.StoreChannel {
|
func (_m *JobStore) GetAllByType(jobType string) ([]*model.Job, *model.AppError) {
|
||||||
ret := _m.Called(jobType)
|
ret := _m.Called(jobType)
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 []*model.Job
|
||||||
if rf, ok := ret.Get(0).(func(string) store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func(string) []*model.Job); ok {
|
||||||
r0 = rf(jobType)
|
r0 = rf(jobType)
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
if ret.Get(0) != nil {
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
r0 = ret.Get(0).([]*model.Job)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func(string) *model.AppError); ok {
|
||||||
|
r1 = rf(jobType)
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetAllByTypePage provides a mock function with given fields: jobType, offset, limit
|
// GetAllByTypePage provides a mock function with given fields: jobType, offset, limit
|
||||||
func (_m *JobStore) GetAllByTypePage(jobType string, offset int, limit int) store.StoreChannel {
|
func (_m *JobStore) GetAllByTypePage(jobType string, offset int, limit int) ([]*model.Job, *model.AppError) {
|
||||||
ret := _m.Called(jobType, offset, limit)
|
ret := _m.Called(jobType, offset, limit)
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 []*model.Job
|
||||||
if rf, ok := ret.Get(0).(func(string, int, int) store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func(string, int, int) []*model.Job); ok {
|
||||||
r0 = rf(jobType, offset, limit)
|
r0 = rf(jobType, offset, limit)
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
if ret.Get(0) != nil {
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
r0 = ret.Get(0).([]*model.Job)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func(string, int, int) *model.AppError); ok {
|
||||||
|
r1 = rf(jobType, offset, limit)
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetAllPage provides a mock function with given fields: offset, limit
|
// GetAllPage provides a mock function with given fields: offset, limit
|
||||||
func (_m *JobStore) GetAllPage(offset int, limit int) store.StoreChannel {
|
func (_m *JobStore) GetAllPage(offset int, limit int) ([]*model.Job, *model.AppError) {
|
||||||
ret := _m.Called(offset, limit)
|
ret := _m.Called(offset, limit)
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 []*model.Job
|
||||||
if rf, ok := ret.Get(0).(func(int, int) store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func(int, int) []*model.Job); ok {
|
||||||
r0 = rf(offset, limit)
|
r0 = rf(offset, limit)
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
if ret.Get(0) != nil {
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
r0 = ret.Get(0).([]*model.Job)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func(int, int) *model.AppError); ok {
|
||||||
|
r1 = rf(offset, limit)
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetCountByStatusAndType provides a mock function with given fields: status, jobType
|
// GetCountByStatusAndType provides a mock function with given fields: status, jobType
|
||||||
func (_m *JobStore) GetCountByStatusAndType(status string, jobType string) store.StoreChannel {
|
func (_m *JobStore) GetCountByStatusAndType(status string, jobType string) (int64, *model.AppError) {
|
||||||
ret := _m.Called(status, jobType)
|
ret := _m.Called(status, jobType)
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 int64
|
||||||
if rf, ok := ret.Get(0).(func(string, string) store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func(string, string) int64); ok {
|
||||||
r0 = rf(status, jobType)
|
r0 = rf(status, jobType)
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
r0 = ret.Get(0).(int64)
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
}
|
||||||
|
|
||||||
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func(string, string) *model.AppError); ok {
|
||||||
|
r1 = rf(status, jobType)
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetNewestJobByStatusAndType provides a mock function with given fields: status, jobType
|
// GetNewestJobByStatusAndType provides a mock function with given fields: status, jobType
|
||||||
func (_m *JobStore) GetNewestJobByStatusAndType(status string, jobType string) store.StoreChannel {
|
func (_m *JobStore) GetNewestJobByStatusAndType(status string, jobType string) (*model.Job, *model.AppError) {
|
||||||
ret := _m.Called(status, jobType)
|
ret := _m.Called(status, jobType)
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 *model.Job
|
||||||
if rf, ok := ret.Get(0).(func(string, string) store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func(string, string) *model.Job); ok {
|
||||||
r0 = rf(status, jobType)
|
r0 = rf(status, jobType)
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
if ret.Get(0) != nil {
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
r0 = ret.Get(0).(*model.Job)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func(string, string) *model.AppError); ok {
|
||||||
|
r1 = rf(status, jobType)
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|
||||||
// Save provides a mock function with given fields: job
|
// Save provides a mock function with given fields: job
|
||||||
func (_m *JobStore) Save(job *model.Job) store.StoreChannel {
|
func (_m *JobStore) Save(job *model.Job) (*model.Job, *model.AppError) {
|
||||||
ret := _m.Called(job)
|
ret := _m.Called(job)
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 *model.Job
|
||||||
if rf, ok := ret.Get(0).(func(*model.Job) store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func(*model.Job) *model.Job); ok {
|
||||||
r0 = rf(job)
|
r0 = rf(job)
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
if ret.Get(0) != nil {
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
r0 = ret.Get(0).(*model.Job)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func(*model.Job) *model.AppError); ok {
|
||||||
|
r1 = rf(job)
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|
||||||
// UpdateOptimistically provides a mock function with given fields: job, currentStatus
|
// UpdateOptimistically provides a mock function with given fields: job, currentStatus
|
||||||
func (_m *JobStore) UpdateOptimistically(job *model.Job, currentStatus string) store.StoreChannel {
|
func (_m *JobStore) UpdateOptimistically(job *model.Job, currentStatus string) (bool, *model.AppError) {
|
||||||
ret := _m.Called(job, currentStatus)
|
ret := _m.Called(job, currentStatus)
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 bool
|
||||||
if rf, ok := ret.Get(0).(func(*model.Job, string) store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func(*model.Job, string) bool); ok {
|
||||||
r0 = rf(job, currentStatus)
|
r0 = rf(job, currentStatus)
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
r0 = ret.Get(0).(bool)
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
}
|
||||||
|
|
||||||
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func(*model.Job, string) *model.AppError); ok {
|
||||||
|
r1 = rf(job, currentStatus)
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|
||||||
// UpdateStatus provides a mock function with given fields: id, status
|
// UpdateStatus provides a mock function with given fields: id, status
|
||||||
func (_m *JobStore) UpdateStatus(id string, status string) store.StoreChannel {
|
func (_m *JobStore) UpdateStatus(id string, status string) (*model.Job, *model.AppError) {
|
||||||
ret := _m.Called(id, status)
|
ret := _m.Called(id, status)
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 *model.Job
|
||||||
if rf, ok := ret.Get(0).(func(string, string) store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func(string, string) *model.Job); ok {
|
||||||
r0 = rf(id, status)
|
r0 = rf(id, status)
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
if ret.Get(0) != nil {
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
r0 = ret.Get(0).(*model.Job)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func(string, string) *model.AppError); ok {
|
||||||
|
r1 = rf(id, status)
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|
||||||
// UpdateStatusOptimistically provides a mock function with given fields: id, currentStatus, newStatus
|
// UpdateStatusOptimistically provides a mock function with given fields: id, currentStatus, newStatus
|
||||||
func (_m *JobStore) UpdateStatusOptimistically(id string, currentStatus string, newStatus string) store.StoreChannel {
|
func (_m *JobStore) UpdateStatusOptimistically(id string, currentStatus string, newStatus string) (bool, *model.AppError) {
|
||||||
ret := _m.Called(id, currentStatus, newStatus)
|
ret := _m.Called(id, currentStatus, newStatus)
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 bool
|
||||||
if rf, ok := ret.Get(0).(func(string, string, string) store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func(string, string, string) bool); ok {
|
||||||
r0 = rf(id, currentStatus, newStatus)
|
r0 = rf(id, currentStatus, newStatus)
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
r0 = ret.Get(0).(bool)
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
}
|
||||||
|
|
||||||
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func(string, string, string) *model.AppError); ok {
|
||||||
|
r1 = rf(id, currentStatus, newStatus)
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|||||||
Ссылка в новой задаче
Block a user