compiler.go

package goen

import (
	"reflect"

	sqr "github.com/Masterminds/squirrel"
	"github.com/kamichidu/goen/dialect"
)

var (
	DefaultCompiler PatchCompiler = &defaultCompiler{}

	BulkCompiler PatchCompiler = &BulkCompilerOptions{}
)

type CompilerHook interface {
	PostInsertBuilder(sqr.InsertBuilder) sqr.Sqlizer
	PostUpdateBuilder(sqr.UpdateBuilder) sqr.Sqlizer
	PostDeleteBuilder(sqr.DeleteBuilder) sqr.Sqlizer
}

type CompilerOptions struct {
	Dialect dialect.Dialect

	Patches *PatchList

	Hook CompilerHook
}

type PatchCompiler interface {
	Compile(*CompilerOptions) *SqlizerList
}

func CompilerWithHook(c PatchCompiler, v CompilerHook) PatchCompiler {
	return PatchCompilerFunc(func(opts *CompilerOptions) *SqlizerList {
		opts.Hook = v
		return c.Compile(opts)
	})
}

type PatchCompilerFunc func(*CompilerOptions) *SqlizerList

func (fn PatchCompilerFunc) Compile(opts *CompilerOptions) (sqlizers *SqlizerList) {
	return fn(opts)
}

type compilerOptionsUtils CompilerOptions

func (opts *compilerOptionsUtils) StatementBuilder() sqr.StatementBuilderType {
	v := sqr.StatementBuilder
	if opts.Dialect != nil {
		v = v.PlaceholderFormat(opts.Dialect.PlaceholderFormat())
	}
	return v
}

func (opts *compilerOptionsUtils) Quote(s string) string {
	if opts.Dialect != nil {
		s = opts.Dialect.Quote(s)
	}
	return s
}

func (opts *compilerOptionsUtils) Quotes(l []string) []string {
	out := make([]string, len(l))
	for i := range l {
		out[i] = opts.Quote(l[i])
	}
	return out
}

func (opts *compilerOptionsUtils) RowKeyToSqlizer(rowKey RowKey) sqr.Sqlizer {
	if opts.Dialect != nil {
		return rowKey.ToSqlizerWithDialect(opts.Dialect)
	} else {
		return rowKey
	}
}

func (opts *compilerOptionsUtils) PostInsertBuilder(stmt sqr.InsertBuilder) sqr.Sqlizer {
	if opts.Hook != nil {
		return opts.Hook.PostInsertBuilder(stmt)
	} else {
		return stmt
	}
}

func (opts *compilerOptionsUtils) PostUpdateBuilder(stmt sqr.UpdateBuilder) sqr.Sqlizer {
	if opts.Hook != nil {
		return opts.Hook.PostUpdateBuilder(stmt)
	} else {
		return stmt
	}
}

func (opts *compilerOptionsUtils) PostDeleteBuilder(stmt sqr.DeleteBuilder) sqr.Sqlizer {
	if opts.Hook != nil {
		return opts.Hook.PostDeleteBuilder(stmt)
	} else {
		return stmt
	}
}

type defaultCompiler struct{}

func (*defaultCompiler) Compile(options *CompilerOptions) (sqlizers *SqlizerList) {
	opts := (*compilerOptionsUtils)(options)
	stmtBuilder := opts.StatementBuilder()
	sqlizers = NewSqlizerList()
	for curr := opts.Patches.Front(); curr != nil; curr = curr.Next() {
		patch := curr.GetValue()
		if len(patch.Columns) != len(patch.Values) {
			panic("goen: number of columns and values are mismatched")
		}
		switch patch.Kind {
		case PatchInsert:
			stmt := stmtBuilder.Insert(opts.Quote(patch.TableName)).
				Columns(opts.Quotes(patch.Columns)...).
				Values(patch.Values...)
			sqlizers.PushBack(opts.PostInsertBuilder(stmt))
		case PatchUpdate:
			stmt := stmtBuilder.Update(opts.Quote(patch.TableName))
			for i := range patch.Columns {
				stmt = stmt.Set(opts.Quote(patch.Columns[i]), patch.Values[i])
			}
			if patch.RowKey != nil {
				stmt = stmt.Where(opts.RowKeyToSqlizer(patch.RowKey))
			}
			sqlizers.PushBack(opts.PostUpdateBuilder(stmt))
		case PatchDelete:
			stmt := stmtBuilder.Delete(opts.Quote(patch.TableName))
			if patch.RowKey != nil {
				stmt = stmt.Where(opts.RowKeyToSqlizer(patch.RowKey))
			}
			sqlizers.PushBack(opts.PostDeleteBuilder(stmt))
		default:
			panic("goen: unable to make sql statement for unknown kind (" + string(patch.Kind) + ")")
		}
	}
	return sqlizers
}

