Function BatchLinkEntities

api/controller/entity.go:63–244

BatchLinkEntities godoc @Summary 批量关联实体 @Description 批量为分块/文件/知识库创建并关联实体。如果实体不存在则创建，如果关联已存在则跳过。 @Tags 实体管理 @Accept json @Produce json @Security BearerAuth @Param request body BatchLinkEntityRequest true "批量关联信息" @Success 200 {object} model.CommonResponse{data=BatchLinkEntityResponse} @Router /api/en

(c *gin.Context)

// @Success 200 {object} model.CommonResponse{data=BatchLinkEntityResponse}
// @Router /api/entities/batch-link [post]
func BatchLinkEntities(c *gin.Context) {
	eid := config.GetEID(c)
	if eid <= 0 {
		c.JSON(http.StatusBadRequest, model.ParamError.ToNewErrorResponse(model.InvalidEnterpriseID))
		return
	}

	var req BatchLinkEntityRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, model.ParamError.ToResponse(err))
		return
	}

	if len(req.Items) == 0 {
		c.JSON(http.StatusOK, model.Success.ToResponse(BatchLinkEntityResponse{0, 0}))
		return
	}

	chunkCache := make(map[int64]*model.DocumentChunk)
	fileCache := make(map[int64]*model.File)
	libraryCache := make(map[int64]*model.Library)
	successCount := 0
	failureCount := 0
	createdEntities := make(map[int64]*model.Entity)

	err := model.DB.Transaction(func(tx *gorm.DB) error {
		for _, item := range req.Items {
			item.Type = strings.TrimSpace(item.Type)
			item.Name = strings.TrimSpace(item.Name)
			if item.Type == "" || item.Name == "" {
				failureCount++
				continue
			}
			if item.ChunkID <= 0 && item.FileID <= 0 && item.LibraryID <= 0 {
				failureCount++
				continue
			}

			// 1. 获取/创建实体
			entity, created, err := model.GetOrCreateEntityWithDBAndCreated(tx, eid, item.Type, item.Name)
			if err != nil {
				failureCount++
				continue
			}
			if created {
				createdEntities[entity.ID] = entity
			}

			var spaceID int64
			var chunkID int64
			var fileID int64
			var libraryID int64
			chunkType := "knowledge"

			switch {
			case item.ChunkID > 0:
				chunk, ok := chunkCache[item.ChunkID]
				if !ok {