CleanupFileRelatedData 清理文件相关数据
(eid int64, fileID int64)
| 28 | |
| 29 | // CleanupFileRelatedData 清理文件相关数据 |
| 30 | func (s *CleanupService) CleanupFileRelatedData(eid int64, fileID int64) error { |
| 31 | fmt.Printf("开始清理文件相关数据 (FileID: %d)\n", fileID) |
| 32 | |
| 33 | vectorInfos, err := s.collectVectorIDs(eid, fileID) |
| 34 | if err != nil { |
| 35 | fmt.Printf("收集向量ID失败 (FileID: %d): %v\n", fileID, err) |
| 36 | } |
| 37 | |
| 38 | err = s.cleanupDatabaseData(eid, fileID) |
| 39 | if err != nil { |
| 40 | return fmt.Errorf("清理数据库数据失败: %v", err) |
| 41 | } |
| 42 | |
| 43 | if len(vectorInfos) > 0 { |
| 44 | if err := s.deleteVectorsByInfos(eid, vectorInfos); err != nil { |
| 45 | fmt.Printf("删除向量数据失败 (FileID: %d): %v\n", fileID, err) |
| 46 | } else { |
| 47 | fmt.Printf("成功删除 %d 个向量 (FileID: %d)\n", len(vectorInfos), fileID) |
| 48 | } |
| 49 | } |
| 50 | |
| 51 | fmt.Printf("文件相关数据清理完成 (FileID: %d)\n", fileID) |
| 52 | return nil |
| 53 | } |
| 54 | |
| 55 | // collectVectorIDs 收集文件相关的所有向量ID及其库ID |
| 56 | func (s *CleanupService) collectVectorIDs(eid int64, fileID int64) ([]vectorInfo, error) { |
no test coverage detected