feat(datatype): support varbyte

Author: Brooke-white

redshift_connector/__init__.py

@@ -84,6 +84,7 @@
     TIMETZ,
     UNKNOWN,
     UUID_TYPE,
+    VARBYTE,
     VARCHAR,
     VARCHAR_ARRAY,
     XID,

redshift_connector/core.py

@@ -58,6 +58,7 @@
     array_recv_binary,
     array_recv_text,
     bh_unpack,
+    bytea_recv,
     cccc_unpack,
     ci_unpack,
     date_in,
@@ -83,6 +84,7 @@
     time_recv_binary,
     timetz_in,
     timetz_recv_binary,
+    varbytehex_recv,
     walk_array,
 )
 from redshift_connector.utils.type_utils import (
@@ -99,6 +101,7 @@
     TIMESTAMP,
     TIMESTAMPTZ,
     TIMETZ,
+    VARBYTE,
     VARCHAR_ARRAY,
 )
 
@@ -686,6 +689,7 @@ def _enable_protocol_based_conversion_funcs(self: "Connection"):
             self.pg_types[REAL_ARRAY] = (FC_BINARY, array_recv_binary)  # FLOAT4[]
             self.pg_types[1028] = (FC_BINARY, array_recv_binary)  # OID[]
             self.pg_types[1034] = (FC_BINARY, array_recv_binary)  # ACLITEM[]
+            self.pg_types[VARBYTE] = (FC_TEXT, bytea_recv)  # VARBYTE
         else:  # text protocol
             self.pg_types[NUMERIC] = (FC_TEXT, numeric_in)
             self.pg_types[TIME] = (FC_TEXT, time_in)
@@ -699,6 +703,7 @@ def _enable_protocol_based_conversion_funcs(self: "Connection"):
             self.pg_types[REAL_ARRAY] = (FC_TEXT, float_array_recv)  # FLOAT4[]
             self.pg_types[1028] = (FC_TEXT, int_array_recv)  # OID[]
             self.pg_types[1034] = (FC_TEXT, array_recv_text)  # ACLITEM[]
+            self.pg_types[VARBYTE] = (FC_TEXT, varbytehex_recv)  # VARBYTE
 
     @property
     def _is_multi_databases_catalog_enable_in_server(self: "Connection") -> bool:

redshift_connector/utils/__init__.py

@@ -16,6 +16,7 @@
     array_recv_binary,
     array_recv_text,
     bh_unpack,
+    bytea_recv,
     cccc_unpack,
     ci_unpack,
     date_in,
@@ -38,4 +39,5 @@
     time_recv_binary,
     timetz_in,
     timetz_recv_binary,
+    varbytehex_recv,
 )

redshift_connector/utils/type_utils.py

@@ -1,5 +1,6 @@
 import enum
 import typing
+from codecs import decode as codecs_decode
 from collections import defaultdict
 from datetime import date
 from datetime import datetime as Datetime
@@ -90,6 +91,7 @@
 UUID_TYPE = 2950
 UUID_ARRAY = 2951
 VARCHAR = 1043
+VARBYTE = 6551
 VARCHAR_ARRAY = 1015
 XID = 28
 
@@ -131,8 +133,8 @@ def bool_recv(data: bytes, offset: int, length: int) -> bool:
 
 
 # bytea
-# def bytea_recv(data: bytearray, offset: int, length: int) -> bytearray:
-#     return data[offset:offset + length]
+def bytea_recv(data: bytearray, offset: int, length: int) -> bytearray:
+    return data[offset : offset + length]
 
 
 def int8_recv(data: bytes, offset: int, length: int) -> int:
@@ -574,6 +576,10 @@ def geometryhex_recv(data: bytes, idx: int, length: int) -> str:
         return result.hex()
 
 
+def varbytehex_recv(data: bytes, idx: int, length: int) -> bytes:
+    return codecs_decode(data[idx : idx + length], "hex_codec")
+
+
 # def inet_in(data: bytes, offset: int, length: int) -> typing.Union[IPv4Address, IPv6Address, IPv4Network, IPv6Network]:
 #     inet_str: str = data[offset: offset + length].decode(
 #         _client_encoding)
@@ -634,6 +640,7 @@ def geometryhex_recv(data: bytes, idx: int, length: int) -> str:
         GEOMETRYHEX: (FC_TEXT, geometryhex_recv),  # GEOMETRYHEX
         # 3802: (FC_TEXT, json_in),  # jsonb
         SUPER: (FC_TEXT, text_recv),  # SUPER
+        VARBYTE: (FC_TEXT, varbytehex_recv),  # VARBYTE
     },
 )