From d8ccc8187199aca8f5d6bde907772540da73cd52 Mon Sep 17 00:00:00 2001
From: Edbert Linardi <edbert.linardi@snowflake.com>
Date: Wed, 11 Oct 2023 14:41:11 -0700
Subject: [PATCH] Close stream when there's an error from target (#347)

---
 proxy/server/target.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/proxy/server/target.go b/proxy/server/target.go
index 21c870ed..b3098de8 100644
--- 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