Method createTasksTx

backend/store/task.go:156–268

createTasksTx creates tasks in a transaction.

(ctx context.Context, txn *sql.Tx, creates ...*TaskMessage)

// createTasksTx creates tasks in a transaction.
func (*Store) createTasksTx(ctx context.Context, txn *sql.Tx, creates ...*TaskMessage) ([]*TaskMessage, error) {
	if len(creates) == 0 {
		return nil, nil
	}

	projectID := creates[0].ProjectID
	baseID, err := nextProjectID(ctx, txn, "task", projectID)
	if err != nil {
		return nil, err
	}

	var (
		ids          []int64
		projects     []string
		planIDs      []int64
		instances    []string
		databases    []*string
		environments []string
		types        []string
		payloads     [][]byte
	)
	for i, create := range creates {
		if create.Payload == nil {
			create.Payload = &storepb.Task{}
		}
		payload, err := protojson.Marshal(create.Payload)
		if err != nil {
			return nil, errors.Wrapf(err, "failed to marshal payload")
		}
		ids = append(ids, baseID+int64(i))
		projects = append(projects, create.ProjectID)
		planIDs = append(planIDs, create.PlanID)
		instances = append(instances, create.InstanceID)
		databases = append(databases, create.DatabaseName)
		types = append(types, create.Type.String())
		environments = append(environments, create.Environment)
		payloads = append(payloads, payload)
	}

	q := qb.Q().Space(`
		INSERT INTO task (
			id,
			project,
			plan_id,
			instance,
			db_name,
			environment,
			type,
			payload
		) SELECT
			unnest(CAST(? AS BIGINT[])),
			unnest(CAST(? AS TEXT[])),
			unnest(CAST(? AS BIGINT[])),
			unnest(CAST(? AS TEXT[])),
			unnest(CAST(? AS TEXT[])),
			unnest(CAST(? AS TEXT[])),
			unnest(CAST(? AS TEXT[])),
			unnest(CAST(? AS JSONB[]))