Add support for jsonb_array_length(), jsonb_pretty(), to_char()

Author: exAspArk

src/common/common_config.go

@@ -1,7 +1,7 @@
 package common
 
 const (
-	VERSION = "1.4.0"
+	VERSION = "1.5.0"
 
 	ENV_LOG_LEVEL                   = "BEMIDB_LOG_LEVEL"
 	ENV_DISABLE_ANONYMOUS_ANALYTICS = "BEMIDB_DISABLE_ANONYMOUS_ANALYTICS"

src/server/query_handler_test.go

@@ -188,6 +188,21 @@ func TestHandleQuery(t *testing.T) {
 				"types":       {uint32ToString(pgtype.Int4OID)},
 				"values":      {""},
 			},
+			"SELECT jsonb_array_length('[1, 2, 3]'::jsonb)": {
+				"description": {"jsonb_array_length"},
+				"types":       {uint32ToString(pgtype.XID8OID)},
+				"values":      {"3"},
+			},
+			"SELECT jsonb_pretty('{\"key\": \"value\"}'::JSONB)": {
+				"description": {"jsonb_pretty"},
+				"types":       {uint32ToString(pgtype.TextOID)},
+				"values":      {"{\n    \"key\": \"value\"\n}"},
+			},
+			"SELECT TO_CHAR('2024-01-15 14:30:00'::timestamp, 'YYYY-MM-DD')": {
+				"description": {"to_char"},
+				"types":       {uint32ToString(pgtype.TextOID)},
+				"values":      {"2024-01-15"},
+			},
 		})
 	})
 

src/server/query_remapper_function.go

@@ -41,6 +41,8 @@ func CreatePgCatalogMacroQueries(config *Config) []string {
 			ELSE json_extract_path_text(from_json, path_elems)::varchar
 		END`,
 		`CREATE MACRO jsonb_object_agg(key, value) AS to_json(map(array_agg(key), array_agg(value)))`,
+		`CREATE MACRO jsonb_array_length(json) AS json_array_length(json)`,
+		`CREATE MACRO jsonb_pretty(json) AS json_pretty(json)`,
 		`CREATE MACRO json_build_object(k1, v1) AS json_object(k1, v1),
 			(k1, v1, k2, v2) AS json_object(k1, v1, k2, v2),
 			(k1, v1, k2, v2, k3, v3) AS json_object(k1, v1, k2, v2, k3, v3),
@@ -50,6 +52,18 @@ func CreatePgCatalogMacroQueries(config *Config) []string {
 			WHEN 1 THEN len(arr)
 			ELSE NULL
 		END`,
+		`CREATE MACRO to_char(timestamp, text) AS
+			CASE text
+			WHEN 'YYYY-MM-DD' THEN strftime(timestamp, '%Y-%m-%d')
+			WHEN 'YYYY-MM-DD HH24:MI:SS' THEN strftime(timestamp, '%Y-%m-%d %H:%M:%S')
+			WHEN 'MM/DD/YYYY' THEN strftime(timestamp, '%m/%d/%Y')
+			WHEN 'DD-MON-YYYY' THEN strftime(timestamp, '%d-%b-%Y')
+			WHEN 'HH24:MI:SS' THEN strftime(timestamp, '%H:%M:%S')
+			WHEN 'YYYY' THEN strftime(timestamp, '%Y')
+			WHEN 'MM' THEN strftime(timestamp, '%m')
+			WHEN 'DD' THEN strftime(timestamp, '%d')
+			ELSE strftime(timestamp, text)
+		END`,
 
 		// Table functions
 		"CREATE MACRO pg_is_in_recovery() AS TABLE SELECT false AS pg_is_in_recovery",

src/server/response_handler.go

@@ -120,7 +120,7 @@ func (responseHandler *ResponseHandler) RowValuePointer(col *sql.ColumnType) int
 		return new(sql.NullInt32)
 	case "int64", "*big.Int":
 		return new(sql.NullInt64)
-	case "float64", "float32":
+	case "uint64", "float64", "float32":
 		return new(sql.NullFloat64)
 	case "string":
 		// return a *sql.NullString so Scan writes into the right type