init: AI培训与智能巡检系统

2026-06-16 00:55:20 +08:00
commit c55598494b
201 changed files with 53131 additions and 0 deletions
@@ -0,0 +1,53 @@
+import Database from 'better-sqlite3';
+import { readFileSync } from 'fs';
+import { join } from 'path';
+
+/** 读取 ETL 的 schema.sql 并在给定 db 上建表 + 注入测试数据。*/
+export function seedDatabase(db: Database.Database): void {
+  const schemaPath = join(
+    __dirname,
+    '..',
+    '..',
+    '..',
+    '..',
+    'app',
+    'etl',
+    'schema.sql',
+  );
+  db.exec(readFileSync(schemaPath, 'utf-8'));
+
+  const cat = db.prepare(
+    'INSERT INTO category(name, subcat) VALUES (?, ?)',
+  );
+  const elec = cat.run('电力机车', '').lastInsertRowid as number;
+  const emu = cat.run('动车组', '复兴号').lastInsertRowid as number;
+  const insp = cat.run('检测车', '普速检测').lastInsertRowid as number;
+
+  const model = db.prepare(
+    `INSERT INTO model(category_id, series, model_code, full_name, manufacturer,
+       country, country_type, first_year, last_year, status, usage,
+       max_speed_value, max_speed_unit, weight_value, weight_unit, raw_json)
+     VALUES (@category_id, @series, @model_code, @full_name, @manufacturer,
+       @country, @country_type, @first_year, @last_year, @status, @usage,
+       @max_speed_value, @max_speed_unit, @weight_value, @weight_unit, @raw_json)`,
+  );
+  model.run({
+    category_id: elec, series: '和谐', model_code: 'HXD1型', full_name: '',
+    manufacturer: '中车株洲', country: '中国', country_type: '国产',
+    first_year: 2006, last_year: null, status: '现役', usage: '干线货运',
+    max_speed_value: 120, max_speed_unit: 'km/h', weight_value: 150,
+    weight_unit: 't', raw_json: JSON.stringify({ model_code: 'HXD1型' }),
+  });
+  model.run({
+    category_id: emu, series: '', model_code: 'CR400AF', full_name: '复兴号',
+    manufacturer: '中车四方', country: '中国', country_type: '国产',
+    first_year: 2017, last_year: null, status: '现役', usage: '高速客运',
+    max_speed_value: 350, max_speed_unit: 'km/h', weight_value: null,
+    weight_unit: '', raw_json: JSON.stringify({ model_code: 'CR400AF' }),
+  });
+
+  db.prepare(
+    `INSERT INTO unit(category_id, car_number, model_name, function, depot, status)
+     VALUES (?, ?, ?, ?, ?, ?)`,
+  ).run(insp, 'SYJZ-0001', '移动式线路动态加载试验车', '试验车', '国铁', '半封存');
+}