Method CreateTasks

backend/store/task.go:499–568

CreateTasks creates tasks for a plan.

(ctx context.Context, projectID string, planUID int64, tasks []*TaskMessage)

// CreateTasks creates tasks for a plan.
func (s *Store) CreateTasks(ctx context.Context, projectID string, planUID int64, tasks []*TaskMessage) ([]*TaskMessage, error) {
	tx, err := s.GetDB().BeginTx(ctx, nil)
	if err != nil {
		return nil, errors.Wrapf(err, "failed to begin tx")
	}
	defer tx.Rollback()

	// Check existing tasks to avoid duplicates
	existingTasks, err := s.listTasksImpl(ctx, tx, &TaskFind{
		ProjectID: projectID,
		PlanID:    &planUID,
	})
	if err != nil {
		return nil, errors.Wrapf(err, "failed to list existing tasks")
	}

	type taskKey struct {
		instance string
		database string
		sheet    string
	}

	createdTasks := map[taskKey]struct{}{}
	for _, task := range existingTasks {
		k := taskKey{
			instance: task.InstanceID,
			sheet:    task.Payload.GetSheetSha256(),
		}
		if task.DatabaseName != nil {
			k.database = *task.DatabaseName
		}
		createdTasks[k] = struct{}{}
	}

	var taskCreateList []*TaskMessage

	for _, taskCreate := range tasks {
		k := taskKey{
			instance: taskCreate.InstanceID,
			sheet:    taskCreate.Payload.GetSheetSha256(),
		}
		if taskCreate.DatabaseName != nil {
			k.database = *taskCreate.DatabaseName
		}

		if _, ok := createdTasks[k]; ok {
			continue
		}
		taskCreate.ProjectID = projectID
		taskCreate.PlanID = planUID
		taskCreateList = append(taskCreateList, taskCreate)
	}

	if len(taskCreateList) > 0 {
		tasks, err := s.createTasksTx(ctx, tx, taskCreateList...)
		if err != nil {
			return nil, errors.Wrap(err, "failed to create tasks")
		}