serve.go

package services

import (
	"os"
	"os/signal"
	"syscall"

	"github.com/pkg/errors"
	log "github.com/sirupsen/logrus"
)

type Serve struct {
	servables []Servable
}

type Servable interface {
	Serve() error
}

func NewServe(s ...Servable) *Serve {
	return &Serve{servables: s}
}

func (s *Serve) Serve() error {

	serveError := make(chan error, 1)

	for _, ss := range s.servables {
		go func(sss Servable) {
			err := sss.Serve()
			serveError <- err
		}(ss)
	}

	sigs := make(chan os.Signal, 1)
	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
	select {
	case sig := <-sigs:
		log.WithField("signal", sig).Info("Got syscall")
	case err := <-serveError:
		return errors.Wrap(err, "Got serve error")
	}
	log.Info("Shooting down... at last!")
	return nil
}