Skip to content

Commit 3800918

Browse files
committed
add buildEnv for storage
1 parent 107713c commit 3800918

File tree

1 file changed

+37
-0
lines changed

1 file changed

+37
-0
lines changed

internal/resources/storage_statefulset.go

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,42 @@ func (b *StorageStatefulSetBuilder) Build(obj client.Object) error {
102102
return nil
103103
}
104104

105+
func (b *StorageStatefulSetBuilder) buildEnv() []corev1.EnvVar {
106+
var envVars []corev1.EnvVar
107+
108+
envVars = append(envVars,
109+
corev1.EnvVar{
110+
Name: "POD_NAME", // for `--grpc-public-host` flag
111+
ValueFrom: &corev1.EnvVarSource{
112+
FieldRef: &corev1.ObjectFieldSelector{
113+
APIVersion: "v1",
114+
FieldPath: "metadata.name",
115+
},
116+
},
117+
},
118+
corev1.EnvVar{
119+
Name: "NODE_NAME", // for `--grpc-public-host` flag
120+
ValueFrom: &corev1.EnvVarSource{
121+
FieldRef: &corev1.ObjectFieldSelector{
122+
APIVersion: "v1",
123+
FieldPath: "metadata.name",
124+
},
125+
},
126+
},
127+
corev1.EnvVar{
128+
Name: "POD_IP", // for `--grpc-public-address-<ip-family>` flag
129+
ValueFrom: &corev1.EnvVarSource{
130+
FieldRef: &corev1.ObjectFieldSelector{
131+
APIVersion: "v1",
132+
FieldPath: "status.podIP",
133+
},
134+
},
135+
},
136+
)
137+
138+
return envVars
139+
}
140+
105141
func (b *StorageStatefulSetBuilder) buildPodTemplateLabels() labels.Labels {
106142
podTemplateLabels := labels.Labels{}
107143

@@ -404,6 +440,7 @@ func (b *StorageStatefulSetBuilder) buildContainer() corev1.Container { // todo
404440
ImagePullPolicy: imagePullPolicy,
405441
Command: command,
406442
Args: args,
443+
Env: b.buildEnv(),
407444

408445
SecurityContext: mergeSecurityContextWithDefaults(b.Spec.SecurityContext),
409446

0 commit comments

Comments
 (0)