options.go

package xsql

import (
	"fmt"
	"time"
)

var DefaultOptions = newDefaultOptions()

func newDefaultOptions() sqlOptions {
	return sqlOptions{
		Tag:          "xsql",
		InsertKey:    "INSERT INTO",
		TableKey:     "${TABLE}",
		Placeholder:  "?",
		ColumnQuotes: "`",
		TimeLayout:   "2006-01-02 15:04:05",
		TimeLocation: time.Local,
		TimeFunc: func(placeholder string) string {
			return placeholder
		},
		DebugFunc: nil,
	}
}

type sqlOptions struct {
	// Default: xsql
	Tag string

	// Default: INSERT INTO
	InsertKey string

	// Default: ${TABLE}
	TableKey string

	// Default: ?
	// For oracle, can be configured as :%d
	Placeholder string

	// Default: `
	// For oracle, can be configured as "
	ColumnQuotes string

	// Default: 2006-01-02 15:04:05
	TimeLayout string

	// Default: time.Local
	TimeLocation *time.Location

	// Default: func(placeholder string) string { return placeholder }
	// For oracle, this closure can be modified to add TO_TIMESTAMP
	TimeFunc TimeFunc

	// Global debug SQL
	DebugFunc DebugFunc
}

func mergeOptions(opts []SqlOption) *sqlOptions {
	cp := DefaultOptions // copy
	for _, o := range opts {
		o.apply(&cp)
	}
	return &cp
}

type SqlOption interface {
	apply(*sqlOptions)
}

type funcSqlOption struct {
	f func(*sqlOptions)
}

func (fdo *funcSqlOption) apply(do *sqlOptions) {
	fdo.f(do)
}

func WithTag(tag string) SqlOption {
	return &funcSqlOption{func(opts *sqlOptions) {
		opts.Tag = tag
	}}
}

func WithInsertKey(insertKey string) SqlOption {
	return &funcSqlOption{func(opts *sqlOptions) {
		opts.InsertKey = insertKey
	}}
}

func WithPlaceholder(placeholder string) SqlOption {
	return &funcSqlOption{func(opts *sqlOptions) {
		opts.Placeholder = placeholder
	}}
}

func WithColumnQuotes(columnQuotes string) SqlOption {
	return &funcSqlOption{func(opts *sqlOptions) {
		opts.ColumnQuotes = columnQuotes
	}}
}

func WithTimeLayout(timeLayout string) SqlOption {
	return &funcSqlOption{func(opts *sqlOptions) {
		opts.TimeLayout = timeLayout
	}}
}

func WithTimeLocation(timeLocation *time.Location) SqlOption {
	return &funcSqlOption{func(opts *sqlOptions) {
		opts.TimeLocation = timeLocation
	}}
}

func WithTimeFunc(f TimeFunc) SqlOption {
	return &funcSqlOption{func(opts *sqlOptions) {
		opts.TimeFunc = f
	}}
}

func WithDebugFunc(f DebugFunc) SqlOption {
	return &funcSqlOption{func(opts *sqlOptions) {
		opts.DebugFunc = f
	}}
}

func UseOracle() SqlOption {
	return &funcSqlOption{func(opts *sqlOptions) {
		opts.Placeholder = `:%d`
		opts.ColumnQuotes = `"`
		opts.TimeFunc = func(placeholder string) string {
			return fmt.Sprintf("TO_TIMESTAMP(%s, 'SYYYY-MM-DD HH24:MI:SS:FF6')", placeholder)
		}
	}}
}

func (t *sqlOptions) doDebug(l *Log) {
	if t.DebugFunc == nil {
		return
	}
	t.DebugFunc(l)
}