Binary input/output for hll type.

Author: alberts

hll--1.0.sql

@@ -32,6 +32,16 @@ RETURNS cstring
 AS 'MODULE_PATHNAME'
 LANGUAGE C STRICT IMMUTABLE;
 
+CREATE FUNCTION hll_recv(internal)
+RETURNS hll
+AS 'MODULE_PATHNAME'
+LANGUAGE C IMMUTABLE STRICT;
+
+CREATE FUNCTION hll_send(hll)
+RETURNS bytea
+AS 'MODULE_PATHNAME'
+LANGUAGE C IMMUTABLE STRICT;
+
 CREATE FUNCTION hll_typmod_in(cstring[])
 RETURNS integer
 AS 'MODULE_PATHNAME'
@@ -53,6 +63,8 @@ CREATE TYPE hll (
         OUTPUT = hll_out,
         TYPMOD_IN = hll_typmod_in,
         TYPMOD_OUT = hll_typmod_out,
+        RECEIVE = hll_recv,
+        SEND = hll_send,
         STORAGE = external
 );
 

hll.c

@@ -33,6 +33,8 @@
 #include "utils/lsyscache.h"
 #include "utils/memutils.h"
 #include "catalog/pg_type.h"
+#include "lib/stringinfo.h"
+#include "libpq/pqformat.h"
 
 #include "MurmurHash3.h"
 
@@ -3332,3 +3334,25 @@ hll_ceil_card_unpacked(PG_FUNCTION_ARGS)
         PG_RETURN_INT64(ceilval);
     }
 }
+
+PG_FUNCTION_INFO_V1(hll_recv);
+Datum hll_recv(PG_FUNCTION_ARGS);
+Datum
+hll_recv(PG_FUNCTION_ARGS)
+{
+    Datum dd = DirectFunctionCall1(bytearecv, PG_GETARG_DATUM(0));
+    return dd;
+}
+
+PG_FUNCTION_INFO_V1(hll_send);
+Datum hll_send(PG_FUNCTION_ARGS);
+Datum
+hll_send(PG_FUNCTION_ARGS)
+{
+    Datum dd = PG_GETARG_DATUM(0);
+    bytea* bp = DatumGetByteaP(dd);
+    StringInfoData buf;
+    pq_begintypsend(&buf);
+    pq_sendbytes(&buf, VARDATA(bp), VARSIZE(bp) - VARHDRSZ);
+    PG_RETURN_BYTEA_P(pq_endtypsend(&buf));
+}

regress/Makefile

@@ -52,6 +52,7 @@ SQL =	\
 		cumulative_union_sparse_full_representation.sql \
 		cumulative_union_sparse_promotion.sql \
 		cumulative_union_sparse_sparse.sql \
+		copy_binary.sql \
 		$(NULL)
 
 TEST_DB = hll_regress
@@ -62,7 +63,7 @@ NOT =	\
 OUT := $(SQL:%.sql=%.out)
 
 # Print NULL values explicitly.
-PSQLOPTS = --echo-all -P null=NULL
+PSQLOPTS = -x --echo-all -P null=NULL
 
 # Disable NOTICE log messages
 export PGOPTIONS := --client-min-messages=warning

regress/copy_binary.ref

@@ -0,0 +1,26 @@
+SELECT hll_set_output_version(1);
+-[ RECORD 1 ]----------+--
+hll_set_output_version | 1
+
+DROP TABLE IF EXISTS test_binary;
+DROP TABLE
+CREATE TABLE test_binary (id SERIAL, v1 hll);
+CREATE TABLE
+INSERT INTO test_binary(id,v1) VALUES (1, hll_empty() || hll_hash_text('A'));
+INSERT 0 1
+SELECT hll_cardinality(v1) FROM test_binary;
+-[ RECORD 1 ]---+--
+hll_cardinality | 1
+
+\COPY test_binary TO 'binary.dat' WITH (FORMAT binary)
+DELETE FROM test_binary;
+DELETE 1
+SELECT hll_cardinality(v1) FROM test_binary;
+(No rows)
+\COPY test_binary FROM 'binary.dat' WITH (FORMAT binary)
+SELECT hll_cardinality(v1) FROM test_binary;
+-[ RECORD 1 ]---+--
+hll_cardinality | 1
+
+DROP TABLE test_binary;
+DROP TABLE

regress/copy_binary.sql

@@ -0,0 +1,21 @@
+SELECT hll_set_output_version(1);
+
+DROP TABLE IF EXISTS test_binary;
+
+CREATE TABLE test_binary (id SERIAL, v1 hll);
+
+INSERT INTO test_binary(id,v1) VALUES (1, hll_empty() || hll_hash_text('A'));
+
+SELECT hll_cardinality(v1) FROM test_binary;
+
+\COPY test_binary TO 'binary.dat' WITH (FORMAT binary)
+
+DELETE FROM test_binary;
+
+SELECT hll_cardinality(v1) FROM test_binary;
+
+\COPY test_binary FROM 'binary.dat' WITH (FORMAT binary)
+
+SELECT hll_cardinality(v1) FROM test_binary;
+
+DROP TABLE test_binary;