database/sql Tx - обнаружение фиксации или отката

Question

Использование пакетов базы данных/sql и драйверов и Tx не представляется возможным, чтобы определить, была ли транзакция выполнена или откат, не пытаясь повторить попытку и получив ошибку как результат, а затем изучить ошибку, чтобы определить тип ошибки. Я хотел бы иметь возможность определить из объекта Tx, независимо от того, было оно совершено или нет. Конечно, я могу определить и установить другую переменную в функции, которая использует Tx, но у меня их довольно много, и каждый раз это время 2 (переменная и назначение). У меня также есть отложенная функция для выполнения отката, если это необходимо, и ему необходимо передать переменную bool.

Было бы приемлемым, чтобы установить переменную Tx в nil после Commit или Rollback, и GC восстановит любую память, или это нет-нет, или есть лучшая альтернатива?

Answer 1

Зачем вам это нужно? Функция, вызывающая Begin(), также должна звонить Commit() или Rollback() и возвращать соответствующую ошибку.

Например, этот код фиксации или отката в зависимости от того, возвращается ошибка:

func (s Service) DoSomething() (err error) { 
    tx, err := s.db.Begin() 
    if err != nil { 
     return 
    } 
    defer func() { 
     if err != nil { 
      tx.Rollback() 
      return 
     } 
     err = tx.Commit() 
    }() 
    if _, err = tx.Exec(...); err != nil { 
     return 
    } 
    if _, err = tx.Exec(...); err != nil { 
     return 
    } 
    // ... 
    return 
} 

Обратите внимание, как я проверяю error видеть, должен ли я совершить или откат. Однако вышеприведенный пример не обрабатывает паники.

Мне не нравится делать логику commit/rollback для каждой процедуры базы данных, поэтому я обычно обертываю их обработчиком транзакций. Что-то вдоль линий этого:

func Transact(db *sql.DB, txFunc func(*sql.Tx) error) (err error) { 
    tx, err := db.Begin() 
    if err != nil { 
     return 
    } 
    defer func() { 
     if p := recover(); p != nil { 
      tx.Rollback() 
      panic(p) // re-throw panic after Rollback 
     } else if err != nil { 
      tx.Rollback() 
     } else { 
      err = tx.Commit() 
     } 
    }() 
    err = txFunc(tx) 
    return err 
} 

Это позволяет мне сделать это вместо:

func (s Service) DoSomething() error { 
    return Transact(s.db, func (tx *sql.Tx) error { 
     if _, err := tx.Exec(...); err != nil { 
      return err 
     } 
     if _, err := tx.Exec(...); err != nil { 
      return err 
     } 
    }) 
} 

Обратите внимание, что если что-то в моей транзакции паникует она автоматически обрабатывается обработчиком транзакций.

В моей реальной реализации я передаю интерфейс вместо * sql.Tx, чтобы предотвратить нежелательные вызовы Commit() или Rollback().

Вот простой фрагмент кода, чтобы продемонстрировать, как defer работы (печать 4, а не 5):

package main 

func test() (i int) { 
    defer func() { 
     i = 4 
    }() 
    return 5 
} 

func main() { 
    println(test()) 
} 

http://play.golang.org/p/0OinYDWFlx