Files
certd/packages/ui/certd-server/db/transform.js
T
2026-05-31 01:41:33 +08:00

108 lines
3.6 KiB
JavaScript

import fs from "fs";
/**
* ## sqlite与postgres不同点
* 1.
* sqlite: AUTOINCREAMENT
* postgresql: GENERATED BY DEFAULT AS IDENTITY
*
* 2.
* sqlite: datetime
* postgresql: timestamp
*
* 3.
* sqlite: update sqlite_sequence set seq = 1000 where name = 'sys_role' ;
* postgresql: select setval('sys_role_id_seq', 1000);
*
* 4.
* sqlite: "disabled" boolean DEFAULT (0)
* postgresql: "disabled" boolean DEFAULT (false)
*
* 5.
* sqlite: last_insert_rowid()
* postgresql: LASTVAL()
*
* 6.
* sqlite: integer
* postgresql: bigint
*/
function transformPG() {
// 读取文件列表
const sqliteFiles = fs.readdirSync("./migration/");
const pgFiles = fs.readdirSync("./migration-pg/");
//找出pg里面没有的文件
const notFiles = sqliteFiles.filter(file => !pgFiles.includes(file));
for (const notFile of notFiles) {
//开始转换
const sqliteSql = fs.readFileSync(`./migration/${notFile}`, "utf-8");
let pgSql = sqliteSql.replaceAll(/AUTOINCREMENT/g, "GENERATED BY DEFAULT AS IDENTITY");
pgSql = pgSql.replaceAll(/datetime/g, "timestamp");
pgSql = pgSql.replaceAll(/boolean DEFAULT \(0\)/g, "boolean DEFAULT (false)");
pgSql = pgSql.replaceAll(/boolean DEFAULT \(1\)/g, "boolean DEFAULT (true)");
pgSql = pgSql.replaceAll(/boolean.*NOT NULL DEFAULT \(0\)/g, "boolean NOT NULL DEFAULT (false)");
pgSql = pgSql.replaceAll(/boolean.*NOT NULL DEFAULT \(1\)/g, "boolean NOT NULL DEFAULT (true)");
pgSql = pgSql.replaceAll(/integer/g, "bigint");
pgSql = pgSql.replaceAll(/INTEGER/g, "bigint");
pgSql = pgSql.replaceAll(/last_insert_rowid\(\)/g, "LASTVAL()");
fs.writeFileSync(`./migration-pg/${notFile}`, pgSql);
}
if (notFiles.length > 0) {
console.log("sqlite->pg 转换完成");
throw new Error("sqlite->pg 转换完成,有更新,需要测试pg");
} else {
console.log("sql无需更新");
}
}
function transformMysql() {
// 读取文件列表
const sqliteFiles = fs.readdirSync("./migration/");
const pgFiles = fs.readdirSync("./migration-mysql/");
//找出pg里面没有的文件
const notFiles = sqliteFiles.filter(file => !pgFiles.includes(file));
for (const notFile of notFiles) {
//开始转换
const sqliteSql = fs.readFileSync(`./migration/${notFile}`, "utf-8");
let pgSql = sqliteSql.replaceAll(/AUTOINCREMENT/g, "AUTO_INCREMENT");
pgSql = pgSql.replaceAll(/datetime/g, "timestamp");
//DEFAULT (xxx) 替换成 DEFAULT xxx
pgSql = pgSql.replaceAll(/DEFAULT \(([^)]*)\)/g, "DEFAULT $1");
pgSql = pgSql.replaceAll(/integer/g, "bigint");
pgSql = pgSql.replaceAll(/INTEGER/g, "bigint");
pgSql = pgSql.replaceAll(/last_insert_rowid\(\)/g, "LAST_INSERT_ID()");
//text 改成longtext
pgSql = pgSql.replaceAll(/text/g, "longtext");
//双引号 替换成反引号
pgSql = pgSql.replaceAll(/"/g, "`");
//提取所有的 create table 的表格name
const tableNames = pgSql.match(/CREATE TABLE `([^`]*)`/g);
if (tableNames && tableNames.length > 0) {
for (const item of tableNames) {
/**
* CREATE TABLE `cd_project`
CREATE TABLE `cd_project_member`
CREATE TABLE `cd_audit_log`
*/
//提取表名
const tableName = item.match(/`([^`]*)`/)[1];
pgSql += `\nALTER TABLE \`${tableName}\` ENGINE = InnoDB;`;
}
}
fs.writeFileSync(`./migration-mysql/${notFile}`, pgSql);
}
if (notFiles.length > 0) {
console.log("sqlite->mysql 转换完成");
throw new Error("sqlite->mysql 转换完成,有更新,需要测试mysql");
} else {
console.log("sql无需更新");
}
}
transformPG();
transformMysql();