Add support for INT2 and INT8 (bigint) arrays

Author: malthe

CHANGES.md

@@ -1,5 +1,7 @@
 In next release ...
 
+- Add support for the INT2 and INT8 array types.
+
 - The `Connect` and `End` events have been removed, in addition to the `Parameter`
   event; the `connect` method now returns an object with information about the
   established connection, namely whether the connection is encrypted and the

src/types.ts

@@ -39,11 +39,13 @@ export enum DataType {
     Inet = 869,
     ArrayBytea = 1001,
     ArrayChar = 1002,
+    ArrayInt2 = 1005,
     ArrayInt4 = 1007,
     ArrayRegprocedure = 1008,
     ArrayText = 1009,
     ArrayBpchar = 1014,
     ArrayVarchar = 1015,
+    ArrayInt8 = 1016,
     ArrayFloat4 = 1021,
     ArrayFloat8 = 1022,
     Aclitem = 1033,
@@ -108,7 +110,9 @@ export const arrayDataTypeMapping: ReadonlyMap<DataType, DataType> = new Map([
     [DataType.ArrayDate, DataType.Date],
     [DataType.ArrayFloat4, DataType.Float4],
     [DataType.ArrayFloat8, DataType.Float8],
+    [DataType.ArrayInt2, DataType.Int2],
     [DataType.ArrayInt4, DataType.Int4],
+    [DataType.ArrayInt8, DataType.Int8],
     [DataType.ArrayJson, DataType.Json],
     [DataType.ArrayJsonb, DataType.Jsonb],
     [DataType.ArrayText, DataType.Text],

test/types.test.ts

@@ -189,6 +189,10 @@ describe('Types', () => {
         'ARRAY[\'123e4567-e89b-12d3-a456-426655440000\'::uuid]',
         ['123e4567-e89b-12d3-a456-426655440000']
     );
+    testType<number[]>(
+        DataType.ArrayInt2,
+        '\'{1,2,3}\'::int2[3]',
+        [1, 2, 3]);
     testType<number[]>(
         DataType.ArrayInt4,
         '\'{1,2,3}\'::int4[3]',
@@ -201,6 +205,10 @@ describe('Types', () => {
         DataType.ArrayInt4,
         '\'{{{1, 2}, {3, 4}}, {{5, 6}, {7, 8}}}\'::int4[]',
         [[[1, 2], [3, 4]], [[5, 6], [7, 8]]]);
+    testType<bigint[]>(
+        DataType.ArrayInt8,
+        '\'{1,2,3}\'::int8[3]',
+        [BigInt(1), BigInt(2), BigInt(3)]);
     testType<number[]>(
         DataType.ArrayFloat4,
         '\'{1.0, 2.0, 3.0}\'::float4[3]',