123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307 |
- package core
- import (
- "fmt"
- "strings"
- "time"
- )
- type DbType string
- type Uri struct {
- DbType DbType
- Proto string
- Host string
- Port string
- DbName string
- User string
- Passwd string
- Charset string
- Laddr string
- Raddr string
- Timeout time.Duration
- Schema string
- }
- // a dialect is a driver's wrapper
- type Dialect interface {
- SetLogger(logger ILogger)
- Init(*DB, *Uri, string, string) error
- URI() *Uri
- DB() *DB
- DBType() DbType
- SqlType(*Column) string
- FormatBytes(b []byte) string
- DriverName() string
- DataSourceName() string
- QuoteStr() string
- IsReserved(string) bool
- Quote(string) string
- AndStr() string
- OrStr() string
- EqStr() string
- RollBackStr() string
- AutoIncrStr() string
- SupportInsertMany() bool
- SupportEngine() bool
- SupportCharset() bool
- SupportDropIfExists() bool
- IndexOnTable() bool
- ShowCreateNull() bool
- IndexCheckSql(tableName, idxName string) (string, []interface{})
- TableCheckSql(tableName string) (string, []interface{})
- IsColumnExist(tableName string, colName string) (bool, error)
- CreateTableSql(table *Table, tableName, storeEngine, charset string) string
- DropTableSql(tableName string) string
- CreateIndexSql(tableName string, index *Index) string
- DropIndexSql(tableName string, index *Index) string
- ModifyColumnSql(tableName string, col *Column) string
- ForUpdateSql(query string) string
- //CreateTableIfNotExists(table *Table, tableName, storeEngine, charset string) error
- //MustDropTable(tableName string) error
- GetColumns(tableName string) ([]string, map[string]*Column, error)
- GetTables() ([]*Table, error)
- GetIndexes(tableName string) (map[string]*Index, error)
- Filters() []Filter
- }
- func OpenDialect(dialect Dialect) (*DB, error) {
- return Open(dialect.DriverName(), dialect.DataSourceName())
- }
- type Base struct {
- db *DB
- dialect Dialect
- driverName string
- dataSourceName string
- logger ILogger
- *Uri
- }
- func (b *Base) DB() *DB {
- return b.db
- }
- func (b *Base) SetLogger(logger ILogger) {
- b.logger = logger
- }
- func (b *Base) Init(db *DB, dialect Dialect, uri *Uri, drivername, dataSourceName string) error {
- b.db, b.dialect, b.Uri = db, dialect, uri
- b.driverName, b.dataSourceName = drivername, dataSourceName
- return nil
- }
- func (b *Base) URI() *Uri {
- return b.Uri
- }
- func (b *Base) DBType() DbType {
- return b.Uri.DbType
- }
- func (b *Base) FormatBytes(bs []byte) string {
- return fmt.Sprintf("0x%x", bs)
- }
- func (b *Base) DriverName() string {
- return b.driverName
- }
- func (b *Base) ShowCreateNull() bool {
- return true
- }
- func (b *Base) DataSourceName() string {
- return b.dataSourceName
- }
- func (b *Base) AndStr() string {
- return "AND"
- }
- func (b *Base) OrStr() string {
- return "OR"
- }
- func (b *Base) EqStr() string {
- return "="
- }
- func (db *Base) RollBackStr() string {
- return "ROLL BACK"
- }
- func (db *Base) SupportDropIfExists() bool {
- return true
- }
- func (db *Base) DropTableSql(tableName string) string {
- return fmt.Sprintf("DROP TABLE IF EXISTS `%s`", tableName)
- }
- func (db *Base) HasRecords(query string, args ...interface{}) (bool, error) {
- db.LogSQL(query, args)
- rows, err := db.DB().Query(query, args...)
- if err != nil {
- return false, err
- }
- defer rows.Close()
- if rows.Next() {
- return true, nil
- }
- return false, nil
- }
- func (db *Base) IsColumnExist(tableName, colName string) (bool, error) {
- query := "SELECT `COLUMN_NAME` FROM `INFORMATION_SCHEMA`.`COLUMNS` WHERE `TABLE_SCHEMA` = ? AND `TABLE_NAME` = ? AND `COLUMN_NAME` = ?"
- query = strings.Replace(query, "`", db.dialect.QuoteStr(), -1)
- return db.HasRecords(query, db.DbName, tableName, colName)
- }
- /*
- func (db *Base) CreateTableIfNotExists(table *Table, tableName, storeEngine, charset string) error {
- sql, args := db.dialect.TableCheckSql(tableName)
- rows, err := db.DB().Query(sql, args...)
- if db.Logger != nil {
- db.Logger.Info("[sql]", sql, args)
- }
- if err != nil {
- return err
- }
- defer rows.Close()
- if rows.Next() {
- return nil
- }
- sql = db.dialect.CreateTableSql(table, tableName, storeEngine, charset)
- _, err = db.DB().Exec(sql)
- if db.Logger != nil {
- db.Logger.Info("[sql]", sql)
- }
- return err
- }*/
- func (db *Base) CreateIndexSql(tableName string, index *Index) string {
- quote := db.dialect.Quote
- var unique string
- var idxName string
- if index.Type == UniqueType {
- unique = " UNIQUE"
- }
- idxName = index.XName(tableName)
- return fmt.Sprintf("CREATE%s INDEX %v ON %v (%v)", unique,
- quote(idxName), quote(tableName),
- quote(strings.Join(index.Cols, quote(","))))
- }
- func (db *Base) DropIndexSql(tableName string, index *Index) string {
- quote := db.dialect.Quote
- var name string
- if index.IsRegular {
- name = index.XName(tableName)
- } else {
- name = index.Name
- }
- return fmt.Sprintf("DROP INDEX %v ON %s", quote(name), quote(tableName))
- }
- func (db *Base) ModifyColumnSql(tableName string, col *Column) string {
- return fmt.Sprintf("alter table %s MODIFY COLUMN %s", tableName, col.StringNoPk(db.dialect))
- }
- func (b *Base) CreateTableSql(table *Table, tableName, storeEngine, charset string) string {
- var sql string
- sql = "CREATE TABLE IF NOT EXISTS "
- if tableName == "" {
- tableName = table.Name
- }
- sql += b.dialect.Quote(tableName)
- sql += " ("
- if len(table.ColumnsSeq()) > 0 {
- pkList := table.PrimaryKeys
- for _, colName := range table.ColumnsSeq() {
- col := table.GetColumn(colName)
- if col.IsPrimaryKey && len(pkList) == 1 {
- sql += col.String(b.dialect)
- } else {
- sql += col.StringNoPk(b.dialect)
- }
- sql = strings.TrimSpace(sql)
- sql += ", "
- }
- if len(pkList) > 1 {
- sql += "PRIMARY KEY ( "
- sql += b.dialect.Quote(strings.Join(pkList, b.dialect.Quote(",")))
- sql += " ), "
- }
- sql = sql[:len(sql)-2]
- }
- sql += ")"
- if b.dialect.SupportEngine() && storeEngine != "" {
- sql += " ENGINE=" + storeEngine
- }
- if b.dialect.SupportCharset() {
- if len(charset) == 0 {
- charset = b.dialect.URI().Charset
- }
- if len(charset) > 0 {
- sql += " DEFAULT CHARSET " + charset
- }
- }
- return sql
- }
- func (b *Base) ForUpdateSql(query string) string {
- return query + " FOR UPDATE"
- }
- func (b *Base) LogSQL(sql string, args []interface{}) {
- if b.logger != nil && b.logger.IsShowSQL() {
- if len(args) > 0 {
- b.logger.Info("[sql]", sql, args)
- } else {
- b.logger.Info("[sql]", sql)
- }
- }
- }
- var (
- dialects = map[string]func() Dialect{}
- )
- // RegisterDialect register database dialect
- func RegisterDialect(dbName DbType, dialectFunc func() Dialect) {
- if dialectFunc == nil {
- panic("core: Register dialect is nil")
- }
- dialects[strings.ToLower(string(dbName))] = dialectFunc // !nashtsai! allow override dialect
- }
- // QueryDialect query if registed database dialect
- func QueryDialect(dbName DbType) Dialect {
- if d, ok := dialects[strings.ToLower(string(dbName))]; ok {
- return d()
- }
- return nil
- }
|