fdb2 1.0.13 → 1.0.15

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (313) hide show
  1. package/README.md +9 -2
  2. package/bin/docker/.env +4 -0
  3. package/package.json +26 -18
  4. package/public/favicon.ico +0 -0
  5. package/public/favicon.png +0 -0
  6. package/view/modules/header.tpl +1 -1
  7. package/.dockerignore +0 -21
  8. package/.editorconfig +0 -11
  9. package/.eslintrc.cjs +0 -14
  10. package/.eslintrc.json +0 -7
  11. package/.prettierrc.js +0 -3
  12. package/.tpl.env +0 -22
  13. package/.vscodeignore +0 -45
  14. package/dist/package.json +0 -115
  15. package/dist/pnpm-lock.yaml +0 -8135
  16. package/dist/public/.vite/manifest.json +0 -82
  17. package/dist/public/bootstrap-icons.woff +0 -0
  18. package/dist/public/bootstrap-icons.woff2 +0 -0
  19. package/dist/public/bootstrap.css +0 -14152
  20. package/dist/public/bootstrap.js +0 -5038
  21. package/dist/public/explorer.css +0 -2379
  22. package/dist/public/explorer.js +0 -50586
  23. package/dist/public/index.css +0 -1099
  24. package/dist/public/index.js +0 -12862
  25. package/dist/public/layout.css +0 -318
  26. package/dist/public/layout.js +0 -25
  27. package/dist/public/vue.css +0 -1
  28. package/dist/public/vue.js +0 -9110
  29. package/dist/scripts/preinstall.js +0 -112
  30. package/dist/server/index.d.ts +0 -2
  31. package/dist/server/index.d.ts.map +0 -1
  32. package/dist/server/index.js +0 -601
  33. package/dist/server/index.js.map +0 -1
  34. package/dist/server/index.ts +0 -681
  35. package/dist/server/model/connection.entity.d.ts +0 -55
  36. package/dist/server/model/connection.entity.d.ts.map +0 -1
  37. package/dist/server/model/connection.entity.js +0 -59
  38. package/dist/server/model/connection.entity.js.map +0 -1
  39. package/dist/server/model/connection.entity.ts +0 -66
  40. package/dist/server/model/database.entity.d.ts +0 -203
  41. package/dist/server/model/database.entity.d.ts.map +0 -1
  42. package/dist/server/model/database.entity.js +0 -211
  43. package/dist/server/model/database.entity.js.map +0 -1
  44. package/dist/server/model/database.entity.ts +0 -246
  45. package/dist/server/service/connection.service.d.ts +0 -84
  46. package/dist/server/service/connection.service.d.ts.map +0 -1
  47. package/dist/server/service/connection.service.js +0 -369
  48. package/dist/server/service/connection.service.js.map +0 -1
  49. package/dist/server/service/connection.service.ts +0 -359
  50. package/dist/server/service/database/base.service.d.ts +0 -183
  51. package/dist/server/service/database/base.service.d.ts.map +0 -1
  52. package/dist/server/service/database/base.service.js +0 -253
  53. package/dist/server/service/database/base.service.js.map +0 -1
  54. package/dist/server/service/database/base.service.ts +0 -407
  55. package/dist/server/service/database/cockroachdb.service.d.ts +0 -116
  56. package/dist/server/service/database/cockroachdb.service.d.ts.map +0 -1
  57. package/dist/server/service/database/cockroachdb.service.js +0 -812
  58. package/dist/server/service/database/cockroachdb.service.js.map +0 -1
  59. package/dist/server/service/database/cockroachdb.service.ts +0 -871
  60. package/dist/server/service/database/database.service.d.ts +0 -495
  61. package/dist/server/service/database/database.service.d.ts.map +0 -1
  62. package/dist/server/service/database/database.service.js +0 -711
  63. package/dist/server/service/database/database.service.js.map +0 -1
  64. package/dist/server/service/database/database.service.ts +0 -776
  65. package/dist/server/service/database/index.d.ts +0 -8
  66. package/dist/server/service/database/index.d.ts.map +0 -1
  67. package/dist/server/service/database/index.js +0 -18
  68. package/dist/server/service/database/index.js.map +0 -1
  69. package/dist/server/service/database/index.ts +0 -7
  70. package/dist/server/service/database/mongodb.service.d.ts +0 -121
  71. package/dist/server/service/database/mongodb.service.d.ts.map +0 -1
  72. package/dist/server/service/database/mongodb.service.js +0 -502
  73. package/dist/server/service/database/mongodb.service.js.map +0 -1
  74. package/dist/server/service/database/mongodb.service.ts +0 -501
  75. package/dist/server/service/database/mssql.service.d.ts +0 -118
  76. package/dist/server/service/database/mssql.service.d.ts.map +0 -1
  77. package/dist/server/service/database/mssql.service.js +0 -871
  78. package/dist/server/service/database/mssql.service.js.map +0 -1
  79. package/dist/server/service/database/mssql.service.ts +0 -932
  80. package/dist/server/service/database/mysql.service.d.ts +0 -114
  81. package/dist/server/service/database/mysql.service.d.ts.map +0 -1
  82. package/dist/server/service/database/mysql.service.js +0 -960
  83. package/dist/server/service/database/mysql.service.js.map +0 -1
  84. package/dist/server/service/database/mysql.service.ts +0 -1026
  85. package/dist/server/service/database/oracle.service.d.ts +0 -126
  86. package/dist/server/service/database/oracle.service.d.ts.map +0 -1
  87. package/dist/server/service/database/oracle.service.js +0 -963
  88. package/dist/server/service/database/oracle.service.js.map +0 -1
  89. package/dist/server/service/database/oracle.service.ts +0 -1036
  90. package/dist/server/service/database/postgres.service.d.ts +0 -122
  91. package/dist/server/service/database/postgres.service.d.ts.map +0 -1
  92. package/dist/server/service/database/postgres.service.js +0 -882
  93. package/dist/server/service/database/postgres.service.js.map +0 -1
  94. package/dist/server/service/database/postgres.service.ts +0 -961
  95. package/dist/server/service/database/sap.service.d.ts +0 -115
  96. package/dist/server/service/database/sap.service.d.ts.map +0 -1
  97. package/dist/server/service/database/sap.service.js +0 -868
  98. package/dist/server/service/database/sap.service.js.map +0 -1
  99. package/dist/server/service/database/sap.service.ts +0 -922
  100. package/dist/server/service/database/sqlite.service.d.ts +0 -112
  101. package/dist/server/service/database/sqlite.service.d.ts.map +0 -1
  102. package/dist/server/service/database/sqlite.service.js +0 -723
  103. package/dist/server/service/database/sqlite.service.js.map +0 -1
  104. package/dist/server/service/database/sqlite.service.ts +0 -787
  105. package/dist/server/service/session.service.ts +0 -158
  106. package/dist/view/index.html +0 -45
  107. package/env.d.ts +0 -1
  108. package/packages/vscode/.vscodeignore +0 -44
  109. package/packages/vscode/README.md +0 -62
  110. package/packages/vscode/out/database-services/base.service.js +0 -236
  111. package/packages/vscode/out/database-services/base.service.js.map +0 -1
  112. package/packages/vscode/out/database-services/cockroachdb.service.js +0 -634
  113. package/packages/vscode/out/database-services/cockroachdb.service.js.map +0 -1
  114. package/packages/vscode/out/database-services/connection.service.js +0 -346
  115. package/packages/vscode/out/database-services/connection.service.js.map +0 -1
  116. package/packages/vscode/out/database-services/database.service.js +0 -571
  117. package/packages/vscode/out/database-services/database.service.js.map +0 -1
  118. package/packages/vscode/out/database-services/index.js +0 -18
  119. package/packages/vscode/out/database-services/index.js.map +0 -1
  120. package/packages/vscode/out/database-services/model/connection.entity.js +0 -11
  121. package/packages/vscode/out/database-services/model/connection.entity.js.map +0 -1
  122. package/packages/vscode/out/database-services/model/database.entity.js +0 -35
  123. package/packages/vscode/out/database-services/model/database.entity.js.map +0 -1
  124. package/packages/vscode/out/database-services/mongodb.service.js +0 -458
  125. package/packages/vscode/out/database-services/mongodb.service.js.map +0 -1
  126. package/packages/vscode/out/database-services/mssql.service.js +0 -694
  127. package/packages/vscode/out/database-services/mssql.service.js.map +0 -1
  128. package/packages/vscode/out/database-services/mysql.service.js +0 -735
  129. package/packages/vscode/out/database-services/mysql.service.js.map +0 -1
  130. package/packages/vscode/out/database-services/oracle.service.js +0 -787
  131. package/packages/vscode/out/database-services/oracle.service.js.map +0 -1
  132. package/packages/vscode/out/database-services/postgres.service.js +0 -696
  133. package/packages/vscode/out/database-services/postgres.service.js.map +0 -1
  134. package/packages/vscode/out/database-services/sap.service.js +0 -695
  135. package/packages/vscode/out/database-services/sap.service.js.map +0 -1
  136. package/packages/vscode/out/database-services/sqlite.service.js +0 -532
  137. package/packages/vscode/out/database-services/sqlite.service.js.map +0 -1
  138. package/packages/vscode/out/extension.js +0 -93
  139. package/packages/vscode/out/extension.js.map +0 -1
  140. package/packages/vscode/out/provider/DatabaseTreeProvider.js +0 -159
  141. package/packages/vscode/out/provider/DatabaseTreeProvider.js.map +0 -1
  142. package/packages/vscode/out/provider/WebViewProvider.js +0 -259
  143. package/packages/vscode/out/provider/WebViewProvider.js.map +0 -1
  144. package/packages/vscode/out/service/ConnectionManager.js +0 -105
  145. package/packages/vscode/out/service/ConnectionManager.js.map +0 -1
  146. package/packages/vscode/out/service/DatabaseServiceBridge.js +0 -395
  147. package/packages/vscode/out/service/DatabaseServiceBridge.js.map +0 -1
  148. package/packages/vscode/out/typings/connection.js +0 -3
  149. package/packages/vscode/out/typings/connection.js.map +0 -1
  150. package/packages/vscode/package.json +0 -144
  151. package/packages/vscode/resources/icon.svg +0 -5
  152. package/packages/vscode/resources/webview/_plugin-vue_export-helper.js +0 -6529
  153. package/packages/vscode/resources/webview/_plugin-vue_export-helper.js.map +0 -1
  154. package/packages/vscode/resources/webview/connection.css +0 -69
  155. package/packages/vscode/resources/webview/connection.js +0 -228
  156. package/packages/vscode/resources/webview/connection.js.map +0 -1
  157. package/packages/vscode/resources/webview/database.css +0 -259
  158. package/packages/vscode/resources/webview/database.js +0 -275
  159. package/packages/vscode/resources/webview/database.js.map +0 -1
  160. package/packages/vscode/resources/webview/favicon.ico +0 -0
  161. package/packages/vscode/resources/webview/index.html +0 -9
  162. package/packages/vscode/resources/webview/modules/header.tpl +0 -14
  163. package/packages/vscode/resources/webview/modules/initial_state.tpl +0 -55
  164. package/packages/vscode/resources/webview/query.css +0 -162
  165. package/packages/vscode/resources/webview/query.js +0 -198
  166. package/packages/vscode/resources/webview/query.js.map +0 -1
  167. package/packages/vscode/src/database-services/base.service.js.map +0 -1
  168. package/packages/vscode/src/database-services/base.service.ts +0 -363
  169. package/packages/vscode/src/database-services/cockroachdb.service.js.map +0 -1
  170. package/packages/vscode/src/database-services/cockroachdb.service.ts +0 -659
  171. package/packages/vscode/src/database-services/connection.service.ts +0 -341
  172. package/packages/vscode/src/database-services/database.service.ts +0 -630
  173. package/packages/vscode/src/database-services/index.ts +0 -7
  174. package/packages/vscode/src/database-services/model/connection.entity.js +0 -11
  175. package/packages/vscode/src/database-services/model/connection.entity.js.map +0 -1
  176. package/packages/vscode/src/database-services/model/connection.entity.ts +0 -66
  177. package/packages/vscode/src/database-services/model/database.entity.js +0 -35
  178. package/packages/vscode/src/database-services/model/database.entity.js.map +0 -1
  179. package/packages/vscode/src/database-services/model/database.entity.ts +0 -246
  180. package/packages/vscode/src/database-services/mongodb.service.js.map +0 -1
  181. package/packages/vscode/src/database-services/mongodb.service.ts +0 -454
  182. package/packages/vscode/src/database-services/mssql.service.js.map +0 -1
  183. package/packages/vscode/src/database-services/mssql.service.ts +0 -723
  184. package/packages/vscode/src/database-services/mysql.service.js.map +0 -1
  185. package/packages/vscode/src/database-services/mysql.service.ts +0 -761
  186. package/packages/vscode/src/database-services/oracle.service.js.map +0 -1
  187. package/packages/vscode/src/database-services/oracle.service.ts +0 -832
  188. package/packages/vscode/src/database-services/postgres.service.js.map +0 -1
  189. package/packages/vscode/src/database-services/postgres.service.ts +0 -741
  190. package/packages/vscode/src/database-services/sap.service.js.map +0 -1
  191. package/packages/vscode/src/database-services/sap.service.ts +0 -713
  192. package/packages/vscode/src/database-services/sqlite.service.js.map +0 -1
  193. package/packages/vscode/src/database-services/sqlite.service.ts +0 -558
  194. package/packages/vscode/src/extension.ts +0 -76
  195. package/packages/vscode/src/provider/DatabaseTreeProvider.ts +0 -167
  196. package/packages/vscode/src/provider/WebViewProvider.ts +0 -277
  197. package/packages/vscode/src/service/DatabaseServiceBridge.ts +0 -414
  198. package/packages/vscode/src/typings/connection.ts +0 -90
  199. package/packages/vscode/tsconfig.json +0 -21
  200. package/scripts/preinstall.js +0 -112
  201. package/server/index.ts +0 -681
  202. package/server/model/connection.entity.js +0 -11
  203. package/server/model/connection.entity.js.map +0 -1
  204. package/server/model/connection.entity.ts +0 -66
  205. package/server/model/database.entity.js +0 -35
  206. package/server/model/database.entity.js.map +0 -1
  207. package/server/model/database.entity.ts +0 -246
  208. package/server/service/connection.service.ts +0 -359
  209. package/server/service/database/base.service.ts +0 -407
  210. package/server/service/database/cockroachdb.service.ts +0 -871
  211. package/server/service/database/database.service.ts +0 -776
  212. package/server/service/database/index.ts +0 -7
  213. package/server/service/database/mongodb.service.ts +0 -501
  214. package/server/service/database/mssql.service.ts +0 -932
  215. package/server/service/database/mysql.service.ts +0 -1026
  216. package/server/service/database/oracle.service.ts +0 -1036
  217. package/server/service/database/postgres.service.ts +0 -961
  218. package/server/service/database/sap.service.ts +0 -922
  219. package/server/service/database/sqlite.service.ts +0 -787
  220. package/server/service/session.service.ts +0 -158
  221. package/server/tsconfig.json +0 -20
  222. package/src/adapter/ajax.ts +0 -135
  223. package/src/assets/base.css +0 -1
  224. package/src/assets/database.css +0 -950
  225. package/src/assets/images/collapse.png +0 -0
  226. package/src/assets/images/no-login.png +0 -0
  227. package/src/assets/images/svg/illustrations/illustration-1.svg +0 -1
  228. package/src/assets/images/svg/illustrations/illustration-2.svg +0 -2
  229. package/src/assets/images/svg/illustrations/illustration-3.svg +0 -50
  230. package/src/assets/images/svg/illustrations/illustration-4.svg +0 -1
  231. package/src/assets/images/svg/illustrations/illustration-5.svg +0 -73
  232. package/src/assets/images/svg/illustrations/illustration-6.svg +0 -89
  233. package/src/assets/images/svg/illustrations/illustration-7.svg +0 -39
  234. package/src/assets/images/svg/illustrations/illustration-8.svg +0 -1
  235. package/src/assets/images/svg/separators/curve-2.svg +0 -3
  236. package/src/assets/images/svg/separators/curve.svg +0 -3
  237. package/src/assets/images/svg/separators/line.svg +0 -3
  238. package/src/assets/images/theme/light/screen-1-1000x800.jpg +0 -0
  239. package/src/assets/images/theme/light/screen-2-1000x800.jpg +0 -0
  240. package/src/assets/login/bg.jpg +0 -0
  241. package/src/assets/login/bg.png +0 -0
  242. package/src/assets/login/left.jpg +0 -0
  243. package/src/assets/logo.svg +0 -73
  244. package/src/assets/logo.webp +0 -0
  245. package/src/assets/main.css +0 -1
  246. package/src/base/config.ts +0 -20
  247. package/src/base/detect.ts +0 -134
  248. package/src/base/entity.ts +0 -92
  249. package/src/base/eventBus.ts +0 -37
  250. package/src/components/connection-editor/index.vue +0 -589
  251. package/src/components/dataGrid/index.vue +0 -163
  252. package/src/components/dataGrid/pagination.vue +0 -106
  253. package/src/components/loading/index.vue +0 -43
  254. package/src/components/modal/index.ts +0 -181
  255. package/src/components/modal/index.vue +0 -560
  256. package/src/components/toast/index.ts +0 -44
  257. package/src/components/toast/toast.vue +0 -58
  258. package/src/components/user/name.vue +0 -104
  259. package/src/components/user/selector.vue +0 -416
  260. package/src/domain/SysConfig.ts +0 -74
  261. package/src/platform/App.vue +0 -8
  262. package/src/platform/database/components/connection-detail.vue +0 -1153
  263. package/src/platform/database/components/data-editor.vue +0 -478
  264. package/src/platform/database/components/data-import-export.vue +0 -1602
  265. package/src/platform/database/components/database-detail.vue +0 -1199
  266. package/src/platform/database/components/database-monitor.vue +0 -1086
  267. package/src/platform/database/components/db-tools.vue +0 -1265
  268. package/src/platform/database/components/query-history.vue +0 -1349
  269. package/src/platform/database/components/sql-executor.vue +0 -738
  270. package/src/platform/database/components/sql-query-editor.vue +0 -1046
  271. package/src/platform/database/components/table-data-grid.vue +0 -273
  272. package/src/platform/database/components/table-detail.vue +0 -1173
  273. package/src/platform/database/components/table-editor.vue +0 -917
  274. package/src/platform/database/explorer.vue +0 -1840
  275. package/src/platform/database/index.vue +0 -1193
  276. package/src/platform/database/layout.vue +0 -367
  277. package/src/platform/database/router.ts +0 -37
  278. package/src/platform/database/styles/common.scss +0 -602
  279. package/src/platform/database/types/common.ts +0 -445
  280. package/src/platform/database/utils/export.ts +0 -232
  281. package/src/platform/database/utils/helpers.ts +0 -437
  282. package/src/platform/index.ts +0 -33
  283. package/src/platform/router.ts +0 -41
  284. package/src/platform/vscode/bridge.ts +0 -121
  285. package/src/platform/vscode/components/ConnectionPanel.vue +0 -272
  286. package/src/platform/vscode/components/DatabasePanel.vue +0 -532
  287. package/src/platform/vscode/components/QueryPanel.vue +0 -371
  288. package/src/platform/vscode/entry/connection.ts +0 -13
  289. package/src/platform/vscode/entry/database.ts +0 -13
  290. package/src/platform/vscode/entry/query.ts +0 -13
  291. package/src/platform/vscode/index.ts +0 -5
  292. package/src/service/base.ts +0 -134
  293. package/src/service/database.ts +0 -506
  294. package/src/service/login.ts +0 -121
  295. package/src/shims-vue.d.ts +0 -7
  296. package/src/stores/connection.ts +0 -266
  297. package/src/stores/session.ts +0 -87
  298. package/src/typings/database-types.ts +0 -413
  299. package/src/typings/database.ts +0 -364
  300. package/src/typings/global.d.ts +0 -58
  301. package/src/typings/pinia.d.ts +0 -8
  302. package/src/utils/clipboard.ts +0 -30
  303. package/src/utils/database-types.ts +0 -243
  304. package/src/utils/modal.ts +0 -124
  305. package/src/utils/request.ts +0 -55
  306. package/src/utils/sleep.ts +0 -4
  307. package/src/utils/toast.ts +0 -73
  308. package/src/utils/util.ts +0 -171
  309. package/src/utils/xlsx.ts +0 -228
  310. package/tsconfig.json +0 -33
  311. package/tsconfig.server.json +0 -19
  312. package/vite.config.ts +0 -424
  313. package/vite.config.vscode.ts +0 -47
