Add jsonbc_get_dict_id() user function

Nikita Glukhov · Nikita Glukhov · commit f26c21e9587e · 2017-05-02T22:13:37.000+03:00
diff --git a/src/backend/utils/adt/jsonbc_util.c b/src/backend/utils/adt/jsonbc_util.c
@@ -1502,3 +1502,16 @@ jsonbc_handler(PG_FUNCTION_ARGS)
 
 	PG_RETURN_POINTER(cmr);
 }
+
+PG_FUNCTION_INFO_V1(jsonbc_get_dict_id);
+
+Datum
+jsonbc_get_dict_id(PG_FUNCTION_ARGS)
+{
+	Json *json = DatumGetJsont(PG_GETARG_DATUM(0));
+
+	if (json->root.ops == &jsonbcContainerOps)
+		PG_RETURN_INT32(jsonbcGetDictId(JsonRoot(json)));
+	else
+		PG_RETURN_NULL();
+}
diff --git a/src/include/catalog/pg_proc.h b/src/include/catalog/pg_proc.h
@@ -4529,6 +4529,8 @@ DATA(insert OID = 3421 (  jsonbc_get_id_by_name	   PGNSP PGUID 12 1 0 0 0 f f f
 DESCR("get name id from jsonbc dictionary");
 DATA(insert OID = 3998 (  jsonbc_get_name_by_id	   PGNSP PGUID 12 1 0 0 0 f f f f t f i s 2 0 25 "26 23" _null_ _null_ _null_ _null_ _null_ jsonbc_get_name_by_id _null_ _null_ _null_ ));
 DESCR("get name id from jsonbc dictionary");
+DATA(insert OID = 4001 (  jsonbc_get_dict_id	   PGNSP PGUID 12 1 0 0 0 f f f f t f i s 1 0 26 "114" _null_ _null_ _null_ _null_ _null_ jsonbc_get_dict_id _null_ _null_ _null_ ));
+DESCR("get jsonbc dictionary id from a json object");
 
 /* uuid */
 DATA(insert OID = 2952 (  uuid_in		   PGNSP PGUID 12 1 0 0 0 f f f f t f i s 1 0 2950 "2275" _null_ _null_ _null_ _null_ _null_ uuid_in _null_ _null_ _null_ ));
