-
Notifications
You must be signed in to change notification settings - Fork 26
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Remove condition managers interfaces
- Loading branch information
Showing
5 changed files
with
192 additions
and
263 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,178 @@ | ||
| package components | ||
|
|
||
| import ( | ||
| "context" | ||
|
|
||
| "k8s.io/apimachinery/pkg/api/meta" | ||
| metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" | ||
| "sigs.k8s.io/controller-runtime/pkg/client" | ||
|
|
||
| ytv1 "github.com/ytsaurus/yt-k8s-operator/api/v1" | ||
| "github.com/ytsaurus/yt-k8s-operator/pkg/apiproxy" | ||
| ) | ||
|
|
||
| type baseStateManager struct { | ||
| client client.Client | ||
| ytsaurus *ytv1.Ytsaurus | ||
| } | ||
|
|
||
| type ConditionManager struct { | ||
| baseStateManager | ||
| } | ||
|
|
||
| type StateManager struct { | ||
| baseStateManager | ||
| } | ||
|
|
||
| func NewConditionManagerFromYtsaurus(ytsaurus *apiproxy.Ytsaurus) *ConditionManager { | ||
| return NewConditionManager( | ||
| ytsaurus.APIProxy().Client(), | ||
| ytsaurus.GetResource(), | ||
| ) | ||
| } | ||
|
|
||
| func NewConditionManager(client client.Client, ytsaurus *ytv1.Ytsaurus) *ConditionManager { | ||
| return &ConditionManager{ | ||
| baseStateManager{ | ||
| client: client, | ||
| ytsaurus: ytsaurus, | ||
| }, | ||
| } | ||
| } | ||
|
|
||
| func NewStateManagerFromYtsaurus(ytsaurus *apiproxy.Ytsaurus) *StateManager { | ||
| return NewStateManager( | ||
| ytsaurus.APIProxy().Client(), | ||
| ytsaurus.GetResource(), | ||
| ) | ||
| } | ||
|
|
||
| func NewStateManager(client client.Client, ytsaurus *ytv1.Ytsaurus) *StateManager { | ||
| return &StateManager{ | ||
| baseStateManager{ | ||
| client: client, | ||
| ytsaurus: ytsaurus, | ||
| }, | ||
| } | ||
| } | ||
|
|
||
| func (m *baseStateManager) updateStatus(ctx context.Context, change func(ytsaurusResource *ytv1.Ytsaurus)) error { | ||
| change(m.ytsaurus) | ||
| return m.client.Status().Update(ctx, m.ytsaurus) | ||
| } | ||
|
|
||
| func (cm *ConditionManager) SetTrue(ctx context.Context, condName ConditionName) error { | ||
| return cm.SetTrueMsg(ctx, condName, "") | ||
| } | ||
| func (cm *ConditionManager) SetTrueMsg(ctx context.Context, condName ConditionName, msg string) error { | ||
| return cm.SetMsg(ctx, condName, true, msg) | ||
| } | ||
| func (cm *ConditionManager) SetFalse(ctx context.Context, condName ConditionName) error { | ||
| return cm.SetFalseMsg(ctx, condName, "") | ||
| } | ||
| func (cm *ConditionManager) SetFalseMsg(ctx context.Context, condName ConditionName, msg string) error { | ||
| return cm.SetMsg(ctx, condName, false, msg) | ||
| } | ||
| func (cm *ConditionManager) Set(ctx context.Context, condName ConditionName, val bool) error { | ||
| return cm.SetMsg(ctx, condName, val, "") | ||
| } | ||
| func (cm *ConditionManager) SetCond(ctx context.Context, cond Condition) error { | ||
| return cm.SetMsg(ctx, cond.Name, cond.Val, "") | ||
| } | ||
| func (cm *ConditionManager) SetCondMany(ctx context.Context, conds ...Condition) error { | ||
| var metaconds []metav1.Condition | ||
| for _, cond := range conds { | ||
| metaconds = append(metaconds, cm.buildCond(cond.Name, cond.Val, "")) | ||
| } | ||
| return cm.updateStatus(ctx, func(ytsaurus *ytv1.Ytsaurus) { | ||
| for _, metacond := range metaconds { | ||
| meta.SetStatusCondition(&ytsaurus.Status.Conditions, metacond) | ||
| } | ||
| }) | ||
| } | ||
| func (cm *ConditionManager) SetCondMsg(ctx context.Context, cond Condition, msg string) error { | ||
| return cm.SetMsg(ctx, cond.Name, cond.Val, msg) | ||
| } | ||
| func (cm *ConditionManager) buildCond(condName ConditionName, val bool, msg string) metav1.Condition { | ||
| return metav1.Condition{ | ||
| Type: string(condName), | ||
| Status: map[bool]metav1.ConditionStatus{ | ||
| true: metav1.ConditionTrue, | ||
| false: metav1.ConditionFalse, | ||
| }[val], | ||
| // DO we need better reason? | ||
| Reason: string(condName), | ||
| Message: msg, | ||
| } | ||
| } | ||
| func (cm *ConditionManager) SetMsg(ctx context.Context, condName ConditionName, val bool, msg string) error { | ||
| metacond := cm.buildCond(condName, val, msg) | ||
| return cm.updateStatus(ctx, func(ytsaurus *ytv1.Ytsaurus) { | ||
| meta.SetStatusCondition(&ytsaurus.Status.Conditions, metacond) | ||
| }) | ||
| } | ||
| func (cm *ConditionManager) IsTrue(condName ConditionName) bool { | ||
| return meta.IsStatusConditionTrue(cm.ytsaurus.Status.Conditions, string(condName)) | ||
| } | ||
| func (cm *ConditionManager) IsFalse(condName ConditionName) bool { | ||
| return !cm.IsTrue(condName) | ||
| } | ||
| func (cm *ConditionManager) Is(cond Condition) bool { | ||
| return cm.IsSatisfied(cond) | ||
| } | ||
| func (cm *ConditionManager) All(conds ...Condition) bool { | ||
| for _, cond := range conds { | ||
| if cm.IsNotSatisfied(cond) { | ||
| return false | ||
| } | ||
| } | ||
| return true | ||
| } | ||
| func (cm *ConditionManager) Any(conds ...Condition) bool { | ||
| for _, cond := range conds { | ||
| if cm.IsSatisfied(cond) { | ||
| return true | ||
| } | ||
| } | ||
| return false | ||
| } | ||
| func (cm *ConditionManager) IsSatisfied(cond Condition) bool { | ||
| return cm.IsTrue(cond.Name) == cond.Val | ||
| } | ||
| func (cm *ConditionManager) IsNotSatisfied(cond Condition) bool { | ||
| return !cm.IsSatisfied(cond) | ||
| } | ||
| func (cm *ConditionManager) Get(condName ConditionName) bool { | ||
| if cm.IsTrue(condName) { | ||
| return true | ||
| } else { | ||
| return false | ||
| } | ||
| } | ||
|
|
||
| func (cm *StateManager) SetClusterState(ctx context.Context, clusterState ytv1.ClusterState) error { | ||
| return cm.updateStatus(ctx, func(ytsaurus *ytv1.Ytsaurus) { | ||
| ytsaurus.Status.State = clusterState | ||
| }) | ||
| } | ||
| func (cm *StateManager) SetClusterUpdateState(ctx context.Context, updateState ytv1.UpdateState) error { | ||
| return cm.updateStatus(ctx, func(ytsaurus *ytv1.Ytsaurus) { | ||
| ytsaurus.Status.UpdateStatus.State = updateState | ||
| }) | ||
| } | ||
| func (cm *StateManager) SetTabletCellBundles(ctx context.Context, cells []ytv1.TabletCellBundleInfo) error { | ||
| return cm.updateStatus(ctx, func(ytsaurus *ytv1.Ytsaurus) { | ||
| ytsaurus.Status.UpdateStatus.TabletCellBundles = cells | ||
| }) | ||
| } | ||
| func (cm *StateManager) SetMasterMonitoringPaths(ctx context.Context, paths []string) error { | ||
| return cm.updateStatus(ctx, func(ytsaurus *ytv1.Ytsaurus) { | ||
| ytsaurus.Status.UpdateStatus.MasterMonitoringPaths = paths | ||
| }) | ||
| } | ||
| func (cm *StateManager) GetTabletCellBundles() []ytv1.TabletCellBundleInfo { | ||
| return cm.ytsaurus.Status.UpdateStatus.TabletCellBundles | ||
| } | ||
| func (cm *StateManager) GetMasterMonitoringPaths() []string { | ||
| return cm.ytsaurus.Status.UpdateStatus.MasterMonitoringPaths | ||
| } |
Oops, something went wrong.