1. const eventBeforeImporting = Date.now();
2. import {
3. DynamoDBClient,
4. GetItemCommand,
5. PutItemCommand,
6. } from "@aws-sdk/client-dynamodb";
7. ​
8. const TABLE_NAME = "";
9. const TOKEN_ID = "";
10. ​
11. const initinfSDK = Date.now();
12. const client = new DynamoDBClient();
13. ​
14. console.log("sdk init successful", Date.now() - initinfSDK);
15. ​
16. export const nowUnix = () => Math.floor(Date.now() / 1000);
17. ​
18. export async function getManagementTokenFromDB() {
19. const startTime = Date.now();
20. ​
21. const command = new GetItemCommand({
22. TableName: TABLE_NAME,
23. Key: {
24. "my-key-name": { S: TOKEN_ID as string },
25. },
26. });
27. const result = await client.send(command);
28. const duration = Date.now() - startTime;
29. console.info(`GetToken Time READ FROM DYNO: ${duration}ms`);
30. ​
31. if (!result.Item) {
32. console.info("No token found in DB.");
33. return null;
34. }
35. const tokenValue = result.Item.tokenValue.S;
36. const expiresAt = parseInt(result.Item.expiresAt.N as string);
37. const now = nowUnix();
38. const age = expiresAt - now;
39. if (now >= expiresAt) {
40. console.info("Token expired.");
41. return null;
42. }
43. console.info(`Token is valid. Expires in ${age}s`); // remove log for prod and even some more top level logs
44. return tokenValue;
45. }
46. ​
47. export async function saveNewToken(token: string) {
48. const expiresAt = nowUnix() + Math.floor(6.5 * 24 * 60 * 60); // 6.5 days
49. const now = nowUnix();
50. ​
51. const command = new PutItemCommand({
52. TableName: TABLE_NAME,
53. Item: {
54. "my-key-name": { S: TOKEN_ID as string },
55. tokenValue: { S: token },
56. createdAt: { N: String(now) },
57. expiresAt: { N: String(expiresAt) },
58. },
59. });
60. await client.send(command);
61. console.info("New token saved to DB.");
62. }