From d4eae8cc0c55a3ac293e5dd65fe0883c22969763 Mon Sep 17 00:00:00 2001
From: Yecheng Fu <cofyc.jackson@gmail.com>
Date: Fri, 17 Aug 2018 21:20:22 +0800
Subject: [PATCH] Wait for all internal discoveries are done before exiting.
 (#4508)

Signed-off-by: Yecheng Fu <cofyc.jackson@gmail.com>
---
 discovery/kubernetes/kubernetes.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/discovery/kubernetes/kubernetes.go b/discovery/kubernetes/kubernetes.go
index 4f6a074f1..4ab431fc5 100644
--- a/discovery/kubernetes/kubernetes.go
+++ b/discovery/kubernetes/kubernetes.go
@@ -370,6 +370,8 @@ func (d *Discovery) Run(ctx context.Context, ch chan<- []*targetgroup.Group) {
 	}
 
 	d.Unlock()
+
+	wg.Wait()
 	<-ctx.Done()
 }