Better json func, and tests

go-jet · Jun 4, 2022 · 11b0a68 · 11b0a68
1 parent 0425e88
commit 11b0a68
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/postgres/literal.go b/postgres/literal.go
@@ -66,8 +66,13 @@ func String(value string) StringExpression {
 }
 
 // Json creates new json literal expression
-func Json(value string) StringExpression {
-	return StringExp(CAST(jet.String(value)).AS("json"))
+func Json(value interface{}) StringExpression {
+	switch value.(type) {
+	case string, []byte:
+	default:
+		panic("Bytea parameter value has to be of the type string or []byte")
+	}
+	return StringExp(CAST(jet.Literal(value)).AS("json"))
 }
 
 // UUID is a helper function to create string literal expression from uuid object

diff --git a/postgres/literal_test.go b/postgres/literal_test.go
@@ -67,6 +67,11 @@ func TestBytea(t *testing.T) {
 	assertSerialize(t, Bytea([]byte("Some byte array")), `$1::bytea`, []byte("Some byte array"))
 }
 
+func TestJson(t *testing.T) {
+	assertSerialize(t, Json("{\"key\": \"value\"}"), `$1::json`, "{\"key\": \"value\"}")
+	assertSerialize(t, Json([]byte("{\"key\": \"value\"}")), `$1::json`, []byte("{\"key\": \"value\"}"))
+}
+
 func TestDate(t *testing.T) {
 	assertSerialize(t, Date(2014, time.January, 2), `$1::date`, "2014-01-02")
 	assertSerialize(t, DateT(time.Now()), `$1::date`)