aa9b62f275
Signed-off-by: 商讯在线 <swhbox@foxmail.com>
69 lines
1.7 KiB
Go
69 lines
1.7 KiB
Go
package xorm
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"reflect"
|
|
"runtime"
|
|
"sync"
|
|
)
|
|
|
|
const (
|
|
Version string = "0.3.1"
|
|
)
|
|
|
|
func close(engine *Engine) {
|
|
engine.Close()
|
|
}
|
|
|
|
// new a db manager according to the parameter. Currently support four
|
|
// drivers
|
|
func NewEngine(driverName string, dataSourceName string) (*Engine, error) {
|
|
engine := &Engine{
|
|
DriverName: driverName,
|
|
DataSourceName: dataSourceName,
|
|
Filters: make([]Filter, 0),
|
|
TimeZone: "UTC",
|
|
}
|
|
engine.SetMapper(SnakeMapper{})
|
|
|
|
if driverName == SQLITE {
|
|
engine.dialect = &sqlite3{}
|
|
} else if driverName == MYSQL {
|
|
engine.dialect = &mysql{}
|
|
} else if driverName == POSTGRES {
|
|
engine.dialect = &postgres{}
|
|
engine.Filters = append(engine.Filters, &PgSeqFilter{})
|
|
engine.Filters = append(engine.Filters, &QuoteFilter{})
|
|
} else if driverName == MYMYSQL {
|
|
engine.dialect = &mymysql{}
|
|
} else if driverName == "odbc" {
|
|
engine.dialect = &mssql{quoteFilter: &QuoteFilter{}}
|
|
engine.Filters = append(engine.Filters, &QuoteFilter{})
|
|
} else if driverName == ORACLE_OCI {
|
|
engine.dialect = &oracle{}
|
|
engine.Filters = append(engine.Filters, &QuoteFilter{})
|
|
} else {
|
|
return nil, errors.New(fmt.Sprintf("Unsupported driver name: %v", driverName))
|
|
}
|
|
err := engine.dialect.Init(driverName, dataSourceName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
engine.Tables = make(map[reflect.Type]*Table)
|
|
engine.mutex = &sync.RWMutex{}
|
|
engine.TagIdentifier = "xorm"
|
|
|
|
engine.Filters = append(engine.Filters, &IdFilter{})
|
|
engine.Logger = os.Stdout
|
|
|
|
//engine.Pool = NewSimpleConnectPool()
|
|
//engine.Pool = NewNoneConnectPool()
|
|
//engine.Cacher = NewLRUCacher()
|
|
err = engine.SetPool(NewSysConnectPool())
|
|
runtime.SetFinalizer(engine, close)
|
|
return engine, err
|
|
}
|