Method cleanupDatabaseData

api/service/cleanup_service.go:84–142

cleanupDatabaseData 清理数据库中的相关数据

(eid int64, fileID int64)

// cleanupDatabaseData 清理数据库中的相关数据
func (s *CleanupService) cleanupDatabaseData(eid int64, fileID int64) error {
	// 开启事务
	tx := s.db.Begin()
	defer func() {
		if r := recover(); r != nil {
			tx.Rollback()
		}
	}()

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

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

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

	// 删除操作日志
	err = tx.Where("eid = ? AND file_id = ?", eid, fileID).Delete(&model.ChunkOperationLog{}).Error
	if err != nil {
		tx.Rollback()
		return fmt.Errorf("删除操作日志失败: %v", err)
	}

	// 删除文件内容
	err = tx.Where("eid = ? AND file_id = ?", eid, fileID).Delete(&model.FileBody{}).Error
	if err != nil {
		tx.Rollback()
		return fmt.Errorf("删除文件内容失败: %v", err)
	}

	err = tx.Where("file_id = ?", fileID).Delete(&model.FileBodyVersion{}).Error
	if err != nil {
		tx.Rollback()
		return fmt.Errorf("删除文件版本记录失败: %v", err)
	}

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

	// 提交事务
	return tx.Commit().Error