100 lines
2.6 KiB
JavaScript
100 lines
2.6 KiB
JavaScript
/**
|
||
* 政府后端模型索引文件
|
||
* @file index.js
|
||
* @description 导出所有模型并建立关联关系
|
||
*/
|
||
const { sequelize } = require('../config/database');
|
||
|
||
// 导入所有模型
|
||
const User = require('./User');
|
||
const AdminStaff = require('./AdminStaff');
|
||
const Department = require('./Department');
|
||
const Position = require('./Position');
|
||
const Farmer = require('./Farmer');
|
||
const HarmlessPlace = require('./HarmlessPlace');
|
||
const HarmlessRegistration = require('./HarmlessRegistration');
|
||
const Material = require('./Material');
|
||
const Slaughterhouse = require('./Slaughterhouse');
|
||
const EpidemicAgency = require('./EpidemicAgency');
|
||
const SmartCollar = require('./SmartCollar');
|
||
const SmartEarmark = require('./SmartEarmark');
|
||
const SmartHost = require('./SmartHost');
|
||
const WarehouseTransaction = require('./WarehouseTransaction');
|
||
|
||
// 初始化所有模型
|
||
const initModels = () => {
|
||
try {
|
||
// 处理直接导出的模型
|
||
if (User && typeof User !== 'function') {
|
||
// User模型已经直接导出，不需要额外处理
|
||
}
|
||
|
||
// 处理需要sequelize参数的模型
|
||
const modelsToInit = [
|
||
AdminStaff,
|
||
Department,
|
||
Position,
|
||
Farmer,
|
||
HarmlessPlace,
|
||
HarmlessRegistration,
|
||
Material,
|
||
Slaughterhouse,
|
||
EpidemicAgency,
|
||
SmartCollar,
|
||
SmartEarmark,
|
||
SmartHost,
|
||
WarehouseTransaction
|
||
];
|
||
|
||
// 初始化模型
|
||
modelsToInit.forEach(modelFactory => {
|
||
if (typeof modelFactory === 'function') {
|
||
// 调用函数获取模型实例
|
||
const model = modelFactory(sequelize);
|
||
// 将初始化后的模型赋值回原来的变量
|
||
Object.assign(module.exports, { [model.name]: model });
|
||
}
|
||
});
|
||
|
||
console.log('✅ 所有政府后端模型初始化完成');
|
||
} catch (error) {
|
||
console.error('❌ 政府后端模型初始化失败:', error);
|
||
}
|
||
};
|
||
|
||
// 初始化模型
|
||
initModels();
|
||
|
||
// 导出sequelize和所有模型
|
||
module.exports = {
|
||
sequelize,
|
||
User,
|
||
AdminStaff,
|
||
Department,
|
||
Position,
|
||
Farmer,
|
||
HarmlessPlace,
|
||
HarmlessRegistration,
|
||
Material,
|
||
Slaughterhouse,
|
||
EpidemicAgency,
|
||
SmartCollar,
|
||
SmartEarmark,
|
||
SmartHost,
|
||
WarehouseTransaction
|
||
};
|
||
|
||
// 同步所有模型到数据库（可选）
|
||
const syncModels = async (options = {}) => {
|
||
try {
|
||
await sequelize.sync(options);
|
||
console.log('✅ 政府后端所有模型已同步到数据库');
|
||
return true;
|
||
} catch (error) {
|
||
console.error('❌ 政府后端模型同步失败:', error);
|
||
return false;
|
||
}
|
||
};
|
||
|
||
// 添加syncModels到导出
|
||
module.exports.syncModels = syncModels; |