Initial commit: 宁夏智慧养殖监管平台
This commit is contained in:
shenquanyi
165
backend/import-data.js
Normal file
165
backend/import-data.js
Normal file
@@ -0,0 +1,165 @@
|
||||
const { sequelize } = require('./config/database-simple');
|
||||
const { QueryTypes } = require('sequelize');
|
||||
|
||||
async function importData() {
|
||||
try {
|
||||
console.log('开始导入数据...');
|
||||
|
||||
// 连接数据库
|
||||
await sequelize.authenticate();
|
||||
console.log('数据库连接成功');
|
||||
|
||||
// 1. 插入养殖场数据
|
||||
const farmData = [
|
||||
{
|
||||
name: '东方养殖场',
|
||||
type: 'pig',
|
||||
location: JSON.stringify({ lat: 39.9042, lng: 116.4074, address: '北京市朝阳区' }),
|
||||
address: '北京市朝阳区东三环北路',
|
||||
contact: '张三',
|
||||
phone: '13800138001',
|
||||
status: 'active',
|
||||
created_at: new Date(),
|
||||
updated_at: new Date()
|
||||
},
|
||||
{
|
||||
name: '西部牧场',
|
||||
type: 'cattle',
|
||||
location: JSON.stringify({ lat: 30.5728, lng: 104.0668, address: '四川省成都市' }),
|
||||
address: '四川省成都市高新区天府大道',
|
||||
contact: '李四',
|
||||
phone: '13800138002',
|
||||
status: 'active',
|
||||
created_at: new Date(),
|
||||
updated_at: new Date()
|
||||
},
|
||||
{
|
||||
name: '南方羊场',
|
||||
type: 'sheep',
|
||||
location: JSON.stringify({ lat: 23.1291, lng: 113.2644, address: '广东省广州市' }),
|
||||
address: '广东省广州市天河区珠江新城',
|
||||
contact: '王五',
|
||||
phone: '13800138003',
|
||||
status: 'active',
|
||||
created_at: new Date(),
|
||||
updated_at: new Date()
|
||||
}
|
||||
];
|
||||
|
||||
await sequelize.query(
|
||||
`INSERT INTO farms (name, type, location, address, contact, phone, status, created_at, updated_at) VALUES ${farmData.map(() => '(?, ?, ?, ?, ?, ?, ?, ?, ?)').join(', ')}`,
|
||||
{
|
||||
replacements: farmData.flatMap(farm => [
|
||||
farm.name, farm.type, farm.location, farm.address, farm.contact,
|
||||
farm.phone, farm.status, farm.created_at, farm.updated_at
|
||||
]),
|
||||
type: QueryTypes.INSERT
|
||||
}
|
||||
);
|
||||
console.log('养殖场数据导入成功');
|
||||
|
||||
// 获取刚插入的养殖场ID
|
||||
const farms = await sequelize.query('SELECT id, name, type FROM farms WHERE name IN ("东方养殖场", "西部牧场", "南方羊场")', { type: QueryTypes.SELECT });
|
||||
console.log('获取到养殖场:', farms);
|
||||
|
||||
// 2. 插入动物数据
|
||||
const animalData = [];
|
||||
const animalTypes = {
|
||||
'pig': ['猪', '母猪', '仔猪'],
|
||||
'cattle': ['肉牛', '奶牛', '小牛'],
|
||||
'sheep': ['山羊', '绵羊', '羔羊']
|
||||
};
|
||||
|
||||
for (const farm of farms) {
|
||||
const types = animalTypes[farm.type] || ['未知'];
|
||||
for (const type of types) {
|
||||
animalData.push({
|
||||
type: type,
|
||||
count: Math.floor(Math.random() * 100) + 50,
|
||||
farm_id: farm.id,
|
||||
health_status: Math.random() > 0.2 ? 'healthy' : 'sick',
|
||||
last_inspection: new Date(Date.now() - Math.random() * 30 * 24 * 60 * 60 * 1000),
|
||||
notes: `${type}群体健康状况良好`,
|
||||
created_at: new Date(),
|
||||
updated_at: new Date()
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
if (animalData.length > 0) {
|
||||
await sequelize.query(
|
||||
`INSERT INTO animals (type, count, farm_id, health_status, last_inspection, notes, created_at, updated_at) VALUES ${animalData.map(() => '(?, ?, ?, ?, ?, ?, ?, ?)').join(', ')}`,
|
||||
{
|
||||
replacements: animalData.flatMap(animal => [
|
||||
animal.type, animal.count, animal.farm_id, animal.health_status,
|
||||
animal.last_inspection, animal.notes, animal.created_at, animal.updated_at
|
||||
]),
|
||||
type: QueryTypes.INSERT
|
||||
}
|
||||
);
|
||||
console.log('动物数据导入成功');
|
||||
}
|
||||
|
||||
// 3. 插入设备数据
|
||||
const deviceData = [];
|
||||
const deviceTypes = ['温度传感器', '湿度传感器', '摄像头', '喂食器', '饮水器'];
|
||||
const deviceStatuses = ['online', 'offline', 'maintenance'];
|
||||
|
||||
for (const farm of farms) {
|
||||
for (let i = 0; i < 5; i++) {
|
||||
const type = deviceTypes[Math.floor(Math.random() * deviceTypes.length)];
|
||||
const status = deviceStatuses[Math.floor(Math.random() * deviceStatuses.length)];
|
||||
|
||||
deviceData.push({
|
||||
name: `${type}_${farm.name}_${String(i + 1).padStart(3, '0')}`,
|
||||
type: type,
|
||||
status: status,
|
||||
farm_id: farm.id,
|
||||
last_maintenance: new Date(Date.now() - Math.random() * 90 * 24 * 60 * 60 * 1000),
|
||||
installation_date: new Date(Date.now() - Math.random() * 730 * 24 * 60 * 60 * 1000),
|
||||
metrics: JSON.stringify({
|
||||
temperature: Math.round((Math.random() * 15 + 15) * 10) / 10,
|
||||
humidity: Math.round((Math.random() * 40 + 40) * 10) / 10,
|
||||
battery: Math.round(Math.random() * 100),
|
||||
signal_strength: Math.round(Math.random() * 100)
|
||||
}),
|
||||
created_at: new Date(),
|
||||
updated_at: new Date()
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
if (deviceData.length > 0) {
|
||||
await sequelize.query(
|
||||
`INSERT INTO devices (name, type, status, farm_id, last_maintenance, installation_date, metrics, created_at, updated_at) VALUES ${deviceData.map(() => '(?, ?, ?, ?, ?, ?, ?, ?, ?)').join(', ')}`,
|
||||
{
|
||||
replacements: deviceData.flatMap(device => [
|
||||
device.name, device.type, device.status, device.farm_id,
|
||||
device.last_maintenance, device.installation_date, device.metrics,
|
||||
device.created_at, device.updated_at
|
||||
]),
|
||||
type: QueryTypes.INSERT
|
||||
}
|
||||
);
|
||||
console.log('设备数据导入成功');
|
||||
}
|
||||
|
||||
// 4. 验证导入结果
|
||||
const [farmCount] = await sequelize.query('SELECT COUNT(*) as count FROM farms', { type: QueryTypes.SELECT });
|
||||
const [animalCount] = await sequelize.query('SELECT COUNT(*) as count FROM animals', { type: QueryTypes.SELECT });
|
||||
const [deviceCount] = await sequelize.query('SELECT COUNT(*) as count FROM devices', { type: QueryTypes.SELECT });
|
||||
|
||||
console.log('\n=== 数据导入完成 ===');
|
||||
console.log(`养殖场总数: ${farmCount.count}`);
|
||||
console.log(`动物总数: ${animalCount.count}`);
|
||||
console.log(`设备总数: ${deviceCount.count}`);
|
||||
|
||||
} catch (error) {
|
||||
console.error('数据导入失败:', error.message);
|
||||
console.error('详细错误:', error);
|
||||
} finally {
|
||||
await sequelize.close();
|
||||
}
|
||||
}
|
||||
|
||||
importData();
|
||||
Reference in New Issue
Block a user