12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091 |
- // 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().BeginTx(session.ctx, nil)
- 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("ROLL BACK")
- session.isCommitedOrRollbacked = true
- session.isAutoCommit = 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
- session.isAutoCommit = true
-
- if err := session.tx.Commit(); err != nil {
- return err
- }
-
- // 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 nil
- }
-
- // IsInTx if current session is in a transaction
- func (session *Session) IsInTx() bool {
- return !session.isAutoCommit
- }
|