第五节 Rollback()实现

Rollback()中，主要对不同事务进行不同操作：

1. 如果当前事务是只读事务，则只需要从db中的txs中找到当前事务，然后移除掉即可。
2. 如果当前事务是读写事务，则需要将空闲列表中和该事务关联的页释放掉，同时重新从freelist中加载空闲页。

1. // Rollback closes the transaction and ignores all previous updates. Read-only
2. // transactions must be rolled back and not committed.
3. func (tx *Tx) Rollback() error {
4. _assert(!tx.managed, "managed tx rollback not allowed")
5. if tx.db == nil {
6. return ErrTxClosed
7. }
8. tx.rollback()
9. return nil
10. }

12. func (tx *Tx) rollback() {
13. if tx.db == nil {
14. return
15. }
16. if tx.writable {
17. // 移除该事务关联的pages
18. tx.db.freelist.rollback(tx.meta.txid)
19. // 重新从freelist页中读取构建空闲列表
20. tx.db.freelist.reload(tx.db.page(tx.db.meta().freelist))
21. }
22. tx.close()
23. }

25. func (tx *Tx) close() {
26. if tx.db == nil {
27. return
28. }
29. if tx.writable {
30. // Grab freelist stats.
31. var freelistFreeN = tx.db.freelist.free_count()
32. var freelistPendingN = tx.db.freelist.pending_count()
33. var freelistAlloc = tx.db.freelist.size()

35. // Remove transaction ref & writer lock.
36. tx.db.rwtx = nil
37. tx.db.rwlock.Unlock()

39. // Merge statistics.
40. tx.db.statlock.Lock()
41. tx.db.stats.FreePageN = freelistFreeN
42. tx.db.stats.PendingPageN = freelistPendingN
43. tx.db.stats.FreeAlloc = (freelistFreeN + freelistPendingN) * tx.db.pageSize
44. tx.db.stats.FreelistInuse = freelistAlloc
45. tx.db.stats.TxStats.add(&tx.stats)
46. tx.db.statlock.Unlock()
47. } else {
48. // 只读事务
49. tx.db.removeTx(tx)
50. }

52. // Clear all references.
53. tx.db = nil
54. tx.meta = nil
55. tx.root = Bucket{tx: tx}
56. tx.pages = nil
57. }

59. // removeTx removes a transaction from the database.
60. func (db *DB) removeTx(tx *Tx) {
61. // Release the read lock on the mmap.
62. db.mmaplock.RUnlock()

64. // Use the meta lock to restrict access to the DB object.
65. db.metalock.Lock()

67. // Remove the transaction.
68. for i, t := range db.txs {
69. if t == tx {
70. last := len(db.txs) - 1
71. db.txs[i] = db.txs[last]
72. db.txs[last] = nil
73. db.txs = db.txs[:last]
74. break
75. }
76. }
77. n := len(db.txs)

79. // Unlock the meta pages.
80. db.metalock.Unlock()

82. // Merge statistics.
83. db.statlock.Lock()
84. db.stats.OpenTxN = n
85. db.stats.TxStats.add(&tx.stats)
86. db.statlock.Unlock()
87. }