buffer_pool.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561
  1. package pool
  2. import (
  3. "sync"
  4. "sync/atomic"
  5. )
  6. // pageEntry 缓冲池中的页条目
  7. type pageEntry struct {
  8. data []byte // 页数据副本
  9. dirty bool // 是否被修改(需要写回磁盘)
  10. pageLen int // 页字节长度(用于计算文件偏移)
  11. ver uint64 // 内容版本号: 每次内容被写入都换一个 (见 BufferPool.stampLocked)
  12. lastTouch int64 // 最近一次访问序号 (用 atomic 读写, 替代 LRU 链表节点)
  13. }
  14. // DirtyPage 脏页快照
  15. //
  16. // ver 是快照当时该页的版本号, 只在 pool 内部使用。提交成功后 MarkClean 靠它判断
  17. // 「这页在提交窗口里有没有被再改过」: 改过就必须继续留脏, 不能被顺手标成干净页。
  18. type DirtyPage struct {
  19. PageID uint32
  20. Data []byte
  21. PageLen int
  22. ver uint64
  23. }
  24. // BufferPool 页面缓冲池
  25. // 拦截 ReadPage/WritePage,将页缓存在内存中,减少磁盘 I/O
  26. // 写回策略:
  27. // - Commit() → 通过 CollectDirty() 收集脏页,走 WAL 流程写回
  28. // - 脏页超过阈值时自动触发 onThreshold 回调 (由 DB 层执行 Commit)
  29. //
  30. // 缓存策略: 按时间戳 LRU —— 每次访问用 atomic 递增 touchCounter 并写入页的
  31. // lastTouch 字段, 淘汰时按采样找 lastTouch 最小的干净页 (最久未用)。
  32. // 用 atomic 时间戳替代双向链表, 让 ReadPage cache-hit 路径完全不需要写锁。
  33. // 脏页绝不淘汰 (宁可临时超容量, 由脏页阈值触发的 Commit 回落)。
  34. type BufferPool struct {
  35. pages map[uint32]*pageEntry // pageID → 页条目
  36. mu sync.RWMutex
  37. capacity int // 最大缓存页数
  38. dirtyThreshold int // 脏页阈值 (0=不限制)
  39. onThreshold func() // 脏页超阈值回调
  40. dirtyCount int // 当前脏页数 (实时追踪)
  41. verSeq uint64 // 版本号发放器 (见 stampLocked)
  42. diskRead func(pageID uint32, buf []byte) error // 底层磁盘读
  43. diskWrite func(pageID uint32, buf []byte) error // 底层磁盘写
  44. touchCounter int64 // LRU 时间戳发放器 (atomic 递增, 全局单调递增)
  45. // 锁表机制
  46. writeLocked bool // 写锁标志
  47. writeChan chan struct{} // 用于通知解锁的 channel
  48. // 事务支持
  49. onPageWrite func(pageID uint32) // 页写入时的回调(用于事务缓存原始数据)
  50. }
  51. // evictSample 淘汰时一次最多考察的候选页数 (采样淘汰, 对照 Redis
  52. // maxmemory-samples=5 —— 我们给到 32, 大池近似 LRU, 小池精确 LRU)。
  53. const evictSample = 32
  54. // NewBufferPool 创建页面缓冲池
  55. // - capacity: 最大缓存页数
  56. // - diskRead/diskWrite: 底层磁盘读写函数
  57. func NewBufferPool(capacity int,
  58. diskRead func(uint32, []byte) error,
  59. diskWrite func(uint32, []byte) error,
  60. ) *BufferPool {
  61. bp := &BufferPool{
  62. pages: make(map[uint32]*pageEntry, capacity),
  63. capacity: capacity,
  64. diskRead: diskRead,
  65. diskWrite: diskWrite,
  66. }
  67. return bp
  68. }
  69. // SetDirtyThreshold 设置脏页阈值和回调
  70. // 当脏页数超过 threshold 时调用 onThreshold (由 DB 层触发 Commit)
  71. func (bp *BufferPool) SetDirtyThreshold(threshold int, onThreshold func()) {
  72. bp.mu.Lock()
  73. defer bp.mu.Unlock()
  74. bp.dirtyThreshold = threshold
  75. bp.onThreshold = onThreshold
  76. }
  77. // LockWrites 锁定写入(Commit/Flush 期间调用)
  78. func (bp *BufferPool) LockWrites() {
  79. bp.mu.Lock()
  80. bp.writeLocked = true
  81. bp.writeChan = make(chan struct{}) // 创建新的 channel
  82. bp.mu.Unlock()
  83. }
  84. // UnlockWrites 解锁写入(Commit/Flush 完成后调用)
  85. func (bp *BufferPool) UnlockWrites() {
  86. bp.mu.Lock()
  87. bp.writeLocked = false
  88. if bp.writeChan != nil {
  89. close(bp.writeChan) // 通知所有等待的 goroutine
  90. bp.writeChan = nil
  91. }
  92. bp.mu.Unlock()
  93. }
  94. // waitForWriteLock 等待写锁释放(在 WritePage 中调用)。
  95. //
  96. // 醒来必须【回头复查】 writeLocked: 锁是成对替换的 (LockWrites 每次新建
  97. // channel), 若本 waiter 持有旧 channel 被放行时, 恰逢下一笔提交已重新加锁,
  98. // 不复查就会穿越进别人的提交窗口写页 —— 快照一致性被破坏。
  99. func (bp *BufferPool) waitForWriteLock() {
  100. for {
  101. bp.mu.Lock()
  102. if !bp.writeLocked {
  103. bp.mu.Unlock()
  104. return // 未锁定,直接返回
  105. }
  106. // 保存 channel 引用
  107. ch := bp.writeChan
  108. bp.mu.Unlock()
  109. // 阻塞等待解锁(channel 关闭时会返回)
  110. <-ch
  111. // 循环回头复查: 可能已被下一笔 LockWrites 重新锁住
  112. }
  113. }
  114. // touch 给页打一个"刚被访问过"的时间戳。
  115. // 用 atomic 递增 touchCounter, 全局单调递增, 替代双向链表移动。
  116. // 调用方【不需要】持有 bp.mu —— atomic Store 本身是并发安全的,
  117. // 这正是让 ReadPage cache-hit 路径能用 RLock 的关键。
  118. func (bp *BufferPool) touch(e *pageEntry) {
  119. ts := atomic.AddInt64(&bp.touchCounter, 1)
  120. atomic.StoreInt64(&e.lastTouch, ts)
  121. }
  122. // touchLocked 兼容旧调用点 (在持锁上下文中也可用), 本质就是 touch。
  123. // 现在不再需要锁保护 (atomic 操作), 保留此函数名是为了 grep 容易定位。
  124. func (bp *BufferPool) touchLocked(e *pageEntry) {
  125. bp.touch(e)
  126. }
  127. // removeLocked 从缓存中摘除一页。调用方必须持有 bp.mu (Lock).
  128. // 原来同时需要从 LRU 链表移除, 现在时间戳方案下只需要 delete map entry。
  129. func (bp *BufferPool) removeLocked(pageID uint32, e *pageEntry) {
  130. delete(bp.pages, pageID)
  131. }
  132. // CollectDirty 取出所有脏页快照 (由 DB.Commit() 调用,配合 WAL 流程统一写回)。
  133. //
  134. // 快照之后【不把页移出缓存】。移出会造出一个「既不在缓存、也还没落盘」的窗口: 从
  135. // 收集完到 wal.Commit() 逐页回写完成之间要经过两次 fsync, 这期间任何缓存未命中的读
  136. // 都只能回源磁盘, 于是拿到的是旧内容 —— 或者, 对一页本次会话刚分配、盘上从来没写过
  137. // 的页, 拿到全零 (文件还没长到那个偏移时直接读出错)。btree 把全零页当成
  138. // EntryLen==0 的节点, 随后在 btPID 里切片越界 panic; 读出错则表现为
  139. // "insert: load failed"。页留在缓存里, 并发读就一定命中最新内容, 不存在两头都不在的时刻。
  140. //
  141. // 也不再清空 dirtyCount: 快照里的页此刻【尚未落盘】, 计数必须继续按"未持久化"算。
  142. // 页的脏标记由提交方在 wal.Commit() 成功之后用 MarkClean 收掉 —— 只有快照后没被再改过
  143. // (ver 未变) 的页才会转成干净页, 改过的继续留脏, 等下一次提交。
  144. // 干净页照旧保留在池中继续命中 (旧实现无论脏净一律 delete, 等于每次提交都清空缓存,
  145. // 提交后的读取全部回落到磁盘); 容量仍由 evictCleanForRoomLocked 约束。
  146. func (bp *BufferPool) CollectDirty() []DirtyPage {
  147. // 锁定写入,防止新的写操作
  148. bp.LockWrites()
  149. defer bp.UnlockWrites()
  150. bp.mu.Lock()
  151. var dirty []DirtyPage
  152. for pid, entry := range bp.pages {
  153. if !entry.dirty {
  154. continue // 干净页: 与磁盘一致, 留在池中继续服务后续读
  155. }
  156. data := make([]byte, len(entry.data))
  157. copy(data, entry.data)
  158. dirty = append(dirty, DirtyPage{PageID: pid, Data: data, PageLen: entry.pageLen, ver: entry.ver})
  159. }
  160. // 这里没有释放任何页, 所以不唤醒等待容量的 goroutine (快照只是复制)。
  161. bp.mu.Unlock()
  162. return dirty
  163. }
  164. // MarkClean 把「确实已经落盘」的快照页转成干净页 (提交成功后由提交方调用)。
  165. //
  166. // 判据是版本号而不是"提交成功了": 提交窗口里写入线程可能已经把这一页又改了一遍
  167. // (缓存里是更新的内容, 而本次 WAL 记的是快照那一份)。这种页必须继续留脏 —— 否则它
  168. // 的新内容既不会被写盘, 脏标记还向引擎报告"已经持久化了", 该页一旦被淘汰就永久丢失。
  169. //
  170. // 快照里的页若在窗口里被别的路径丢弃 (回收 / 直写盘), 这里跳过即可: 那些路径自己
  171. // 负责缓存一致性。
  172. func (bp *BufferPool) MarkClean(pages []DirtyPage) {
  173. if len(pages) == 0 {
  174. return
  175. }
  176. bp.mu.Lock()
  177. defer bp.mu.Unlock()
  178. for _, p := range pages {
  179. entry, ok := bp.pages[p.PageID]
  180. if !ok || !entry.dirty || entry.ver != p.ver {
  181. continue
  182. }
  183. entry.dirty = false
  184. bp.dirtyCount--
  185. }
  186. if bp.dirtyCount < 0 {
  187. bp.dirtyCount = 0
  188. }
  189. }
  190. // stampLocked 给页条目盖一个新的版本号。调用方必须持有 bp.mu。
  191. //
  192. // 版本号只用于回答一个问题: "从我拍快照到现在, 这一页被改过吗"。
  193. func (bp *BufferPool) stampLocked(e *pageEntry) {
  194. bp.verSeq++
  195. e.ver = bp.verSeq
  196. }
  197. // RestoreDirty 提交失败时兜底: 保证这批页在缓存里仍然是脏的。
  198. //
  199. // CollectDirty 现在不再把页移出缓存, 所以正常路径上这里无事可做 —— 留着是因为错误
  200. // 处理期间别的动作 (页回收 / 直写盘) 也可能碰过这些页。只补【缺失】的条目: 页已经在
  201. // 缓存里就原样不动, 免得用一份旧快照盖掉提交窗口里写入线程刚改出来的新内容。
  202. //
  203. // 这里刻意不触发阈值回调, 避免在错误处理路径上递归触发 Commit。
  204. func (bp *BufferPool) RestoreDirty(pages []DirtyPage) {
  205. if len(pages) == 0 {
  206. return
  207. }
  208. bp.mu.Lock()
  209. defer bp.mu.Unlock()
  210. for _, p := range pages {
  211. if _, ok := bp.pages[p.PageID]; ok {
  212. continue // 已在缓存 (很可能还带着更新的内容), 不要覆盖
  213. }
  214. e := &pageEntry{data: p.Data, dirty: true, pageLen: p.PageLen}
  215. bp.stampLocked(e)
  216. bp.pages[p.PageID] = e
  217. bp.touchLocked(e)
  218. bp.dirtyCount++
  219. }
  220. }
  221. // DiscardPages 丢弃指定页的缓存 (不落盘), 并释放其占用容量。
  222. //
  223. // 事务 Rollback 使用: 事务期间 autoCommit 被 currentTxn 拦住, 任何脏页都不会刷到磁盘,
  224. // 所以「撤销」= 直接把本事务碰过的页从缓存里丢掉, 后续读取自然回到磁盘旧值。
  225. // 只丢弃传入的页 —— 事务开始前遗留的脏页不属于本事务, 必须原样保留。
  226. func (bp *BufferPool) DiscardPages(pageIDs map[uint32]struct{}) {
  227. if len(pageIDs) == 0 {
  228. return
  229. }
  230. bp.mu.Lock()
  231. defer bp.mu.Unlock()
  232. removedDirty := 0
  233. for pid := range pageIDs {
  234. entry, ok := bp.pages[pid]
  235. if !ok {
  236. continue
  237. }
  238. if entry.dirty {
  239. removedDirty++
  240. }
  241. bp.removeLocked(pid, entry)
  242. }
  243. bp.dirtyCount -= removedDirty
  244. if bp.dirtyCount < 0 {
  245. bp.dirtyCount = 0
  246. }
  247. }
  248. // DiscardPage 丢弃单页缓存 (不落盘), 并释放其占用容量。
  249. //
  250. // 用于「绕过缓冲池直写磁盘」(WritePageDirect) 之后, 让缓存中该页的旧副本立即失效,
  251. // 避免随后 ReadPage 命中与磁盘不一致的陈旧内容。
  252. func (bp *BufferPool) DiscardPage(pageID uint32) {
  253. bp.mu.Lock()
  254. entry, ok := bp.pages[pageID]
  255. if !ok {
  256. bp.mu.Unlock()
  257. return
  258. }
  259. if entry.dirty {
  260. bp.dirtyCount--
  261. if bp.dirtyCount < 0 {
  262. bp.dirtyCount = 0
  263. }
  264. }
  265. bp.removeLocked(pageID, entry)
  266. bp.mu.Unlock()
  267. }
  268. // Flush 将所有脏页写入磁盘 (Close / 维护操作时使用), 并把落盘成功的页转成干净页。
  269. //
  270. // 收尾必须走 MarkClean: CollectDirty 不再移出缓存, 少了这一步, 刚写完的页会一直带着
  271. // 脏标记, 下一次 Commit 会把同一批内容再写一遍 (并且写进 WAL)。
  272. // 写盘失败的页【保持 dirty】并计入返回值: 标干净等于宣布"已持久化", 该页一旦被淘汰
  273. // 就永久丢失 (静默丢数据)。返回第一个写盘错误, 全部成功返回 nil —— 调用方 (如
  274. // db.Close) 据此判断收尾是否可信; ver 比对仍会放过提交窗口里被改过的页, 它们继续
  275. // 留脏, 交给下一次 Flush / Commit。
  276. func (bp *BufferPool) Flush() error {
  277. dirty := bp.CollectDirty()
  278. var firstErr error
  279. var written []DirtyPage
  280. for _, item := range dirty {
  281. if err := bp.diskWrite(item.PageID, item.Data); err != nil {
  282. if firstErr == nil {
  283. firstErr = err
  284. }
  285. continue // 写失败的页保持 dirty, 等下一次 Flush / Commit 重试
  286. }
  287. written = append(written, item)
  288. }
  289. // 只把确实落盘的页标干净
  290. bp.MarkClean(written)
  291. return firstErr
  292. }
  293. // Close 关闭缓冲池 (刷盘后停止); 刷盘错误被丢弃 —— 需要错误判定请直接调 Flush
  294. func (bp *BufferPool) Close() {
  295. _ = bp.Flush()
  296. }
  297. // UpdatePage 更新缓存页 (WAL 提交后调用, 保持缓存与磁盘一致)
  298. func (bp *BufferPool) UpdatePage(pageID uint32, data []byte) {
  299. bp.mu.Lock()
  300. defer bp.mu.Unlock()
  301. cached := make([]byte, len(data))
  302. copy(cached, data)
  303. if old, ok := bp.pages[pageID]; ok {
  304. old.data = cached
  305. old.dirty = false
  306. old.pageLen = len(data)
  307. bp.stampLocked(old) // 内容换过了, 版本号必须往前走, 否则旧快照会误认为这页没变
  308. bp.touchLocked(old)
  309. } else {
  310. e := &pageEntry{data: cached, dirty: false, pageLen: len(data)}
  311. bp.stampLocked(e)
  312. bp.pages[pageID] = e
  313. bp.touchLocked(e)
  314. }
  315. // 回填的页是干净页, 不受 dirtyCount 约束; 不加这一步, 大量回收页会把缓存
  316. // 无限撑大 (回收 N 页 → 缓存增加 N 页), 并让 WritePage 的容量判断形同虚设。
  317. bp.evictCleanForRoomLocked()
  318. }
  319. // evictCleanForRoomLocked 容量已满时按时间戳 LRU 淘汰干净页, 直到降到容量以下。
  320. //
  321. // 采样淘汰 (对照 Redis 的 maxmemory-samples): Go map 的遍历顺序本身随机,
  322. // 从头取 evictSample 个候选、在其中选 lastTouch 最小的干净页 (最久未用) 淘汰。
  323. // 池子大于样本量时是 O(K) 近似 LRU; 小于等于样本量时覆盖全池, 就是精确 LRU。
  324. //
  325. // 为什么不做全表精确最小: 本函数在 WritePage 的持锁路径上, 池满后每次写入都触发,
  326. // 且一轮可能连续淘汰几十上百页 —— 全表扫是 O(n)/页, 采样把整轮压到 O(K×淘汰数)。
  327. // 4000 页池从每次 ~200-400μs 降到 ~3μs。读路径的 RLock + atomic touch 不受影响,
  328. // "一写多读"的并发收益原样保留。
  329. //
  330. // 脏页跳过 (计入样本数, 同 Redis) —— 淘汰脏页会静默丢掉尚未落盘的修改。
  331. // 若抽到的候选全是脏页, 本次不做任何事 —— 允许缓存临时超出容量, 由脏页阈值触发的
  332. // Commit 尽快回落; 下一次写入会重新随机抽样。
  333. //
  334. // 调用方必须持有 bp.mu (Lock).
  335. func (bp *BufferPool) evictCleanForRoomLocked() {
  336. for bp.capacity > 0 && len(bp.pages) >= bp.capacity {
  337. // 采样找 lastTouch 最小的干净页: Go map 遍历顺序随机, 取前
  338. // evictSample 个候选即随机样本, 池子 ≤ 样本量时退化为精确 LRU。
  339. var victimPID uint32
  340. var victim *pageEntry
  341. var minTouch int64 = -1
  342. n := 0
  343. for pid, e := range bp.pages {
  344. if n >= evictSample {
  345. break
  346. }
  347. n++
  348. if e.dirty {
  349. continue
  350. }
  351. if ts := atomic.LoadInt64(&e.lastTouch); victim == nil || ts < minTouch {
  352. minTouch, victimPID, victim = ts, pid, e
  353. }
  354. }
  355. if victim == nil {
  356. return // 样本内全是脏页, 允许临时超容量
  357. }
  358. bp.removeLocked(victimPID, victim)
  359. }
  360. }
  361. // ClearAll 清除所有脏标记, 保留页在缓存中。
  362. //
  363. // ⚠️ 这是"无差别"操作, 只适用于确定没有并发写入的维护场景 (关闭 / 建表删表等)。
  364. // 提交路径上【不能】用它: 提交窗口里被写入线程重新改脏的页并不在本笔 WAL 里, 一并
  365. // 标干净就等于宣布"已持久化", 那份新内容以后既不会被写盘, 也会被淘汰逻辑当作可以
  366. // 直接丢弃的干净页。提交成功要收脏标记请用 MarkClean (按版本号逐页判定)。
  367. func (bp *BufferPool) ClearAll() {
  368. bp.mu.Lock()
  369. defer bp.mu.Unlock()
  370. for _, entry := range bp.pages {
  371. entry.dirty = false
  372. }
  373. bp.dirtyCount = 0
  374. }
  375. // ReadPage 读页 — 先查缓存, 未命中则读磁盘并作为【干净页载入缓存】。
  376. //
  377. // ⚡ 并发读核心: cache-hit 路径完全用 RLock + atomic touch!
  378. // 多个 goroutine 同时 GetPage(不同页) 不再互相阻塞 —— 这是让 TsDB2
  379. // 真正支持"一写多读"的关键一步。
  380. //
  381. // 读缓存让重复扫描/点查不必每次都走系统调用; 载入的页与磁盘一致, 可被 LRU 淘汰。
  382. // 读盘期间另一线程可能已把该页写进缓存 (写缓存优先): 重新检查后以池内内容为准,
  383. // 绝不拿这份可能更旧的读盘结果覆盖脏页。直写盘路径 (WritePageDirect) 自带缓存失效,
  384. // 读缓存不会滞留陈旧内容。
  385. func (bp *BufferPool) ReadPage(pageID uint32, buf []byte) error {
  386. // cache-hit 路径: 用读锁查 map, 复制数据, atomic 打时间戳 —— 全程不需要写锁
  387. bp.mu.RLock()
  388. if entry, ok := bp.pages[pageID]; ok {
  389. copy(buf, entry.data)
  390. bp.touch(entry) // atomic Store, 不需要写锁
  391. bp.mu.RUnlock()
  392. return nil
  393. }
  394. bp.mu.RUnlock()
  395. // 缓存未命中 → 读磁盘 (锁外!)
  396. if err := bp.diskRead(pageID, buf); err != nil {
  397. return err
  398. }
  399. // 作为干净页入池。入池前 double-check: 读盘这段时间里别的线程可能已经写入了这一页
  400. // (缓存里是更新/未落盘的内容), 此时以池内为准, 丢弃读盘结果。
  401. bp.mu.Lock()
  402. defer bp.mu.Unlock()
  403. if entry, ok := bp.pages[pageID]; ok {
  404. copy(buf, entry.data)
  405. bp.touchLocked(entry)
  406. return nil
  407. }
  408. bp.evictCleanForRoomLocked()
  409. entry := &pageEntry{data: make([]byte, len(buf)), dirty: false, pageLen: len(buf)}
  410. copy(entry.data, buf)
  411. bp.pages[pageID] = entry
  412. bp.touchLocked(entry)
  413. return nil
  414. }
  415. // WritePage 写页 — 写入缓存并标记脏,不立即落盘
  416. func (bp *BufferPool) WritePage(pageID uint32, buf []byte) error {
  417. // 1. 等待写锁释放(Commit/Flush 期间阻塞)
  418. bp.waitForWriteLock()
  419. // 事务支持:通知 Transaction 缓存原始数据
  420. if bp.onPageWrite != nil {
  421. bp.onPageWrite(pageID)
  422. }
  423. var triggerThreshold bool
  424. bp.mu.Lock()
  425. // 2. 容量控制: 满了先按 LRU 淘汰干净页, 不阻塞等待。
  426. // 旧实现阻塞在 capacityWait 上等 Commit/Discard 释放空间, 但这两者都不可能
  427. // 在"当前这个正在阻塞的 goroutine"内被调用 —— 单写入者模型下必然死锁
  428. // (fatal error: all goroutines are asleep)。批量删除回收旧页时, realFreePage
  429. // 会把每个回收页经 SyncCache→UpdatePage 以干净页回填缓存(不计 dirtyCount),
  430. // 缓存被干净页迅速撑满, 紧接其后的写入即触发该死锁。
  431. bp.evictCleanForRoomLocked()
  432. existing, exists := bp.pages[pageID]
  433. switch {
  434. case exists && existing.dirty:
  435. // 已是脏页, 直接覆盖, 不增加脏页计数。
  436. //
  437. // 【原地覆写】旧实现在这里每次都新分配一块 len(buf) 的副本再整个换掉
  438. // existing.data —— 对逐行写入路径 (B+ 树叶 + 数据页 + 表信息页, 每行约 6 次
  439. // 16KB 写) 意味着每行 ~96KB 的纯垃圾, 5 万行灌数实测 ~4.7GB 分配, GC 压力
  440. // 吃掉约 1/3 的插入吞吐。这里改为: 页长一致时直接 copy 进旧缓冲 (旧缓冲没有
  441. // 任何外部别名 —— CollectDirty 的快照是独立副本, UpdatePage/RestoreDirty 各自
  442. // 持有自己的数据), 只留一次 16KB memcpy, 分配量从"每次写"降到"每张页"。
  443. if len(existing.data) == len(buf) {
  444. copy(existing.data, buf)
  445. } else {
  446. cached := make([]byte, len(buf))
  447. copy(cached, buf)
  448. existing.data = cached
  449. }
  450. bp.stampLocked(existing) // 内容变了: 让提交窗口里的旧快照认不出这一页
  451. bp.touchLocked(existing)
  452. case exists:
  453. // 干净页 (读缓存/回收回填的页) 被改写: 复用条目, 标脏
  454. if len(existing.data) == len(buf) {
  455. copy(existing.data, buf)
  456. } else {
  457. cached := make([]byte, len(buf))
  458. copy(cached, buf)
  459. existing.data = cached
  460. }
  461. existing.dirty = true
  462. existing.pageLen = len(buf)
  463. bp.stampLocked(existing)
  464. bp.dirtyCount++
  465. bp.touchLocked(existing)
  466. default:
  467. // 新页, 增加脏页计数
  468. cached := make([]byte, len(buf))
  469. copy(cached, buf)
  470. e := &pageEntry{data: cached, dirty: true, pageLen: len(buf)}
  471. bp.stampLocked(e)
  472. bp.pages[pageID] = e
  473. bp.touchLocked(e)
  474. bp.dirtyCount++
  475. }
  476. // 3. 检查是否超过阈值 —— 只按「脏页数」判定。
  477. //
  478. // 不能再用 len(bp.pages): CollectDirty 现在会保留干净页, 读缓存也会让干净页
  479. // 持续驻留, 总页数会持续偏高。若仍以总页数触发, 阈值会被这些干净页反复顶爆,
  480. // 提交频率凭空放大数倍。脏页数才是真正"尚未落盘"的数据量, 也正是崩溃时可能
  481. // 丢失的上限; 干净页的容量由 evictCleanForRoomLocked 按 capacity 约束。
  482. if bp.dirtyThreshold > 0 && bp.onThreshold != nil && bp.dirtyCount >= bp.dirtyThreshold {
  483. triggerThreshold = true
  484. }
  485. bp.mu.Unlock()
  486. // 4. 在锁外触发回调, 避免死锁
  487. if triggerThreshold {
  488. bp.onThreshold()
  489. }
  490. return nil
  491. }
  492. // Len 当前缓存页数 (调试用)
  493. func (bp *BufferPool) Len() int {
  494. bp.mu.Lock()
  495. defer bp.mu.Unlock()
  496. return len(bp.pages)
  497. }
  498. // DirtyCount 脏页数 (调试用)
  499. func (bp *BufferPool) DirtyCount() int {
  500. bp.mu.Lock()
  501. defer bp.mu.Unlock()
  502. return bp.dirtyCount
  503. }
  504. // SetCapacity 设置 BufferPool 容量(页数)
  505. func (bp *BufferPool) SetCapacity(capacity int) {
  506. bp.mu.Lock()
  507. bp.capacity = capacity
  508. bp.mu.Unlock()
  509. }
  510. // SetOnPageWriteCallback 设置页写入回调(用于事务缓存原始数据)
  511. func (bp *BufferPool) SetOnPageWriteCallback(callback func(uint32)) {
  512. bp.mu.Lock()
  513. bp.onPageWrite = callback
  514. bp.mu.Unlock()
  515. }