mirror of https://github.com/ouqiang/gocron
84 lines
2.3 KiB
Go
84 lines
2.3 KiB
Go
|
// Copyright 2016 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
|
||
|
|
||
|
// Begin a transaction
|
||
|
func (session *Session) Begin() error {
|
||
|
if session.IsAutoCommit {
|
||
|
tx, err := session.DB().Begin()
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
session.IsAutoCommit = false
|
||
|
session.IsCommitedOrRollbacked = false
|
||
|
session.Tx = tx
|
||
|
session.saveLastSQL("BEGIN TRANSACTION")
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// Rollback When using transaction, you can rollback if any error
|
||
|
func (session *Session) Rollback() error {
|
||
|
if !session.IsAutoCommit && !session.IsCommitedOrRollbacked {
|
||
|
session.saveLastSQL(session.Engine.dialect.RollBackStr())
|
||
|
session.IsCommitedOrRollbacked = true
|
||
|
return session.Tx.Rollback()
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// Commit When using transaction, Commit will commit all operations.
|
||
|
func (session *Session) Commit() error {
|
||
|
if !session.IsAutoCommit && !session.IsCommitedOrRollbacked {
|
||
|
session.saveLastSQL("COMMIT")
|
||
|
session.IsCommitedOrRollbacked = true
|
||
|
var err error
|
||
|
if err = session.Tx.Commit(); err == nil {
|
||
|
// handle processors after tx committed
|
||
|
|
||
|
closureCallFunc := func(closuresPtr *[]func(interface{}), bean interface{}) {
|
||
|
|
||
|
if closuresPtr != nil {
|
||
|
for _, closure := range *closuresPtr {
|
||
|
closure(bean)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
for bean, closuresPtr := range session.afterInsertBeans {
|
||
|
closureCallFunc(closuresPtr, bean)
|
||
|
|
||
|
if processor, ok := interface{}(bean).(AfterInsertProcessor); ok {
|
||
|
processor.AfterInsert()
|
||
|
}
|
||
|
}
|
||
|
for bean, closuresPtr := range session.afterUpdateBeans {
|
||
|
closureCallFunc(closuresPtr, bean)
|
||
|
|
||
|
if processor, ok := interface{}(bean).(AfterUpdateProcessor); ok {
|
||
|
processor.AfterUpdate()
|
||
|
}
|
||
|
}
|
||
|
for bean, closuresPtr := range session.afterDeleteBeans {
|
||
|
closureCallFunc(closuresPtr, bean)
|
||
|
|
||
|
if processor, ok := interface{}(bean).(AfterDeleteProcessor); ok {
|
||
|
processor.AfterDelete()
|
||
|
}
|
||
|
}
|
||
|
cleanUpFunc := func(slices *map[interface{}]*[]func(interface{})) {
|
||
|
if len(*slices) > 0 {
|
||
|
*slices = make(map[interface{}]*[]func(interface{}), 0)
|
||
|
}
|
||
|
}
|
||
|
cleanUpFunc(&session.afterInsertBeans)
|
||
|
cleanUpFunc(&session.afterUpdateBeans)
|
||
|
cleanUpFunc(&session.afterDeleteBeans)
|
||
|
}
|
||
|
return err
|
||
|
}
|
||
|
return nil
|
||
|
}
|