Dialect refactor improvements and clean up.

postgres/dialect.go

@@ -1,24 +1,30 @@
 package postgres
 
 import (
+	"errors"
 	"github.com/go-jet/jet/internal/jet"
 	"strconv"
+	"strings"
 )
 
 var Dialect = NewDialect()
 
 func NewDialect() jet.Dialect {
 
+	serializeOverrides := map[string]jet.SerializeOverride{}
+	serializeOverrides["REGEXP_LIKE"] = postgres_REGEXP_LIKE_function
+
 	dialectParams := jet.DialectParams{
 		Name:                "PostgreSQL",
 		PackageName:         "postgres",
 		CastOverride:        castFunc,
+		SerializeOverrides:  serializeOverrides,
 		AliasQuoteChar:      '"',
 		IdentifierQuoteChar: '"',
 		ArgumentPlaceholder: func(ord int) string {
 			return "$" + strconv.Itoa(ord)
 		},
-		UpdateAssigment:   postgresUpdateAssigment,
+		SetClause:         postgresSetClause,
 		SupportsReturning: true,
 	}
 
@@ -35,7 +41,7 @@ func castFunc(expression jet.Expression, castType string) jet.SerializeFunc {
 	}
 }
 
-func postgresUpdateAssigment(columns []jet.IColumn, values []jet.Clause, out *jet.SqlBuilder) (err error) {
+func postgresSetClause(columns []jet.IColumn, values []jet.Clause, out *jet.SqlBuilder) (err error) {
 	if len(columns) > 1 {
 		out.WriteString("(")
 	}
@@ -68,3 +74,37 @@ func postgresUpdateAssigment(columns []jet.IColumn, values []jet.Clause, out *je
 
 	return
 }
+
+func postgres_REGEXP_LIKE_function(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")
+		}
+
+		if err := jet.Serialize(expressions[0], statement, out, options...); err != nil {
+			return err
+		}
+
+		caseSensitive := false
+
+		if len(expressions) >= 3 {
+			if stringLiteral, ok := expressions[2].(jet.LiteralExpression); ok {
+				matchType := stringLiteral.Value().(string)
+
+				caseSensitive = !strings.Contains(matchType, "i")
+			}
+		}
+
+		if caseSensitive {
+			out.WriteString("~")
+		} else {
+			out.WriteString("~*")
+		}
+
+		if err := jet.Serialize(expressions[1], statement, out, options...); err != nil {
+			return err
+		}
+
+		return nil
+	}
+}