65 lines
2.0 KiB
Go
65 lines
2.0 KiB
Go
package builder
|
|
|
|
import corev1api "k8s.io/api/core/v1"
|
|
|
|
// NodeSelectorBuilder builds NodeSelector objects
|
|
type NodeSelectorBuilder struct {
|
|
object *corev1api.NodeSelector
|
|
}
|
|
|
|
// ForNodeSelector returns the NodeSelectorBuilder instance with given terms
|
|
func ForNodeSelector(term ...corev1api.NodeSelectorTerm) *NodeSelectorBuilder {
|
|
return &NodeSelectorBuilder{
|
|
object: &corev1api.NodeSelector{
|
|
NodeSelectorTerms: term,
|
|
},
|
|
}
|
|
}
|
|
|
|
// Result returns the built NodeSelector
|
|
func (b *NodeSelectorBuilder) Result() *corev1api.NodeSelector {
|
|
return b.object
|
|
}
|
|
|
|
// NodeSelectorTermBuilder builds NodeSelectorTerm objects.
|
|
type NodeSelectorTermBuilder struct {
|
|
object *corev1api.NodeSelectorTerm
|
|
}
|
|
|
|
// NewNodeSelectorTermBuilder initializes an instance of NodeSelectorTermBuilder
|
|
func NewNodeSelectorTermBuilder() *NodeSelectorTermBuilder {
|
|
return &NodeSelectorTermBuilder{
|
|
object: &corev1api.NodeSelectorTerm{
|
|
MatchExpressions: make([]corev1api.NodeSelectorRequirement, 0),
|
|
MatchFields: make([]corev1api.NodeSelectorRequirement, 0),
|
|
},
|
|
}
|
|
}
|
|
|
|
// WithMatchExpression appends the MatchExpression to the NodeSelectorTerm
|
|
func (ntb *NodeSelectorTermBuilder) WithMatchExpression(key string, op string, values ...string) *NodeSelectorTermBuilder {
|
|
req := corev1api.NodeSelectorRequirement{
|
|
Key: key,
|
|
Operator: corev1api.NodeSelectorOperator(op),
|
|
Values: values,
|
|
}
|
|
ntb.object.MatchExpressions = append(ntb.object.MatchExpressions, req)
|
|
return ntb
|
|
}
|
|
|
|
// WithMatchField appends the MatchField to the NodeSelectorTerm
|
|
func (ntb *NodeSelectorTermBuilder) WithMatchField(key string, op string, values ...string) *NodeSelectorTermBuilder {
|
|
req := corev1api.NodeSelectorRequirement{
|
|
Key: key,
|
|
Operator: corev1api.NodeSelectorOperator(op),
|
|
Values: values,
|
|
}
|
|
ntb.object.MatchFields = append(ntb.object.MatchFields, req)
|
|
return ntb
|
|
}
|
|
|
|
// Result returns the built NodeSelectorTerm
|
|
func (ntb *NodeSelectorTermBuilder) Result() *corev1api.NodeSelectorTerm {
|
|
return ntb.object
|
|
}
|