feat: cleanup goroutines management (#14)

stern/main.go

@@ -195,6 +195,7 @@ OUTER:
 						}
 						tailsMutex.Lock()
 						tails[id].Close()
+						tails[id].PrintClose()
 						delete(tails, id)
 						tailsMutex.Unlock()
 					}
@@ -204,17 +205,18 @@ OUTER:
 				}
 			}()
 
-			go func() {
-			OUTER:
-				for {
-					select {
-					case str := <-logC:
-						fmt.Fprint(os.Stdout, str)
-					case <-ctx.Done():
-						break OUTER
+			if config.GraylogServer == "" {
+				go func() {
+					for {
+						select {
+						case str := <-logC:
+							fmt.Fprint(os.Stdout, str)
+						case <-ctx.Done():
+							return
+						}
 					}
-				}
-			}()
+				}()
+			}
 
 			go func() {
 				defer w.Done()

stern/tail.go

@@ -225,14 +225,15 @@ func (t *Tail) Start(ctx context.Context, i v1.PodInterface, gelfWriter *gelf.TC
 			t.Active = false
 			return
 		}
-		t.PrintOpen()
 		defer stream.Close()
+		t.PrintOpen()
 
 		go func() {
 			<-t.closed
 			stream.Close()
 			t.Active = false
 		}()
+
 		reader := bufio.NewReader(stream)
 
 	OUTER:
@@ -296,7 +297,6 @@ func (t *Tail) Start(ctx context.Context, i v1.PodInterface, gelfWriter *gelf.TC
 					}
 				}
 			} else {
-
 				vm := Log{
 					Message:        logLine,
 					Namespace:      t.Namespace,
@@ -318,11 +318,6 @@ func (t *Tail) Start(ctx context.Context, i v1.PodInterface, gelfWriter *gelf.TC
 			}
 		}
 	}()
-
-	go func() {
-		<-ctx.Done()
-		close(t.closed)
-	}()
 }
 
 // Close stops tailing