tcp: closeConnImmediately on write err

tcp/conn.go

@@ -98,7 +98,11 @@ func (pool *connPool) Send(to net.Addr, m protocol.Message) error {
 		go pool.closeConn(toStr)
 	}
 
-	return c.session.WriteMessage(c.conn, m)
+	if err := c.session.WriteMessage(c.conn, m); err != nil {
+		pool.logger.Errorf("error in session: %v, closing connection...", err)
+		pool.closeConnImmediately(toStr)
+	}
+	return nil
 }
 
 func (pool *connPool) connect(to net.Addr) (conn, error) {
@@ -145,3 +149,13 @@ func (pool *connPool) closeConn(to string) {
 	}
 	delete(pool.conns, to)
 }
+
+func (pool *connPool) closeConnImmediately(to string) {
+	pool.mu.Lock()
+	defer pool.mu.Unlock()
+
+	if err := pool.conns[to].conn.Close(); err != nil {
+		pool.logger.Errorf("error closing connection to %v: %v", to, err)
+	}
+	delete(pool.conns, to)
+}