[MySQL] Optimizer hints

mysql/delete_statement.go

@@ -6,6 +6,8 @@ import "github.com/go-jet/jet/v2/internal/jet"
 type DeleteStatement interface {
 	Statement
 
+	OPTIMIZER_HINTS(hints ...OptimizerHint) DeleteStatement
+
 	USING(tables ...ReadableTable) DeleteStatement
 	WHERE(expression BoolExpression) DeleteStatement
 	ORDER_BY(orderByClauses ...OrderByClause) DeleteStatement
@@ -15,7 +17,7 @@ type DeleteStatement interface {
 type deleteStatementImpl struct {
 	jet.SerializerStatement
 
-	Delete  jet.ClauseStatementBegin
+	Delete  jet.ClauseDelete
 	Using   jet.ClauseFrom
 	Where   jet.ClauseWhere
 	OrderBy jet.ClauseOrderBy
@@ -29,17 +31,22 @@ func newDeleteStatement(table Table) DeleteStatement {
 		&newDelete.Using,
 		&newDelete.Where,
 		&newDelete.OrderBy,
-		&newDelete.Limit)
+		&newDelete.Limit,
+	)
 
-	newDelete.Delete.Name = "DELETE FROM"
+	newDelete.Delete.Table = table
 	newDelete.Using.Name = "USING"
-	newDelete.Delete.Tables = append(newDelete.Delete.Tables, table)
 	newDelete.Where.Mandatory = true
 	newDelete.Limit.Count = -1
 
 	return newDelete
 }
 
+func (d *deleteStatementImpl) OPTIMIZER_HINTS(hints ...OptimizerHint) DeleteStatement {
+	d.Delete.OptimizerHints = hints
+	return d
+}
+
 func (d *deleteStatementImpl) USING(tables ...ReadableTable) DeleteStatement {
 	d.Using.Tables = readableTablesToSerializerList(tables)
 	return d

mysql/insert_statement.go

@@ -6,6 +6,8 @@ import "github.com/go-jet/jet/v2/internal/jet"
 type InsertStatement interface {
 	Statement
 
+	OPTIMIZER_HINTS(hints ...OptimizerHint) InsertStatement
+
 	// Insert row of values
 	VALUES(value interface{}, values ...interface{}) InsertStatement
 	// Insert row of values, where value for each column is extracted from filed of structure data.
@@ -22,7 +24,10 @@ type InsertStatement interface {
 func newInsertStatement(table Table, columns []jet.Column) InsertStatement {
 	newInsert := &insertStatementImpl{}
 	newInsert.SerializerStatement = jet.NewStatementImpl(Dialect, jet.InsertStatementType, newInsert,
-		&newInsert.Insert, &newInsert.ValuesQuery, &newInsert.OnDuplicateKey)
+		&newInsert.Insert,
+		&newInsert.ValuesQuery,
+		&newInsert.OnDuplicateKey,
+	)
 
 	newInsert.Insert.Table = table
 	newInsert.Insert.Columns = columns
@@ -38,6 +43,11 @@ type insertStatementImpl struct {
 	OnDuplicateKey onDuplicateKeyUpdateClause
 }
 
+func (is *insertStatementImpl) OPTIMIZER_HINTS(hints ...OptimizerHint) InsertStatement {
+	is.Insert.OptimizerHints = hints
+	return is
+}
+
 func (is *insertStatementImpl) VALUES(value interface{}, values ...interface{}) InsertStatement {
 	is.ValuesQuery.Rows = append(is.ValuesQuery.Rows, jet.UnwindRowFromValues(value, values))
 	return is

mysql/optimizer_hints.go

@@ -0,0 +1,19 @@
+package mysql
+
+import (
+	"fmt"
+	"github.com/go-jet/jet/v2/internal/jet"
+)
+
+// OptimizerHint provides a way to optimize query execution per-statement basis
+type OptimizerHint = jet.OptimizerHint
+
+// MAX_EXECUTION_TIME limits statement execution time
+func MAX_EXECUTION_TIME(miliseconds int) OptimizerHint {
+	return OptimizerHint(fmt.Sprintf("MAX_EXECUTION_TIME(%d)", miliseconds))
+}
+
+// QB_NAME assigns name to query block
+func QB_NAME(name string) OptimizerHint {
+	return OptimizerHint(fmt.Sprintf("QB_NAME(%s)", name))
+}

mysql/select_statement.go

@@ -40,6 +40,8 @@ type SelectStatement interface {
 	jet.HasProjections
 	Expression
 
+	OPTIMIZER_HINTS(hints ...OptimizerHint) SelectStatement
+
 	DISTINCT() SelectStatement
 	FROM(tables ...ReadableTable) SelectStatement
 	WHERE(expression BoolExpression) SelectStatement
@@ -65,9 +67,19 @@ func SELECT(projection Projection, projections ...Projection) SelectStatement {
 
 func newSelectStatement(table ReadableTable, projections []Projection) SelectStatement {
 	newSelect := &selectStatementImpl{}
-	newSelect.ExpressionStatement = jet.NewExpressionStatementImpl(Dialect, jet.SelectStatementType, newSelect, &newSelect.Select,
-		&newSelect.From, &newSelect.Where, &newSelect.GroupBy, &newSelect.Having, &newSelect.Window, &newSelect.OrderBy,
-		&newSelect.Limit, &newSelect.Offset, &newSelect.For, &newSelect.ShareLock)
+	newSelect.ExpressionStatement = jet.NewExpressionStatementImpl(Dialect, jet.SelectStatementType, newSelect,
+		&newSelect.Select,
+		&newSelect.From,
+		&newSelect.Where,
+		&newSelect.GroupBy,
+		&newSelect.Having,
+		&newSelect.Window,
+		&newSelect.OrderBy,
+		&newSelect.Limit,
+		&newSelect.Offset,
+		&newSelect.For,
+		&newSelect.ShareLock,
+	)
 
 	newSelect.Select.ProjectionList = projections
 	if table != nil {
@@ -100,6 +112,11 @@ type selectStatementImpl struct {
 	ShareLock jet.ClauseOptional
 }
 
+func (s *selectStatementImpl) OPTIMIZER_HINTS(hints ...OptimizerHint) SelectStatement {
+	s.Select.OptimizerHints = hints
+	return s
+}
+
 func (s *selectStatementImpl) DISTINCT() SelectStatement {
 	s.Select.Distinct = true
 	return s

mysql/update_statement.go

@@ -6,6 +6,8 @@ import "github.com/go-jet/jet/v2/internal/jet"
 type UpdateStatement interface {
 	jet.Statement
 
+	OPTIMIZER_HINTS(hints ...OptimizerHint) UpdateStatement
+
 	SET(value interface{}, values ...interface{}) UpdateStatement
 	MODEL(data interface{}) UpdateStatement
 
@@ -36,6 +38,11 @@ func newUpdateStatement(table Table, columns []jet.Column) UpdateStatement {
 	return update
 }
 
+func (u *updateStatementImpl) OPTIMIZER_HINTS(hints ...OptimizerHint) UpdateStatement {
+	u.Update.OptimizerHints = hints
+	return u
+}
+
 func (u *updateStatementImpl) SET(value interface{}, values ...interface{}) UpdateStatement {
 	columnAssigment, isColumnAssigment := value.(ColumnAssigment)