Add uniquification for binary jsons in JsonbGetDatum()

Nikita Glukhov · Nikita Glukhov · commit a749f8843461 · 2020-11-24T18:49:54.000+03:00
diff --git a/src/backend/utils/adt/json.c b/src/backend/utils/adt/json.c
@@ -66,6 +66,7 @@
 #define jsonb_to_recordset			json_to_recordset
 
 #define JsonxContainerOps			(&jsontContainerOps)
+#define JsonxGetUniquified(json)	(json)
 #define JsonxPGetDatum(json)		JsontPGetDatum(json)
 
 #include "postgres.h"
diff --git a/src/backend/utils/adt/json_gin.c b/src/backend/utils/adt/json_gin.c
@@ -19,7 +19,8 @@
 #define gin_triconsistent_jsonb_path	gin_triconsistent_json_path
 
 #define JsonxContainerOps				(&jsontContainerOps)
-#define JsonxGetDatum(json)	JsontGetDatum(json)
+#define JsonxGetUniquified(json)		(json)
+#define JsonxPGetDatum(json)			JsontGetDatum(json)
 
 #include "utils/json_generic.h"
 
diff --git a/src/backend/utils/adt/json_op.c b/src/backend/utils/adt/json_op.c
@@ -24,7 +24,8 @@
 #define jsonb_hash_extended	json_hash_extended
 
 #define JsonxContainerOps			(&jsontContainerOps)
-#define JsonxGetDatum(json)	JsontGetDatum(json)
+#define JsonxGetUniquified(json)	(json)
+#define JsonxPGetDatum(json)		JsontGetDatum(json)
 
 #include "utils/json_generic.h"
 
diff --git a/src/backend/utils/adt/jsonb_util.c b/src/backend/utils/adt/jsonb_util.c
@@ -2133,6 +2133,9 @@ JsonValueUniquify(JsonValue *res, const JsonValue *val)
 {
 	check_stack_depth();
 
+	if (!res)
+		res = (JsonValue *) palloc(sizeof(JsonValue));
+
 	if (val->type == jbvObject &&
 		(!val->val.object.valuesUniquified || !val->val.object.uniquified))
 	{
@@ -2186,6 +2189,31 @@ JsonValueUniquify(JsonValue *res, const JsonValue *val)
 	return res;
 }
 
+Json *
+JsonUniquify(Json *json)
+{
+	if (JsonRoot(json)->ops == &jsontContainerOps)
+	{
+		JsonValue	val;
+		Json	   *res = JsonValueToJson(JsonValueUnpackBinary(JsonToJsonValue(json, &val)));
+		res->is_json = json->is_json;
+		return res;
+	}
+	else if (JsonRoot(json)->ops == &jsonvContainerOps)
+	{
+		const JsonValue *val = (const JsonValue *) JsonRoot(json)->data;
+
+		if (!JsonValueIsUniquified(val))
+		{
+			Json	   *res = JsonValueToJson(JsonValueUniquify(NULL, val));
+			res->is_json = json->is_json;
+			return res;
+		}
+	}
+
+	return json;
+}
+
 static void
 jsonbInitContainer(JsonContainerData *jc, JsonbContainer *jbc, int len)
 {
diff --git a/src/include/utils/json_generic.h b/src/include/utils/json_generic.h
@@ -115,14 +115,10 @@ typedef struct Json
 */
 
 #undef JsonbPGetDatum
-#define JsonbPGetDatum(json)		JsonFlattenToJsonbDatum(json)
-
-#ifndef JsonxPGetDatum
-# ifdef JSON_FLATTEN_INTO_TARGET
-#  define JsonxPGetDatum(json)		JsonbPGetDatum(json)
-# else
-#  define JsonxPGetDatum(json)		JsonGetEOHDatum(json)
-# endif
+#ifdef JSON_FLATTEN_INTO_TARGET
+# define JsonbPGetDatum(json)		JsonFlattenToJsonbDatum(JsonGetUniquified(json))
+#else
+# define JsonbPGetDatum(json)		JsonGetEOHDatum(JsonGetUniquified(json))
 #endif
 
 #ifdef JSON_FLATTEN_INTO_TARGET
@@ -137,7 +133,13 @@ JsontPGetDatum(Json *json)
 }
 #endif
 
-#define JsonGetDatum(json)			JsonxPGetDatum(json)
+#ifdef JsonxPGetDatum
+# define JsonGetDatum(json)			JsonxPGetDatum(json)
+#elif defined(JsonxGetUniquified)
+# define JsonGetDatum(json)			JsonGetEOHDatum(JsonxGetUniquified(json))
+#else
+# define JsonGetDatum(json)			JsonbPGetDatum(json)
+#endif
 
 #undef DatumGetJsonbP
 #define DatumGetJsonbP(datum)		DatumGetJson(datum, &jsonbContainerOps, NULL)
@@ -191,14 +193,15 @@ JsontPGetDatum(Json *json)
 		 ((jc)->ops != &jsonvContainerOps || \
 		  JsonValueIsUniquified((JsonValue *) jc->data)))
 
+#define JsonIsUniquified(json)		JsonContainerIsUniquified(JsonRoot(json))
+
 #define JsonValueIsScalar(jsval)	IsAJsonbScalar(jsval)
 
 #define JsonContainerGetType(jc) ((jc)->ops->type)
 #define JsonContainerGetOpsByType(type) \
 		((type) == JsonContainerJsont ? &jsontContainerOps : \
 		 (type) == JsonContainerJsonb ? &jsonbContainerOps : NULL)
 
-
 #ifdef JSONB_UTIL_C
 #define JsonbValueToJsonb JsonValueToJsonb
 #else
@@ -270,10 +273,15 @@ JsonIteratorNext(JsonIterator **it, JsonValue *val, bool skipNested)
 #define compareJsonbContainers			JsonCompareContainers
 #define equalsJsonbScalarValue			JsonValueScalarEquals
 
+extern JsonContainerOps jsonbContainerOps;
+extern JsonContainerOps jsontContainerOps;
+extern JsonContainerOps jsonvContainerOps;
+
 extern Json *DatumGetJson(Datum val, JsonContainerOps *ops, Json *tmp);
 
 extern void JsonFree(Json *json);
 extern Json *JsonCopyTemporary(Json *tmp);
+extern Json *JsonUniquify(Json *json);
 
 #define JsonContainerAlloc() \
 	((JsonContainerData *) palloc(sizeof(JsonContainerData)))
@@ -315,6 +323,12 @@ JsonGetNonTemporary(Json *json)
 	return JsonIsTemporary(json) ? JsonCopyTemporary(json) : json;
 }
 
+static inline Json *
+JsonGetUniquified(Json *json)
+{
+	return JsonIsUniquified(json) ? json : JsonUniquify(json);
+}
+
 static inline JsonValue *
 JsonValueInitObject(JsonValue *val, int nPairs, int nPairsAllocated,
 					bool uniquified)
@@ -425,8 +439,4 @@ extern int lengthCompareJsonbStringValue(const void *a, const void *b);
 extern int lengthCompareJsonbString(const char *val1, int len1,
 									const char *val2, int len2);
 
-extern JsonContainerOps jsonbContainerOps;
-extern JsonContainerOps jsontContainerOps;
-extern JsonContainerOps jsonvContainerOps;
-
 #endif /* UTILS_JSON_GENERIC_H */
