-
Notifications
You must be signed in to change notification settings - Fork 3.1k
/
always.go
37 lines (30 loc) · 973 Bytes
/
always.go
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
package fake
import (
wfv1 "github.com/argoproj/argo/v3/pkg/apis/workflow/v1alpha1"
testutil "github.com/argoproj/argo/v3/test/util"
"github.com/argoproj/argo/v3/workflow/hydrator"
)
// this test fake is nearly a Reference Implementation
type always struct{}
func (i always) IsHydrated(wf *wfv1.Workflow) bool {
return wf.Status.OffloadNodeStatusVersion == ""
}
func (i always) Hydrate(wf *wfv1.Workflow) error {
if !i.IsHydrated(wf) {
testutil.MustUnmarshallJSON(wf.Status.OffloadNodeStatusVersion, &wf.Status.Nodes)
wf.Status.OffloadNodeStatusVersion = ""
}
return nil
}
func (i always) Dehydrate(wf *wfv1.Workflow) error {
if i.IsHydrated(wf) {
wf.Status.OffloadNodeStatusVersion = testutil.MustMarshallJSON(&wf.Status.Nodes)
wf.Status.Nodes = nil
}
return nil
}
func (i always) HydrateWithNodes(wf *wfv1.Workflow, nodes wfv1.Nodes) {
wf.Status.Nodes = nodes
wf.Status.OffloadNodeStatusVersion = ""
}
var Always hydrator.Interface = &always{}