type BulkCompilerOptions struct {
	// MaxPatches limits values per bulk operations.
	// MaxPatches<=0 means unlimited.
	MaxPatches int
}

func (c *BulkCompilerOptions) Compile(options *CompilerOptions) (sqlizers *SqlizerList) {
	opts := (*compilerOptionsUtils)(options)
	stmtBuilder := opts.StatementBuilder()
	sqlizers = NewSqlizerList()
	for curr := opts.Patches.Front(); curr != nil; curr = curr.Next() {
		patch := curr.GetValue()
		if len(patch.Columns) != len(patch.Values) {
			panic("goen: number of columns and values are mismatched")
		}

		switch patch.Kind {
		case PatchInsert:
			stmt := stmtBuilder.Insert(opts.Quote(patch.TableName)).Columns(opts.Quotes(patch.Columns)...).Values(patch.Values...)
			chunks := 1
			for c.canTakeMoreChunks(chunks) && curr.Next() != nil && c.isCompat(patch, curr.Next().GetValue()) {
				curr = curr.Next()
				stmt = stmt.Values(curr.GetValue().Values...)
				chunks++
			}
			sqlizers.PushBack(opts.PostInsertBuilder(stmt))
		case PatchDelete:
			stmt := stmtBuilder.Delete(opts.Quote(patch.TableName))
			cond := sqr.Or{}
			if patch.RowKey != nil {
				cond = append(cond, opts.RowKeyToSqlizer(patch.RowKey))
			}
			chunks := 1
			for c.canTakeMoreChunks(chunks) && curr.Next() != nil && c.isCompat(patch, curr.Next().GetValue()) {
				curr = curr.Next()
				if np := curr.GetValue(); np.RowKey != nil {
					cond = append(cond, opts.RowKeyToSqlizer(np.RowKey))
					chunks++
				}
			}
			stmt = stmt.Where(cond)
			sqlizers.PushBack(opts.PostDeleteBuilder(stmt))
		case PatchUpdate:
			stmt := stmtBuilder.Update(opts.Quote(patch.TableName))
			for i := range patch.Columns {
				stmt = stmt.Set(opts.Quote(patch.Columns[i]), patch.Values[i])
			}
			cond := sqr.Or{}
			if patch.RowKey != nil {
				cond = append(cond, opts.RowKeyToSqlizer(patch.RowKey))
			}
			chunks := 1
			for c.canTakeMoreChunks(chunks) && curr.Next() != nil && c.isCompat(patch, curr.Next().GetValue()) {
				curr = curr.Next()
				if np := curr.GetValue(); np.RowKey != nil {
					cond = append(cond, opts.RowKeyToSqlizer(np.RowKey))
					chunks++
				}
			}
			stmt = stmt.Where(cond)
			sqlizers.PushBack(opts.PostUpdateBuilder(stmt))
		default:
			fallbackOpts := &CompilerOptions{
				Dialect: opts.Dialect,
				Patches: NewPatchList(),
				Hook:    opts.Hook,
			}
			// fallback a patch by patch, for keeping its order
			fallbackOpts.Patches.PushBack(patch)
			sqlizers.PushBackList(DefaultCompiler.Compile(fallbackOpts))
		}
	}
	return sqlizers
}

func (c *BulkCompilerOptions) isCompat(p1, p2 *Patch) bool {
	if p1.Kind != p2.Kind {
		return false
	}
	if p1.TableName != p2.TableName {
		return false
	}
	if len(p1.Columns) != len(p2.Columns) {
		return false
	} else {
		for i := range p1.Columns {
			if p1.Columns[i] != p2.Columns[i] {
				return false
			}
		}
	}
	switch p1.Kind {
	case PatchUpdate:
		// do not use "database/sql/driver".Valuer.
		// it's for converting go type to sql type; type converting.
		// if converts the actual value, maybe illegal implementation.
		if !reflect.DeepEqual(p1.Values, p2.Values) {
			return false
		}
	}
	return true
}

func (c *BulkCompilerOptions) canTakeMoreChunks(chunks int) bool {
	if c.MaxPatches <= 0 {
		return true
	}
	return chunks < c.MaxPatches
}