Add GSON method setPath()

Author: Nikita Glukhov

contrib/jsonb_toaster/jsonb_toaster.c

@@ -2061,6 +2061,14 @@ jsonxzCopy(JsonContainer *jc)
 	return res;
 }
 
+static Datum
+jsonxSetPath(JsonContainer *js, Datum *path_elems,
+			 bool *path_nulls, int path_len,
+			 JsonValue *newval, int flags)
+{
+	return JsonSetPathGeneric(js, path_elems, path_nulls, path_len, newval, flags);
+}
+
 static JsonContainerOps
 jsonxzContainerOps =
 {
@@ -2074,7 +2082,8 @@ jsonxzContainerOps =
 	JsonbToCStringRaw,
 	jsonxzCopy,
 	jsonxzFree,
-	jsonxzEncode
+	jsonxzEncode,
+	jsonxSetPath
 };
 
 static JsonContainerOps
@@ -2090,7 +2099,8 @@ jsonxContainerOps =
 	JsonbToCStringRaw,
 	JsonCopyFlat,
 	NULL,
-	jsonxEncode
+	jsonxEncode,
+	jsonxSetPath
 };
 
 static JsonContainer *

src/backend/utils/adt/jsonb_util.c

@@ -2578,7 +2578,8 @@ jsonbContainerOps =
 	JsonbToCStringRaw,
 	JsonCopyFlat,
 	NULL,
-	NULL
+	NULL,
+	JsonSetPathGeneric
 };
 
 #ifndef JSONB_DETOAST_ITERATOR
@@ -3169,7 +3170,8 @@ jsonbzContainerOps =
 	JsonbToCStringRaw,
 	JsonCopyFlat,	// FIXME
 	jsonbzFree,
-	NULL
+	NULL,
+	JsonSetPathGeneric
 };
 
 Json *

src/backend/utils/adt/jsonfuncs.c

@@ -4532,6 +4532,23 @@ jsonb_delete_idx(PG_FUNCTION_ARGS)
 	PG_RETURN_DATUM(JsonbValueToOrigJsonbDatum(res, in));
 }
 
+Datum
+JsonSetPathGeneric(JsonContainer *js,
+				   Datum *path_elems, bool *path_nulls, int path_len,
+				   JsonValue *newval, int flags)
+{
+	JsonbParseState *st = NULL;
+	JsonbIterator *it = JsonbIteratorInit(js);
+	JsonbValue *res;
+
+	res = setPath(&it, path_elems, path_nulls, path_len, &st, 0,
+				  newval, flags);
+
+	Assert(res != NULL);
+
+	return JsonbValueToOrigJsonbDatum2(res, js);
+}
+
 /*
  * SQL function jsonb_set(jsonb, text[], jsonb, boolean)
  */
@@ -4543,12 +4560,10 @@ jsonb_set(PG_FUNCTION_ARGS)
 	Jsonb	   *newjsonb = PG_GETARG_JSONB_P(2);
 	JsonbValue	newval;
 	bool		create = PG_GETARG_BOOL(3);
-	JsonbValue *res = NULL;
+	Datum		res;
 	Datum	   *path_elems;
 	bool	   *path_nulls;
 	int			path_len;
-	JsonbIterator *it;
-	JsonbParseState *st = NULL;
 
 	JsonToJsonValue(newjsonb, &newval);
 
@@ -4570,14 +4585,10 @@ jsonb_set(PG_FUNCTION_ARGS)
 	if (path_len == 0)
 		PG_RETURN_JSONB_P(in);
 
-	it = JsonbIteratorInit(JsonbRoot(in));
-
-	res = setPath(&it, path_elems, path_nulls, path_len, &st,
-				  0, &newval, create ? JB_PATH_CREATE : JB_PATH_REPLACE);
-
-	Assert(res != NULL);
+	res = JsonSetPath(JsonbRoot(in), path_elems, path_nulls, path_len,
+					  &newval, create ? JB_PATH_CREATE : JB_PATH_REPLACE);
 
-	PG_RETURN_DATUM(JsonbValueToOrigJsonbDatum(res, in));
+	PG_RETURN_DATUM(res);
 }
 
 

src/include/utils/json_generic.h

@@ -74,6 +74,9 @@ struct JsonContainerOps
 	JsonContainer  *(*copy)(JsonContainer *jc);
 	void			(*free)(JsonContainer *jc);
 	void		   *(*encode)(JsonValue *jc, JsonContainerOps *ops, Oid toasterid);
+	Datum			(*setPath)(JsonContainer *js, Datum *path_elems,
+							   bool *path_nulls, int path_len,
+							   JsonValue *newval, int flags);
 };
 
 typedef struct CompressedObject
@@ -166,6 +169,9 @@ typedef JsonContainer JsonbContainer;
 #define JsonOp4(op, jscontainer, arg1, arg2, arg3, arg4) \
 		JsonOp(op, jscontainer)(jscontainer, arg1, arg2, arg3, arg4)
 
+#define JsonOp5(op, jscontainer, arg1, arg2, arg3, arg4, arg5) \
+		JsonOp(op, jscontainer)(jscontainer, arg1, arg2, arg3, arg4, arg5)
+
 #define JsonIteratorInit(jscontainer) \
 		JsonOp0(iteratorInit, jscontainer)
 
@@ -195,6 +201,8 @@ typedef JsonContainer JsonbContainer;
 			(jc)->ops->free(jc); \
 	} while (0)
 
+#define JsonSetPath(jc, path_elems, path_nulls, path_len, newval, flags) \
+		JsonOp5(setPath, jc, path_elems, path_nulls, path_len, newval, flags)
 
 static inline JsonIteratorToken
 JsonIteratorNext(JsonIterator **it, JsonValue *val, bool skipNested)
@@ -303,6 +311,9 @@ extern JsonValue *JsonValueCopy(JsonValue *res, const JsonValue *val);
 extern const JsonValue *JsonValueUnwrap(const JsonValue *val, JsonValue *buf);
 extern JsonContainer *JsonCopyFlat(JsonContainer *flatContainer);
 extern JsonValue *JsonExtractScalar(JsonContainer *jc, JsonValue *scalar);
+extern Datum JsonSetPathGeneric(JsonContainer *js, Datum *path_elems,
+								bool *path_nulls, int path_len,
+								JsonValue *newval, int flags);
 
 extern Jsonb *JsonbMakeEmptyArray(void);
 extern Jsonb *JsonbMakeEmptyObject(void);