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

Cast dialect clean up.

Browse source
This commit is contained in:
go-jet 2019-08-06 12:12:48 +02:00
parent 6ee2f45548
commit cb606f9026
3 changed files with 22 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

22
postgres/dialect.go
View file

@ -13,11 +13,11 @@ func NewDialect() jet.Dialect {
serializeOverrides := map[string]jet.SerializeOverride{}
serializeOverrides["REGEXP_LIKE"] = postgres_REGEXP_LIKE_function
serializeOverrides["CAST"] = postgresCAST
dialectParams := jet.DialectParams{
Name: "PostgreSQL",
PackageName: "postgres",
CastOverride: castFunc,
SerializeOverrides: serializeOverrides,
AliasQuoteChar: '"',
IdentifierQuoteChar: '"',
@ -31,8 +31,26 @@ func NewDialect() jet.Dialect {
return jet.NewDialect(dialectParams)
}
func castFunc(expression jet.Expression, castType string) jet.SerializeFunc {
func postgresCAST(expressions ...jet.Expression) jet.SerializeFunc {
return func(statement jet.StatementType, out *jet.SqlBuilder, options ...jet.SerializeOption) error {
if len(expressions) < 2 {
return errors.New("jet: invalid number of expressions for operator")
}
expression := expressions[0]
litExpr, ok := expressions[1].(jet.LiteralExpression)
if !ok {
return errors.New("jet: cast invalid cast type")
}
castType, ok := litExpr.Value().(string)
if !ok {
return errors.New("jet: cast type is not string")
}
if err := jet.Serialize(expression, statement, out, options...); err != nil {
return err
}