harness-drone/drone/agent/exec.go

81 lines
1.5 KiB
Go
Raw Normal View History

2016-04-20 01:37:53 +00:00
package agent
import (
"io"
"time"
"github.com/Sirupsen/logrus"
"github.com/drone/drone/agent"
"github.com/drone/drone/build/docker"
2016-04-20 01:37:53 +00:00
"github.com/drone/drone/client"
"github.com/samalba/dockerclient"
)
2016-04-22 00:10:19 +00:00
type config struct {
platform string
namespace string
privileged []string
pull bool
logs int64
timeout time.Duration
2016-04-22 00:10:19 +00:00
}
type pipeline struct {
drone client.Client
docker dockerclient.Client
config config
2016-04-20 01:37:53 +00:00
}
2016-04-22 00:10:19 +00:00
func (r *pipeline) run() error {
w, err := r.drone.Pull("linux", "amd64")
2016-04-20 01:37:53 +00:00
if err != nil {
return err
}
logrus.Infof("Starting build %s/%s#%d.%d",
w.Repo.Owner, w.Repo.Name, w.Build.Number, w.Job.Number)
cancel := make(chan bool, 1)
engine := docker.NewClient(r.docker)
2016-04-20 01:37:53 +00:00
// streaming the logs
rc, wc := io.Pipe()
defer func() {
wc.Close()
rc.Close()
}()
2016-04-20 01:37:53 +00:00
a := agent.Agent{
Update: agent.NewClientUpdater(r.drone),
Logger: agent.NewClientLogger(r.drone, w.Job.ID, rc, wc, r.config.logs),
Engine: engine,
Timeout: r.config.timeout,
Platform: "linux/amd64",
Namespace: r.config.namespace,
Escalate: r.config.privileged,
Pull: r.config.pull,
2016-04-20 01:37:53 +00:00
}
// signal for canceling the build.
2016-04-22 00:10:19 +00:00
wait := r.drone.Wait(w.Job.ID)
defer wait.Cancel()
2016-04-20 01:37:53 +00:00
go func() {
if _, err := wait.Done(); err == nil {
cancel <- true
2016-04-20 01:37:53 +00:00
logrus.Infof("Cancel build %s/%s#%d.%d",
w.Repo.Owner, w.Repo.Name, w.Build.Number, w.Job.Number)
}
}()
a.Run(w, cancel)
2016-04-20 01:37:53 +00:00
wc.Close()
rc.Close()
logrus.Infof("Finished build %s/%s#%d.%d",
w.Repo.Owner, w.Repo.Name, w.Build.Number, w.Job.Number)
return nil
}