This repository has been archived by the owner on Oct 9, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 60
/
utils.go
91 lines (72 loc) · 1.89 KB
/
utils.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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package k8s
import (
"math"
v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"github.com/golang/protobuf/ptypes"
"github.com/lyft/flyteidl/gen/pb-go/flyteidl/core"
"github.com/lyft/flytepropeller/pkg/apis/flyteworkflow/v1alpha1"
)
func refInt(i int) *int {
return &i
}
func refStr(s string) *string {
return &s
}
func computeRetryStrategy(n *core.Node, t *core.TaskTemplate) *v1alpha1.RetryStrategy {
if n.GetMetadata() != nil && n.GetMetadata().GetRetries() != nil {
return &v1alpha1.RetryStrategy{
MinAttempts: refInt(int(n.GetMetadata().GetRetries().Retries + 1)),
}
}
if t != nil && t.GetMetadata() != nil && t.GetMetadata().GetRetries() != nil {
return &v1alpha1.RetryStrategy{
MinAttempts: refInt(int(t.GetMetadata().GetRetries().Retries + 1)),
}
}
return nil
}
func computeDeadline(n *core.Node) (*v1.Duration, error) {
var deadline *v1.Duration
if n.GetMetadata() != nil && n.GetMetadata().GetTimeout() != nil {
duration, err := ptypes.Duration(n.GetMetadata().GetTimeout())
if err != nil {
return nil, err
}
deadline = &v1.Duration{
Duration: duration,
}
}
return deadline, nil
}
func getResources(task *core.TaskTemplate) *core.Resources {
if task == nil {
return nil
}
if task.GetContainer() == nil {
return nil
}
return task.GetContainer().Resources
}
func toAliasValueArray(aliases []*core.Alias) []v1alpha1.Alias {
if aliases == nil {
return nil
}
res := make([]v1alpha1.Alias, 0, len(aliases))
for _, alias := range aliases {
res = append(res, v1alpha1.Alias{Alias: *alias})
}
return res
}
func toBindingValueArray(bindings []*core.Binding) []*v1alpha1.Binding {
if bindings == nil {
return nil
}
res := make([]*v1alpha1.Binding, 0, len(bindings))
for _, binding := range bindings {
res = append(res, &v1alpha1.Binding{Binding: binding})
}
return res
}
func minInt(i, j int) int {
return int(math.Min(float64(i), float64(j)))
}