make beauty

Author: feodor

src/backend/utils/adt/jsonpath.c

@@ -19,6 +19,10 @@
 #include "utils/jsonpath.h"
 
 /*****************************INPUT/OUTPUT************************************/
+
+/*
+ * Convert AST to flat jsonpath type representation
+ */
 static int
 flattenJsonPathParseItem(StringInfo buf, JsonPathParseItem *item,
 						 bool forbiddenRoot)
@@ -33,6 +37,11 @@ flattenJsonPathParseItem(StringInfo buf, JsonPathParseItem *item,
 	alignStringInfoInt(buf);
 
 	next = (item->next) ? buf->len : 0;
+
+	/*
+	 * actual value will be recorded later, after next and
+	 * children processing
+	 */
 	appendBinaryStringInfo(buf, (char*)&next /* fake value */, sizeof(next));
 
 	switch(item->type)
@@ -83,6 +92,11 @@ flattenJsonPathParseItem(StringInfo buf, JsonPathParseItem *item,
 				int32	left, right;
 
 				left = buf->len;
+
+				/*
+				 * first, reserve place for left/right arg's positions, then
+				 * record both args and sets actual position in reserved places
+				 */
 				appendBinaryStringInfo(buf, (char*)&left /* fake value */, sizeof(left));
 				right = buf->len;
 				appendBinaryStringInfo(buf, (char*)&right /* fake value */, sizeof(right));
@@ -408,6 +422,10 @@ jsonpath_out(PG_FUNCTION_ARGS)
 
 /********************Support functions for JsonPath****************************/
 
+/*
+ * Support macroses to read stored values
+ */
+
 #define read_byte(v, b, p) do {			\
 	(v) = *(uint8*)((b) + (p));			\
 	(p) += 1;							\
@@ -423,13 +441,19 @@ jsonpath_out(PG_FUNCTION_ARGS)
 	(p) += sizeof(int32) * (n);			\
 } while(0)								\
 
+/*
+ * Read root node and fill root node representation
+ */
 void
 jspInit(JsonPathItem *v, JsonPath *js)
 {
 	Assert(js->header == JSONPATH_VERSION);
 	jspInitByBuffer(v, js->data, 0);
 }
 
+/*
+ * Read node from buffer and fill its representation
+ */
 void
 jspInitByBuffer(JsonPathItem *v, char *base, int32 pos)
 {

src/backend/utils/adt/jsonpath_exec.c

@@ -13,11 +13,13 @@
 #include "postgres.h"
 #include "funcapi.h"
 #include "miscadmin.h"
+#include "catalog/pg_collation.h"
 #include "catalog/pg_type.h"
 #include "lib/stringinfo.h"
 #include "utils/builtins.h"
 #include "utils/json.h"
 #include "utils/jsonpath.h"
+#include "utils/varlena.h"
 
 typedef struct JsonPathExecContext
 {
@@ -31,6 +33,9 @@ static JsonPathExecResult recursiveExecute(JsonPathExecContext *cxt,
 
 /********************Execute functions for JsonPath***************************/
 
+/*
+ * Find value of jsonpath variable in a list of passing params
+ */
 static void
 computeJsonPathVariable(JsonPathItem *variable, List *vars, JsonbValue *value)
 {
@@ -132,6 +137,9 @@ computeJsonPathVariable(JsonPathItem *variable, List *vars, JsonbValue *value)
 	}
 }
 
+/*
+ * Convert jsonpath's scalar or variable node to actual jsonb value
+ */
 static void
 computeJsonPathItem(JsonPathExecContext *cxt, JsonPathItem *item, JsonbValue *value)
 {
@@ -161,6 +169,12 @@ computeJsonPathItem(JsonPathExecContext *cxt, JsonPathItem *item, JsonbValue *va
 }
 
 
+/*
+ * Returns jbv* type of of JsonbValue. Note, it never returns
+ * jbvBinary as is - jbvBinary is used as mark of store naked
+ * scalar value. To improve readability it defines jbvScalar
+ * as alias to jbvBinary
+ */
 #define jbvScalar jbvBinary
 static int
 JsonbType(JsonbValue *jb)
@@ -196,31 +210,10 @@ compareNumeric(Numeric a, Numeric b)
 			);
 }
 
-static bool
-checkScalarEquality(JsonbValue *jb1, JsonbValue *jb2)
-{
-	switch (jb1->type)
-	{
-		case jbvNull:
-			return true;
-		case jbvString:
-			return (jb1->val.string.len == jb2->val.string.len &&
-					memcmp(jb2->val.string.val, jb1->val.string.val,
-						   jb1->val.string.len) == 0);
-		case jbvBool:
-			return (jb2->val.boolean == jb1->val.boolean);
-		case jbvNumeric:
-			return (compareNumeric(jb1->val.numeric, jb2->val.numeric) == 0);
-		default:
-			elog(ERROR,"1Wrong state");
-			return false;
-	}
-}
-
 static JsonPathExecResult
 checkEquality(JsonbValue *jb1, JsonbValue *jb2, bool not)
 {
-	bool	eq;
+	bool	eq = false;
 
 	if (jb1->type != jb2->type)
 	{
@@ -232,14 +225,28 @@ checkEquality(JsonbValue *jb1, JsonbValue *jb2, bool not)
 
 	if (jb1->type == jbvBinary)
 		return jperError;
-	/*
-		eq = compareJsonbContainers(jb1->val.binary.data,
-									jb2->val.binary.data) == 0;
-	*/
-	else
-		eq = checkScalarEquality(jb1, jb2);
 
-	return !!not ^ !!eq ? jperOk : jperNotFound;
+	switch (jb1->type)
+	{
+		case jbvNull:
+			eq = true;
+			break;
+		case jbvString:
+			eq = (jb1->val.string.len == jb2->val.string.len &&
+					memcmp(jb2->val.string.val, jb1->val.string.val,
+						   jb1->val.string.len) == 0);
+			break;
+		case jbvBool:
+			eq = (jb2->val.boolean == jb1->val.boolean);
+			break;
+		case jbvNumeric:
+			eq = (compareNumeric(jb1->val.numeric, jb2->val.numeric) == 0);
+			break;
+		default:
+			elog(ERROR,"1Wrong state");
+	}
+
+	return (not ^ eq) ? jperOk : jperNotFound;
 }
 
 static JsonPathExecResult
@@ -269,13 +276,11 @@ makeCompare(int32 op, JsonbValue *jb1, JsonbValue *jb2)
 		case jbvNumeric:
 			cmp = compareNumeric(jb1->val.numeric, jb2->val.numeric);
 			break;
-		/*
 		case jbvString:
 			cmp = varstr_cmp(jb1->val.string.val, jb1->val.string.len,
 							 jb2->val.string.val, jb2->val.string.len,
-							 collationId);
+							 DEFAULT_COLLATION_OID);
 			break;
-		*/
 		default:
 			return jperError;
 	}
@@ -532,6 +537,9 @@ executeUnaryArithmExpr(JsonPathExecContext *cxt, JsonPathItem *jsp,
 	return jper;
 }
 
+/*
+ * implements jpiAny node (** operator)
+ */
 static JsonPathExecResult
 recursiveAny(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 			 List **found, uint32 level, uint32 first, uint32 last)
@@ -548,6 +556,9 @@ recursiveAny(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 
 	it = JsonbIteratorInit(jb->val.binary.data);
 
+	/*
+	 * Recursivly iterate over jsonb objects/arrays
+	 */
 	while((r = JsonbIteratorNext(&it, &v, true)) != WJB_DONE)
 	{
 		if (r == WJB_KEY)
@@ -590,12 +601,23 @@ recursiveAny(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 	return res;
 }
 
+/*
+ * Main executor function: walks on jsonpath structure and tries to find
+ * correspoding parts of jsonb. Note, jsonb and jsonpath values should be
+ * avaliable and untoasted during work because JsonPathItem, JsonbValue
+ * and found could have pointers into input values. If caller wants just to
+ * check matching of json by jsonpath then it doesn't provide a found arg.
+ * In this case executor works till first positive result and does not check
+ * the rest if it is possible. In other case it tries to find all satisfied
+ * results
+ */
 static JsonPathExecResult
 recursiveExecute(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 				 List **found)
 {
 	JsonPathItem		elem;
 	JsonPathExecResult	res = jperNotFound;
+	bool				hasNext;
 
 	check_stack_depth();
 
@@ -607,10 +629,15 @@ recursiveExecute(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 			{
 				JsonPathExecResult res2;
 
+				/*
+				 * SQL/JSON says that we should check second arg
+				 * in case of jperError
+				 */
+
 				jspGetRightArg(jsp, &elem);
 				res2 = recursiveExecute(cxt, &elem, jb, NULL);
 
-				res = res2 == jperOk ? res : res2;
+				res = (res2 == jperOk) ? res : res2;
 			}
 			break;
 		case jpiOr:
@@ -623,7 +650,7 @@ recursiveExecute(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 				jspGetRightArg(jsp, &elem);
 				res2 = recursiveExecute(cxt, &elem, jb, NULL);
 
-				res = res2 == jperNotFound ? res : res2;
+				res = (res2 == jperNotFound) ? res : res2;
 			}
 			break;
 		case jpiNot:
@@ -643,7 +670,7 @@ recursiveExecute(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 		case jpiIsUnknown:
 			jspGetArg(jsp, &elem);
 			res = recursiveExecute(cxt, &elem, jb, NULL);
-			res = res == jperError ? jperOk : jperNotFound;
+			res = (res == jperError) ? jperOk : jperNotFound;
 			break;
 		case jpiKey:
 			if (JsonbType(jb) == jbvObject)
@@ -676,6 +703,7 @@ recursiveExecute(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 		case jpiCurrent:
 			if (!jspGetNext(jsp, &elem))
 			{
+				/* we are last in chain of node */
 				res = jperOk;
 				if (found)
 				{
@@ -685,7 +713,7 @@ recursiveExecute(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 						v = JsonbExtractScalar(jb->val.binary.data,
 											   palloc(sizeof(*v)));
 					else
-						v = copyJsonbValue(jb); /* FIXME */
+						v = copyJsonbValue(jb);
 
 					*found = lappend(*found, v);
 				}
@@ -709,7 +737,6 @@ recursiveExecute(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 				JsonbIterator	*it;
 				int32			r;
 				JsonbValue		v;
-				bool			hasNext;
 
 				hasNext = jspGetNext(jsp, &elem);
 				it = JsonbIteratorInit(jb->val.binary.data);
@@ -746,7 +773,6 @@ recursiveExecute(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 			if (JsonbType(jb) == jbvArray)
 			{
 				JsonbValue		*v;
-				bool			hasNext;
 				int				i;
 
 				hasNext = jspGetNext(jsp, &elem);
@@ -788,7 +814,6 @@ recursiveExecute(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 				JsonbIterator	*it;
 				int32			r;
 				JsonbValue		v;
-				bool			hasNext;
 
 				hasNext = jspGetNext(jsp, &elem);
 				it = JsonbIteratorInit(jb->val.binary.data);
@@ -915,6 +940,9 @@ recursiveExecute(JsonPathExecContext *cxt, JsonPathItem *jsp, JsonbValue *jb,
 	return res;
 }
 
+/*
+ * Public interface to jsonpath executor
+ */
 JsonPathExecResult
 executeJsonPath(JsonPath *path, List *vars, Jsonb *json, List **foundJson)
 {
@@ -948,6 +976,9 @@ returnNULL(void *arg, bool *isNull)
 	return Int32GetDatum(0);
 }
 
+/*
+ * Convert jsonb object into list of vars for executor
+ */
 static List*
 makePassingVars(Jsonb *jb)
 {