Preserve json formatting in json_strip_nulls()

Nikita Glukhov · Nikita Glukhov · commit c7372eaf7a98 · 2017-05-02T21:38:30.000+03:00
diff --git a/src/backend/utils/adt/json.c b/src/backend/utils/adt/json.c
@@ -59,7 +59,6 @@
 #define jsonb_populate_recordset	json_populate_recordset
 #define jsonb_pretty				json_pretty
 #define jsonb_set					json_set
-#define jsonb_strip_nulls			json_strip_nulls
 #define jsonb_to_record				json_to_record
 #define jsonb_to_recordset			json_to_recordset
 
diff --git a/src/backend/utils/adt/jsonfuncs.c b/src/backend/utils/adt/jsonfuncs.c
@@ -505,6 +505,8 @@ static void transform_string_values_array_element_start(void *state, bool isnull
 static void transform_string_values_scalar(void *state, char *token, JsonTokenType tokentype);
 #endif
 
+static Datum jsonb_strip_nulls_internal(Jsonb *jb);
+
 /*
  * SQL function json_object_keys
  *
@@ -3854,7 +3856,9 @@ populate_recordset_object_field_end(void *state, char *fname, bool isnull)
 		hashentry->val = _state->saved_scalar;
 	}
 }
+#endif
 
+#ifdef JSON_C
 /*
  * Semantic actions for json_strip_nulls.
  *
@@ -3957,12 +3961,24 @@ sn_scalar(void *state, char *token, JsonTokenType tokentype)
 Datum
 json_strip_nulls(PG_FUNCTION_ARGS)
 {
-	text	   *json = PG_GETARG_TEXT_PP(0);
+#ifdef JSON_GENERIC
+	Json	   *json = PG_GETARG_JSONB(0);
+#else
+	text	   *json = PG_GETARG_TEXT_P(0);
+#endif
 	StripnullState *state;
 	JsonLexContext *lex;
 	JsonSemAction *sem;
 
+#ifdef JSON_GENERIC
+	if (json->root.ops != &jsontContainerOps)
+		return jsonb_strip_nulls_internal(json);
+
+	lex = makeJsonLexContextCstringLen(json->root.data, json->root.len, true);
+#else
 	lex = makeJsonLexContext(json, true);
+#endif
+
 	state = palloc0(sizeof(StripnullState));
 	sem = palloc0(sizeof(JsonSemAction));
 
@@ -3985,15 +4001,21 @@ json_strip_nulls(PG_FUNCTION_ARGS)
 											  state->strval->len));
 
 }
-#endif
+#else
 
 /*
  * SQL function jsonb_strip_nulls(jsonb) -> jsonb
  */
 Datum
 jsonb_strip_nulls(PG_FUNCTION_ARGS)
 {
-	Jsonb	   *jb = PG_GETARG_JSONB(0);
+	return jsonb_strip_nulls_internal(PG_GETARG_JSONB(0));
+}
+#endif
+
+static Datum
+jsonb_strip_nulls_internal(Jsonb *jb)
+{
 	JsonbIterator *it;
 	JsonbParseState *parseState = NULL;
 	JsonbValue *res = NULL;
