fix MSSQL bug on org (#3405)
parent
a0c397df08
commit
97fe773491
@ -0,0 +1,26 @@ |
|||||||
|
// Copyright 2017 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" |
||||||
|
|
||||||
|
// PingContext tests if database is alive
|
||||||
|
func (engine *Engine) PingContext(ctx context.Context) error { |
||||||
|
session := engine.NewSession() |
||||||
|
defer session.Close() |
||||||
|
return session.PingContext(ctx) |
||||||
|
} |
||||||
|
|
||||||
|
// PingContext test if database is ok
|
||||||
|
func (session *Session) PingContext(ctx context.Context) error { |
||||||
|
if session.isAutoClose { |
||||||
|
defer session.Close() |
||||||
|
} |
||||||
|
|
||||||
|
session.engine.logger.Infof("PING DATABASE %v", session.engine.DriverName()) |
||||||
|
return session.DB().PingContext(ctx) |
||||||
|
} |
@ -0,0 +1,194 @@ |
|||||||
|
// Copyright 2017 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.
|
||||||
|
|
||||||
|
package xorm |
||||||
|
|
||||||
|
import ( |
||||||
|
"github.com/go-xorm/core" |
||||||
|
) |
||||||
|
|
||||||
|
// EngineGroup defines an engine group
|
||||||
|
type EngineGroup struct { |
||||||
|
*Engine |
||||||
|
slaves []*Engine |
||||||
|
policy GroupPolicy |
||||||
|
} |
||||||
|
|
||||||
|
// NewEngineGroup creates a new engine group
|
||||||
|
func NewEngineGroup(args1 interface{}, args2 interface{}, policies ...GroupPolicy) (*EngineGroup, error) { |
||||||
|
var eg EngineGroup |
||||||
|
if len(policies) > 0 { |
||||||
|
eg.policy = policies[0] |
||||||
|
} else { |
||||||
|
eg.policy = RoundRobinPolicy() |
||||||
|
} |
||||||
|
|
||||||
|
driverName, ok1 := args1.(string) |
||||||
|
conns, ok2 := args2.([]string) |
||||||
|
if ok1 && ok2 { |
||||||
|
engines := make([]*Engine, len(conns)) |
||||||
|
for i, conn := range conns { |
||||||
|
engine, err := NewEngine(driverName, conn) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
engine.engineGroup = &eg |
||||||
|
engines[i] = engine |
||||||
|
} |
||||||
|
|
||||||
|
eg.Engine = engines[0] |
||||||
|
eg.slaves = engines[1:] |
||||||
|
return &eg, nil |
||||||
|
} |
||||||
|
|
||||||
|
master, ok3 := args1.(*Engine) |
||||||
|
slaves, ok4 := args2.([]*Engine) |
||||||
|
if ok3 && ok4 { |
||||||
|
master.engineGroup = &eg |
||||||
|
for i := 0; i < len(slaves); i++ { |
||||||
|
slaves[i].engineGroup = &eg |
||||||
|
} |
||||||
|
eg.Engine = master |
||||||
|
eg.slaves = slaves |
||||||
|
return &eg, nil |
||||||
|
} |
||||||
|
return nil, ErrParamsType |
||||||
|
} |
||||||
|
|
||||||
|
// Close the engine
|
||||||
|
func (eg *EngineGroup) Close() error { |
||||||
|
err := eg.Engine.Close() |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
for i := 0; i < len(eg.slaves); i++ { |
||||||
|
err := eg.slaves[i].Close() |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
// Master returns the master engine
|
||||||
|
func (eg *EngineGroup) Master() *Engine { |
||||||
|
return eg.Engine |
||||||
|
} |
||||||
|
|
||||||
|
// Ping tests if database is alive
|
||||||
|
func (eg *EngineGroup) Ping() error { |
||||||
|
if err := eg.Engine.Ping(); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
for _, slave := range eg.slaves { |
||||||
|
if err := slave.Ping(); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
// SetColumnMapper set the column name mapping rule
|
||||||
|
func (eg *EngineGroup) SetColumnMapper(mapper core.IMapper) { |
||||||
|
eg.Engine.ColumnMapper = mapper |
||||||
|
for i := 0; i < len(eg.slaves); i++ { |
||||||
|
eg.slaves[i].ColumnMapper = mapper |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// SetDefaultCacher set the default cacher
|
||||||
|
func (eg *EngineGroup) SetDefaultCacher(cacher core.Cacher) { |
||||||
|
eg.Engine.SetDefaultCacher(cacher) |
||||||
|
for i := 0; i < len(eg.slaves); i++ { |
||||||
|
eg.slaves[i].SetDefaultCacher(cacher) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// SetLogger set the new logger
|
||||||
|
func (eg *EngineGroup) SetLogger(logger core.ILogger) { |
||||||
|
eg.Engine.SetLogger(logger) |
||||||
|
for i := 0; i < len(eg.slaves); i++ { |
||||||
|
eg.slaves[i].SetLogger(logger) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// SetLogLevel sets the logger level
|
||||||
|
func (eg *EngineGroup) SetLogLevel(level core.LogLevel) { |
||||||
|
eg.Engine.SetLogLevel(level) |
||||||
|
for i := 0; i < len(eg.slaves); i++ { |
||||||
|
eg.slaves[i].SetLogLevel(level) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// SetMapper set the name mapping rules
|
||||||
|
func (eg *EngineGroup) SetMapper(mapper core.IMapper) { |
||||||
|
eg.Engine.SetMapper(mapper) |
||||||
|
for i := 0; i < len(eg.slaves); i++ { |
||||||
|
eg.slaves[i].SetMapper(mapper) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// SetMaxIdleConns set the max idle connections on pool, default is 2
|
||||||
|
func (eg *EngineGroup) SetMaxIdleConns(conns int) { |
||||||
|
eg.Engine.db.SetMaxIdleConns(conns) |
||||||
|
for i := 0; i < len(eg.slaves); i++ { |
||||||
|
eg.slaves[i].db.SetMaxIdleConns(conns) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// SetMaxOpenConns is only available for go 1.2+
|
||||||
|
func (eg *EngineGroup) SetMaxOpenConns(conns int) { |
||||||
|
eg.Engine.db.SetMaxOpenConns(conns) |
||||||
|
for i := 0; i < len(eg.slaves); i++ { |
||||||
|
eg.slaves[i].db.SetMaxOpenConns(conns) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// SetPolicy set the group policy
|
||||||
|
func (eg *EngineGroup) SetPolicy(policy GroupPolicy) *EngineGroup { |
||||||
|
eg.policy = policy |
||||||
|
return eg |
||||||
|
} |
||||||
|
|
||||||
|
// SetTableMapper set the table name mapping rule
|
||||||
|
func (eg *EngineGroup) SetTableMapper(mapper core.IMapper) { |
||||||
|
eg.Engine.TableMapper = mapper |
||||||
|
for i := 0; i < len(eg.slaves); i++ { |
||||||
|
eg.slaves[i].TableMapper = mapper |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// ShowExecTime show SQL statement and execute time or not on logger if log level is great than INFO
|
||||||
|
func (eg *EngineGroup) ShowExecTime(show ...bool) { |
||||||
|
eg.Engine.ShowExecTime(show...) |
||||||
|
for i := 0; i < len(eg.slaves); i++ { |
||||||
|
eg.slaves[i].ShowExecTime(show...) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// ShowSQL show SQL statement or not on logger if log level is great than INFO
|
||||||
|
func (eg *EngineGroup) ShowSQL(show ...bool) { |
||||||
|
eg.Engine.ShowSQL(show...) |
||||||
|
for i := 0; i < len(eg.slaves); i++ { |
||||||
|
eg.slaves[i].ShowSQL(show...) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Slave returns one of the physical databases which is a slave according the policy
|
||||||
|
func (eg *EngineGroup) Slave() *Engine { |
||||||
|
switch len(eg.slaves) { |
||||||
|
case 0: |
||||||
|
return eg.Engine |
||||||
|
case 1: |
||||||
|
return eg.slaves[0] |
||||||
|
} |
||||||
|
return eg.policy.Slave(eg) |
||||||
|
} |
||||||
|
|
||||||
|
// Slaves returns all the slaves
|
||||||
|
func (eg *EngineGroup) Slaves() []*Engine { |
||||||
|
return eg.slaves |
||||||
|
} |
@ -0,0 +1,116 @@ |
|||||||
|
// Copyright 2017 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.
|
||||||
|
|
||||||
|
package xorm |
||||||
|
|
||||||
|
import ( |
||||||
|
"math/rand" |
||||||
|
"sync" |
||||||
|
"time" |
||||||
|
) |
||||||
|
|
||||||
|
// GroupPolicy is be used by chosing the current slave from slaves
|
||||||
|
type GroupPolicy interface { |
||||||
|
Slave(*EngineGroup) *Engine |
||||||
|
} |
||||||
|
|
||||||
|
// GroupPolicyHandler should be used when a function is a GroupPolicy
|
||||||
|
type GroupPolicyHandler func(*EngineGroup) *Engine |
||||||
|
|
||||||
|
// Slave implements the chosen of slaves
|
||||||
|
func (h GroupPolicyHandler) Slave(eg *EngineGroup) *Engine { |
||||||
|
return h(eg) |
||||||
|
} |
||||||
|
|
||||||
|
// RandomPolicy implmentes randomly chose the slave of slaves
|
||||||
|
func RandomPolicy() GroupPolicyHandler { |
||||||
|
var r = rand.New(rand.NewSource(time.Now().UnixNano())) |
||||||
|
return func(g *EngineGroup) *Engine { |
||||||
|
return g.Slaves()[r.Intn(len(g.Slaves()))] |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// WeightRandomPolicy implmentes randomly chose the slave of slaves
|
||||||
|
func WeightRandomPolicy(weights []int) GroupPolicyHandler { |
||||||
|
var rands = make([]int, 0, len(weights)) |
||||||
|
for i := 0; i < len(weights); i++ { |
||||||
|
for n := 0; n < weights[i]; n++ { |
||||||
|
rands = append(rands, i) |
||||||
|
} |
||||||
|
} |
||||||
|
var r = rand.New(rand.NewSource(time.Now().UnixNano())) |
||||||
|
|
||||||
|
return func(g *EngineGroup) *Engine { |
||||||
|
var slaves = g.Slaves() |
||||||
|
idx := rands[r.Intn(len(rands))] |
||||||
|
if idx >= len(slaves) { |
||||||
|
idx = len(slaves) - 1 |
||||||
|
} |
||||||
|
return slaves[idx] |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func RoundRobinPolicy() GroupPolicyHandler { |
||||||
|
var pos = -1 |
||||||
|
var lock sync.Mutex |
||||||
|
return func(g *EngineGroup) *Engine { |
||||||
|
var slaves = g.Slaves() |
||||||
|
|
||||||
|
lock.Lock() |
||||||
|
defer lock.Unlock() |
||||||
|
pos++ |
||||||
|
if pos >= len(slaves) { |
||||||
|
pos = 0 |
||||||
|
} |
||||||
|
|
||||||
|
return slaves[pos] |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func WeightRoundRobinPolicy(weights []int) GroupPolicyHandler { |
||||||
|
var rands = make([]int, 0, len(weights)) |
||||||
|
for i := 0; i < len(weights); i++ { |
||||||
|
for n := 0; n < weights[i]; n++ { |
||||||
|
rands = append(rands, i) |
||||||
|
} |
||||||
|
} |
||||||
|
var pos = -1 |
||||||
|
var lock sync.Mutex |
||||||
|
|
||||||
|
return func(g *EngineGroup) *Engine { |
||||||
|
var slaves = g.Slaves() |
||||||
|
lock.Lock() |
||||||
|
defer lock.Unlock() |
||||||
|
pos++ |
||||||
|
if pos >= len(rands) { |
||||||
|
pos = 0 |
||||||
|
} |
||||||
|
|
||||||
|
idx := rands[pos] |
||||||
|
if idx >= len(slaves) { |
||||||
|
idx = len(slaves) - 1 |
||||||
|
} |
||||||
|
return slaves[idx] |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// LeastConnPolicy implements GroupPolicy, every time will get the least connections slave
|
||||||
|
func LeastConnPolicy() GroupPolicyHandler { |
||||||
|
return func(g *EngineGroup) *Engine { |
||||||
|
var slaves = g.Slaves() |
||||||
|
connections := 0 |
||||||
|
idx := 0 |
||||||
|
for i := 0; i < len(slaves); i++ { |
||||||
|
openConnections := slaves[i].DB().Stats().OpenConnections |
||||||
|
if i == 0 { |
||||||
|
connections = openConnections |
||||||
|
idx = i |
||||||
|
} else if openConnections <= connections { |
||||||
|
connections = openConnections |
||||||
|
idx = i |
||||||
|
} |
||||||
|
} |
||||||
|
return slaves[idx] |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,103 @@ |
|||||||
|
// Copyright 2017 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.
|
||||||
|
|
||||||
|
package xorm |
||||||
|
|
||||||
|
import ( |
||||||
|
"database/sql" |
||||||
|
"reflect" |
||||||
|
"time" |
||||||
|
|
||||||
|
"github.com/go-xorm/core" |
||||||
|
) |
||||||
|
|
||||||
|
// Interface defines the interface which Engine, EngineGroup and Session will implementate.
|
||||||
|
type Interface interface { |
||||||
|
AllCols() *Session |
||||||
|
Alias(alias string) *Session |
||||||
|
Asc(colNames ...string) *Session |
||||||
|
BufferSize(size int) *Session |
||||||
|
Cols(columns ...string) *Session |
||||||
|
Count(...interface{}) (int64, error) |
||||||
|
CreateIndexes(bean interface{}) error |
||||||
|
CreateUniques(bean interface{}) error |
||||||
|
Decr(column string, arg ...interface{}) *Session |
||||||
|
Desc(...string) *Session |
||||||
|
Delete(interface{}) (int64, error) |
||||||
|
Distinct(columns ...string) *Session |
||||||
|
DropIndexes(bean interface{}) error |
||||||
|
Exec(string, ...interface{}) (sql.Result, error) |
||||||
|
Exist(bean ...interface{}) (bool, error) |
||||||
|
Find(interface{}, ...interface{}) error |
||||||
|
Get(interface{}) (bool, error) |
||||||
|
GroupBy(keys string) *Session |
||||||
|
ID(interface{}) *Session |
||||||
|
In(string, ...interface{}) *Session |
||||||
|
Incr(column string, arg ...interface{}) *Session |
||||||
|
Insert(...interface{}) (int64, error) |
||||||
|
InsertOne(interface{}) (int64, error) |
||||||
|
IsTableEmpty(bean interface{}) (bool, error) |
||||||
|
IsTableExist(beanOrTableName interface{}) (bool, error) |
||||||
|
Iterate(interface{}, IterFunc) error |
||||||
|
Limit(int, ...int) *Session |
||||||
|
NoAutoCondition(...bool) *Session |
||||||
|
NotIn(string, ...interface{}) *Session |
||||||
|
Join(joinOperator string, tablename interface{}, condition string, args ...interface{}) *Session |
||||||
|
Omit(columns ...string) *Session |
||||||
|
OrderBy(order string) *Session |
||||||
|
Ping() error |
||||||
|
Query(sqlOrAgrs ...interface{}) (resultsSlice []map[string][]byte, err error) |
||||||
|
QueryInterface(sqlorArgs ...interface{}) ([]map[string]interface{}, error) |
||||||
|
QueryString(sqlorArgs ...interface{}) ([]map[string]string, error) |
||||||
|
Rows(bean interface{}) (*Rows, error) |
||||||
|
SetExpr(string, string) *Session |
||||||
|
SQL(interface{}, ...interface{}) *Session |
||||||
|
Sum(bean interface{}, colName string) (float64, error) |
||||||
|
SumInt(bean interface{}, colName string) (int64, error) |
||||||
|
Sums(bean interface{}, colNames ...string) ([]float64, error) |
||||||
|
SumsInt(bean interface{}, colNames ...string) ([]int64, error) |
||||||
|
Table(tableNameOrBean interface{}) *Session |
||||||
|
Unscoped() *Session |
||||||
|
Update(bean interface{}, condiBeans ...interface{}) (int64, error) |
||||||
|
UseBool(...string) *Session |
||||||
|
Where(interface{}, ...interface{}) *Session |
||||||
|
} |
||||||
|
|
||||||
|
// EngineInterface defines the interface which Engine, EngineGroup will implementate.
|
||||||
|
type EngineInterface interface { |
||||||
|
Interface |
||||||
|
|
||||||
|
Before(func(interface{})) *Session |
||||||
|
Charset(charset string) *Session |
||||||
|
CreateTables(...interface{}) error |
||||||
|
DBMetas() ([]*core.Table, error) |
||||||
|
Dialect() core.Dialect |
||||||
|
DropTables(...interface{}) error |
||||||
|
DumpAllToFile(fp string, tp ...core.DbType) error |
||||||
|
GetColumnMapper() core.IMapper |
||||||
|
GetDefaultCacher() core.Cacher |
||||||
|
GetTableMapper() core.IMapper |
||||||
|
GetTZDatabase() *time.Location |
||||||
|
GetTZLocation() *time.Location |
||||||
|
NewSession() *Session |
||||||
|
NoAutoTime() *Session |
||||||
|
Quote(string) string |
||||||
|
SetDefaultCacher(core.Cacher) |
||||||
|
SetLogLevel(core.LogLevel) |
||||||
|
SetMapper(core.IMapper) |
||||||
|
SetTZDatabase(tz *time.Location) |
||||||
|
SetTZLocation(tz *time.Location) |
||||||
|
ShowSQL(show ...bool) |
||||||
|
Sync(...interface{}) error |
||||||
|
Sync2(...interface{}) error |
||||||
|
StoreEngine(storeEngine string) *Session |
||||||
|
TableInfo(bean interface{}) *Table |
||||||
|
UnMapType(reflect.Type) |
||||||
|
} |
||||||
|
|
||||||
|
var ( |
||||||
|
_ Interface = &Session{} |
||||||
|
_ EngineInterface = &Engine{} |
||||||
|
_ EngineInterface = &EngineGroup{} |
||||||
|
) |
Loading…
Reference in new issue