Press n or j to go to the next uncovered block, b, p or k for the previous block.
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 | 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 4x 4x 1x 1x 1x 1x 1x 4x 2x 2x 2x 2x 2x 1x 1x 1x 1x 1x 4x 4x 4x 1x 3x 3x 3x 3x 3x 3x 3x 3x 3x 3x 2x 3x 1x 1x 1x 1x 1x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 2x 3x 3x 2x 2x 2x 2x 2x 1x 1x 1x 1x | import {
CloudWatchClient, PutMetricDataCommand
} from '@aws-sdk/client-cloudwatch';
import {
LambdaApiFunction,
handleResourceApi
} from './_base';
import {
parseJsonBody
} from './_utils';
import {
api401Body, api403Body,
generateApi400Body
} from '@/types/api/_shared';
import {
AddHeartbeatApi,
GetAllHeartbeatsApi, Heartbeat,
addHeartbeatBodyValidator
} from '@/types/api/heartbeats';
import {
TABLE_STATUS, typedScan,
typedUpdate
} from '@/utils/backend/dynamoTyped';
import { getLogger } from '@/utils/common/logger';
const logger = getLogger('resources/api/v2/heartbeats');
const cloudWatch = new CloudWatchClient();
const GET: LambdaApiFunction<GetAllHeartbeatsApi> = async function (event, user, userPerms) {
logger.trace('GET', ...arguments);
// Authorize the user
if (user === null) {
return [
401,
api401Body,
];
}
if (!userPerms.isAdmin) {
return [
403,
api403Body,
];
}
// Get the items to return
const heartbeats = await typedScan<Heartbeat>({
TableName: TABLE_STATUS,
});
return [
200,
heartbeats.Items || [],
];
};
const POST: LambdaApiFunction<AddHeartbeatApi> = async function (event) {
logger.trace('POST', ...arguments);
// Parse and validate the body
const [
body,
bodyErrors,
] = parseJsonBody(
event.body,
addHeartbeatBodyValidator
);
if (
body === null ||
bodyErrors.length > 0
) {
return [
400,
generateApi400Body(bodyErrors),
];
}
// Send the metric
const metricPromise = cloudWatch.send(new PutMetricDataCommand({
Namespace: 'VHF Metrics',
MetricData: [ {
MetricName: body.Server,
Timestamp: new Date(),
Unit: 'Count',
Value: 1,
}, ],
}));
// Insert the update into the status table
await typedUpdate<Heartbeat>({
TableName: TABLE_STATUS,
Key: {
Server: body.Server,
},
ExpressionAttributeNames: {
'#IsPrimary': 'IsPrimary',
'#IsActive': 'IsActive',
'#LastHeartbeat': 'LastHeartbeat',
},
ExpressionAttributeValues: {
':IsPrimary': body.IsPrimary,
':IsActive': body.IsActive,
':LastHeartbeat': Date.now(),
},
UpdateExpression: 'SET #IsPrimary = :IsPrimary, #IsActive = :IsActive, #LastHeartbeat = :LastHeartbeat',
});
// Get the other heartbeats
const data = (await typedScan<Heartbeat>({
TableName: TABLE_STATUS,
})).Items || [];
// Wait for everything to finish and send the response
await metricPromise;
return [
200,
data,
];
};
export const main = handleResourceApi.bind(null, {
GET,
POST,
});
|