(7.0) Add ability to override FSM spec (#2037)

gravitational · Aug 24, 2020 · f4ae2c6 · f4ae2c6
1 parent 80cccfb
commit f4ae2c6
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/e b/e
diff --git a/lib/install/fsmspec.go b/lib/install/fsmspec.go
@@ -29,9 +29,18 @@ import (
 	"k8s.io/client-go/kubernetes"
 )
 
-// FSMSpec returns a function that returns an appropriate phase executor
-// based on the provided params
-func FSMSpec(config FSMConfig) fsm.FSMSpecFunc {
+// FSMSpecFunc defines a function that returns install FSM spec based on the config.
+type FSMSpecFunc func(FSMConfig) fsm.FSMSpecFunc
+
+// FSMSpec is the install FSM spec.
+//
+// It may be overriden by external implementations to support additional
+// install operation phases (e.g. by the enterprise version).
+var FSMSpec FSMSpecFunc = DefaultFSMSpec
+
+// DefaultFSMSpec returns a function that returns an the default install FSM
+// spec for the provided install FSM config.
+func DefaultFSMSpec(config FSMConfig) fsm.FSMSpecFunc {
 	return func(p fsm.ExecutorParams, remote fsm.Remote) (fsm.PhaseExecutor, error) {
 		switch {
 		case strings.HasPrefix(p.Phase.ID, phases.InitPhase):