Skip to content

Commit b4a2355

Browse files
authored
fix: enable snowflake auth using key or token (#3878)
1 parent 87ba88c commit b4a2355

3 files changed

Lines changed: 13 additions & 15 deletions

File tree

services/apps/snowflake_connectors/src/core/snowflakeExporter.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ export class SnowflakeExporter {
2828
private readonly snowflake: SnowflakeClient
2929

3030
constructor() {
31-
this.snowflake = SnowflakeClient.fromToken({ parentLog: log })
31+
this.snowflake = SnowflakeClient.fromEnv({ parentLog: log })
3232
}
3333

3434
async destroy(): Promise<void> {

services/apps/snowflake_connectors/src/main.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ import { Options, ServiceWorker } from '@crowd/archetype-worker'
99

1010
const config: Config = {
1111
envvars: [
12-
'CROWD_SNOWFLAKE_ENABLED_PLATFORMS',
12+
// 'CROWD_SNOWFLAKE_ENABLED_PLATFORMS', // allow optional to enable/disable worker
1313
'CROWD_SNOWFLAKE_S3_BUCKET_PATH',
1414
'CROWD_SNOWFLAKE_STORAGE_INTEGRATION',
1515
'CROWD_SNOWFLAKE_S3_REGION',

services/libs/snowflake/src/client.ts

Lines changed: 11 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -179,28 +179,26 @@ export class SnowflakeClient {
179179
}
180180

181181
public static fromEnv(extraConfig: any = {}) {
182-
return new SnowflakeClient({
183-
privateKeyString: process.env.CROWD_SNOWFLAKE_PRIVATE_KEY,
182+
const base = {
184183
account: process.env.CROWD_SNOWFLAKE_ACCOUNT,
185184
username: process.env.CROWD_SNOWFLAKE_USERNAME,
186185
database: process.env.CROWD_SNOWFLAKE_DATABASE,
187186
warehouse: process.env.CROWD_SNOWFLAKE_WAREHOUSE,
188187
role: process.env.CROWD_SNOWFLAKE_ROLE,
189188
maxConnections: 1,
190189
...extraConfig,
191-
})
192-
}
190+
}
191+
192+
if (process.env.CROWD_SNOWFLAKE_TOKEN) {
193+
return new SnowflakeClient({
194+
token: process.env.CROWD_SNOWFLAKE_TOKEN,
195+
...base,
196+
})
197+
}
193198

194-
public static fromToken(extraConfig: any = {}) {
195199
return new SnowflakeClient({
196-
token: process.env.CROWD_SNOWFLAKE_TOKEN,
197-
account: process.env.CROWD_SNOWFLAKE_ACCOUNT,
198-
username: process.env.CROWD_SNOWFLAKE_USERNAME,
199-
database: process.env.CROWD_SNOWFLAKE_DATABASE,
200-
warehouse: process.env.CROWD_SNOWFLAKE_WAREHOUSE,
201-
role: process.env.CROWD_SNOWFLAKE_ROLE,
202-
maxConnections: 1,
203-
...extraConfig,
200+
privateKeyString: process.env.CROWD_SNOWFLAKE_PRIVATE_KEY,
201+
...base,
204202
})
205203
}
206204
}

0 commit comments

Comments
 (0)