feat: add more decorators to track

Hweinstock · Hweinstock · commit 595b6d55ff63 · 2025-02-10T11:15:18.000-05:00
diff --git a/packages/core/src/auth/auth.ts b/packages/core/src/auth/auth.ts
@@ -417,6 +417,7 @@ export class Auth implements AuthService, ConnectionManager {
         await setContext('aws.isInternalUser', false)
     }
 
+    @withTelemetryContext({ name: 'getConnection', class: authClassName })
     public async getConnection(connection: Pick<Connection, 'id'>): Promise<Connection | undefined> {
         const connections = await this.listConnections()
 
@@ -524,6 +525,7 @@ export class Auth implements AuthService, ConnectionManager {
         await this.thrownOnConn(id, 'exists')
     }
 
+    @withTelemetryContext({ name: 'thrownOnConn', class: authClassName })
     private async thrownOnConn(id: CredentialsId, throwOn: 'exists' | 'not-exists') {
         const idAsString = asString(id)
         const conns = await this.listConnections() // triggers loading of profile in to store

Original file line number	Diff line number	Diff line change
`@@ -417,6 +417,7 @@ export class Auth implements AuthService, ConnectionManager {`
`417`	`417`	`await setContext('aws.isInternalUser', false)`
`418`	`418`	`}`
`419`	`419`
	`420`	`+ @withTelemetryContext({ name: 'getConnection', class: authClassName })`
`420`	`421`	`public async getConnection(connection: Pick<Connection, 'id'>): Promise<Connection \| undefined> {`
`421`	`422`	`const connections = await this.listConnections()`
`422`	`423`
`@@ -524,6 +525,7 @@ export class Auth implements AuthService, ConnectionManager {`
`524`	`525`	`await this.thrownOnConn(id, 'exists')`
`525`	`526`	`}`
`526`	`527`
	`528`	`+ @withTelemetryContext({ name: 'thrownOnConn', class: authClassName })`
`527`	`529`	`private async thrownOnConn(id: CredentialsId, throwOn: 'exists' \| 'not-exists') {`
`528`	`530`	`const idAsString = asString(id)`
`529`	`531`	`const conns = await this.listConnections() // triggers loading of profile in to store`