host: Separate out the jobStream cleanup logic.

This makes it easier to comprehend, since we can now move it to where the
variable is actually being used.

Signed-off-by: Blaž Hrastnik <blaz.hrast@gmail.com>

host/host.go

@@ -175,13 +175,7 @@ func runDaemon(args *docopt.Args) {
 		sh.Fatal(err)
 	}
 
-	var jobStream stream.Stream
-	sh.BeforeExit(func() {
-		if jobStream != nil {
-			jobStream.Close()
-		}
-		backend.Cleanup()
-	})
+	sh.BeforeExit(func() { backend.Cleanup() })
 
 	if force {
 		if err := backend.Cleanup(); err != nil {
@@ -281,6 +275,12 @@ func runDaemon(args *docopt.Args) {
 		h.Metadata[kv[0]] = kv[1]
 	}
 
+	var jobStream stream.Stream
+	sh.BeforeExit(func() {
+		if jobStream != nil {
+			jobStream.Close()
+		}
+	})
 	for {
 		newLeader := cluster.NewLeaderSignal()