Skip to content

Commit 6a4a452

Browse files
committed
make naming more consistent
1 parent dc315c0 commit 6a4a452

File tree

2 files changed

+17
-17
lines changed

2 files changed

+17
-17
lines changed

packages/core/src/awsService/ec2/model.ts

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ export class Ec2ConnectionManager {
4444
protected ssmClient: SsmClient
4545
protected ec2Client: Ec2Client
4646
protected iamClient: DefaultIamClient
47-
protected envManager: Ec2RemoteSessionManager
47+
protected sessionManager: Ec2RemoteSessionManager
4848

4949
private policyDocumentationUri = vscode.Uri.parse(
5050
'https://docs.aws.amazon.com/systems-manager/latest/userguide/session-manager-getting-started-instance-profile.html'
@@ -58,7 +58,7 @@ export class Ec2ConnectionManager {
5858
this.ssmClient = this.createSsmSdkClient()
5959
this.ec2Client = this.createEc2SdkClient()
6060
this.iamClient = this.createIamSdkClient()
61-
this.envManager = new Ec2RemoteSessionManager(regionCode, this.ssmClient)
61+
this.sessionManager = new Ec2RemoteSessionManager(regionCode, this.ssmClient)
6262
}
6363

6464
protected createSsmSdkClient(): SsmClient {
@@ -73,16 +73,16 @@ export class Ec2ConnectionManager {
7373
return new DefaultIamClient(this.regionCode)
7474
}
7575

76-
public async addActiveEnv(sessionId: SSM.SessionId, instanceId: EC2.InstanceId): Promise<void> {
77-
await this.envManager.addEnv(instanceId, sessionId)
76+
public async addActiveSession(sessionId: SSM.SessionId, instanceId: EC2.InstanceId): Promise<void> {
77+
await this.sessionManager.addSession(instanceId, sessionId)
7878
}
7979

8080
public async closeConnections(): Promise<void> {
81-
await this.envManager.closeConnections()
81+
await this.sessionManager.closeConnections()
8282
}
8383

8484
public isConnectedTo(instanceId: string): boolean {
85-
return this.envManager.isConnectedTo(instanceId)
85+
return this.sessionManager.isConnectedTo(instanceId)
8686
}
8787

8888
public async getAttachedIamRole(instanceId: string): Promise<IAM.Role | undefined> {
@@ -225,7 +225,7 @@ export class Ec2ConnectionManager {
225225
throw err
226226
}
227227
const ssmSession = await this.ssmClient.startSession(selection.instanceId, 'AWS-StartSSHSession')
228-
await this.envManager.addEnv(selection.instanceId, ssmSession.SessionId!)
228+
await this.addActiveSession(selection.instanceId, ssmSession.SessionId!)
229229

230230
const vars = getEc2SsmEnv(selection, ssm, ssmSession)
231231
const envProvider = async () => {

packages/core/src/awsService/ec2/remoteSessionManager.ts

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -7,35 +7,35 @@ import { EC2, SSM } from 'aws-sdk'
77
import { SsmClient } from '../../shared/clients/ssmClient'
88

99
export class Ec2RemoteSessionManager {
10-
private activeEnvs: Map<EC2.InstanceId, SSM.SessionId>
10+
private activeSessions: Map<EC2.InstanceId, SSM.SessionId>
1111

1212
public constructor(
1313
readonly regionCode: string,
1414
protected ssmClient: SsmClient
1515
) {
16-
this.activeEnvs = new Map<EC2.InstanceId, SSM.SessionId>()
16+
this.activeSessions = new Map<EC2.InstanceId, SSM.SessionId>()
1717
}
1818

19-
public async addEnv(instanceId: EC2.InstanceId, sessionId: SSM.SessionId): Promise<void> {
19+
public async addSession(instanceId: EC2.InstanceId, sessionId: SSM.SessionId): Promise<void> {
2020
if (this.isConnectedTo(instanceId)) {
21-
const existingSessionId = this.activeEnvs.get(instanceId)!
21+
const existingSessionId = this.activeSessions.get(instanceId)!
2222
await this.ssmClient.terminateSessionFromId(existingSessionId)
23-
this.activeEnvs.set(instanceId, sessionId)
23+
this.activeSessions.set(instanceId, sessionId)
2424
} else {
25-
this.activeEnvs.set(instanceId, sessionId)
25+
this.activeSessions.set(instanceId, sessionId)
2626
}
2727
}
2828

2929
private async disconnectEnv(instanceId: EC2.InstanceId): Promise<void> {
30-
await this.ssmClient.terminateSessionFromId(this.activeEnvs.get(instanceId)!)
31-
this.activeEnvs.delete(instanceId)
30+
await this.ssmClient.terminateSessionFromId(this.activeSessions.get(instanceId)!)
31+
this.activeSessions.delete(instanceId)
3232
}
3333

3434
public async closeConnections(): Promise<void> {
35-
this.activeEnvs.forEach(async (_sessionId, instanceId) => await this.disconnectEnv(instanceId))
35+
this.activeSessions.forEach(async (_sessionId, instanceId) => await this.disconnectEnv(instanceId))
3636
}
3737

3838
public isConnectedTo(instanceId: EC2.InstanceId): boolean {
39-
return this.activeEnvs.has(instanceId)
39+
return this.activeSessions.has(instanceId)
4040
}
4141
}

0 commit comments

Comments
 (0)