dialect.go

// Copyright 2014 Eryx <evorui аt gmаil dοt cοm>, All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package mysqlgo

import (
	"strings"

	"github.com/lynkdb/iomix/rdb"
	"github.com/lynkdb/iomix/rdb/modeler"
)

const (
	dialectQuote = "`"
)

var dialectColumnTypes = map[string]string{
	"bool":            "bool",
	"string":          "varchar(%v)",
	"string-text":     "longtext",
	"date":            "date",
	"datetime":        "datetime",
	"int8":            "tinyint",
	"int16":           "smallint",
	"int32":           "integer",
	"int64":           "bigint",
	"uint8":           "tinyint unsigned",
	"uint16":          "smallint unsigned",
	"uint32":          "integer unsigned",
	"uint64":          "bigint unsigned",
	"float64":         "double precision",
	"float64-decimal": "numeric(%v, %v)",
	"bytes":           "LONGBLOB",
}

var dialectStmts = map[string]string{
	"insertIgnore": "INSERT IGNORE INTO %s (%s) VALUES (%s)",
}

var dialectAllowFuncs = map[string]bool{
	"COUNT":      true,
	"SUM":        true,
	"LENGTH":     true,
	"MIN":        true,
	"MAX":        true,
	"MOD":        true,
	"COMPRESS":   true,
	"UNCOMPRESS": true,
}

func dialectQuoteStr(name string) string {

	if name == "*" {
		return name
	}

	if n := strings.IndexByte(name, '('); n > 0 {
		upName := strings.ToUpper(name)
		if dialectAllowFuncs[upName[:n]] {
			return name
		}
	}

	return dialectQuote + name + dialectQuote
}

type Dialect struct {
	rdb.Base
	dbName string
}

func (dc *Dialect) DBName() string {
	return dc.dbName
}

func (dc *Dialect) Modeler() (modeler.Modeler, error) {
	return &DialectModeler{
		base: dc,
	}, nil
}

func (dc *Dialect) QuoteStr(str string) string {
	return dialectQuote + str + dialectQuote
}

func (dc *Dialect) NewFilter() rdb.Filter {
	return NewFilter()
}

func (dc *Dialect) NewQueryer() rdb.Queryer {
	return NewQueryer()
}

func (dc *Dialect) Close() {
	dc.Base.Close()
}