Function cascadeDeleteFileData

api/model/file.go:1093–1219

cascadeDeleteFileData 级联删除文件相关数据

(eid int64, fileID int64)

// cascadeDeleteFileData 级联删除文件相关数据
func cascadeDeleteFileData(eid int64, fileID int64) error {
	var sandboxOutputKey string
	var sandboxOutputFileID int64
	file, err := GetFileByID(eid, fileID)
	if err == nil && file != nil && file.OriginType == FileOriginTypeAIGenerated && file.OriginRefID > 0 {
		sandboxOutputFileID = file.OriginRefID
		if outputFile, outputErr := GetAIUploadFileByID(file.OriginRefID); outputErr == nil && outputFile != nil {
			sandboxOutputKey = outputFile.Key
		}
	}

	// 在删除数据之前，先收集需要清理的向量ID
	vectorIDs, err := getVectorIDsForFile(eid, fileID)
	if err != nil {
		log.Printf("获取文件向量ID失败，将跳过向量清理 - EID: %d, FileID: %d, Error: %v", eid, fileID, err)
		vectorIDs = []string{} // 设置为空数组，避免后续处理出错
	}

	// 开启数据库事务
	tx := DB.Begin()
	defer func() {
		if r := recover(); r != nil {
			tx.Rollback()
		}
	}()

	// 按依赖关系顺序删除：关联关系 -> 检索块 -> 文档分块 -> 其他数据 -> 文件

	var entityIDs []int64
	if err := tx.Model(&EntityChunkRelation{}).
		Distinct("entity_id").
		Where("eid = ? AND file_id = ?", eid, fileID).
		Pluck("entity_id", &entityIDs).Error; err != nil {
		tx.Rollback()
		return fmt.Errorf("查询实体关联失败: %v", err)
	}

	if err := tx.Where("eid = ? AND file_id = ?", eid, fileID).Delete(&EntityChunkRelation{}).Error; err != nil {
		tx.Rollback()
		return fmt.Errorf("删除实体关联失败: %v", err)
	}

	// 1. 删除分块关联关系
	if err := tx.Where("eid = ? AND file_id = ?", eid, fileID).Delete(&ChunkRelation{}).Error; err != nil {
		tx.Rollback()
		return fmt.Errorf("删除分块关联关系失败: %v", err)
	}

	// 2. 删除检索块
	if err := tx.Where("eid = ? AND file_id = ?", eid, fileID).Delete(&RetrievalChunk{}).Error; err != nil {
		tx.Rollback()
		return fmt.Errorf("删除检索块失败: %v", err)
	}

	// 3. 删除文档分块
	if err := tx.Where("eid = ? AND file_id = ?", eid, fileID).Delete(&DocumentChunk{}).Error; err != nil {
		tx.Rollback()
		return fmt.Errorf("删除文档分块失败: %v", err)