Skip to content

Commit 6f2d477

Browse files
committed
Add argument dth (datetime and hour) in createPartition for user invoke
1 parent 6510bdf commit 6f2d477

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

functions/createPartitions.js

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,13 @@ const database = process.env.DATABASE;
88

99
// creates partitions for the hour after the current hour
1010
exports.handler = async (event, context, callback) => {
11-
var nextHour = new Date(Date.now() + 60 * 60 * 1000);
11+
12+
// format of dth is '2021-01-01T00'
13+
var input = 'dth' in event ? `${event.dth}:00:00Z` : Date.now() + 60 * 60 * 1000;
14+
var nextHour = new Date(input);
15+
if (isNaN(nextHour))
16+
throw new Error('invalid dth')
17+
1218
var year = nextHour.getUTCFullYear();
1319
var month = (nextHour.getUTCMonth() + 1).toString().padStart(2, '0');
1420
var day = nextHour.getUTCDate().toString().padStart(2, '0');

0 commit comments

Comments
 (0)