fix

Author: afiskon

Makefile

@@ -0,0 +1,17 @@
+EXTENSION = zson
+MODULES = zson
+DATA = zson--1.0.sql
+OBJS = zson.o
+REGRESS = zson
+
+ifdef USE_PGXS
+PGXS := $(shell pg_config --pgxs)
+include $(PGXS)
+else
+subdir = contrib/zson
+top_builddir = ../..
+include $(top_builddir)/src/Makefile.global
+include $(top_srcdir)/contrib/contrib-global.mk
+endif
+
+

expected/zson.out

@@ -0,0 +1,13 @@
+CREATE EXTENSION zson;
+SELECT '{"a": true}'::zson;
+    zson     
+-------------
+ {"a": true}
+(1 row)
+
+SELECT '{"a": true}'::zson -> 'a';
+ ?column? 
+----------
+ true
+(1 row)
+

sql/zson.sql

@@ -0,0 +1,5 @@
+CREATE EXTENSION zson;
+
+SELECT '{"a": true}'::zson;
+
+SELECT '{"a": true}'::zson -> 'a';

zson--1.0.sql

@@ -0,0 +1,35 @@
+-- complain if script is sourced in psql, rather than via CREATE EXTENSION
+\echo Use "CREATE EXTENSION zson" to load this file. \quit
+
+CREATE TYPE zson;
+
+CREATE FUNCTION zson_in(cstring)
+	RETURNS zson
+	AS 'MODULE_PATHNAME'
+	LANGUAGE C STRICT IMMUTABLE;
+
+CREATE FUNCTION zson_out(zson)
+	RETURNS cstring
+	AS 'MODULE_PATHNAME'
+	LANGUAGE C STRICT IMMUTABLE;
+
+CREATE TYPE zson (
+	INTERNALLENGTH = -1,
+	INPUT = zson_in,
+	OUTPUT = zson_out,
+	STORAGE = extended
+);
+
+CREATE FUNCTION z2j(zson)
+	RETURNS jsonb
+	AS 'MODULE_PATHNAME'
+	LANGUAGE C STRICT IMMUTABLE;
+
+CREATE FUNCTION j2z(jsonb)
+	RETURNS zson
+	AS 'MODULE_PATHNAME'
+	LANGUAGE C STRICT IMMUTABLE;
+
+CREATE CAST (zson AS jsonb) WITH FUNCTION z2j(zson) AS IMPLICIT;
+CREATE CAST (jsonb AS zson) WITH FUNCTION j2z(jsonb) AS ASSIGNMENT;
+

zson.c

@@ -0,0 +1,76 @@
+#include <postgres.h>
+#include <utils/builtins.h>
+#include <utils/jsonb.h>
+
+PG_MODULE_MAGIC;
+
+PG_FUNCTION_INFO_V1(zson_in);
+PG_FUNCTION_INFO_V1(zson_out);
+PG_FUNCTION_INFO_V1(z2j);
+PG_FUNCTION_INFO_V1(j2z);
+
+Datum
+zson_in(PG_FUNCTION_ARGS)
+{
+	Jsonb	*j = DatumGetJsonb(
+					DirectFunctionCall1(
+						jsonb_in,
+						PG_GETARG_DATUM(0)
+					)
+				);
+
+	PG_RETURN_BYTEA_P(
+		DatumGetByteaP(
+			DirectFunctionCall1(
+				j2z,
+				JsonbGetDatum(j)
+			)
+		)
+	);
+}
+
+
+Datum
+zson_out(PG_FUNCTION_ARGS)
+{
+	bytea	*z = PG_GETARG_BYTEA_P(0);
+
+	PG_RETURN_CSTRING(
+		DatumGetCString(
+			DirectFunctionCall1(
+				jsonb_out,
+				PointerGetDatum(VARDATA_ANY(z))
+			)
+		)
+	);
+}
+
+
+Datum
+z2j(PG_FUNCTION_ARGS)
+{
+	bytea	*z = PG_GETARG_BYTEA_P(0);
+	Jsonb	*j;
+
+	j = palloc(VARSIZE_ANY(z));
+
+	memcpy(j, VARDATA_ANY(z), VARSIZE_ANY_EXHDR(z));
+
+	PG_RETURN_JSONB(j);
+}
+
+Datum
+j2z(PG_FUNCTION_ARGS)
+{
+	Jsonb	*j = PG_GETARG_JSONB(0);
+	bytea	*z;
+
+	z = palloc(VARSIZE_ANY(j) + VARHDRSZ);
+	SET_VARSIZE(z, VARSIZE_ANY(j) + VARHDRSZ);
+
+	memcpy(VARDATA_ANY(z), j, VARSIZE_ANY(j));
+
+	PG_RETURN_BYTEA_P(z);
+}
+
+

zson.control

@@ -0,0 +1,4 @@
+comment = 'ZSON'
+default_version = '1.0'
+module_pathname = '$libdir/zson'
+relocatable = true