@@ -1,1026 +0,0 @@
1
- import { DataSource } from 'typeorm';
2
- import * as fs from 'fs';
3
- import * as path from 'path';
4
- import { execSync } from 'child_process';
5
- import { BaseDatabaseService } from './base.service';
6
- import {
7
- TableEntity,
8
- ColumnEntity,
9
- IndexEntity,
10
- ForeignKeyEntity
11
- } from '../../model/database.entity';
12
-
13
- /**
14
- * MySQL数据库服务实现
15
- */
16
- export class MySQLService extends BaseDatabaseService {
17
-
18
- getDatabaseType(): string {
19
- return 'mysql';
20
- }
21
-
22
- /**
23
- * 导出表数据到 CSV 文件
24
- */
25
- async exportTableDataToCSV(dataSource: DataSource, databaseName: string, tableName: string, options?: any): Promise<string> {
26
- try {
27
- const exportPath = options?.path || path.join(__dirname, '..', '..', '..', 'data', 'exports');
28
- if (!fs.existsSync(exportPath)) {
29
- fs.mkdirSync(exportPath, { recursive: true });
30
- }
31
- const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
32
- const exportFile = path.join(exportPath, `${tableName}_data_${timestamp}.csv`);
33
-
34
- const columns = await this.getColumns(dataSource, databaseName, tableName);
35
- const columnNames = columns.map(column => column.name);
36
-
37
- // 写入 CSV 头部
38
- const header = columnNames.join(',') + '\n';
39
- fs.writeFileSync(exportFile, '\ufeff' + header, 'utf8');
40
-
41
- // 分批处理数据
42
- const batchSize = options?.batchSize || 10000;
43
- let offset = 0;
44
- let hasMoreData = true;
45
-
46
- while (hasMoreData) {
47
- const query = `SELECT * FROM ${this.quoteIdentifier(tableName)} LIMIT ${batchSize} OFFSET ${offset}`;
48
- const data = await dataSource.query(query);
49
-
50
- if (data.length === 0) {
51
- hasMoreData = false;
52
- break;
53
- }
54
-
55
- // 生成当前批次的 CSV 行
56
- let batchContent = '';
57
- data.forEach((row: any) => {
58
- const values = columnNames.map(column => {
59
- const value = row[column];
60
- if (value === null || value === undefined) return '';
61
- if (typeof value === 'string') {
62
- const stringValue = String(value);
63
- if (stringValue.includes(',') || stringValue.includes('"') || stringValue.includes('\n')) {
64
- return `"${stringValue.replace(/"/g, '""')}"`;
65
- }
66
- return stringValue;
67
- }
68
- if (typeof value === 'object') {
69
- try {
70
- const stringValue = JSON.stringify(value);
71
- if (stringValue.includes(',') || stringValue.includes('"') || stringValue.includes('\n')) {
72
- return `"${stringValue.replace(/"/g, '""')}"`;
73
- }
74
- return stringValue;
75
- } catch {
76
- return `"${String(value).replace(/"/g, '""')}"`;
77
- }
78
- }
79
- return String(value);
80
- });
81
- batchContent += values.join(',') + '\n';
82
- });
83
-
84
- fs.appendFileSync(exportFile, batchContent, 'utf8');
85
- offset += batchSize;
86
- console.log(`MySQL导出CSV进度: ${tableName} - 已处理 ${offset} 行`);
87
- }
88
-
89
- return exportFile;
90
- } catch (error) {
91
- console.error('MySQL导出CSV失败:', error);
92
- throw new Error(`导出CSV失败: ${error.message}`);
93
- }
94
- }
95
-
96
- /**
97
- * 导出表数据到 JSON 文件
98
- */
99
- async exportTableDataToJSON(dataSource: DataSource, databaseName: string, tableName: string, options?: any): Promise<string> {
100
- try {
101
- const exportPath = options?.path || path.join(__dirname, '..', '..', '..', 'data', 'exports');
102
- if (!fs.existsSync(exportPath)) {
103
- fs.mkdirSync(exportPath, { recursive: true });
104
- }
105
- const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
106
- const exportFile = path.join(exportPath, `${tableName}_data_${timestamp}.json`);
107
-
108
- const columns = await this.getColumns(dataSource, databaseName, tableName);
109
- const columnNames = columns.map(column => column.name);
110
-
111
- // 分批处理数据,生成 JSON 数组
112
- const batchSize = options?.batchSize || 10000;
113
- let offset = 0;
114
- let hasMoreData = true;
115
- let isFirstBatch = true;
116
-
117
- // 写入 JSON 开始
118
- fs.writeFileSync(exportFile, '[', 'utf8');
119
-
120
- while (hasMoreData) {
121
- const query = `SELECT * FROM ${this.quoteIdentifier(tableName)} LIMIT ${batchSize} OFFSET ${offset}`;
122
- const data = await dataSource.query(query);
123
-
124
- if (data.length === 0) {
125
- hasMoreData = false;
126
- break;
127
- }
128
-
129
- // 生成当前批次的 JSON 数据
130
- let batchContent = '';
131
- data.forEach((row: any, index: number) => {
132
- if (!isFirstBatch || index > 0) {
133
- batchContent += ',';
134
- }
135
- // 处理每个字段,避免对已经是JSON字符串的值进行双重序列化
136
- const processedRow: any = {};
137
- for (const key in row) {
138
- const value = row[key];
139
- if (typeof value === 'string') {
140
- // 检查是否是JSON字符串
141
- try {
142
- const parsed = JSON.parse(value);
143
- // 如果解析成功且是对象或数组,说明是JSON字符串,直接使用
144
- if (typeof parsed === 'object' && parsed !== null) {
145
- processedRow[key] = parsed;
146
- } else {
147
- processedRow[key] = value;
148
- }
149
- } catch {
150
- // 不是JSON字符串,直接使用
151
- processedRow[key] = value;
152
- }
153
- } else {
154
- processedRow[key] = value;
155
- }
156
- }
157
- batchContent += JSON.stringify(processedRow);
158
- });
159
-
160
- fs.appendFileSync(exportFile, batchContent, 'utf8');
161
- isFirstBatch = false;
162
- offset += batchSize;
163
- console.log(`MySQL导出JSON进度: ${tableName} - 已处理 ${offset} 行`);
164
- }
165
-
166
- // 写入 JSON 结束
167
- fs.appendFileSync(exportFile, ']', 'utf8');
168
-
169
- return exportFile;
170
- } catch (error) {
171
- console.error('MySQL导出JSON失败:', error);
172
- throw new Error(`导出JSON失败: ${error.message}`);
173
- }
174
- }
175
-
176
- /**
177
- * 导出表数据到 Excel 文件
178
- */
179
- async exportTableDataToExcel(dataSource: DataSource, databaseName: string, tableName: string, options?: any): Promise<string> {
180
- try {
181
- // 由于 Excel 文件需要特殊处理,这里先导出为 CSV,然后在前端转换为 Excel
182
- // 或者使用专门的库生成 Excel 文件
183
- // 这里暂时实现为导出为 CSV,后续可以根据需要优化
184
- return this.exportTableDataToCSV(dataSource, databaseName, tableName, options);
185
- } catch (error) {
186
- console.error('MySQL导出Excel失败:', error);
187
- throw new Error(`导出Excel失败: ${error.message}`);
188
- }
189
- }
190
-
191
- /**
192
- * 获取MySQL数据库列表
193
- */
194
- async getDatabases(dataSource: DataSource): Promise<string[]> {
195
- const result = await dataSource.query('SHOW DATABASES');
196
- // 过滤掉系统数据库
197
- //const systemDatabases = ['information_schema', 'performance_schema', 'mysql', 'sys'];
198
- return result
199
- .map((row: any) => row.Database)
200
- //.filter((db: string) => !systemDatabases.includes(db));
201
- }
202
-
203
- /**
204
- * 获取MySQL表列表
205
- */
206
- async getTables(dataSource: DataSource, database: string): Promise<TableEntity[]> {
207
- const result = await dataSource.query(`
208
- SELECT
209
- TABLE_NAME as name,
210
- TABLE_TYPE as type,
211
- ENGINE as engine,
212
- TABLE_ROWS as rowCount,
213
- DATA_LENGTH as dataSize,
214
- INDEX_LENGTH as indexSize,
215
- TABLE_COLLATION as collation,
216
- CREATE_TIME as createdAt,
217
- UPDATE_TIME as updatedAt,
218
- TABLE_COMMENT as comment
219
- FROM information_schema.TABLES
220
- WHERE TABLE_SCHEMA = ?
221
- `, [database]);
222
-
223
- return result.map((row: any) => ({
224
- name: row.name,
225
- type: row.type,
226
- engine: row.engine,
227
- rowCount: row.rowCount || 0,
228
- dataSize: row.dataSize || 0,
229
- indexSize: row.indexSize || 0,
230
- collation: row.collation,
231
- createdAt: row.createdAt,
232
- updatedAt: row.updatedAt,
233
- comment: row.comment
234
- }));
235
- }
236
-
237
- /**
238
- * 获取MySQL列信息
239
- */
240
- async getColumns(dataSource: DataSource, database: string, table: string): Promise<ColumnEntity[]> {
241
- // 使用兼容的SQL查询,避免使用某些MySQL版本不支持的NUMERIC_PRECISION和NUMERIC_SCALE
242
- const result = await dataSource.query(`
243
- SELECT
244
- COLUMN_NAME as name,
245
- COLUMN_TYPE as type,
246
- IS_NULLABLE as nullable,
247
- COLUMN_DEFAULT as defaultValue,
248
- COLUMN_KEY as columnKey,
249
- EXTRA as extra,
250
- CHARACTER_MAXIMUM_LENGTH as length,
251
- CHARACTER_SET_NAME as charset,
252
- COLLATION_NAME as collation,
253
- COLUMN_COMMENT as comment
254
- FROM information_schema.COLUMNS
255
- WHERE TABLE_SCHEMA = ? AND TABLE_NAME = ?
256
- `, [database, table]);
257
-
258
- return result.map((row: any) => {
259
- // 从COLUMN_TYPE中解析精度信息
260
- const columnType = row.type || '';
261
- let precision = undefined;
262
- let scale = undefined;
263
-
264
- // 解析DECIMAL(M,D)或NUMERIC(M,D)类型的精度
265
- const decimalMatch = columnType.match(/(DECIMAL|NUMERIC)\s*\(\s*(\d+)\s*,\s*(\d+)\s*\)/i);
266
- if (decimalMatch) {
267
- precision = parseInt(decimalMatch[2]);
268
- scale = parseInt(decimalMatch[3]);
269
- }
270
-
271
- return {
272
- name: row.name,
273
- type: row.type,
274
- nullable: row.nullable === 'YES',
275
- defaultValue: row.defaultValue,
276
- isPrimary: row.columnKey === 'PRI',
277
- isAutoIncrement: row.extra?.includes('auto_increment') || false,
278
- length: row.length,
279
- precision: precision,
280
- scale: scale,
281
- charset: row.charset,
282
- collation: row.collation,
283
- comment: row.comment
284
- };
285
- });
286
- }
287
-
288
- /**
289
- * 获取MySQL索引信息
290
- */
291
- async getIndexes(dataSource: DataSource, database: string, table: string): Promise<IndexEntity[]> {
292
- // 使用更兼容的SQL查询,避免使用保留关键字作为别名
293
- const result = await dataSource.query(`
294
- SELECT
295
- INDEX_NAME as name,
296
- INDEX_TYPE as type,
297
- COLUMN_NAME as columnName
298
- FROM information_schema.STATISTICS
299
- WHERE TABLE_SCHEMA = ? AND TABLE_NAME = ?
300
- ORDER BY INDEX_NAME, SEQ_IN_INDEX
301
- `, [database, table]);
302
-
303
- // 按索引名分组并判断唯一性
304
- const indexMap = new Map<string, IndexEntity>();
305
- result.forEach((row: any) => {
306
- if (!indexMap.has(row.name)) {
307
- // 主键索引是唯一的,其他索引需要通过SHOW INDEX查询确定
308
- let isUnique = false;
309
- if (row.name === 'PRIMARY') {
310
- isUnique = true;
311
- } else {
312
- // 对于非主键索引,使用更简单的方法:检查索引名称是否包含UNIQUE关键字
313
- isUnique = row.name.toUpperCase().includes('UNIQUE') || row.type === 'UNIQUE';
314
- }
315
-
316
- indexMap.set(row.name, {
317
- name: row.name,
318
- type: row.type,
319
- columns: [],
320
- unique: isUnique
321
- });
322
- }
323
- indexMap.get(row.name)!.columns.push(row.columnName);
324
- });
325
-
326
- return Array.from(indexMap.values());
327
- }
328
-
329
- /**
330
- * 获取MySQL外键信息
331
- */
332
- async getForeignKeys(dataSource: DataSource, database: string, table: string): Promise<ForeignKeyEntity[]> {
333
- const result = await dataSource.query(`
334
- SELECT
335
- kcu.CONSTRAINT_NAME as name,
336
- kcu.COLUMN_NAME as columnName,
337
- kcu.REFERENCED_TABLE_NAME as referencedTable,
338
- kcu.REFERENCED_COLUMN_NAME as referencedColumn,
339
- rc.DELETE_RULE as onDelete,
340
- rc.UPDATE_RULE as onUpdate
341
- FROM information_schema.KEY_COLUMN_USAGE kcu
342
- LEFT JOIN information_schema.REFERENTIAL_CONSTRAINTS rc
343
- ON kcu.CONSTRAINT_NAME = rc.CONSTRAINT_NAME
344
- AND kcu.CONSTRAINT_SCHEMA = rc.CONSTRAINT_SCHEMA
345
- WHERE kcu.TABLE_SCHEMA = ?
346
- AND kcu.TABLE_NAME = ?
347
- AND kcu.REFERENCED_TABLE_NAME IS NOT NULL
348
- `, [database, table]);
349
-
350
- return result.map((row: any) => ({
351
- name: row.name,
352
- column: row.columnName,
353
- referencedTable: row.referencedTable,
354
- referencedColumn: row.referencedColumn,
355
- onDelete: row.onDelete || 'NO ACTION',
356
- onUpdate: row.onUpdate || 'NO ACTION'
357
- }));
358
- }
359
-
360
- /**
361
- * 获取MySQL数据库大小
362
- */
363
- async getDatabaseSize(dataSource: DataSource, database: string): Promise<number> {
364
- const result = await dataSource.query(`
365
- SELECT SUM(data_length + index_length) as size
366
- FROM information_schema.tables
367
- WHERE table_schema = ?
368
- `, [database]);
369
- return result[0]?.size || 0;
370
- }
371
-
372
- /**
373
- * MySQL使用反引号标识符
374
- */
375
- public quoteIdentifier(identifier: string): string {
376
- return `\`${identifier}\``;
377
- }
378
-
379
- /**
380
- * 获取MySQL视图列表
381
- */
382
- async getViews(dataSource: DataSource, database: string): Promise<any[]> {
383
- // 完全移除TABLE_COMMENT字段,因为某些MySQL版本中不存在该字段
384
- const result = await dataSource.query(`
385
- SELECT
386
- TABLE_NAME as name,
387
- TABLE_SCHEMA as schemaName
388
- FROM information_schema.VIEWS
389
- WHERE TABLE_SCHEMA = ?
390
- ORDER BY TABLE_NAME
391
- `, [database]);
392
-
393
- return result.map((row: any) => ({
394
- name: row.name,
395
- comment: '',
396
- schemaName: row.schemaName
397
- }));
398
- }
399
-
400
- /**
401
- * 获取MySQL视图定义
402
- */
403
- async getViewDefinition(dataSource: DataSource, database: string, viewName: string): Promise<string> {
404
- const result = await dataSource.query(`
405
- SELECT VIEW_DEFINITION as definition
406
- FROM information_schema.VIEWS
407
- WHERE TABLE_SCHEMA = ?
408
- AND TABLE_NAME = ?
409
- `, [database, viewName]);
410
-
411
- return result[0]?.definition || '';
412
- }
413
-
414
- /**
415
- * 获取MySQL存储过程列表
416
- */
417
- async getProcedures(dataSource: DataSource, database: string): Promise<any[]> {
418
- const result = await dataSource.query(`
419
- SELECT
420
- ROUTINE_NAME as name,
421
- ROUTINE_TYPE as type,
422
- ROUTINE_COMMENT as comment
423
- FROM information_schema.ROUTINES
424
- WHERE ROUTINE_SCHEMA = ?
425
- ORDER BY ROUTINE_NAME
426
- `, [database]);
427
-
428
- return result.map((row: any) => ({
429
- name: row.name,
430
- comment: row.comment || '',
431
- type: row.type,
432
- returnType: '',
433
- language: 'SQL'
434
- }));
435
- }
436
-
437
- /**
438
- * 获取MySQL存储过程定义
439
- */
440
- async getProcedureDefinition(dataSource: DataSource, database: string, procedureName: string): Promise<string> {
441
- const result = await dataSource.query(`
442
- SELECT ROUTINE_DEFINITION as definition
443
- FROM information_schema.ROUTINES
444
- WHERE ROUTINE_SCHEMA = ?
445
- AND ROUTINE_NAME = ?
446
- `, [database, procedureName]);
447
-
448
- return result[0]?.definition || '';
449
- }
450
-
451
- /**
452
- * 创建MySQL数据库
453
- */
454
- async createDatabase(dataSource: DataSource, databaseName: string, options?: any): Promise<void> {
455
- let sql = `CREATE DATABASE ${this.quoteIdentifier(databaseName)}`;
456
-
457
- if (options) {
458
- const clauses = [];
459
-
460
- if (options.charset) {
461
- clauses.push(`CHARACTER SET ${options.charset}`);
462
- }
463
-
464
- if (options.collation) {
465
- clauses.push(`COLLATE ${options.collation}`);
466
- }
467
-
468
- if (clauses.length > 0) {
469
- sql += ' ' + clauses.join(' ');
470
- }
471
- }
472
-
473
- await dataSource.query(sql);
474
- }
475
-
476
- /**
477
- * 删除MySQL数据库
478
- */
479
- async dropDatabase(dataSource: DataSource, databaseName: string): Promise<void> {
480
- const sql = `DROP DATABASE ${this.quoteIdentifier(databaseName)}`;
481
- await dataSource.query(sql);
482
- }
483
-
484
- /**
485
- * 导出数据库架构
486
- */
487
- async exportSchema(dataSource: DataSource, databaseName: string): Promise<string> {
488
- // 获取所有表
489
- const tables = await this.getTables(dataSource, databaseName);
490
- let schemaSql = `-- 数据库架构导出 - ${databaseName}\n`;
491
- schemaSql += `-- 导出时间: ${new Date().toISOString()}\n\n`;
492
-
493
- // 为每个表生成CREATE TABLE语句
494
- for (const table of tables) {
495
- // 获取表结构
496
- const columns = await this.getColumns(dataSource, databaseName, table.name);
497
- const indexes = await this.getIndexes(dataSource, databaseName, table.name);
498
- const foreignKeys = await this.getForeignKeys(dataSource, databaseName, table.name);
499
-
500
- // 生成CREATE TABLE语句
501
- schemaSql += `-- 表结构: ${table.name}\n`;
502
- schemaSql += `CREATE TABLE IF NOT EXISTS \`${table.name}\` (\n`;
503
-
504
- // 添加列定义
505
- const columnDefinitions = columns.map(column => {
506
- let definition = ` \`${column.name}\` ${column.type}`;
507
- if (!column.nullable) definition += ' NOT NULL';
508
- if (column.defaultValue !== undefined) {
509
- // 特殊关键字处理
510
- const upperDefault = column.defaultValue.toString().toUpperCase();
511
- if (upperDefault === 'CURRENT_TIMESTAMP' || upperDefault === 'NOW()' || upperDefault === 'CURRENT_DATE') {
512
- definition += ` DEFAULT ${upperDefault}`;
513
- } else {
514
- definition += ` DEFAULT ${column.defaultValue === null ? 'NULL' : `'${column.defaultValue}'`}`;
515
- }
516
- }
517
- if (column.isAutoIncrement) definition += ' AUTO_INCREMENT';
518
- return definition;
519
- });
520
-
521
- // 添加主键
522
- const primaryKeyColumns = columns.filter(column => column.isPrimary);
523
- if (primaryKeyColumns.length > 0) {
524
- const primaryKeyNames = primaryKeyColumns.map(column => `\`${column.name}\``).join(', ');
525
- columnDefinitions.push(` PRIMARY KEY (${primaryKeyNames})`);
526
- }
527
-
528
- schemaSql += columnDefinitions.join(',\n');
529
- schemaSql += '\n) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;\n\n';
530
-
531
- // 添加索引
532
- for (const index of indexes) {
533
- if (index.type === 'PRIMARY' || index.name.toUpperCase() === 'PRIMARY') continue; // 跳过主键索引
534
- schemaSql += `-- 索引: ${index.name} on ${table.name}\n`;
535
- schemaSql += `CREATE ${index.unique ? 'UNIQUE ' : ''}INDEX \`${index.name}\` ON \`${table.name}\` (${index.columns.map(col => `\`${col}\``).join(', ')})\n`;
536
- }
537
-
538
- if (indexes.length > 0) schemaSql += '\n';
539
-
540
- // 添加外键
541
- for (const foreignKey of foreignKeys) {
542
- schemaSql += `-- 外键: ${foreignKey.name} on ${table.name}\n`;
543
- schemaSql += `ALTER TABLE \`${table.name}\` ADD CONSTRAINT \`${foreignKey.name}\` FOREIGN KEY (\`${foreignKey.column}\`) REFERENCES \`${foreignKey.referencedTable}\` (\`${foreignKey.referencedColumn}\`)${foreignKey.onDelete ? ` ON DELETE ${foreignKey.onDelete}` : ''}${foreignKey.onUpdate ? ` ON UPDATE ${foreignKey.onUpdate}` : ''};\n`;
544
- }
545
-
546
- if (foreignKeys.length > 0) schemaSql += '\n';
547
- }
548
-
549
- return schemaSql;
550
- }
551
-
552
- /**
553
- * 查看数据库日志
554
- */
555
- async viewLogs(dataSource: DataSource, database?: string, limit: number = 100): Promise<any[]> {
556
- // MySQL查看错误日志
557
- try {
558
- const logs = await dataSource.query(`SHOW ERROR LOGS LIMIT ${limit}`);
559
- return logs;
560
- } catch (error) {
561
- // 如果SHOW ERROR LOGS失败,尝试其他方式
562
- try {
563
- // 尝试查看通用日志
564
- const logs = await dataSource.query(`SHOW GLOBAL VARIABLES LIKE '%log%'`);
565
- return logs;
566
- } catch (e) {
567
- return [{ message: '无法获取MySQL日志,请确保具有适当的权限' }];
568
- }
569
- }
570
- }
571
-
572
- /**
573
- * 备份数据库
574
- */
575
- async backupDatabase(dataSource: DataSource, databaseName: string, options?: any): Promise<string> {
576
- // MySQL备份数据库
577
- try {
578
- // 使用mysqldump命令备份
579
- const backupPath = options?.path || path.join(__dirname, '..', '..', 'backups');
580
-
581
- // 确保备份目录存在
582
- if (!fs.existsSync(backupPath)) {
583
- fs.mkdirSync(backupPath, { recursive: true });
584
- }
585
-
586
- const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
587
- const backupFile = path.join(backupPath, `${databaseName}_${timestamp}.sql`);
588
-
589
- // 执行备份命令
590
- const connectionOptions = dataSource.options as any;
591
- const host = connectionOptions.host || 'localhost';
592
- const port = connectionOptions.port || 3306;
593
- const user = connectionOptions.username;
594
- const password = connectionOptions.password;
595
-
596
- // 构建mysqldump命令
597
- let command = `mysqldump -h ${host} -P ${port} -u ${user}`;
598
- if (password) {
599
- command += ` -p${password}`;
600
- }
601
- command += ` ${databaseName} > ${backupFile}`;
602
-
603
- // 执行命令
604
- execSync(command);
605
-
606
- return `备份成功:${backupFile}`;
607
- } catch (error) {
608
- console.error('MySQL备份失败:', error);
609
- throw new Error(`备份失败: ${error.message}`);
610
- }
611
- }
612
-
613
- /**
614
- * 恢复数据库
615
- */
616
- async restoreDatabase(dataSource: DataSource, databaseName: string, filePath: string, options?: any): Promise<void> {
617
- // MySQL恢复数据库
618
- try {
619
- // 执行恢复命令
620
- const connectionOptions = dataSource.options as any;
621
- const host = connectionOptions.host || 'localhost';
622
- const port = connectionOptions.port || 3306;
623
- const user = connectionOptions.username;
624
- const password = connectionOptions.password;
625
-
626
- // 构建mysql命令
627
- let command = `mysql -h ${host} -P ${port} -u ${user}`;
628
- if (password) {
629
- command += ` -p${password}`;
630
- }
631
- command += ` ${databaseName} < ${filePath}`;
632
-
633
- // 执行命令
634
- execSync(command);
635
- } catch (error) {
636
- console.error('MySQL恢复失败:', error);
637
- throw new Error(`恢复失败: ${error.message}`);
638
- }
639
- }
640
-
641
- /**
642
- * 导出表数据到 SQL 文件
643
- */
644
- async exportTableDataToSQL(dataSource: DataSource, databaseName: string, tableName: string, options?: any): Promise<string> {
645
- try {
646
- // 创建导出目录
647
- const exportPath = options?.path || path.join(__dirname, '..', '..', '..', 'data', 'exports');
648
- if (!fs.existsSync(exportPath)) {
649
- fs.mkdirSync(exportPath, { recursive: true });
650
- }
651
-
652
- // 生成文件名
653
- const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
654
- const exportFile = path.join(exportPath, `${tableName}_data_${timestamp}.sql`);
655
-
656
- // 获取表结构
657
- const columns = await this.getColumns(dataSource, databaseName, tableName);
658
- const columnNames = columns.map(column => column.name);
659
-
660
- // 生成文件头部
661
- const header = `-- 表数据导出 - ${tableName}\n` +
662
- `-- 导出时间: ${new Date().toISOString()}\n\n`;
663
- fs.writeFileSync(exportFile, header, 'utf8');
664
-
665
- // 分批处理数据,避免一次性加载大量数据到内存
666
- const batchSize = options?.batchSize || 10000; // 每批处理10000行
667
- let offset = 0;
668
- let hasMoreData = true;
669
-
670
- while (hasMoreData) {
671
- // 分批查询数据
672
- const query = `SELECT * FROM ${this.quoteIdentifier(tableName)} LIMIT ${batchSize} OFFSET ${offset}`;
673
- const data = await dataSource.query(query);
674
-
675
- if (data.length === 0) {
676
- hasMoreData = false;
677
- break;
678
- }
679
-
680
- // 生成当前批次的 INSERT 语句
681
- let batchSql = '';
682
- data.forEach((row: any) => {
683
- const values = columnNames.map(column => {
684
- const value = row[column];
685
- if (value === null || value === undefined) {
686
- return 'NULL';
687
- } else if (typeof value === 'string') {
688
- return `'${value.replace(/'/g, "''")}'`;
689
- } else if (typeof value === 'boolean') {
690
- return value ? '1' : '0';
691
- } else if (value instanceof Date) {
692
- return `'${value.toISOString().slice(0, 19).replace('T', ' ')}'`;
693
- } else if (typeof value === 'object') {
694
- try {
695
- // 递归处理对象,确保所有嵌套的JSON字符串都被正确转义
696
- const processValue = (val) => {
697
- if (val === null || val === undefined) {
698
- return val;
699
- } else if (typeof val === 'string') {
700
- // 检查是否是JSON字符串
701
- try {
702
- const parsed = JSON.parse(val);
703
- // 如果是JSON字符串,递归处理
704
- if (typeof parsed === 'object') {
705
- return processValue(parsed);
706
- }
707
- } catch {
708
- // 不是JSON字符串,直接返回
709
- }
710
- return val;
711
- } else if (typeof val === 'object') {
712
- if (Array.isArray(val)) {
713
- return val.map(processValue);
714
- } else {
715
- const processedObj = {};
716
- for (const key in val) {
717
- processedObj[key] = processValue(val[key]);
718
- }
719
- return processedObj;
720
- }
721
- } else {
722
- return val;
723
- }
724
- };
725
-
726
- // 处理对象
727
- const processedValue = processValue(value);
728
- // 序列化处理后的对象
729
- let stringValue = JSON.stringify(processedValue);
730
- // 对JSON字符串中的单引号进行转义,确保在SQL中正确处理
731
- stringValue = stringValue.replace(/'/g, "''");
732
- // 返回用单引号包裹的字符串
733
- return `'${stringValue}'`;
734
- } catch {
735
- return `'${String(value).replace(/'/g, "''")}'`;
736
- }
737
- } else {
738
- return String(value);
739
- }
740
- });
741
-
742
- batchSql += `INSERT INTO ${this.quoteIdentifier(tableName)} (${columnNames.map(col => this.quoteIdentifier(col)).join(', ')}) VALUES (${values.join(', ')});\n`;
743
- });
744
-
745
- // 追加写入文件
746
- fs.appendFileSync(exportFile, batchSql, 'utf8');
747
-
748
- // 增加偏移量
749
- offset += batchSize;
750
-
751
- // 打印进度信息
752
- console.log(`MySQL导出表数据进度: ${tableName} - 已处理 ${offset} 行`);
753
- }
754
-
755
- return exportFile;
756
- } catch (error) {
757
- console.error('MySQL导出表数据失败:', error);
758
- throw new Error(`导出表数据失败: ${error.message}`);
759
- }
760
- }
761
-
762
- /**
763
- * 修改表结构
764
- */
765
- async alterTable(dataSource: DataSource, databaseName: string, tableDiff: any): Promise<any> {
766
- try {
767
- const tableName = tableDiff.tableName;
768
- const sqlStatements: string[] = [];
769
-
770
- // 修改表注释
771
- if (tableDiff.tableCommentChanged) {
772
- sqlStatements.push(`ALTER TABLE \`${tableName}\` COMMENT='${tableDiff.tableComment}';`);
773
- }
774
-
775
- // 添加新列
776
- tableDiff.addedColumns.forEach((column: any) => {
777
- // 检查type是否已经包含长度信息(括号)
778
- const typeHasLength = /\(\d+\)/.test(column.type);
779
-
780
- let columnSQL = `ALTER TABLE \`${tableName}\` ADD COLUMN \`${column.name}\` ${column.type}`;
781
-
782
- // 处理长度和精度(仅当type中没有指定长度时才添加)
783
- if (!typeHasLength) {
784
- if (column.length) {
785
- columnSQL += `(${column.length})`;
786
- } else if (column.precision) {
787
- if (column.scale) {
788
- columnSQL += `(${column.precision},${column.scale})`;
789
- } else {
790
- columnSQL += `(${column.precision})`;
791
- }
792
- }
793
- }
794
-
795
- // 处理NULL约束
796
- if (!column.nullable) {
797
- columnSQL += ' NOT NULL';
798
- }
799
-
800
- // 处理默认值
801
- if (column.defaultValue) {
802
- const upperDefault = column.defaultValue.toString().toUpperCase();
803
- if (['CURRENT_TIMESTAMP', 'NOW()', 'CURRENT_DATE', 'CURRENT_TIME'].includes(upperDefault)) {
804
- columnSQL += ` DEFAULT ${upperDefault}`;
805
- } else {
806
- columnSQL += ` DEFAULT '${column.defaultValue}'`;
807
- }
808
- }
809
-
810
- // 处理自增
811
- if (column.isAutoIncrement) {
812
- columnSQL += ' AUTO_INCREMENT';
813
- }
814
-
815
- // 处理注释
816
- if (column.comment) {
817
- columnSQL += ` COMMENT '${column.comment}'`;
818
- }
819
-
820
- sqlStatements.push(columnSQL + ';');
821
- });
822
-
823
- // 修改列
824
- tableDiff.modifiedColumns.forEach((modification: any) => {
825
- const { oldColumn, newColumn } = modification;
826
-
827
- // 检查type是否已经包含长度信息(括号)
828
- const typeHasLength = /\(\d+\)/.test(newColumn.type);
829
-
830
- let columnSQL = `ALTER TABLE \`${tableName}\` MODIFY COLUMN \`${newColumn.name}\` ${newColumn.type}`;
831
-
832
- // 处理长度和精度(仅当type中没有指定长度时才添加)
833
- if (!typeHasLength) {
834
- if (newColumn.length) {
835
- columnSQL += `(${newColumn.length})`;
836
- } else if (newColumn.precision) {
837
- if (newColumn.scale) {
838
- columnSQL += `(${newColumn.precision},${newColumn.scale})`;
839
- } else {
840
- columnSQL += `(${newColumn.precision})`;
841
- }
842
- }
843
- }
844
-
845
- // 处理NULL约束
846
- if (!newColumn.nullable) {
847
- columnSQL += ' NOT NULL';
848
- } else {
849
- columnSQL += ' NULL';
850
- }
851
-
852
- // 处理默认值
853
- if (newColumn.defaultValue) {
854
- const upperDefault = newColumn.defaultValue.toString().toUpperCase();
855
- if (['CURRENT_TIMESTAMP', 'NOW()', 'CURRENT_DATE', 'CURRENT_TIME'].includes(upperDefault)) {
856
- columnSQL += ` DEFAULT ${upperDefault}`;
857
- } else {
858
- columnSQL += ` DEFAULT '${newColumn.defaultValue}'`;
859
- }
860
- }
861
-
862
- // 处理自增
863
- if (newColumn.isAutoIncrement) {
864
- columnSQL += ' AUTO_INCREMENT';
865
- }
866
-
867
- // 处理注释
868
- if (newColumn.comment) {
869
- columnSQL += ` COMMENT '${newColumn.comment}'`;
870
- }
871
-
872
- sqlStatements.push(columnSQL + ';');
873
- });
874
-
875
- // 删除列
876
- tableDiff.deletedColumns.forEach((column: any) => {
877
- sqlStatements.push(`ALTER TABLE \`${tableName}\` DROP COLUMN \`${column.name}\`;`);
878
- });
879
-
880
- // 执行SQL语句
881
- if (sqlStatements.length > 0) {
882
- await this.executeBatchQuery(dataSource, sqlStatements, { useTransaction: true });
883
- }
884
-
885
- return { ret: 0, message: '表结构修改成功' };
886
- } catch (error) {
887
- console.error('MySQL修改表结构失败:', error);
888
- return { ret: 1, message: `修改表结构失败: ${error instanceof Error ? error.message : String(error)}` };
889
- }
890
- }
891
-
892
- /**
893
- * 批量插入数据
894
- */
895
- async bulkInsert(dataSource: DataSource, databaseName: string, tableName: string, data: any[], overrideExisting: boolean = false): Promise<void> {
896
- if (data.length === 0) return;
897
-
898
- const columns = Object.keys(data[0]);
899
- const placeholders = data.map(() =>
900
- `(${columns.map(() => '?').join(', ')})`
901
- ).join(', ');
902
-
903
- // 处理JSON类型的数据,将对象序列化为JSON字符串
904
- const values = data.flatMap(row =>
905
- columns.map(column => {
906
- const value = row[column];
907
- // 如果值是对象且不是null,则序列化为JSON字符串
908
- if (value !== null && typeof value === 'object') {
909
- return JSON.stringify(value);
910
- }
911
- return value;
912
- })
913
- );
914
-
915
- let sql = `INSERT INTO \`${tableName}\` (${columns.map(col => `\`${col}\``).join(', ')}) VALUES ${placeholders}`;
916
-
917
- if (overrideExisting) {
918
- // 使用 ON DUPLICATE KEY UPDATE 来更新已存在的数据
919
- const updateClauses = columns.map(col => `\`${col}\` = VALUES(\`${col}\`)`).join(', ');
920
- sql += ` ON DUPLICATE KEY UPDATE ${updateClauses}`;
921
- } else {
922
- // 使用 INSERT IGNORE 来忽略已存在的数据
923
- sql = sql.replace('INSERT INTO', 'INSERT IGNORE INTO');
924
- }
925
-
926
- await dataSource.query(sql, values);
927
- }
928
-
929
- /**
930
- * 插入单条数据
931
- */
932
- async insertData(dataSource: DataSource, databaseName: string, tableName: string, data: any, overrideExisting: boolean = false): Promise<void> {
933
- const columns = Object.keys(data);
934
- const placeholders = columns.map(() => '?').join(', ');
935
-
936
- // 处理JSON类型的数据,将对象序列化为JSON字符串
937
- const values = columns.map(column => {
938
- const value = data[column];
939
- // 如果值是对象且不是null,则序列化为JSON字符串
940
- if (value !== null && typeof value === 'object') {
941
- return JSON.stringify(value);
942
- }
943
- return value;
944
- });
945
-
946
- let sql = `INSERT INTO \`${tableName}\` (${columns.map(col => `\`${col}\``).join(', ')}) VALUES (${placeholders})`;
947
-
948
- if (overrideExisting) {
949
- // 使用 ON DUPLICATE KEY UPDATE 来更新已存在的数据
950
- const updateClauses = columns.map(col => `\`${col}\` = VALUES(\`${col}\`)`).join(', ');
951
- sql += ` ON DUPLICATE KEY UPDATE ${updateClauses}`;
952
- } else {
953
- // 使用 INSERT IGNORE 来忽略已存在的数据
954
- sql = sql.replace('INSERT INTO', 'INSERT IGNORE INTO');
955
- }
956
-
957
- await dataSource.query(sql, values);
958
- }
959
-
960
- /**
961
- * 删除表
962
- */
963
- async dropTable(dataSource: DataSource, databaseName: string, tableName: string): Promise<void> {
964
- const sql = `DROP TABLE IF EXISTS \`${tableName}\``;
965
- await dataSource.query(sql);
966
- }
967
-
968
- /**
969
- * 创建表
970
- */
971
- async createTable(dataSource: DataSource, databaseName: string, table: any): Promise<void> {
972
- const { name, columns, comment } = table;
973
-
974
- let sql = `CREATE TABLE \`${name}\` (\n`;
975
- const columnDefs: string[] = [];
976
- const primaryKeys: string[] = [];
977
-
978
- columns.forEach((column: any) => {
979
- let columnDef = ` \`${column.name}\` ${column.type}`;
980
-
981
- if (!column.nullable) {
982
- columnDef += ' NOT NULL';
983
- }
984
-
985
- if (column.defaultValue) {
986
- const upperDefault = column.defaultValue.toString().toUpperCase();
987
- if (['CURRENT_TIMESTAMP', 'NOW()', 'CURRENT_DATE', 'CURRENT_TIME'].includes(upperDefault)) {
988
- columnDef += ` DEFAULT ${upperDefault}`;
989
- } else {
990
- columnDef += ` DEFAULT '${column.defaultValue}'`;
991
- }
992
- }
993
-
994
- if (column.isAutoIncrement) {
995
- columnDef += ' AUTO_INCREMENT';
996
- }
997
-
998
- if (column.comment) {
999
- columnDef += ` COMMENT '${column.comment}'`;
1000
- }
1001
-
1002
- columnDefs.push(columnDef);
1003
-
1004
- // 收集主键列
1005
- if (column.isPrimary) {
1006
- primaryKeys.push(`\`${column.name}\``);
1007
- }
1008
- });
1009
-
1010
- // 添加主键约束
1011
- if (primaryKeys.length > 0) {
1012
- columnDefs.push(` PRIMARY KEY (${primaryKeys.join(', ')})`);
1013
- }
1014
-
1015
- sql += columnDefs.join(',\n');
1016
- sql += '\n)';
1017
-
1018
- if (comment) {
1019
- sql += ` COMMENT='${comment}'`;
1020
- }
1021
-
1022
- sql += ';';
1023
-
1024
- await dataSource.query(sql);
1025
- }
1026
- }