Add context to JsonValueFlatten()

Author: Nikita Glukhov

src/backend/utils/adt/jsonb_util.c

@@ -165,7 +165,7 @@ JsonContainerFlatten(JsonContainer *jc, JsonValueEncoder encoder,
  */
 void *
 JsonValueFlatten(const JsonValue *val, JsonValueEncoder encoder,
-				 JsonContainerOps *ops)
+				 JsonContainerOps *ops, void *cxt)
 {
 	if (val->type == jbvBinary)
 		return JsonContainerFlatten(val->val.binary.data, encoder, ops, val);
@@ -180,7 +180,7 @@ JsonValueFlatten(const JsonValue *val, JsonValueEncoder encoder,
 		Assert(val->type == jbvObject || val->type == jbvArray);
 	}
 
-	return JsonEncode(val, encoder, NULL);
+	return JsonEncode(val, encoder, cxt);
 }
 
 /*

src/include/utils/json_generic.h

@@ -92,7 +92,7 @@ typedef JsonContainer JsonbContainer;
 		PointerGetDatum(JsonFlatten(json, JsonbEncode, &jsonbContainerOps))
 
 #define JsonValueToJsonbDatum(jv) \
-	PointerGetDatum(JsonValueFlatten(jv, JsonbEncode, &jsonbContainerOps))
+	PointerGetDatum(JsonValueFlatten(jv, JsonbEncode, &jsonbContainerOps, NULL))
 
 #define JsonbPGetDatum(json)		JsonFlattenToJsonbDatum(json)
 
@@ -294,7 +294,7 @@ extern void *JsonContainerFlatten(JsonContainer *jc, JsonValueEncoder encoder,
 								  JsonContainerOps *ops, const JsonValue *binary);
 
 extern void *JsonValueFlatten(const JsonValue *val, JsonValueEncoder encoder,
-							  JsonContainerOps *ops);
+							  JsonContainerOps *ops, void *cxt);
 extern void *JsonEncode(const JsonbValue *val, JsonValueEncoder encoder, void *cxt);
 
 static inline void *
@@ -306,7 +306,7 @@ JsonFlatten(Json *json, JsonValueEncoder encoder, JsonContainerOps *ops)
 extern void JsonbEncode(StringInfo, const JsonValue *, void *cxt);
 
 #define JsonValueToJsonb(val) \
-		JsonValueFlatten(val, JsonbEncode, &jsonbContainerOps)
+		JsonValueFlatten(val, JsonbEncode, &jsonbContainerOps, NULL)
 
 extern int lengthCompareJsonbStringValue(const void *a, const void *b);
 extern int lengthCompareJsonbString(const char *val1, int len1,