Simplify and move to jsonb_utils.c clone_parse_state()

Nikita Glukhov · Nikita Glukhov · commit 7a9b2c1cae32 · 2020-12-07T19:31:01.000+03:00
diff --git a/src/backend/utils/adt/jsonb.c b/src/backend/utils/adt/jsonb.c
@@ -86,7 +86,6 @@ static void datum_to_jsonb(Datum val, bool is_null, JsonbInState *result,
 						   bool key_scalar);
 static void add_jsonb(Datum val, bool is_null, JsonbInState *result,
 					  Oid val_type, bool key_scalar);
-static JsonbParseState *clone_parse_state(JsonbParseState *state);
 static char *JsonbToCStringWorker(StringInfo out, JsonbContainer *in, int estimated_len, bool indent);
 static void add_indent(StringInfo out, bool indent, int level);
 
@@ -1465,40 +1464,6 @@ jsonb_object_two_arg(PG_FUNCTION_ARGS)
 }
 
 
-/*
- * shallow clone of a parse state, suitable for use in aggregate
- * final functions that will only append to the values rather than
- * change them.
- */
-static JsonbParseState *
-clone_parse_state(JsonbParseState *state)
-{
-	JsonbParseState *result,
-			   *icursor,
-			   *ocursor;
-
-	if (state == NULL)
-		return NULL;
-
-	result = palloc(sizeof(JsonbParseState));
-	icursor = state;
-	ocursor = result;
-	for (;;)
-	{
-		ocursor->contVal = icursor->contVal;
-		ocursor->size = icursor->size;
-		icursor = icursor->next;
-		if (icursor == NULL)
-			break;
-		ocursor->next = palloc(sizeof(JsonbParseState));
-		ocursor = ocursor->next;
-	}
-	ocursor->next = NULL;
-
-	return result;
-}
-
-
 /*
  * jsonb_agg aggregate function
  */
@@ -1642,7 +1607,7 @@ jsonb_agg_finalfn(PG_FUNCTION_ARGS)
 	 * values, just add the final array end marker.
 	 */
 
-	result.parseState = clone_parse_state(arg->res->parseState);
+	result.parseState = JsonbParseStateClone(arg->res->parseState);
 
 	result.res = pushJsonbValue(&result.parseState,
 								WJB_END_ARRAY, NULL);
@@ -1874,7 +1839,7 @@ jsonb_object_agg_finalfn(PG_FUNCTION_ARGS)
 	 * marker.
 	 */
 
-	result.parseState = clone_parse_state(arg->res->parseState);
+	result.parseState = JsonbParseStateClone(arg->res->parseState);
 
 	result.res = pushJsonbValue(&result.parseState,
 								WJB_END_OBJECT, NULL);
diff --git a/src/backend/utils/adt/jsonb_util.c b/src/backend/utils/adt/jsonb_util.c
@@ -538,6 +538,32 @@ fillJsonbValue(JsonbContainer *container, int index,
 	}
 }
 
+/*
+ * shallow clone of a parse state, suitable for use in aggregate
+ * final functions that will only append to the values rather than
+ * change them.
+ */
+JsonbParseState *
+JsonbParseStateClone(JsonbParseState *state)
+{
+	JsonbParseState	   *result,
+					   *icursor,
+					   *ocursor,
+					  **pocursor = &result;
+
+	for (icursor = state; icursor; icursor = icursor->next)
+	{
+		*pocursor = ocursor = palloc(sizeof(JsonbParseState));
+		ocursor->contVal = icursor->contVal;
+		ocursor->size = icursor->size;
+		pocursor = &ocursor->next;
+	}
+
+	*pocursor = NULL;
+
+	return result;
+}
+
 /*
  * Push JsonbValue into JsonbParseState.
  *
diff --git a/src/include/utils/jsonb.h b/src/include/utils/jsonb.h
@@ -387,6 +387,7 @@ extern JsonbValue *getIthJsonbValueFromContainer(JsonbContainer *sheader,
 												 uint32 i);
 extern JsonbValue *pushJsonbValue(JsonbParseState **pstate,
 								  JsonbIteratorToken seq, JsonbValue *jbval);
+extern JsonbParseState *JsonbParseStateClone(JsonbParseState *state);
 extern JsonbIterator *JsonbIteratorInit(JsonbContainer *container);
 extern JsonbIteratorToken JsonbIteratorNext(JsonbIterator **it, JsonbValue *val,
 											bool skipNested);
