Skip to content

Commit f6fc43d

Browse files
refactor(pool): expose provider identity
1 parent 870ce90 commit f6fc43d

3 files changed

Lines changed: 4 additions & 2 deletions

File tree

lambdas/functions/control-plane/src/pool/ec2-pool.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ export function createEc2PoolProvider(): PoolRunnerProvider {
8282
const config = loadEc2PoolProviderConfig();
8383

8484
return {
85+
type: 'ec2',
8586
listRunners: listEc2PoolRunners,
8687
countAvailableRunners: calculateEc2PoolSize,
8788
createRunners: createEc2PoolRunners.bind(undefined, config),

lambdas/functions/control-plane/src/pool/pool-provider.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import type { Octokit } from '@octokit/rest';
22

3+
import type { RunnerProvider } from '../runner-provider';
34
import type { CreateGitHubRunnerConfig, GitHubRunnerType } from '../scale-runners/types';
45

56
export interface RunnerStatus {
@@ -19,7 +20,7 @@ export interface CreatePoolRunnersInput {
1920
githubInstallationClient: Octokit;
2021
}
2122

22-
export interface PoolRunnerProvider<TRunner = unknown> {
23+
export interface PoolRunnerProvider<TRunner = unknown> extends RunnerProvider {
2324
listRunners(input: ListPoolRunnersInput): Promise<TRunner[]>;
2425
countAvailableRunners(
2526
runners: TRunner[],

lambdas/functions/control-plane/src/pool/pool.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ export interface PoolEvent {
1818
export async function adjust(event: PoolEvent): Promise<void> {
1919
const runnerProviderType = resolveRunnerProviderType(event.type);
2020
const runnerProvider = createPoolRunnerProvider(runnerProviderType);
21-
logger.info(`Checking current ${runnerProviderType} pool size against pool of size: ${event.poolSize}`);
21+
logger.info(`Checking current ${runnerProvider.type} pool size against pool of size: ${event.poolSize}`);
2222
const runnerLabels = process.env.RUNNER_LABELS || '';
2323
const runnerGroup = process.env.RUNNER_GROUP_NAME || '';
2424
const runnerNamePrefix = process.env.RUNNER_NAME_PREFIX || '';

0 commit comments

Comments
 (0)