From 997923223c0574dba2e87df10e09af94e6bd9d2f Mon Sep 17 00:00:00 2001
From: YangKian <1207783292@qq.com>
Date: Fri, 21 Jul 2023 17:02:24 +0800
Subject: [PATCH] fix shardReader for grpc-haskell

---
 hstream/src/HStream/Server/Handler/ShardReader.hs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/hstream/src/HStream/Server/Handler/ShardReader.hs b/hstream/src/HStream/Server/Handler/ShardReader.hs
index 960e4ec6c..571dcd87b 100644
--- a/hstream/src/HStream/Server/Handler/ShardReader.hs
+++ b/hstream/src/HStream/Server/Handler/ShardReader.hs
@@ -116,7 +116,7 @@ readShardStreamHandler sc (ServerWriterRequest _meta req streamSend) =
   defaultServerStreamExceptionHandle $ do
     Log.debug $ "Receive read shard stream Request: " <> Log.build (show req)
     C.readShardStream sc req streamWrite
-    return $ ServerWriterResponse mempty StatusUnknown "should not reach here"
+    returnServerStreamingResp StatusOk ""
   where
     streamWrite x = first show <$> streamSend x
 
@@ -135,7 +135,7 @@ readStreamHandler sc (ServerWriterRequest _meta req streamSend) =
   defaultServerStreamExceptionHandle $ do
     Log.debug $ "Receive read shard stream Request: " <> Log.build (show req)
     C.readStream sc req streamWrite
-    return $ ServerWriterResponse mempty StatusUnknown "should not reach here"
+    returnServerStreamingResp StatusOk ""
   where
     streamWrite x = first show <$> streamSend x
 
@@ -154,7 +154,7 @@ readSingleShardStreamHandler sc (ServerWriterRequest _meta req streamSend) =
   defaultServerStreamExceptionHandle $ do
     Log.debug $ "Receive read single shard stream Request: " <> Log.build (show req)
     C.readSingleShardStream sc req streamWrite
-    return $ ServerWriterResponse mempty StatusUnknown "should not reach here"
+    returnServerStreamingResp StatusOk ""
   where
     streamWrite x = first show <$> streamSend x