feat(M01-B): 增加迁移执行器与旧库只读兼容层
This commit is contained in:
Codex
@@ -0,0 +1,145 @@
|
||||
import type { MySqlPool } from './mysql.js';
|
||||
|
||||
type LegacyEntity = 'stores' | 'rooms' | 'orders' | 'devices';
|
||||
|
||||
export interface LegacyTableMapping {
|
||||
table: string;
|
||||
idColumn: string;
|
||||
tenantColumn: string;
|
||||
parentColumn?: string;
|
||||
nameColumn?: string;
|
||||
statusColumn?: string;
|
||||
codeColumn?: string;
|
||||
startColumn?: string;
|
||||
endColumn?: string;
|
||||
}
|
||||
|
||||
export interface LegacyReadOptions {
|
||||
tenantId: number;
|
||||
parentId?: number;
|
||||
limit?: number;
|
||||
}
|
||||
|
||||
export interface LegacyRecord {
|
||||
legacyId: string;
|
||||
tenantId: string;
|
||||
parentId: string | null;
|
||||
name: string | null;
|
||||
code: string | null;
|
||||
status: string | null;
|
||||
startAt: Date | string | null;
|
||||
endAt: Date | string | null;
|
||||
}
|
||||
|
||||
export const defaultLegacyMappings: Record<LegacyEntity, LegacyTableMapping> = {
|
||||
stores: {
|
||||
table: 'member_store_info',
|
||||
idColumn: 'id',
|
||||
tenantColumn: 'tenant_id',
|
||||
nameColumn: 'store_name',
|
||||
statusColumn: 'status'
|
||||
},
|
||||
rooms: {
|
||||
table: 'member_room_info',
|
||||
idColumn: 'id',
|
||||
tenantColumn: 'tenant_id',
|
||||
parentColumn: 'store_id',
|
||||
nameColumn: 'room_name',
|
||||
codeColumn: 'room_no',
|
||||
statusColumn: 'status'
|
||||
},
|
||||
orders: {
|
||||
table: 'member_order_info',
|
||||
idColumn: 'id',
|
||||
tenantColumn: 'tenant_id',
|
||||
parentColumn: 'room_id',
|
||||
codeColumn: 'order_no',
|
||||
statusColumn: 'status',
|
||||
startColumn: 'start_time',
|
||||
endColumn: 'end_time'
|
||||
},
|
||||
devices: {
|
||||
table: 'member_device_info',
|
||||
idColumn: 'id',
|
||||
tenantColumn: 'tenant_id',
|
||||
parentColumn: 'room_id',
|
||||
nameColumn: 'device_name',
|
||||
codeColumn: 'device_id',
|
||||
statusColumn: 'status'
|
||||
}
|
||||
};
|
||||
|
||||
const identifierPattern = /^[A-Za-z][A-Za-z0-9_]*$/;
|
||||
|
||||
function quoteIdentifier(identifier: string): string {
|
||||
if (!identifierPattern.test(identifier)) {
|
||||
throw new Error(`Unsafe legacy SQL identifier: ${identifier}`);
|
||||
}
|
||||
return `\`${identifier}\``;
|
||||
}
|
||||
|
||||
function selectedColumn(column: string | undefined, alias: string): string {
|
||||
return column ? `${quoteIdentifier(column)} AS ${quoteIdentifier(alias)}` : `NULL AS ${quoteIdentifier(alias)}`;
|
||||
}
|
||||
|
||||
function normalizeLimit(limit = 100): number {
|
||||
if (!Number.isInteger(limit) || limit < 1 || limit > 500) {
|
||||
throw new Error('Legacy read limit must be an integer between 1 and 500.');
|
||||
}
|
||||
return limit;
|
||||
}
|
||||
|
||||
export class LegacyReadRepository {
|
||||
constructor(
|
||||
private readonly pool: Pick<MySqlPool, 'query'>,
|
||||
private readonly mappings: Record<LegacyEntity, LegacyTableMapping> = defaultLegacyMappings
|
||||
) {}
|
||||
|
||||
listStores(options: LegacyReadOptions): Promise<LegacyRecord[]> {
|
||||
return this.list('stores', options);
|
||||
}
|
||||
|
||||
listRooms(options: LegacyReadOptions): Promise<LegacyRecord[]> {
|
||||
return this.list('rooms', options);
|
||||
}
|
||||
|
||||
listOrders(options: LegacyReadOptions): Promise<LegacyRecord[]> {
|
||||
return this.list('orders', options);
|
||||
}
|
||||
|
||||
listDevices(options: LegacyReadOptions): Promise<LegacyRecord[]> {
|
||||
return this.list('devices', options);
|
||||
}
|
||||
|
||||
private async list(entity: LegacyEntity, options: LegacyReadOptions): Promise<LegacyRecord[]> {
|
||||
const mapping = this.mappings[entity];
|
||||
const limit = normalizeLimit(options.limit);
|
||||
const clauses = [`${quoteIdentifier(mapping.tenantColumn)} = ?`];
|
||||
const parameters: Array<number> = [options.tenantId];
|
||||
|
||||
if (mapping.parentColumn && options.parentId !== undefined) {
|
||||
clauses.push(`${quoteIdentifier(mapping.parentColumn)} = ?`);
|
||||
parameters.push(options.parentId);
|
||||
}
|
||||
|
||||
parameters.push(limit);
|
||||
const sql = [
|
||||
'SELECT',
|
||||
`${quoteIdentifier(mapping.idColumn)} AS ${quoteIdentifier('legacyId')},`,
|
||||
`${quoteIdentifier(mapping.tenantColumn)} AS ${quoteIdentifier('tenantId')},`,
|
||||
`${selectedColumn(mapping.parentColumn, 'parentId')},`,
|
||||
`${selectedColumn(mapping.nameColumn, 'name')},`,
|
||||
`${selectedColumn(mapping.codeColumn, 'code')},`,
|
||||
`${selectedColumn(mapping.statusColumn, 'status')},`,
|
||||
`${selectedColumn(mapping.startColumn, 'startAt')},`,
|
||||
selectedColumn(mapping.endColumn, 'endAt'),
|
||||
`FROM ${quoteIdentifier(mapping.table)}`,
|
||||
`WHERE ${clauses.join(' AND ')}`,
|
||||
`ORDER BY ${quoteIdentifier(mapping.idColumn)} ASC`,
|
||||
'LIMIT ?'
|
||||
].join(' ');
|
||||
|
||||
const [rows] = await this.pool.query(sql, parameters);
|
||||
return rows as LegacyRecord[];
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,49 @@
|
||||
import { loadConfig } from '../config.js';
|
||||
import { closeMySqlPool, createMySqlPool } from './mysql.js';
|
||||
import {
|
||||
executeMigrationPlan,
|
||||
loadMigrationPlan,
|
||||
type MigrationDirection
|
||||
} from './migration-runner.js';
|
||||
|
||||
const command = process.argv[2] ?? 'plan';
|
||||
const validCommands = new Set(['plan', 'up', 'verify', 'down']);
|
||||
|
||||
if (!validCommands.has(command)) {
|
||||
console.error('Usage: migrate-cli.js <plan|up|verify|down>');
|
||||
process.exitCode = 2;
|
||||
} else {
|
||||
const direction: MigrationDirection = command === 'plan' ? 'up' : command as MigrationDirection;
|
||||
const plan = await loadMigrationPlan(direction);
|
||||
|
||||
if (command === 'plan') {
|
||||
console.log(JSON.stringify({
|
||||
mode: 'dry-run',
|
||||
direction: plan.direction,
|
||||
file: plan.file,
|
||||
checksum: plan.checksum,
|
||||
statementCount: plan.statements.length
|
||||
}, null, 2));
|
||||
} else {
|
||||
const config = loadConfig();
|
||||
if (!config.mysql.passwordConfigured) {
|
||||
console.error('QIPAI_MYSQL_PASSWORD is required for live migration commands.');
|
||||
process.exitCode = 2;
|
||||
} else {
|
||||
const pool = createMySqlPool(config);
|
||||
try {
|
||||
const result = await executeMigrationPlan(pool, plan);
|
||||
console.log(JSON.stringify({
|
||||
mode: 'live',
|
||||
direction: result.direction,
|
||||
file: result.file,
|
||||
checksum: result.checksum,
|
||||
statementCount: result.statements.length,
|
||||
affectedRows: result.affectedRows
|
||||
}, null, 2));
|
||||
} finally {
|
||||
await closeMySqlPool(pool);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,153 @@
|
||||
import { createHash } from 'node:crypto';
|
||||
import { readFile } from 'node:fs/promises';
|
||||
import { dirname, resolve } from 'node:path';
|
||||
import { fileURLToPath } from 'node:url';
|
||||
import type { MySqlPool } from './mysql.js';
|
||||
|
||||
export type MigrationDirection = 'up' | 'verify' | 'down';
|
||||
|
||||
export interface MigrationPlan {
|
||||
direction: MigrationDirection;
|
||||
file: string;
|
||||
checksum: string;
|
||||
statements: readonly string[];
|
||||
}
|
||||
|
||||
export interface MigrationExecutionResult extends MigrationPlan {
|
||||
executed: boolean;
|
||||
affectedRows: number;
|
||||
}
|
||||
|
||||
const repoRoot = resolve(dirname(fileURLToPath(import.meta.url)), '../../..');
|
||||
const migrationFiles: Record<MigrationDirection, string> = {
|
||||
up: 'database/migrations/2026061601_m01b_core_schema.up.sql',
|
||||
verify: 'database/migrations/2026061601_m01b_core_schema.verify.sql',
|
||||
down: 'database/migrations/2026061601_m01b_core_schema.down.sql'
|
||||
};
|
||||
|
||||
export function splitSqlStatements(sql: string): string[] {
|
||||
const statements: string[] = [];
|
||||
let current = '';
|
||||
let quote: "'" | '"' | '`' | null = null;
|
||||
let escaped = false;
|
||||
let lineComment = false;
|
||||
let blockComment = false;
|
||||
|
||||
for (let index = 0; index < sql.length; index += 1) {
|
||||
const character = sql[index];
|
||||
const next = sql[index + 1];
|
||||
|
||||
if (lineComment) {
|
||||
if (character === '\n') {
|
||||
lineComment = false;
|
||||
current += character;
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
if (blockComment) {
|
||||
if (character === '*' && next === '/') {
|
||||
blockComment = false;
|
||||
index += 1;
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!quote && character === '-' && next === '-' && (index === 0 || /\s/.test(sql[index - 1]))) {
|
||||
lineComment = true;
|
||||
index += 1;
|
||||
continue;
|
||||
}
|
||||
|
||||
if (!quote && character === '/' && next === '*') {
|
||||
blockComment = true;
|
||||
index += 1;
|
||||
continue;
|
||||
}
|
||||
|
||||
current += character;
|
||||
|
||||
if (quote) {
|
||||
if (escaped) {
|
||||
escaped = false;
|
||||
} else if (character === '\\') {
|
||||
escaped = true;
|
||||
} else if (character === quote) {
|
||||
if (next === quote) {
|
||||
current += next;
|
||||
index += 1;
|
||||
} else {
|
||||
quote = null;
|
||||
}
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
if (character === "'" || character === '"' || character === '`') {
|
||||
quote = character;
|
||||
continue;
|
||||
}
|
||||
|
||||
if (character === ';') {
|
||||
const statement = current.slice(0, -1).trim();
|
||||
if (statement) {
|
||||
statements.push(statement);
|
||||
}
|
||||
current = '';
|
||||
}
|
||||
}
|
||||
|
||||
const trailing = current.trim();
|
||||
if (trailing) {
|
||||
statements.push(trailing);
|
||||
}
|
||||
|
||||
if (quote || blockComment) {
|
||||
throw new Error('Migration SQL contains an unterminated quote or comment.');
|
||||
}
|
||||
|
||||
return statements;
|
||||
}
|
||||
|
||||
export async function loadMigrationPlan(direction: MigrationDirection): Promise<MigrationPlan> {
|
||||
const relativeFile = migrationFiles[direction];
|
||||
const sql = await readFile(resolve(repoRoot, relativeFile), 'utf8');
|
||||
|
||||
return {
|
||||
direction,
|
||||
file: relativeFile,
|
||||
checksum: createHash('sha256').update(sql).digest('hex'),
|
||||
statements: splitSqlStatements(sql)
|
||||
};
|
||||
}
|
||||
|
||||
export async function executeMigrationPlan(
|
||||
pool: Pick<MySqlPool, 'query'>,
|
||||
plan: MigrationPlan,
|
||||
dryRun = false
|
||||
): Promise<MigrationExecutionResult> {
|
||||
if (dryRun) {
|
||||
return { ...plan, executed: false, affectedRows: 0 };
|
||||
}
|
||||
|
||||
let affectedRows = 0;
|
||||
for (const [index, statement] of plan.statements.entries()) {
|
||||
const [result] = await pool.query(statement);
|
||||
if (plan.direction === 'verify') {
|
||||
const minimumRows = [10, 26, 1][index] ?? 1;
|
||||
if (!Array.isArray(result) || result.length < minimumRows) {
|
||||
throw new Error(
|
||||
`Migration verification statement ${index + 1} returned fewer than ${minimumRows} rows.`
|
||||
);
|
||||
}
|
||||
}
|
||||
if (result && typeof result === 'object' && 'affectedRows' in result) {
|
||||
const value = Reflect.get(result, 'affectedRows');
|
||||
if (typeof value === 'number') {
|
||||
affectedRows += value;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return { ...plan, executed: true, affectedRows };
|
||||
}
|
||||
@@ -23,3 +23,7 @@ export function toPoolOptions(config: AppConfig): PoolOptions {
|
||||
dateStrings: false
|
||||
};
|
||||
}
|
||||
|
||||
export async function closeMySqlPool(pool: MySqlPool): Promise<void> {
|
||||
await pool.end();
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user