connection: Read back and parse server response

Author: ohsayan

src/skytable_py/connection.py

@@ -16,6 +16,7 @@
 from asyncio import StreamReader, StreamWriter
 from .query import Query
 from .protocol import Protocol
+from .response import Response
 
 
 class Connection:
@@ -51,11 +52,18 @@ async def close(self):
         self._writer.close()
         await self._writer.wait_closed()
 
-    async def run_simple_query(self, query: Query):
+    async def run_simple_query(self, query: Query) -> Response:
         query_window_str = str(query._q_window)
         total_packet_size = len(query_window_str) + 1 + len(query._buffer)
         # write metaframe
         metaframe = f"S{str(total_packet_size)}\n{query_window_str}\n"
         await self._write_all(metaframe.encode())
         # write dataframe
         await self._write_all(query._buffer)
+        # read response
+        while True:
+            new_block = await self._reader.read(1024)
+            self._protocol.push_additional_bytes(new_block)
+            resp = self._protocol.parse()
+            if resp:
+                return resp

src/skytable_py/protocol.py

@@ -16,7 +16,7 @@
 from typing import Union
 from .exception import ProtocolException
 from .response import Value, UInt8, UInt16, UInt32, UInt64, SInt8, SInt16, SInt32, SInt64, Float32, Float64, Empty, \
-    ErrorCode, Row
+    ErrorCode, Row, Response
 
 
 class Protocol:
@@ -209,6 +209,11 @@ def parse_rows(self) -> Union[None, list[Row]]:
                 return None
         return rows
 
+    def parse(self) -> Response:
+        e = self.parse_next_element()
+        if e:
+            return Response(e)
+
     def parse_next_element(self) -> Union[None, Value, Empty, ErrorCode]:
         if self.__is_eof():
             return None