Package structure refactor.

mysql/dialect.go

@@ -0,0 +1,139 @@
+package mysql
+
+import (
+	"errors"
+	"github.com/go-jet/jet/internal/jet"
+)
+
+var Dialect = NewDialect()
+
+func NewDialect() jet.Dialect {
+
+	serializeOverrides := map[string]jet.SerializeOverride{}
+	serializeOverrides["IS DISTINCT FROM"] = mysql_IS_DISTINCT_FROM
+	serializeOverrides["IS NOT DISTINCT FROM"] = mysql_IS_NOT_DISTINCT_FROM
+	serializeOverrides["/"] = mysql_DIVISION
+	serializeOverrides["#"] = mysql_BIT_XOR
+
+	mySQLDialectParams := jet.DialectParams{
+		Name:                "MySQL",
+		PackageName:         "mysql",
+		SerializeOverrides:  serializeOverrides,
+		AliasQuoteChar:      '"',
+		IdentifierQuoteChar: '`',
+		ArgumentPlaceholder: func(int) string {
+			return "?"
+		},
+		UpdateAssigment:   mysqlUpdateAssigment,
+		SupportsReturning: false,
+	}
+
+	return jet.NewDialect(mySQLDialectParams)
+}
+
+func mysqlUpdateAssigment(columns []jet.IColumn, values []jet.Clause, out *jet.SqlBuilder) (err error) {
+
+	if len(columns) != len(values) {
+		return errors.New("jet: mismatch in numers of columns and values")
+	}
+
+	for i, column := range columns {
+		if i > 0 {
+			out.WriteString(", ")
+		}
+
+		out.WriteString(column.Name())
+
+		out.WriteString(" = ")
+
+		if err = jet.Serialize(values[i], jet.UpdateStatementType, out); err != nil {
+			return err
+		}
+	}
+
+	return nil
+}
+
+func mysql_BIT_XOR(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")
+		}
+
+		lhs := expressions[0]
+		rhs := expressions[1]
+
+		if err := jet.Serialize(lhs, statement, out, options...); err != nil {
+			return err
+		}
+
+		out.WriteString("^")
+
+		if err := jet.Serialize(rhs, statement, out, options...); err != nil {
+			return err
+		}
+		return nil
+	}
+}
+
+func mysql_DIVISION(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")
+		}
+
+		lhs := expressions[0]
+		rhs := expressions[1]
+
+		if err := jet.Serialize(lhs, statement, out, options...); err != nil {
+			return err
+		}
+
+		_, isLhsInt := lhs.(IntegerExpression)
+		_, isRhsInt := rhs.(IntegerExpression)
+
+		if isLhsInt && isRhsInt {
+			out.WriteString("DIV")
+		} else {
+			out.WriteString("/")
+		}
+
+		if err := jet.Serialize(rhs, statement, out, options...); err != nil {
+			return err
+		}
+		return nil
+	}
+}
+
+func mysql_IS_NOT_DISTINCT_FROM(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); err != nil {
+			return err
+		}
+
+		out.WriteString("<=>")
+
+		if err := jet.Serialize(expressions[1], statement, out); err != nil {
+			return err
+		}
+
+		return nil
+	}
+}
+
+func mysql_IS_DISTINCT_FROM(expressions ...jet.Expression) jet.SerializeFunc {
+	return func(statement jet.StatementType, out *jet.SqlBuilder, options ...jet.SerializeOption) error {
+		out.WriteString("NOT")
+
+		err := mysql_IS_NOT_DISTINCT_FROM(expressions...)(statement, out, options...)
+
+		if err != nil {
+			return err
+		}
+
+		return nil
+	}
+}