69 lines
2.4 KiB
JavaScript
69 lines
2.4 KiB
JavaScript
import Fastify from "fastify";
|
|
import sequelizePlugin from "./plugins/sequelize.js";
|
|
import dotenv from "dotenv";
|
|
dotenv.config();
|
|
|
|
import { S3Client, ListBucketsCommand } from "@aws-sdk/client-s3";
|
|
const s3Client = new S3Client({ region: process.env.AWS_REGION, profile: 'default' });
|
|
const listCommand = new ListBucketsCommand({});
|
|
const response = await s3Client.send(listCommand);
|
|
const server = Fastify({ logger: true });
|
|
server.register(sequelizePlugin);
|
|
|
|
server.get("/", async (request, reply) => {
|
|
const [results, metadata] = await server.sequelize.query('SELECT 1 + 2 AS result');
|
|
console.log(results);
|
|
console.log(metadata);
|
|
return { hello: "world" };
|
|
});
|
|
|
|
const productQuery = `SELECT DISTINCT
|
|
line_item_resource_id AS resourceId,
|
|
line_item_product_code AS productCode,
|
|
line_item_usage_account_id AS accountId
|
|
FROM ${process.env.ATHENA_CU_TABLE};`;
|
|
|
|
import { AthenaClient, StartQueryExecutionCommand, GetQueryExecutionCommand, GetQueryResultsCommand } from "@aws-sdk/client-athena";
|
|
const athenaClient = new AthenaClient({ region: process.env.AWS_REGION, profile: 'default' });
|
|
|
|
const startQueryCommand = new StartQueryExecutionCommand({
|
|
QueryString: productQuery,
|
|
QueryExecutionContext: { Database: process.env.ATHENA_CU_DATABASE },
|
|
ResultConfiguration: { OutputLocation: process.env.ATHENA_OUTPUT_S3_BUCKET },
|
|
});
|
|
|
|
const startQueryResponse = await athenaClient.send(startQueryCommand);
|
|
const queryExecutionId = startQueryResponse.QueryExecutionId;
|
|
|
|
let queryExecutionStatus;
|
|
do {
|
|
const getQueryExecutionCommand = new GetQueryExecutionCommand({
|
|
QueryExecutionId: queryExecutionId,
|
|
});
|
|
|
|
const queryExecutionResponse = await athenaClient.send(getQueryExecutionCommand);
|
|
queryExecutionStatus = queryExecutionResponse.QueryExecution.Status.State;
|
|
|
|
if (queryExecutionStatus === "FAILED") {
|
|
console.error(`Query Failed: ${JSON.stringify(queryExecutionResponse)}`);
|
|
break;
|
|
}
|
|
|
|
await new Promise((resolve) => setTimeout(resolve, 5000));
|
|
} while (queryExecutionStatus !== "SUCCEEDED");
|
|
|
|
const getQueryResultsCommand = new GetQueryResultsCommand({
|
|
QueryExecutionId: queryExecutionId,
|
|
});
|
|
|
|
console.log(getQueryResultsCommand);
|
|
const result = await athenaClient.send(getQueryResultsCommand);
|
|
console.log(result.ResultSet.Rows);
|
|
|
|
try {
|
|
await server.listen({ port: 3000 })
|
|
} catch (err) {
|
|
server.log.error(err)
|
|
process.exit(1)
|
|
}
|