!60 build: wait all builder goroutines when stop daemon

Merge pull request !60 from zvier/bep

daemon/backend.go

@@ -26,6 +26,7 @@ type Backend struct {
 	sync.RWMutex
 	daemon *Daemon
 	status map[string]*status
+	wg     sync.WaitGroup
 }
 
 // NewBackend create an instance of backend
@@ -33,6 +34,7 @@ func (d *Daemon) NewBackend() {
 	d.backend = &Backend{
 		daemon: d,
 		status: make(map[string]*status),
+		wg:     sync.WaitGroup{},
 	}
 }
 

daemon/build.go

@@ -31,6 +31,8 @@ import (
 
 // Build receives a build request and build an image
 func (b *Backend) Build(req *pb.BuildRequest, stream pb.Control_BuildServer) (err error) { // nolint:gocyclo
+	b.wg.Add(1)
+	defer b.wg.Done()
 	logrus.WithFields(logrus.Fields{
 		"BuildType": req.GetBuildType(),
 		"BuildID":   req.GetBuildID(),

daemon/daemon.go

@@ -126,6 +126,7 @@ func (d *Daemon) Run() error {
 
 	systemd.NotifySystemStopping()
 	d.grpc.server.GracefulStop()
+	d.backend.wg.Wait()
 	return err
 }