mirror of https://github.com/VinGarcia/ksql.git
119 lines
2.3 KiB
Go
119 lines
2.3 KiB
Go
package ksql
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
)
|
|
|
|
type insertMethod int
|
|
|
|
const (
|
|
insertWithReturning insertMethod = iota
|
|
insertWithOutput
|
|
insertWithLastInsertID
|
|
insertWithNoIDRetrieval
|
|
)
|
|
|
|
var supportedDialects = map[string]Dialect{
|
|
"postgres": &postgresDialect{},
|
|
"sqlite3": &sqlite3Dialect{},
|
|
"mysql": &mysqlDialect{},
|
|
"sqlserver": &sqlserverDialect{},
|
|
}
|
|
|
|
// Dialect is used to represent the different ways
|
|
// of writing SQL queries used by each SQL driver.
|
|
type Dialect interface {
|
|
InsertMethod() insertMethod
|
|
Escape(str string) string
|
|
Placeholder(idx int) string
|
|
DriverName() string
|
|
}
|
|
|
|
type postgresDialect struct{}
|
|
|
|
func (postgresDialect) DriverName() string {
|
|
return "postgres"
|
|
}
|
|
|
|
func (postgresDialect) InsertMethod() insertMethod {
|
|
return insertWithReturning
|
|
}
|
|
|
|
func (postgresDialect) Escape(str string) string {
|
|
return `"` + str + `"`
|
|
}
|
|
|
|
func (postgresDialect) Placeholder(idx int) string {
|
|
return "$" + strconv.Itoa(idx+1)
|
|
}
|
|
|
|
type sqlite3Dialect struct{}
|
|
|
|
func (sqlite3Dialect) DriverName() string {
|
|
return "sqlite3"
|
|
}
|
|
|
|
func (sqlite3Dialect) InsertMethod() insertMethod {
|
|
return insertWithLastInsertID
|
|
}
|
|
|
|
func (sqlite3Dialect) Escape(str string) string {
|
|
return "`" + str + "`"
|
|
}
|
|
|
|
func (sqlite3Dialect) Placeholder(idx int) string {
|
|
return "?"
|
|
}
|
|
|
|
// GetDriverDialect instantiantes the dialect for the
|
|
// provided driver string, if the drive is not supported
|
|
// it returns an error
|
|
func GetDriverDialect(driver string) (Dialect, error) {
|
|
dialect, found := map[string]Dialect{
|
|
"postgres": &postgresDialect{},
|
|
"sqlite3": &sqlite3Dialect{},
|
|
}[driver]
|
|
if !found {
|
|
return nil, fmt.Errorf("unsupported driver `%s`", driver)
|
|
}
|
|
|
|
return dialect, nil
|
|
}
|
|
|
|
type mysqlDialect struct{}
|
|
|
|
func (mysqlDialect) DriverName() string {
|
|
return "mysql"
|
|
}
|
|
|
|
func (mysqlDialect) InsertMethod() insertMethod {
|
|
return insertWithLastInsertID
|
|
}
|
|
|
|
func (mysqlDialect) Escape(str string) string {
|
|
return "`" + str + "`"
|
|
}
|
|
|
|
func (mysqlDialect) Placeholder(idx int) string {
|
|
return "?"
|
|
}
|
|
|
|
type sqlserverDialect struct{}
|
|
|
|
func (sqlserverDialect) DriverName() string {
|
|
return "sqlserver"
|
|
}
|
|
|
|
func (sqlserverDialect) InsertMethod() insertMethod {
|
|
return insertWithOutput
|
|
}
|
|
|
|
func (sqlserverDialect) Escape(str string) string {
|
|
return `[` + str + `]`
|
|
}
|
|
|
|
func (sqlserverDialect) Placeholder(idx int) string {
|
|
return "@p" + strconv.Itoa(idx+1)
|
|
}
|