74 lines
2.5 KiB
JavaScript
74 lines
2.5 KiB
JavaScript
/**
|
||
* 为 biz_plans / biz_usage_monthly 补齐套餐 API 权限相关字段（可重复执行）
|
||
*/
|
||
const mysql = require("mysql2/promise");
|
||
const config = require("../config/config");
|
||
|
||
async function column_exists(conn, schema, table, column) {
|
||
const [rows] = await conn.query(
|
||
`SELECT COUNT(*) AS c FROM information_schema.COLUMNS
|
||
WHERE TABLE_SCHEMA = ? AND TABLE_NAME = ? AND COLUMN_NAME = ?`,
|
||
[schema, table, column]
|
||
);
|
||
return Number(rows[0].c) > 0;
|
||
}
|
||
|
||
async function table_exists(conn, schema, table) {
|
||
const [rows] = await conn.query(
|
||
`SELECT COUNT(*) AS c FROM information_schema.TABLES
|
||
WHERE TABLE_SCHEMA = ? AND TABLE_NAME = ?`,
|
||
[schema, table]
|
||
);
|
||
return Number(rows[0].c) > 0;
|
||
}
|
||
|
||
async function main() {
|
||
const db = config.db;
|
||
if (!db || !db.database) {
|
||
throw new Error("config.db.database 未配置");
|
||
}
|
||
const conn = await mysql.createConnection({
|
||
host: db.host,
|
||
port: db.port || 3306,
|
||
user: db.username,
|
||
password: db.password,
|
||
database: db.database,
|
||
});
|
||
const schema = db.database;
|
||
try {
|
||
if (!(await table_exists(conn, schema, "biz_plans"))) {
|
||
console.warn("跳过 biz_plans：当前库中不存在该表（请确认连接的是已部署订阅模块的库）");
|
||
} else if (!(await column_exists(conn, schema, "biz_plans", "allowed_apis"))) {
|
||
await conn.query(
|
||
"ALTER TABLE `biz_plans` ADD COLUMN `allowed_apis` JSON DEFAULT NULL COMMENT '可访问的接口路径列表，null=不限制'"
|
||
);
|
||
console.log("已添加 biz_plans.allowed_apis");
|
||
}
|
||
if (
|
||
(await table_exists(conn, schema, "biz_plans")) &&
|
||
!(await column_exists(conn, schema, "biz_plans", "api_call_quota"))
|
||
) {
|
||
await conn.query(
|
||
"ALTER TABLE `biz_plans` ADD COLUMN `api_call_quota` INT NOT NULL DEFAULT 0 COMMENT '每月API总调用次数上限，0=不限制'"
|
||
);
|
||
console.log("已添加 biz_plans.api_call_quota");
|
||
}
|
||
if (!(await table_exists(conn, schema, "biz_usage_monthly"))) {
|
||
console.warn("跳过 biz_usage_monthly：当前库中不存在该表");
|
||
} else if (!(await column_exists(conn, schema, "biz_usage_monthly", "api_call_count"))) {
|
||
await conn.query(
|
||
"ALTER TABLE `biz_usage_monthly` ADD COLUMN `api_call_count` INT NOT NULL DEFAULT 0 COMMENT '当月API转发总调用次数'"
|
||
);
|
||
console.log("已添加 biz_usage_monthly.api_call_count");
|
||
}
|
||
console.log("字段检查完成");
|
||
} finally {
|
||
await conn.end();
|
||
}
|
||
}
|
||
|
||
main().catch((e) => {
|
||
console.error(e);
|
||
process.exit(1);
|
||
});
|