87 lines
3.0 KiB
JavaScript
87 lines
3.0 KiB
JavaScript
/**
|
||
* Migration 004 — PID & Slug 字段
|
||
*
|
||
* - internal_pid: 系统内部编号，如 VPS-000001（唯一，自动生成）
|
||
* - provider_pid: 商家产品 ID，从购买链接自动解析（如 pid=28）
|
||
* - slug: 前台友好 URL，唯一
|
||
*/
|
||
const Database = require('better-sqlite3');
|
||
const path = require('path');
|
||
require('dotenv').config({ path: path.join(__dirname, '..', '.env') });
|
||
|
||
const dbPath = path.resolve(__dirname, '..', process.env.DB_PATH || 'db/monitor.sqlite');
|
||
const db = new Database(dbPath);
|
||
db.pragma('journal_mode = WAL');
|
||
|
||
function ensureColumn(table, column, sql) {
|
||
const cols = db.prepare(`PRAGMA table_info(${table})`).all().map(c => c.name);
|
||
if (!cols.includes(column)) {
|
||
db.exec(`ALTER TABLE ${table} ADD COLUMN ${sql}`);
|
||
console.log(`+ ${table}.${column}`);
|
||
} else {
|
||
console.log(` ${table}.${column} (already exists)`);
|
||
}
|
||
}
|
||
|
||
// 添加新字段
|
||
ensureColumn('products', 'internal_pid', 'internal_pid TEXT');
|
||
ensureColumn('products', 'provider_pid', 'provider_pid TEXT');
|
||
ensureColumn('products', 'slug', 'slug TEXT');
|
||
|
||
// 创建唯一索引（如果尚不存在）
|
||
try {
|
||
db.exec('CREATE UNIQUE INDEX IF NOT EXISTS idx_products_internal_pid ON products(internal_pid) WHERE internal_pid IS NOT NULL');
|
||
console.log('+ index: idx_products_internal_pid');
|
||
} catch (e) {
|
||
console.log(' index idx_products_internal_pid:', e.message);
|
||
}
|
||
try {
|
||
db.exec('CREATE UNIQUE INDEX IF NOT EXISTS idx_products_slug ON products(slug) WHERE slug IS NOT NULL');
|
||
console.log('+ index: idx_products_slug');
|
||
} catch (e) {
|
||
console.log(' index idx_products_slug:', e.message);
|
||
}
|
||
|
||
// 为已有产品回填 internal_pid 和 slug
|
||
const { generateInternalPid, generateSlug, parseProviderPid } = require('../src/utils/pidHelper');
|
||
|
||
const products = db.prepare('SELECT id, name, merchant_id, url, buy_url, internal_pid, slug, provider_pid FROM products').all();
|
||
const updateStmt = db.prepare('UPDATE products SET internal_pid = ?, slug = ?, provider_pid = ? WHERE id = ?');
|
||
|
||
const getMerchantName = db.prepare('SELECT name FROM merchants WHERE id = ?');
|
||
const existingSlugs = new Set(
|
||
db.prepare("SELECT slug FROM products WHERE slug IS NOT NULL AND slug != ''").all().map(r => r.slug)
|
||
);
|
||
|
||
db.transaction(() => {
|
||
for (const p of products) {
|
||
let ipid = p.internal_pid;
|
||
let slug = p.slug;
|
||
let ppid = p.provider_pid;
|
||
|
||
// 生成 internal_pid
|
||
if (!ipid) {
|
||
ipid = generateInternalPid(db);
|
||
}
|
||
|
||
// 生成 slug
|
||
if (!slug) {
|
||
const merchant = getMerchantName.get(p.merchant_id);
|
||
const merchantName = merchant ? merchant.name : '';
|
||
slug = generateSlug(p.name, merchantName, existingSlugs);
|
||
existingSlugs.add(slug);
|
||
}
|
||
|
||
// 解析 provider_pid
|
||
if (!ppid) {
|
||
ppid = parseProviderPid(p.buy_url) || parseProviderPid(p.url) || null;
|
||
}
|
||
|
||
updateStmt.run(ipid, slug, ppid, p.id);
|
||
console.log(` product #${p.id}: internal_pid=${ipid}, slug=${slug}, provider_pid=${ppid || '(none)'}`);
|
||
}
|
||
})();
|
||
|
||
console.log('✅ migration-004 done:', dbPath);
|
||
db.close();
|