fixup! Try to asynchronously connect to Fluentd before writing

fluent/fluent.go

@@ -367,7 +367,6 @@ func (f *Fluent) connect(ctx context.Context) (err error) {
 }
 
 func (f *Fluent) connectAsync(ctx context.Context, stopAsyncConnect <-chan bool) {
-	f.wg.Add(1)
 	defer f.wg.Done()
 
 	waiter := time.After(time.Duration(0))
@@ -381,14 +380,14 @@ func (f *Fluent) connectAsync(ctx context.Context, stopAsyncConnect <-chan bool)
 			err := f.connect(ctx)
 			if err == nil {
 				f.ready <- true
-				break
+				return
 			}
 
 			if _, ok := err.(*ErrUnknownNetwork); ok {
 				// No need to retry on unknown network error. Thus false is passed
 				// to ready channel to let the other end drain the message queue.
 				f.ready <- false
-				break
+				return
 			}
 
 			waitTime := f.Config.RetryWait * e(defaultReconnectWaitIncreRate, float64(i-1))
@@ -398,7 +397,7 @@ func (f *Fluent) connectAsync(ctx context.Context, stopAsyncConnect <-chan bool)
 
 			waiter = time.After(time.Duration(waitTime) * time.Millisecond)
 		case <-stopAsyncConnect:
-			break
+			return
 		}
 	}
 }
@@ -414,7 +413,9 @@ func (f *Fluent) run() {
 	// the select on f.stopRunning to signal its end to this goroutine.
 	ctx, cancelDialing := context.WithCancel(context.Background())
 	stopAsyncConnect := make(chan bool)
+	f.wg.Add(1)
 	go f.connectAsync(ctx, stopAsyncConnect)
+
 	select {
 	case <-f.stopRunning:
 		drainEvents = true