choose ambassador by platform

This commit is contained in:
Brad Rydzewski 2016-08-25 11:35:28 -07:00
parent 6d2c525064
commit 0f7253febb
2 changed files with 38 additions and 5 deletions

View file

@ -167,7 +167,7 @@ func (a *Agent) prep(w *queue.Work) (*yaml.Config, error) {
transform.ImageVolume(conf, []string{a.Local + ":" + conf.Workspace.Path})
}
transform.Pod(conf)
transform.Pod(conf, a.Platform)
return conf, nil
}

View file

@ -9,21 +9,54 @@ import (
"github.com/gorilla/securecookie"
)
type ambassador struct {
image string
entrypoint []string
command []string
}
// default linux amd64 ambassador
var defaultAmbassador = ambassador{
image: "busybox:latest",
entrypoint: []string{"/bin/sleep"},
command: []string{"86400"},
}
// lookup ambassador configuration by architecture and os
var lookupAmbassador = map[string]ambassador{
"linux/amd64": {
image: "busybox:latest",
entrypoint: []string{"/bin/sleep"},
command: []string{"86400"},
},
"linux/arm": {
image: "armhf/alpine:latest",
entrypoint: []string{"/bin/sleep"},
command: []string{"86400"},
},
}
// Pod transforms the containers in the Yaml to use Pod networking, where every
// container shares the localhost connection.
func Pod(c *yaml.Config) error {
func Pod(c *yaml.Config, platform string) error {
rand := base64.RawURLEncoding.EncodeToString(
securecookie.GenerateRandomKey(8),
)
// choose the ambassador configuration based on os and architecture
conf, ok := lookupAmbassador[platform]
if !ok {
conf = defaultAmbassador
}
ambassador := &yaml.Container{
ID: fmt.Sprintf("drone_ambassador_%s", rand),
Name: "ambassador",
Image: "busybox:latest",
Image: conf.image,
Detached: true,
Entrypoint: []string{"/bin/sleep"},
Command: []string{"86400"},
Entrypoint: conf.entrypoint,
Command: conf.command,
Volumes: []string{c.Workspace.Path, c.Workspace.Base},
Environment: map[string]string{},
}