Skip to content

Commit bf9747d

Browse files
committed
feat(podselector): update parameter format to match other annotation
1 parent 67cfd5d commit bf9747d

3 files changed

Lines changed: 10 additions & 11 deletions

File tree

charts/builder/values.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@ pull_policy: "Always"
33
docker_tag: canary
44
# limits_cpu: "100m"
55
# limits_memory: "50Mi"
6-
# builder_pod_node_selector: "cloud.google.com/gke-nodepool=builder"
6+
# builder_pod_node_selector: "disk:ssd"

pkg/gitreceive/build.go

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
package gitreceive
22

33
import (
4-
"bufio"
54
"bytes"
65
"encoding/json"
76
"fmt"
@@ -330,13 +329,12 @@ func build(
330329
func buildBuilderPodNodeSelector(config string) (map[string]string, error) {
331330
selector := make(map[string]string)
332331
if config != "" {
333-
line := bufio.NewScanner(strings.NewReader(config))
334-
for line.Scan() {
335-
splits := strings.Split(line.Text(), "=")
336-
if len(splits) != 2 {
332+
for _, line := range strings.Split(config, ",") {
333+
param := strings.Split(line, ":")
334+
if len(param) != 2 {
337335
return nil, fmt.Errorf("Invalid BuilderPodNodeSelector value format: %s", config)
338336
}
339-
selector[splits[0]] = splits[1]
337+
selector[strings.TrimSpace(param[0])] = strings.TrimSpace(param[1])
340338
}
341339
}
342340
return selector, nil

pkg/gitreceive/build_test.go

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -237,8 +237,9 @@ func TestBuildBuilderPodNodeSelector(t *testing.T) {
237237

238238
cazes := []podSelectorBuildCase{
239239
{"", emptyNodeSelector},
240-
{"pool=worker", map[string]string{"pool": "worker"}},
241-
{"pool=worker\nnetwork=fast", map[string]string{"pool": "worker", "network": "fast"}},
240+
{"pool:worker", map[string]string{"pool": "worker"}},
241+
{"pool:worker,network:fast", map[string]string{"pool": "worker", "network": "fast"}},
242+
{"pool:worker ,network:fast, disk:ssd", map[string]string{"pool": "worker", "network": "fast", "disk": "ssd"}},
242243
}
243244

244245
for _, caze := range cazes {
@@ -247,6 +248,6 @@ func TestBuildBuilderPodNodeSelector(t *testing.T) {
247248
assert.Equal(t, output, caze.Output, "pod selector")
248249
}
249250

250-
_, err := buildBuilderPodNodeSelector("failtest")
251-
assert.ExistsErr(t, err, "fail test")
251+
_, err := buildBuilderPodNodeSelector("invalidformat")
252+
assert.ExistsErr(t, err, "invalid format")
252253
}

0 commit comments

Comments
 (0)