102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
// Copyright 2015 The Xorm Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// +build go1.8
|
|
|
|
package xorm
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"runtime"
|
|
"time"
|
|
|
|
"github.com/xormplus/xorm/caches"
|
|
"github.com/xormplus/xorm/core"
|
|
"github.com/xormplus/xorm/dialects"
|
|
"github.com/xormplus/xorm/log"
|
|
"github.com/xormplus/xorm/names"
|
|
"github.com/xormplus/xorm/schemas"
|
|
"github.com/xormplus/xorm/tags"
|
|
)
|
|
|
|
const (
|
|
// Version show the xorm's version
|
|
Version string = "1.0.5.0912"
|
|
)
|
|
|
|
func close(engine *Engine) {
|
|
engine.Close()
|
|
}
|
|
|
|
// NewEngine new a db manager according to the parameter. Currently support four
|
|
// drivers
|
|
func NewEngine(driverName string, dataSourceName string) (*Engine, error) {
|
|
dialect, err := dialects.OpenDialect(driverName, dataSourceName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
db, err := core.Open(driverName, dataSourceName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return newEngine(driverName, dataSourceName, dialect, db)
|
|
}
|
|
|
|
func newEngine(driverName, dataSourceName string, dialect dialects.Dialect, db *core.DB) (*Engine, error) {
|
|
cacherMgr := caches.NewManager()
|
|
mapper := names.NewCacheMapper(new(names.SnakeMapper))
|
|
tagParser := tags.NewParser("xorm", dialect, mapper, mapper, cacherMgr)
|
|
|
|
engine := &Engine{
|
|
dialect: dialect,
|
|
TZLocation: time.Local,
|
|
defaultContext: context.Background(),
|
|
cacherMgr: cacherMgr,
|
|
tagParser: tagParser,
|
|
driverName: driverName,
|
|
dataSourceName: dataSourceName,
|
|
db: db,
|
|
logSessionID: false,
|
|
}
|
|
|
|
if dialect.URI().DBType == schemas.SQLITE {
|
|
engine.DatabaseTZ = time.UTC
|
|
} else {
|
|
engine.DatabaseTZ = time.Local
|
|
}
|
|
|
|
logger := log.NewSimpleLogger(os.Stdout)
|
|
logger.SetLevel(log.LOG_INFO)
|
|
engine.SetLogger(log.NewLoggerAdapter(logger))
|
|
|
|
runtime.SetFinalizer(engine, func(engine *Engine) {
|
|
_ = engine.Close()
|
|
})
|
|
|
|
return engine, nil
|
|
}
|
|
|
|
// NewEngineWithParams new a db manager with params. The params will be passed to dialects.
|
|
func NewEngineWithParams(driverName string, dataSourceName string, params map[string]string) (*Engine, error) {
|
|
engine, err := NewEngine(driverName, dataSourceName)
|
|
engine.dialect.SetParams(params)
|
|
return engine, err
|
|
}
|
|
|
|
// NewEngineWithDialectAndDB new a db manager according to the parameter.
|
|
// If you do not want to use your own dialect or db, please use NewEngine.
|
|
// For creating dialect, you can call dialects.OpenDialect. And, for creating db,
|
|
// you can call core.Open or core.FromDB.
|
|
func NewEngineWithDialectAndDB(driverName, dataSourceName string, dialect dialects.Dialect, db *core.DB) (*Engine, error) {
|
|
return newEngine(driverName, dataSourceName, dialect, db)
|
|
}
|
|
|
|
// Clone clone an engine
|
|
func (engine *Engine) Clone() (*Engine, error) {
|
|
return NewEngine(engine.DriverName(), engine.DataSourceName())
|
|
}
|