57 lines
2.3 KiB
TypeScript
57 lines
2.3 KiB
TypeScript
/**
|
||
* 一次性迁移：把「当前全库 Ypt 三分位」定格写入每个材料的 health_level 字段。
|
||
*
|
||
* - 公共 + 自建材料合为一个总体计算三分位（与线上分档口径一致）
|
||
* - 数据完整的材料写入 A/B/C；任一 be_area 缺失的材料写入 null（"—"）
|
||
* - 覆盖自建材料原有的手动 health_level，并给公共材料补上
|
||
* - 幂等：可重复运行（每次按当前数据重新定格）
|
||
*
|
||
* 运行（本地）：npx dotenv -e .env.development -- npx ts-node scripts/migrate-health-level.ts
|
||
* 运行（线上 iapips2）：npx dotenv -e .env -- npx ts-node scripts/migrate-health-level.ts
|
||
* （DATABASE_URL 由 dotenv 注入；确认指向目标库后再执行）
|
||
*/
|
||
import { PrismaClient } from '@prisma/client'
|
||
import { computeYpt, buildTierClassifier } from '../src/common/material-ranking'
|
||
|
||
const db = new PrismaClient()
|
||
|
||
const BE_SELECT = {
|
||
methanal_be_area: true,
|
||
tvoc_be_area: true,
|
||
benzene_be_area: true,
|
||
toluene_be_area: true,
|
||
p_xylene_be_area: true
|
||
} as const
|
||
|
||
async function main() {
|
||
// SelfMaterial 主键是复合键 [material_id, creator_id]，需一并取出
|
||
const [pub, self] = await Promise.all([
|
||
db.publicMaterial.findMany({ select: { material_id: true, ...BE_SELECT } }),
|
||
db.selfMaterial.findMany({ select: { material_id: true, creator_id: true, ...BE_SELECT } })
|
||
])
|
||
const classify = buildTierClassifier([...pub, ...self])
|
||
|
||
const dist: Record<string, number> = { A: 0, B: 0, C: 0, 'null': 0 }
|
||
|
||
for (const m of pub) {
|
||
const tier = classify(computeYpt(m))
|
||
dist[tier ?? 'null']++
|
||
await db.publicMaterial.update({ where: { material_id: m.material_id }, data: { health_level: tier } })
|
||
}
|
||
for (const m of self) {
|
||
const tier = classify(computeYpt(m))
|
||
dist[tier ?? 'null']++
|
||
await db.selfMaterial.updateMany({ where: { material_id: m.material_id, creator_id: m.creator_id }, data: { health_level: tier } })
|
||
}
|
||
|
||
console.log(`migrate-health-level done: public=${pub.length}, self=${self.length}`)
|
||
console.log(`分档分布 -> A:${dist.A} B:${dist.B} C:${dist.C} 缺值(null):${dist['null']}`)
|
||
}
|
||
|
||
main()
|
||
.catch(e => {
|
||
console.error('migrate failed:', e)
|
||
process.exitCode = 1
|
||
})
|
||
.finally(() => db.$disconnect())
|