Add jsonb support for custom toasters

Nikita Glukhov · Nikita Glukhov · commit f649da10b38e · 2022-07-27T00:46:13.000+03:00
diff --git a/src/backend/utils/adt/json_generic.c b/src/backend/utils/adt/json_generic.c
@@ -10,6 +10,7 @@
 
 #include "postgres.h"
 
+#include "access/toasterapi.h"
 #include "miscadmin.h"
 #include "utils/builtins.h"
 #include "utils/jsonb.h"
@@ -101,9 +102,24 @@ JsonExpandDatum(Datum value, JsonContainerOps *ops, Json *tmp)
 Json *
 DatumGetJson(Datum value, JsonContainerOps *ops, Json *tmp)
 {
-	Json	   *json = JsonExpandDatum(value, ops, tmp);
+	Json	   *json;
 
-	//JsonInit(json);
+	if (VARATT_IS_CUSTOM(value))
+	{
+		Oid			toasterid = VARATT_CUSTOM_GET_TOASTERID(value);
+		TsrRoutine *toaster = SearchTsrCache(toasterid);
+		JsonToastRoutine *routine = toaster->get_vtable(toasterid);
+
+		if (routine->magic == JSON_TOASTER_MAGIC)
+		{
+			json = JsonExpand(tmp, value, false, routine->ops);
+			routine->ops->init(JsonRoot(json), value);
+
+			return json;
+		}
+	}
+
+	json = JsonExpandDatum(value, ops, tmp);
 	json->root.ops->init(&json->root, json->obj.value);
 
 	return json;
diff --git a/src/include/utils/json_generic.h b/src/include/utils/json_generic.h
@@ -310,4 +310,12 @@ extern int lengthCompareJsonbString(const char *val1, int len1,
 
 extern JsonContainerOps jsonbContainerOps;
 
+#define JSON_TOASTER_MAGIC 0x20211223
+
+typedef struct JsonToastRoutine
+{
+	uint32		magic;
+	JsonContainerOps *ops;
+} JsonToastRoutine;
+
 #endif /* UTILS_JSON_GENERIC_H */
