support array of inet

Author: leafo

pgmoon/init.lua

@@ -116,6 +116,7 @@ local PG_TYPES = {
   [1016] = "array_number",
   [1021] = "array_number",
   [1022] = "array_number",
+  [1041] = "array_inet",
   [1231] = "array_number",
   [1009] = "array_string",
   [1015] = "array_string",
@@ -177,6 +178,11 @@ do
       bytea = function(self, val, name)
         return self:decode_bytea(val)
       end,
+      array_inet = function(self, val, name)
+        local decode_array
+        decode_array = require("pgmoon.arrays").decode_array
+        return decode_array(val, nil, self)
+      end,
       array_boolean = function(self, val, name)
         local decode_array
         decode_array = require("pgmoon.arrays").decode_array

pgmoon/init.moon

@@ -110,6 +110,7 @@ PG_TYPES = {
   [1016]: "array_number" -- int8 array
   [1021]: "array_number" -- float4 array
   [1022]: "array_number" -- float8 array
+  [1041]: "array_inet" -- array of ip addresses
   [1231]: "array_number" -- numeric array
 
   [1009]: "array_string" -- text array
@@ -173,6 +174,10 @@ class Postgres
     bytea: (val, name) =>
       @decode_bytea val
 
+    array_inet: (val, name) =>
+      import decode_array from require "pgmoon.arrays"
+      decode_array val, nil, @
+
     array_boolean: (val, name) =>
       import decode_array from require "pgmoon.arrays"
       decode_array val, tobool, @

spec/pgmoon_spec.moon

@@ -1018,6 +1018,16 @@ describe "pgmoon with server", ->
               } }
             }, (pg\query "select * from arrays_test")
 
+          it "loads inet arrays from table", ->
+            assert pg\query "create table arrays_test (addresses inet[])"
+            assert pg\query "insert into arrays_test (addresses) values ('{192.168.1.1,10.0.0.1}')"
+            assert pg\query "insert into arrays_test (addresses) values ('{127.0.0.1,::1,2001:db8::1}')"
+
+            assert.same {
+              { addresses: {"192.168.1.1", "10.0.0.1"} }
+              { addresses: {"127.0.0.1", "::1", "2001:db8::1"} }
+            }, (pg\query "select * from arrays_test")
+
 
       it "converts null", ->
         pg.convert_null = true