xorm/xorm.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
}