luxolus/open-nomad
2
0
Fork 0
Code Issues 1 Pull requests Packages Releases Wiki Activity

compile

Browse source
This commit is contained in:
Alex Dadgar 2018-07-17 13:42:45 -07:00 committed by Michael Schurter
parent 6ebdf532ea
commit e1ba73b515
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
client/allocrunnerv2/taskrunner/task_runner.go
View file

@ -494,7 +494,7 @@ func (tr *TaskRunner) Restore(tx *bolt.Tx) error {
if err := clientstate.GetObject(bkt, taskStateKey, &ts); err != nil { if err := clientstate.GetObject(bkt, taskStateKey, &ts); err != nil {
return fmt.Errorf("failed to read task state: %v", err) return fmt.Errorf("failed to read task state: %v", err)
} }
tr.state = &tr tr.state = &ts
return nil return nil
} }
@ -512,7 +512,7 @@ func (tr *TaskRunner) SetState(state string, event *structs.TaskEvent) {
} }
// Append the event // Append the event
tr.emitEventImpl(tx, event) tr.emitEventImpl(event)
// Handle the state transition. // Handle the state transition.
switch state { switch state {
@ -558,7 +558,7 @@ func (tr *TaskRunner) SetState(state string, event *structs.TaskEvent) {
return err return err
} }
return bkt.PutObject(bkt, taskStateKey, tr.state) return clientstate.PutObject(bkt, taskStateKey, tr.state)
}) })
if err != nil { if err != nil {
// Only a warning because the next event/state-transition will // Only a warning because the next event/state-transition will
@ -591,8 +591,9 @@ func (tr *TaskRunner) EmitEvent(event *structs.TaskEvent) {
return err return err
} }
return bkt.PutObject(bkt, taskStateKey, tr.state) return clientstate.PutObject(bkt, taskStateKey, tr.state)
}) })
if err != nil { if err != nil {
// Only a warning because the next event/state-transition will // Only a warning because the next event/state-transition will
// try to persist it again. // try to persist it again.
@ -625,6 +626,8 @@ func (tr *TaskRunner) emitEventImpl(event *structs.TaskEvent) error {
// Append event to slice // Append event to slice
appendTaskEvent(tr.state, event) appendTaskEvent(tr.state, event)
return nil
} }
// WaitCh is closed when TaskRunner.Run exits. // WaitCh is closed when TaskRunner.Run exits.