forked from openshift/origin
-
Notifications
You must be signed in to change notification settings - Fork 1
/
buildpodutil.go
73 lines (60 loc) · 1.95 KB
/
buildpodutil.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
package common
import (
"errors"
"fmt"
"github.com/openshift/origin/pkg/build/buildscheme"
corev1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/runtime"
buildv1 "github.com/openshift/api/build/v1"
)
// GetBuildFromPod returns a build object encoded in a pod's BUILD environment variable along with
// its encoding version
func GetBuildFromPod(pod *corev1.Pod) (*buildv1.Build, error) {
if len(pod.Spec.Containers) == 0 {
return nil, errors.New("unable to get build from pod: pod has no containers")
}
buildEnvVar := getEnvVar(&pod.Spec.Containers[0], "BUILD")
if len(buildEnvVar) == 0 {
return nil, errors.New("unable to get build from pod: BUILD environment variable is empty")
}
obj, err := runtime.Decode(buildscheme.Decoder, []byte(buildEnvVar))
if err != nil {
return nil, fmt.Errorf("unable to get build from pod: %v", err)
}
build, ok := obj.(*buildv1.Build)
if !ok {
return nil, fmt.Errorf("unable to get build from pod: %v", errors.New("decoded object is not of type Build"))
}
return build, nil
}
// SetBuildInPod encodes a build object and sets it in a pod's BUILD environment variable
func SetBuildInPod(pod *corev1.Pod, build *buildv1.Build) error {
encodedBuild, err := runtime.Encode(buildscheme.Encoder, build)
if err != nil {
return fmt.Errorf("unable to set build in pod: %v", err)
}
for i := range pod.Spec.Containers {
setEnvVar(&pod.Spec.Containers[i], "BUILD", string(encodedBuild))
}
for i := range pod.Spec.InitContainers {
setEnvVar(&pod.Spec.InitContainers[i], "BUILD", string(encodedBuild))
}
return nil
}
func getEnvVar(c *corev1.Container, name string) string {
for _, envVar := range c.Env {
if envVar.Name == name {
return envVar.Value
}
}
return ""
}
func setEnvVar(c *corev1.Container, name, value string) {
for i, envVar := range c.Env {
if envVar.Name == name {
c.Env[i] = corev1.EnvVar{Name: name, Value: value}
return
}
}
c.Env = append(c.Env, corev1.EnvVar{Name: name, Value: value})
}