Merge pull request #8 from sandeepbhojwani/retry

fix for the reconnect issue
wavefrontHQ · May 9, 2019 · bc43231 · bc43231
2 parents 2ff1217 + d73b8a7
commit bc43231
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/backend/pool.go b/backend/pool.go
@@ -59,7 +59,12 @@ func (cp *Pool) Get() (net.Conn, error) {
 	}
 }
 
-func (cp *Pool) Return(c net.Conn) {
+func (cp *Pool) Return(c net.Conn, failed bool) {
+
+	if failed {
+		<-cp.createsem
+		return
+	}
 	select {
 	case cp.connections <- c:
 	default:

diff --git a/backend/writer.go b/backend/writer.go
@@ -46,9 +46,7 @@ func (w *MetricWriter) Write(rq prompb.WriteRequest) error {
 	bw := bufio.NewWriter(out)
 	defer func() {
 		bw.Flush()
-		if !fail {
-			w.pool.Return(out)
-		}
+		w.pool.Return(out, fail)
 	}()
 	for _, ts := range rq.Timeseries {
 		if err := w.writeMetrics(bw, &ts); err != nil {