GoでDBのトランザクション処理を書いてみた

今回は、Goでトランザクション処理を書いてみます。

import (
    "database/sql"
    "log"
)

func withTransaction(ctx context.Context, db *sql.DB, txFunc func(*sql.Tx) error) error {
	tx, err := db.BeginTx(ctx, nil)
	if err != nil {
		log.Println(err.Error())
		return err
	}
	defer func() {
		if p := recover(); p != nil {
			tx.Rollback()
			panic(p) // re-throw panic after Rollback
		} else if err != nil {
			log.Println(err)
			tx.Rollback()
		} else {
			err = tx.Commit()
		}
	}()
	err = txFunc(tx)
	return err
}

第３引数には、第１引数に*sql.Txが入り、error型が返り値の無名関数が入ります。
実際の使われ方をみた方が早いので以下を参考にしてください。
GoのクエリビルダSquirrelを使用しています。

type userRepository struct {
	DBClient *sql.DB
}

func (r *userRepository) CreateUser(ctx context.Context, userID string) error {
    query := sq.Insert(usersTable).
          SetMap(sq.Eq{
             "id":          userID,
          })
    return withTransaction(ctx, r.DBClient, func(tx *sql.Tx) error {
		if _, err := query.RunWith(tx).ExecContext(ctx); err != nil {
    	    log.Println(err)
            return err
         }
        return nil
      }

txFuncでエラーやpanicが発生したらdefer funcの中でtx.Rollbackが呼ばれ、dbの内容の修正が取り消され、正常終了した場合はtx.Commitがされます。