return writerError on writer.go:write function when w.dial() returns …

…error to avoid locking on writer.go:WriteMessages method (because it expects always to have a res message) (segmentio#39)
ghostsusan · Dec 27, 2017 · e57b88a · e57b88a
1 parent f88c22d
commit e57b88a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/writer.go b/writer.go
@@ -505,6 +505,9 @@ func (w *writer) dial() (conn *Conn, err error) {
 func (w *writer) write(conn *Conn, batch []Message, resch [](chan<- error)) (ret *Conn, err error) {
 	if conn == nil {
 		if conn, err = w.dial(); err != nil {
+			for i, res := range resch {
+				res <- &writerError{msg: batch[i], err: err}
+			}
 			return
 		}
 	}