トランザクション制御

[kyokomi]

txをdaoで持ち回す感じで、txがnillじゃなければ使うようにする感じ

type XxxxDao struct {
    // ...
    tx *sql.Tx
}

func (d *XxxxDao) SetTx(tx *sql.Tx) {
    d.tx = tx
}

if d.tx != nil {
    d.tx.Query()
} else {
    d.db.Query()
}

[kyokomi]

実装した。サンプルはこちら

goma/test/mysql/main.go

Lines 191 to 270 in 8cac5bd

 func txTest(g Goma) { 
  
 	id := int64(1234567890) 
  
 	tx, err := g.Begin() 
 	if err != nil { 
 		log.Fatalln(err) 
 	} 
  
 	// string 
 	dao := g.GomaStringTypes 
 	dao.SetTx(tx) 
  
 	e := entity.GomaStringTypesEntity{ 
 		ID:                id, 
 		TextColumns:       "あいうえおかきくけこ", 
 		TinytextColumns:   "abc", 
 		MediumtextColumns: "abcdefg", 
 		LongtextColumns:   "鉄1234567890abcdefghijkelmnopqrstuvwxyz1234567890abcdefghijkelmnopqrstuvwxyz柱", 
 		CharColumns:       "a", 
 		VarcharColumns:    "1234567890abcdefghijkelmnopqrstuvwxyz", 
 	} 
  
 	_, err = dao.Insert(e) 
 	if err != nil { 
 		log.Fatalln(err) 
 	} 
  
 	// Rollback（insertを無効にする） 
 	if err := tx.Rollback(); err != nil { 
 		log.Fatalln(err) 
 	} 
  
 	// Rollback後に使うならResetもしくは新しいtxを設定する 
 	dao.ResetTx() 
  
 	// 再度トランザクションをはる 
 	tx, err = g.Begin() 
 	if err != nil { 
 		log.Fatalln(err) 
 	} 
 	dao.SetTx(tx) 
  
 	// Rollbackでnilのはず 
 	if e, err := dao.SelectByID(id); err != nil { 
 		log.Fatalln(err) 
 	} else { 
 		fmt.Printf("Rollbackでnilのはず => %s: %+v\n", dao.TableName, e) 
 	} 
  
 	// Insertする 
 	_, err = dao.Insert(e) 
 	if err != nil { 
 		log.Fatalln(err) 
 	} 
  
 	// Commit 
 	if err := tx.Commit(); err != nil { 
 		log.Fatalln(err) 
 	} 
 	// Commit後後に使うならResetもしくは新しいtxを設定する 
 	dao.ResetTx() 
  
 	// Commitしたのでnilじゃない 
 	if e, err := dao.SelectByID(id); err != nil { 
 		log.Fatalln(err) 
 	} else { 
 		fmt.Printf("Commitしたのでnilじゃない => %s: %+v\n", dao.TableName, e) 
 	} 
  
 	_, err = dao.Delete(id) 
 	if err != nil { 
 		log.Fatalln(err) 
 	} 
  
 	if e, err := dao.SelectByID(id); err != nil { 
 		log.Fatalln(err) 
 	} else { 
 		fmt.Printf("普通にDeleteしてnilのはず => %s: %+v\n", dao.TableName, e) 
 	} 

[kyokomi]

db.Beginでtx取得して

dao.SetTx(tx)でトランザクションを設定する。以降このDaoはトランザクション制御内になる。

tx.Rollback()もしくはtx.Commit()で通常通り反映できる。

もし、RollbackやCommit後にもDaoを使いまわしたい時はdao.ResetTx()を呼び出す。