516d641db0
Restoring calls NewAR -> Restore -> Run NewAR now calls NewTR AR.Restore calls TR.Restore AR.Run calls TR.Run
72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package state
|
|
|
|
import (
|
|
"github.com/hashicorp/nomad/client/structs"
|
|
"github.com/hashicorp/nomad/helper"
|
|
)
|
|
|
|
var (
|
|
// taskRunnerStateAllKey holds all the task runners state. At the moment
|
|
// there is no need to split it
|
|
//XXX refactor out of client/state and taskrunner
|
|
taskRunnerStateAllKey = []byte("simple-all")
|
|
)
|
|
|
|
// LocalState is Task state which is persisted for use when restarting Nomad
|
|
// agents.
|
|
type LocalState struct {
|
|
Hooks map[string]*HookState
|
|
|
|
// DriverNetwork is the network information returned by the task
|
|
// driver's Start method
|
|
DriverNetwork *structs.DriverNetwork
|
|
}
|
|
|
|
func NewLocalState() *LocalState {
|
|
return &LocalState{
|
|
Hooks: make(map[string]*HookState),
|
|
}
|
|
}
|
|
|
|
// Copy should be called with the lock held
|
|
func (s *LocalState) Copy() *LocalState {
|
|
// Create a copy
|
|
c := &LocalState{
|
|
Hooks: make(map[string]*HookState, len(s.Hooks)),
|
|
DriverNetwork: s.DriverNetwork,
|
|
}
|
|
|
|
// Copy the hooks
|
|
for h, state := range s.Hooks {
|
|
c.Hooks[h] = state.Copy()
|
|
}
|
|
|
|
return c
|
|
}
|
|
|
|
type HookState struct {
|
|
// Prestart is true if the hook has run Prestart successfully and does
|
|
// not need to run again
|
|
PrestartDone bool
|
|
Data map[string]string
|
|
}
|
|
|
|
func (h *HookState) Copy() *HookState {
|
|
c := new(HookState)
|
|
*c = *h
|
|
c.Data = helper.CopyMapStringString(c.Data)
|
|
return c
|
|
}
|
|
|
|
func (h *HookState) Equal(o *HookState) bool {
|
|
if h == nil || o == nil {
|
|
return h == o
|
|
}
|
|
|
|
if h.PrestartDone != o.PrestartDone {
|
|
return false
|
|
}
|
|
|
|
return helper.CompareMapStringString(h.Data, o.Data)
|
|
}
|