XuJiacheng
eb94aaf92b
- 新增 heartbeat 数据库与表结构文档，描述心跳明细表设计及字段约束。 - 新增 OpenSpec 符合性说明文档，指出与规范的一致点及偏差。 - 新增 Kafka 心跳数据推送说明文档，定义消息格式与推送方式。 - 更新数据库创建脚本，支持 UTF-8 编码与中文排序规则。 - 更新心跳表结构脚本，定义主表及索引，采用 ts_ms 日分区。 - 实现自动分区机制，确保按天创建分区以支持高吞吐写入。 - 添加数据库应用脚本，自动执行 SQL 文件并验证表结构。 - 添加运行时烟雾测试脚本，验证数据库连接与基本操作。 - 添加完整的烟雾测试脚本，验证数据插入与分区创建。
68 lines
1.6 KiB
JavaScript
68 lines
1.6 KiB
JavaScript
import { Client } from 'pg';
|
|
import config from '../../src/config/config.js';
|
|
|
|
async function main() {
|
|
const client = new Client({
|
|
host: config.db.host,
|
|
port: config.db.port,
|
|
user: config.db.user,
|
|
password: config.db.password,
|
|
database: config.db.database,
|
|
});
|
|
|
|
await client.connect();
|
|
|
|
// 预创建今日分区，避免“无分区时 INSERT 直接失败”
|
|
await client.query('SELECT heartbeat.ensure_partitions(current_date, current_date)');
|
|
|
|
const ts = Date.now();
|
|
await client.query(
|
|
`INSERT INTO heartbeat.heartbeat_events (
|
|
ts_ms, hotel_id, room_id, device_id, ip,
|
|
power_state, guest_type, cardless_state, service_mask,
|
|
pms_state, carbon_state, device_count, comm_seq, extra
|
|
) VALUES ($1,$2,$3,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13,$14)`,
|
|
[
|
|
ts,
|
|
1,
|
|
101,
|
|
'dev-1',
|
|
'192.168.0.1',
|
|
1,
|
|
0,
|
|
0,
|
|
5,
|
|
0,
|
|
0,
|
|
1,
|
|
1,
|
|
{ source: 'smoke-test' },
|
|
]
|
|
);
|
|
|
|
const partitions = await client.query(
|
|
`SELECT c.relname AS partition
|
|
FROM pg_inherits i
|
|
JOIN pg_class c ON c.oid = i.inhrelid
|
|
JOIN pg_class p ON p.oid = i.inhparent
|
|
JOIN pg_namespace n ON n.oid = p.relnamespace
|
|
WHERE n.nspname = 'heartbeat'
|
|
AND p.relname = 'heartbeat_events'
|
|
ORDER BY c.relname`
|
|
);
|
|
|
|
const cnt = await client.query(
|
|
'SELECT count(*)::int AS n FROM heartbeat.heartbeat_events'
|
|
);
|
|
|
|
console.log('partitions:', partitions.rows.map((r) => r.partition));
|
|
console.log('rows:', cnt.rows[0].n);
|
|
|
|
await client.end();
|
|
}
|
|
|
|
main().catch((err) => {
|
|
console.error('smoke test failed:', err);
|
|
process.exit(1);
|
|
});
|