重构:为核心组件实现 Reset 方法优化重置机制
为所有核心组件添加 Reset() 方法: - LogWriter.Reset(): 删除并重新创建日志文件,保持 index 和 wbuf 引用不变 - RecordIndex.Reset(): 清空索引数据并重新创建索引文件 - RecordQuery.Reset(): 关闭并重新打开日志文件 - ProcessCursor.Reset(): 删除位置文件并重置游标位置 - LogTailer.Reset(): 重置内部 channel 状态 优化 TopicProcessor.Reset() 实现: - 不再销毁和重建组件对象 - 通过调用各组件的 Reset() 方法重置状态 - 保持组件间引用关系稳定 - 减少代码行数约 20 行 - 避免空指针风险和内存分配开销 代码改进: - LogWriter 添加 path 字段用于重置 - 移除 topic_processor.go 中未使用的 os import - 职责分离更清晰,每个组件管理自己的重置逻辑 测试结果: - TestTopicReset: PASS - TestTopicResetWithPendingRecords: PASS - 所有 TopicProcessor 相关测试通过 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
38
index.go
38
index.go
@@ -249,6 +249,44 @@ func (ri *RecordIndex) Close() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// Reset 重置索引,清空所有数据并重新创建索引文件
|
||||
func (ri *RecordIndex) Reset() error {
|
||||
ri.mu.Lock()
|
||||
defer ri.mu.Unlock()
|
||||
|
||||
// 关闭索引文件
|
||||
if ri.indexFile != nil {
|
||||
if err := ri.indexFile.Close(); err != nil {
|
||||
return err
|
||||
}
|
||||
ri.indexFile = nil
|
||||
}
|
||||
|
||||
// 删除索引文件
|
||||
if err := os.Remove(ri.indexPath); err != nil && !os.IsNotExist(err) {
|
||||
return err
|
||||
}
|
||||
|
||||
// 清空内存中的索引数据
|
||||
ri.offsets = make([]int64, 0, 1024)
|
||||
ri.dirtyCount = 0
|
||||
ri.lastSync = time.Now()
|
||||
|
||||
// 保存空索引(创建文件并写入头部)
|
||||
if err := ri.save(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// 重新打开索引文件用于追加
|
||||
f, err := os.OpenFile(ri.indexPath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
|
||||
if err != nil {
|
||||
return fmt.Errorf("reopen index file: %w", err)
|
||||
}
|
||||
ri.indexFile = f
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// Sync 同步索引文件到磁盘(立即同步,不考虑批量策略)
|
||||
func (ri *RecordIndex) Sync() error {
|
||||
return ri.Flush()
|
||||
|
||||
Reference in New Issue
Block a user