Explicit list of switch roles and IsSpine helper

githedgehog · Nov 10, 2023 · 57e0a5c · 57e0a5c
1 parent 8fa1a16
commit 57e0a5c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/api/wiring/v1alpha2/switch_types.go b/api/wiring/v1alpha2/switch_types.go
@@ -38,6 +38,16 @@ const (
 	SwitchRoleBorderLeaf SwitchRole = "border-leaf"
 )
 
+var SwitchRoles = []SwitchRole{
+	SwitchRoleSpine,
+	SwitchRoleServerLeaf,
+	SwitchRoleBorderLeaf,
+}
+
+func (r SwitchRole) IsSpine() bool {
+	return r == SwitchRoleSpine
+}
+
 func (r SwitchRole) IsLeaf() bool {
 	return r == SwitchRoleServerLeaf || r == SwitchRoleBorderLeaf
 }