Add JsonFreeIfCopy()

Nikita Glukhov · Nikita Glukhov · commit 9c5809a784d8 · 2017-05-02T21:38:31.000+03:00
diff --git a/src/include/utils/json_generic.h b/src/include/utils/json_generic.h
@@ -156,14 +156,16 @@ typedef struct Json
 #undef	PG_GETARG_JSONB
 #define PG_GETARG_JSONB(n)			PG_GETARG_JSONX_TMP(n, alloca(sizeof(Json))) /* FIXME conditional alloca() */
 
-#define PG_FREE_IF_COPY_JSONB(json, n) \
-	do { \
-		if (!VARATT_IS_EXTERNAL_EXPANDED(PG_GETARG_POINTER(n))) \
-			JsonFree(json); \
-		else \
-			Assert(DatumGetEOHP(PG_GETARG_DATUM(n)) == &(json)->obj.eoh); \
-	} while (0)
+#define JsonFreeIfCopy(json, datum) \
+		do { \
+			if (!VARATT_IS_EXTERNAL_EXPANDED(DatumGetPointer(datum))) \
+				JsonFree(json); \
+			else \
+				Assert(DatumGetEOHP(datum) == &(json)->obj.eoh); \
+		} while (0)
 
+#define PG_FREE_IF_COPY_JSONB(json, n) \
+		JsonFreeIfCopy(json, PG_GETARG_DATUM(n))
 
 #define JsonRoot(json)				(&(json)->root)
 #define JsonGetSize(json)			(JsonRoot(json)->len)
