Method filterDeletedFiles

api/service/rag/search.go:4337–4403

filterDeletedFiles 过滤已删除的文件和已删除的知识库

(results []SearchResultItem)

// filterDeletedFiles 过滤已删除的文件和已删除的知识库
func (s *SearchService) filterDeletedFiles(results []SearchResultItem) []SearchResultItem {
	if len(results) == 0 {
		return results
	}

	// 收集所有需要检查的文件ID和知识库ID
	fileIDs := make([]int64, 0, len(results))
	libraryIDs := make([]int64, 0, len(results))

	for _, result := range results {
		if result.FileID != 0 {
			fileIDs = append(fileIDs, result.FileID)
		}
		if result.LibraryID != 0 {
			libraryIDs = append(libraryIDs, result.LibraryID)
		}
	}

	// 创建已删除资源ID的集合
	deletedFileIDs := make(map[int64]bool)
	deletedLibraryIDs := make(map[int64]bool)

	// 批量查询所有被删除的文件
	if len(fileIDs) > 0 {
		var deletedFiles []model.File
		err := s.db.Where("id IN ? AND is_deleted = ?", fileIDs, true).Find(&deletedFiles).Error
		if err != nil {
			logger.SysLogf("批量查询已删除文件时出错: %v", err)
			// 出错时继续处理知识库
		} else {
			for _, file := range deletedFiles {
				deletedFileIDs[file.ID] = true
				logger.SysLogf("标记已删除的文件: file_id=%d", file.ID)
			}
		}
	}

	// 批量查询所有被删除的知识库
	// if len(libraryIDs) > 0 {
	// 	var deletedLibraries []model.Library
	// 	err := s.db.Where("id IN ?", libraryIDs).Find(&deletedLibraries).Error
	// 	if err != nil {
	// 		logger.SysLogf("批量查询已删除知识库时出错: %v", err)
	// 		// 出错时继续处理
	// 	} else {
	// 		for _, library := range deletedLibraries {
	// 			deletedLibraryIDs[library.ID] = true
	// 			logger.SysLogf("标记已删除的知识库: library_id=%d", library.ID)
	// 		}
	// 	}
	// }

	// 过滤结果
	var filteredResults []SearchResultItem
	for _, result := range results {
		// 如果文件或知识库被删除，则过滤掉该结果
		if (result.FileID != 0 && deletedFileIDs[result.FileID]) ||
			(result.LibraryID != 0 && deletedLibraryIDs[result.LibraryID]) {