Gleipnir/jet
4
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Better json func, and tests

Browse source
This commit is contained in:
Vladimír Zahradník 2022-06-04 14:22:33 +02:00
parent 0425e8895c
commit 11b0a6858a
2 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
postgres/literal.go
View file

@ -66,8 +66,13 @@ func String(value string) StringExpression {
} }
// Json creates new json literal expression // Json creates new json literal expression
func Json(value string) StringExpression { func Json(value interface{}) StringExpression {
return StringExp(CAST(jet.String(value)).AS("json")) 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 // UUID is a helper function to create string literal expression from uuid object

5
postgres/literal_test.go
View file

@ -67,6 +67,11 @@ func TestBytea(t *testing.T) {
assertSerialize(t, Bytea([]byte("Some byte array")), `$1::bytea`, []byte("Some byte array")) 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) { func TestDate(t *testing.T) {
assertSerialize(t, Date(2014, time.January, 2), `$1::date`, "2014-01-02") assertSerialize(t, Date(2014, time.January, 2), `$1::date`, "2014-01-02")
assertSerialize(t, DateT(time.Now()), `$1::date`) assertSerialize(t, DateT(time.Now()), `$1::date`)