Initial commit: 宁夏智慧养殖监管平台

backend/models/Animal.js

@@ -0,0 +1,115 @@
+/**
+ * Animal 模型定义
+ * @file Animal.js
+ * @description 定义动物模型，用于数据库操作
+ */
+const { DataTypes } = require('sequelize');
+const BaseModel = require('./BaseModel');
+const { sequelize } = require('../config/database-simple');
+
+/**
+ * 动物模型
+ * @typedef {Object} Animal
+ * @property {number} id - 动物唯一标识
+ * @property {string} type - 动物类型
+ * @property {number} count - 数量
+ * @property {number} farmId - 所属养殖场ID
+ * @property {Date} created_at - 创建时间
+ * @property {Date} updated_at - 更新时间
+ */
+class Animal extends BaseModel {
+  /**
+   * 获取动物所属的养殖场
+   * @returns {Promise<Object>} 养殖场信息
+   */
+  async getFarm() {
+    return await this.getFarm();
+  }
+
+  /**
+   * 更新动物数量
+   * @param {Number} count 新数量
+   * @returns {Promise<Boolean>} 更新结果
+   */
+  async updateCount(count) {
+    try {
+      if (count < 0) {
+        throw new Error('数量不能为负数');
+      }
+      
+      this.count = count;
+      await this.save();
+      return true;
+    } catch (error) {
+      console.error('更新动物数量失败:', error);
+      return false;
+    }
+  }
+
+  /**
+   * 更新健康状态
+   * @param {String} status 新状态
+   * @returns {Promise<Boolean>} 更新结果
+   */
+  async updateHealthStatus(status) {
+    try {
+      this.health_status = status;
+      await this.save();
+      return true;
+    } catch (error) {
+      console.error('更新健康状态失败:', error);
+      return false;
+    }
+  }
+}
+
+// 初始化Animal模型
+Animal.init({
+  id: {
+    type: DataTypes.INTEGER,
+    primaryKey: true,
+    autoIncrement: true
+  },
+  type: {
+    type: DataTypes.STRING(50),
+    allowNull: false
+  },
+  count: {
+    type: DataTypes.INTEGER,
+    allowNull: false,
+    defaultValue: 0
+  },
+  farm_id: {
+    type: DataTypes.INTEGER,
+    allowNull: false,
+    references: {
+      model: 'farms',
+      key: 'id'
+    }
+  },
+  health_status: {
+    type: DataTypes.ENUM('healthy', 'sick', 'quarantine', 'treatment'),
+    defaultValue: 'healthy'
+  },
+  last_inspection: {
+    type: DataTypes.DATE,
+    allowNull: true
+  },
+  notes: {
+    type: DataTypes.TEXT,
+    allowNull: true
+  }
+}, {
+  sequelize,
+  tableName: 'animals',
+  modelName: 'Animal',
+  timestamps: true,
+  createdAt: 'created_at',
+  updatedAt: 'updated_at'
+});
+
+/**
+ * 导出动物模型
+ * @exports Animal
+ */
+module.exports = Animal;