Boonlive_RD_Web/Web_BLS_OnOffLine_Server
9
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 添加当前或未来日期检查功能以优化分区创建逻辑

Browse Source
This commit is contained in:
徐
2026-03-03 20:21:57 +08:00
parent c6d7cea8cf
commit ba61a540da
2 changed files with 25 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
bls-onoffline-backend/dist/index.js vendored
View File

@@ -174,6 +174,13 @@ class DatabaseManager {
}
const dbManager = new DatabaseManager(config.db);
class PartitionManager {
isCurrentOrFutureDate(date) {
const normalizedDate = new Date(date);
normalizedDate.setHours(0, 0, 0, 0);
const today = /* @__PURE__ */ new Date();
today.setHours(0, 0, 0, 0);
return normalizedDate.getTime() >= today.getTime();
}
/**
* Calculate the start and end timestamps (milliseconds) for a given date.
* @param {Date} date - The date to calculate for.
@@ -218,10 +225,11 @@ class PartitionManager {
if (!checkRes.rows[0].exists) {
logger.info(`Creating partition ${partitionName} for range [${startMs}, ${endMs})`);
console.log(`Creating partition ${partitionName} for range [${startMs}, ${endMs})`);
const tablespaceClause = this.isCurrentOrFutureDate(targetDate) ? " TABLESPACE ts_hot" : "";
const createSql = `
CREATE TABLE IF NOT EXISTS ${partitionName}
PARTITION OF ${schema}.${table}
FOR VALUES FROM (${startMs}) TO (${endMs});
FOR VALUES FROM (${startMs}) TO (${endMs})${tablespaceClause};
`;
await client.query(createSql);
}
@@ -263,10 +271,11 @@ class PartitionManager {
const checkRes = await client.query(`SELECT to_regclass($1) as exists;`, [partitionName]);
if (!checkRes.rows[0].exists) {
logger.info(`Creating partition ${partitionName} for range [${startMs}, ${endMs})`);
const tablespaceClause = this.isCurrentOrFutureDate(targetDate) ? " TABLESPACE ts_hot" : "";
await client.query(`
CREATE TABLE IF NOT EXISTS ${partitionName}
PARTITION OF ${schema}.${table}
FOR VALUES FROM (${startMs}) TO (${endMs});
FOR VALUES FROM (${startMs}) TO (${endMs})${tablespaceClause};
`);
}
}

16
bls-onoffline-backend/src/db/partitionManager.js
View File

@@ -3,6 +3,16 @@ import { config } from '../config/config.js';
import dbManager from './databaseManager.js';
class PartitionManager {
isCurrentOrFutureDate(date) {
const normalizedDate = new Date(date);
normalizedDate.setHours(0, 0, 0, 0);
const today = new Date();
today.setHours(0, 0, 0, 0);
return normalizedDate.getTime() >= today.getTime();
}
/**
* Calculate the start and end timestamps (milliseconds) for a given date.
* @param {Date} date - The date to calculate for.
@@ -56,10 +66,11 @@ class PartitionManager {
if (!checkRes.rows[0].exists) {
logger.info(`Creating partition ${partitionName} for range [${startMs}, ${endMs})`);
console.log(`Creating partition ${partitionName} for range [${startMs}, ${endMs})`);
const tablespaceClause = this.isCurrentOrFutureDate(targetDate) ? ' TABLESPACE ts_hot' : '';
const createSql = `
CREATE TABLE IF NOT EXISTS ${partitionName}
PARTITION OF ${schema}.${table}
FOR VALUES FROM (${startMs}) TO (${endMs});
FOR VALUES FROM (${startMs}) TO (${endMs})${tablespaceClause};
`;
await client.query(createSql);
}
@@ -108,10 +119,11 @@ class PartitionManager {
const checkRes = await client.query(`SELECT to_regclass($1) as exists;`, [partitionName]);
if (!checkRes.rows[0].exists) {
logger.info(`Creating partition ${partitionName} for range [${startMs}, ${endMs})`);
const tablespaceClause = this.isCurrentOrFutureDate(targetDate) ? ' TABLESPACE ts_hot' : '';
await client.query(`
CREATE TABLE IF NOT EXISTS ${partitionName}
PARTITION OF ${schema}.${table}
FOR VALUES FROM (${startMs}) TO (${endMs});
FOR VALUES FROM (${startMs}) TO (${endMs})${tablespaceClause};
`);
}
}