Close stream when there's an error from target (#347)

Snowflake-Labs · Oct 11, 2023 · d8ccc81 · d8ccc81
1 parent 9d86da1
commit d8ccc81
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/proxy/server/target.go b/proxy/server/target.go
@@ -226,6 +226,7 @@ func (s *TargetStream) Run(nonce uint32, replyChan chan *pb.ProxyReply) {
 					return nil
 				}
 				if err != nil {
+					s.CloseWith(err)
 					return err
 				}
 				// otherwise, this is a streamData reply