Method CreateEntityScaleSet

runner/scalesets.go:209–284

(ctx context.Context, entityType params.ForgeEntityType, entityID string, param params.CreateScaleSetParams)

}

func (r *Runner) CreateEntityScaleSet(ctx context.Context, entityType params.ForgeEntityType, entityID string, param params.CreateScaleSetParams) (scaleSetRet params.ScaleSet, err error) {
	if !auth.IsAdmin(ctx) {
		return params.ScaleSet{}, runnerErrors.ErrUnauthorized
	}

	if param.RunnerBootstrapTimeout == 0 {
		param.RunnerBootstrapTimeout = appdefaults.DefaultRunnerBootstrapTimeout
	}

	if param.GitHubRunnerGroup == "" {
		param.GitHubRunnerGroup = "Default"
	}

	entity, err := r.store.GetForgeEntity(ctx, entityType, entityID)
	if err != nil {
		return params.ScaleSet{}, fmt.Errorf("error getting entity: %w", err)
	}

	if entity.Credentials.ForgeType != params.GithubEndpointType {
		return params.ScaleSet{}, runnerErrors.NewBadRequestError("scale sets are only supported for github entities")
	}

	template, err := r.findTemplate(ctx, entity, param.OSType, param.TemplateID)
	if err != nil {
		return params.ScaleSet{}, fmt.Errorf("failed to find suitable template: %w", err)
	}

	param.TemplateID = &template.ID

	ghCli, err := github.Client(ctx, entity)
	if err != nil {
		return params.ScaleSet{}, fmt.Errorf("error creating github client: %w", err)
	}

	scalesetCli, err := scalesets.NewClient(ghCli)
	if err != nil {
		return params.ScaleSet{}, fmt.Errorf("error getting scaleset client: %w", err)
	}

	runnerGroupID, err := ghCli.GetEntityRunnerGroupIDByName(ctx, param.GitHubRunnerGroup)
	if err != nil {
		return params.ScaleSet{}, fmt.Errorf("failed to get github runner group for entity %s: %w", entity.ID, err)
	}

	createParam := &params.RunnerScaleSet{
		Name:          param.Name,
		RunnerGroupID: runnerGroupID,
		Labels:        param.GitHubLabels(),
		RunnerSetting: params.RunnerSetting{
			Ephemeral:     true,
			DisableUpdate: param.DisableUpdate,
		},
		Enabled: &param.Enabled,
	}

	runnerScaleSet, err := scalesetCli.CreateRunnerScaleSet(ctx, createParam)
	if err != nil {
		return params.ScaleSet{}, fmt.Errorf("error creating runner scale set: %w", err)