drop-lobster-summary-config.js 4.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
  1. /**
  2. * Drop unused lobster summary config menus + company_lobster_summary_config table.
  3. * Usage:
  4. * node scripts/drop-lobster-summary-config.js # all tenants + master
  5. * node scripts/drop-lobster-summary-config.js fs_tenant_cs1 # single tenant only
  6. */
  7. const fs = require('fs')
  8. const path = require('path')
  9. const mysql = require('mysql2/promise')
  10. const DB_CFG = {
  11. host: process.env.DB_HOST || 'cq-cdb-8fjmemkb.sql.tencentcdb.com',
  12. port: Number(process.env.DB_PORT || 27220),
  13. user: process.env.DB_USER || 'root',
  14. password: process.env.DB_PASSWORD || 'Ylrz_1q2w3e4r5t6y',
  15. multipleStatements: true,
  16. charset: 'utf8mb4'
  17. }
  18. const MASTER_DB = process.env.DB_NAME || 'ylrz_saas'
  19. const ROOT = path.join(__dirname, '../../java')
  20. const MASTER_SQL = fs.readFileSync(
  21. path.join(ROOT, 'sql/master/V20260626_01__drop_lobster_summary_config.sql'),
  22. 'utf8'
  23. )
  24. const TENANT_SQL = fs.readFileSync(
  25. path.join(ROOT, 'fs-agent/src/main/resources/db/migration/tenant/V20260626_01__drop_lobster_summary_config.sql'),
  26. 'utf8'
  27. )
  28. function dbNameFromUrl(url) {
  29. if (!url) return null
  30. const m = String(url).match(/\/([^/?]+)(?:\?|$)/)
  31. return m ? m[1] : null
  32. }
  33. async function runSql(database, sql, label) {
  34. const conn = await mysql.createConnection({ ...DB_CFG, database })
  35. try {
  36. const statements = sql
  37. .split(';')
  38. .map(s => s.replace(/--[^\n]*/g, '').trim())
  39. .filter(Boolean)
  40. let affected = 0
  41. for (const stmt of statements) {
  42. if (!(await canRun(conn, stmt))) continue
  43. const [r] = await conn.query(stmt)
  44. affected += r && r.affectedRows ? r.affectedRows : 0
  45. }
  46. const [menus] = await conn.query(
  47. "SELECT COUNT(*) AS c FROM information_schema.tables WHERE table_schema = ? AND table_name IN ('sys_menu','company_menu','tenant_sys_menu','tenant_company_menu')",
  48. [database]
  49. )
  50. let summaryMenu = 0
  51. if (menus[0].c > 0) {
  52. const checks = []
  53. if (await tableExists(conn, 'sys_menu')) {
  54. checks.push("SELECT COUNT(*) AS c FROM sys_menu WHERE menu_id IN (29923,2999) OR component='lobster/summary-config/index'")
  55. }
  56. if (await tableExists(conn, 'company_menu')) {
  57. checks.push("SELECT COUNT(*) AS c FROM company_menu WHERE menu_id IN (29923,2999) OR component='lobster/summary-config/index'")
  58. }
  59. if (await tableExists(conn, 'tenant_sys_menu')) {
  60. checks.push("SELECT COUNT(*) AS c FROM tenant_sys_menu WHERE menu_id IN (29923,2999) OR component='lobster/summary-config/index'")
  61. }
  62. if (await tableExists(conn, 'tenant_company_menu')) {
  63. checks.push("SELECT COUNT(*) AS c FROM tenant_company_menu WHERE menu_id IN (29923,2999) OR component='lobster/summary-config/index'")
  64. }
  65. for (const q of checks) {
  66. const [r] = await conn.query(q)
  67. summaryMenu += Number(r[0].c || 0)
  68. }
  69. }
  70. const [tbl] = await conn.query(
  71. "SELECT COUNT(*) AS c FROM information_schema.tables WHERE table_schema = ? AND table_name = 'company_lobster_summary_config'",
  72. [database]
  73. )
  74. console.log(`[${label}] ${database}: affected=${affected}, summary_menu_left=${summaryMenu}, summary_table=${Number(tbl[0].c)}`)
  75. } finally {
  76. await conn.end()
  77. }
  78. }
  79. async function tableExists(conn, table) {
  80. const [r] = await conn.query('SHOW TABLES LIKE ?', [table])
  81. return r.length > 0
  82. }
  83. async function canRun(conn, stmt) {
  84. const m = stmt.match(/^DELETE\s+FROM\s+([a-zA-Z0-9_]+)/i)
  85. if (!m) return true
  86. return tableExists(conn, m[1])
  87. }
  88. async function listTenantDbs() {
  89. const conn = await mysql.createConnection({ ...DB_CFG, database: MASTER_DB })
  90. try {
  91. const [rows] = await conn.query(
  92. "SELECT id, tenant_code, db_url FROM tenant_info WHERE db_url IS NOT NULL AND db_url <> ''"
  93. )
  94. const dbs = new Map()
  95. for (const t of rows) {
  96. const name = dbNameFromUrl(t.db_url)
  97. if (name) dbs.set(name, `${t.id}:${t.tenant_code}`)
  98. }
  99. return dbs
  100. } finally {
  101. await conn.end()
  102. }
  103. }
  104. async function main() {
  105. const onlyTenant = process.argv[2]
  106. if (!onlyTenant) {
  107. console.log('=== master ===')
  108. await runSql(MASTER_DB, MASTER_SQL, 'master')
  109. }
  110. const tenantDbs = await listTenantDbs()
  111. const targets = onlyTenant
  112. ? [[onlyTenant, onlyTenant]]
  113. : [...tenantDbs.entries()]
  114. console.log('=== tenants === count', targets.length)
  115. for (const [db, label] of targets) {
  116. try {
  117. await runSql(db, TENANT_SQL, label)
  118. } catch (e) {
  119. console.error(`[FAIL] ${db}: ${e.message}`)
  120. }
  121. }
  122. console.log('Done.')
  123. }
  124. main().catch(e => {
  125. console.error(e)
  126. process.exit(1)
  127. })