signal handling for source job

cmd/config_job_source.go

@@ -3,9 +3,13 @@ package cmd
 import (
 	"time"
 
-	"github.com/zrepl/zrepl/rpc"
 	mapstructure "github.com/mitchellh/mapstructure"
 	"github.com/pkg/errors"
+	"github.com/zrepl/zrepl/rpc"
+	"io"
+	"os"
+	"os/signal"
+	"syscall"
 )
 
 type SourceJob struct {
@@ -64,20 +68,36 @@ func (j *SourceJob) JobName() string {
 
 func (j *SourceJob) JobDo(log Logger) (err error) {
 
-	// Setup automatic snapshotting
-
 	listener, err := j.Serve.Listen()
 	if err != nil {
 		return err
 	}
 
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
+
+	rwcChan := make(chan io.ReadWriteCloser)
+
+	// Serve connections until interrupted or error
+outer:
 	for {
 
-		// listener does auth for us
+		go func() {
 			rwc, err := listener.Accept()
 			if err != nil {
-			// if err != AuthError...
+				log.Printf("error accepting connection: %s", err)
-			panic(err) // TODO
+				close(rwcChan)
+				return
+			}
+			rwcChan <- rwc
+		}()
+
+		select {
+
+		case rwc, notClosed := <-rwcChan:
+
+			if !notClosed {
+				break outer // closed because of accept error
 			}
 
 			// construct connection handler
@@ -88,13 +108,31 @@ func (j *SourceJob) JobDo(log Logger) (err error) {
 			}
 
 			// handle connection
-		server := rpc.NewServer(rwc)
+			rpcServer := rpc.NewServer(rwc)
-		registerEndpoints(server, handler)
+			registerEndpoints(rpcServer, handler)
-		if err = server.Serve(); err != nil {
+			if err = rpcServer.Serve(); err != nil {
 				log.Printf("error serving connection: %s", err)
 			}
-
 			rwc.Close()
+
+		case sig := <-sigChan:
+
+			log.Printf("%s received", sig)
+			break outer
+
 		}
 
 	}
+
+	signal.Stop(sigChan)
+	close(sigChan)
+
+	log.Printf("closing listener")
+	err = listener.Close()
+	if err != nil {
+		log.Printf("error closing listener: %s", err)
+	}
+
+	return nil
+
+}