summaryrefslogtreecommitdiff
path: root/worker/state.go
blob: bf152e8b598a9fdf3e10521be629fdd98ba3d109 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package worker

import (
	"sync/atomic"
)

// SYNC WITH worker_watcher.GET
const (
	// StateInactive - no associated process
	StateInactive int64 = iota

	// StateReady - ready for job.
	StateReady

	// StateWorking - working on given payload.
	StateWorking

	// StateInvalid - indicates that WorkerProcess is being disabled and will be removed.
	StateInvalid

	// StateStopping - process is being softly stopped.
	StateStopping

	// StateKilling - process is being forcibly stopped
	StateKilling

	// StateDestroyed State of worker, when no need to allocate new one
	StateDestroyed

	// StateMaxJobsReached State of worker, when it reached executions limit
	StateMaxJobsReached

	// StateStopped - process has been terminated.
	StateStopped

	// StateErrored - error StateImpl (can't be used).
	StateErrored
)

type StateImpl struct {
	value    int64
	numExecs uint64
	// to be lightweight, use UnixNano
	lastUsed uint64
}

// NewWorkerState initializes a state for the sync.Worker
func NewWorkerState(value int64) *StateImpl {
	return &StateImpl{value: value}
}

// String returns current StateImpl as string.
func (s *StateImpl) String() string {
	switch s.Value() {
	case StateInactive:
		return "inactive"
	case StateReady:
		return "ready"
	case StateWorking:
		return "working"
	case StateInvalid:
		return "invalid"
	case StateStopping:
		return "stopping"
	case StateStopped:
		return "stopped"
	case StateKilling:
		return "killing"
	case StateErrored:
		return "errored"
	case StateDestroyed:
		return "destroyed"
	}

	return "undefined"
}

// NumExecs returns number of registered WorkerProcess execs.
func (s *StateImpl) NumExecs() uint64 {
	return atomic.LoadUint64(&s.numExecs)
}

// Value StateImpl returns StateImpl value
func (s *StateImpl) Value() int64 {
	return atomic.LoadInt64(&s.value)
}

// IsActive returns true if WorkerProcess not Inactive or Stopped
func (s *StateImpl) IsActive() bool {
	val := s.Value()
	return val == StateWorking || val == StateReady
}

// Set change StateImpl value (status)
func (s *StateImpl) Set(value int64) {
	atomic.StoreInt64(&s.value, value)
}

// RegisterExec register new execution atomically
func (s *StateImpl) RegisterExec() {
	atomic.AddUint64(&s.numExecs, 1)
}

// SetLastUsed Update last used time
func (s *StateImpl) SetLastUsed(lu uint64) {
	atomic.StoreUint64(&s.lastUsed, lu)
}

func (s *StateImpl) LastUsed() uint64 {
	return atomic.LoadUint64(&s.lastUsed)
}