pig/pig-farm-controller
1
0
Fork 0
Code Issues 15 Pull Requests Actions Packages Projects Releases Wiki Activity

实现配方领域关于猪模型和营养需求的增删改查

Browse Source
This commit is contained in:
huang
2025-11-21 15:03:42 +08:00
parent a669bfda6c
commit 9996fcfd74
8 changed files with 565 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
design/archive/recipe-management/index.md
View File

@@ -53,4 +53,5 @@ http://git.huangwc.com/pig/pig-farm-controller/issues/66
3. 定义配方领域, 实现营养元素的增删改查
4. 实现原材料的增删改查和仓库层的原料库存记录表增查
5. 定义猪的模型和营养需求模型
6. 实现从json读取猪营养需求并写入数据库
6. 实现从json读取猪营养需求并写入数据库
7. 实现配方领域关于猪模型和营养需求的增删改查

34
docs/docs.go
View File

@@ -1826,7 +1826,6 @@ const docTemplate = `{
},
{
"enum": [
7,
-1,
0,
1,
@@ -1836,12 +1835,12 @@ const docTemplate = `{
5,
-1,
5,
6
6,
7
],
"type": "integer",
"format": "int32",
"x-enum-varnames": [
"_numLevels",
"DebugLevel",
"InfoLevel",
"WarnLevel",
@@ -1851,7 +1850,8 @@ const docTemplate = `{
"FatalLevel",
"_minLevel",
"_maxLevel",
"InvalidLevel"
"InvalidLevel",
"_numLevels"
],
"name": "level",
"in": "query"
@@ -7563,32 +7563,24 @@ const docTemplate = `{
"models.PigBatchStatus": {
"type": "string",
"enum": [
"保育",
"生长",
"育肥",
"生产中",
"待售",
"已出售",
"已归档"
],
"x-enum-comments": {
"BatchStatusActive": "饲养中",
"BatchStatusArchived": "批次结束（如全群淘汰等）",
"BatchStatusFinishing": "最后的育肥阶段",
"BatchStatusForSale": "达到出栏标准",
"BatchStatusGrowing": "生长育肥阶段",
"BatchStatusWeaning": "从断奶到保育结束"
"BatchStatusForSale": "达到出栏标准"
},
"x-enum-descriptions": [
"从断奶到保育结束",
"生长育肥阶段",
"最后的育肥阶段",
"饲养中",
"达到出栏标准",
"",
"批次结束（如全群淘汰等）"
],
"x-enum-varnames": [
"BatchStatusWeaning",
"BatchStatusGrowing",
"BatchStatusFinishing",
"BatchStatusActive",
"BatchStatusForSale",
"BatchStatusSold",
"BatchStatusArchived"
@@ -7837,7 +7829,6 @@ const docTemplate = `{
"type": "integer",
"format": "int32",
"enum": [
7,
-1,
0,
1,
@@ -7847,10 +7838,10 @@ const docTemplate = `{
5,
-1,
5,
6
6,
7
],
"x-enum-varnames": [
"_numLevels",
"DebugLevel",
"InfoLevel",
"WarnLevel",
@@ -7860,7 +7851,8 @@ const docTemplate = `{
"FatalLevel",
"_minLevel",
"_maxLevel",
"InvalidLevel"
"InvalidLevel",
"_numLevels"
]
}
},

34
docs/swagger.json
View File

@@ -1818,7 +1818,6 @@
},
{
"enum": [
7,
-1,
0,
1,
@@ -1828,12 +1827,12 @@
5,
-1,
5,
6
6,
7
],
"type": "integer",
"format": "int32",
"x-enum-varnames": [
"_numLevels",
"DebugLevel",
"InfoLevel",
"WarnLevel",
@@ -1843,7 +1842,8 @@
"FatalLevel",
"_minLevel",
"_maxLevel",
"InvalidLevel"
"InvalidLevel",
"_numLevels"
],
"name": "level",
"in": "query"
@@ -7555,32 +7555,24 @@
"models.PigBatchStatus": {
"type": "string",
"enum": [
"保育",
"生长",
"育肥",
"生产中",
"待售",
"已出售",
"已归档"
],
"x-enum-comments": {
"BatchStatusActive": "饲养中",
"BatchStatusArchived": "批次结束（如全群淘汰等）",
"BatchStatusFinishing": "最后的育肥阶段",
"BatchStatusForSale": "达到出栏标准",
"BatchStatusGrowing": "生长育肥阶段",
"BatchStatusWeaning": "从断奶到保育结束"
"BatchStatusForSale": "达到出栏标准"
},
"x-enum-descriptions": [
"从断奶到保育结束",
"生长育肥阶段",
"最后的育肥阶段",
"饲养中",
"达到出栏标准",
"",
"批次结束（如全群淘汰等）"
],
"x-enum-varnames": [
"BatchStatusWeaning",
"BatchStatusGrowing",
"BatchStatusFinishing",
"BatchStatusActive",
"BatchStatusForSale",
"BatchStatusSold",
"BatchStatusArchived"
@@ -7829,7 +7821,6 @@
"type": "integer",
"format": "int32",
"enum": [
7,
-1,
0,
1,
@@ -7839,10 +7830,10 @@
5,
-1,
5,
6
6,
7
],
"x-enum-varnames": [
"_numLevels",
"DebugLevel",
"InfoLevel",
"WarnLevel",
@@ -7852,7 +7843,8 @@
"FatalLevel",
"_minLevel",
"_maxLevel",
"InvalidLevel"
"InvalidLevel",
"_numLevels"
]
}
},

24
docs/swagger.yaml
View File

@@ -1905,30 +1905,22 @@ definitions:
- TreatmentLocationSickBay
models.PigBatchStatus:
enum:
- 保育
- 生长
- 育肥
- 生产中
- 待售
- 已出售
- 已归档
type: string
x-enum-comments:
BatchStatusActive: 饲养中
BatchStatusArchived: 批次结束（如全群淘汰等）
BatchStatusFinishing: 最后的育肥阶段
BatchStatusForSale: 达到出栏标准
BatchStatusGrowing: 生长育肥阶段
BatchStatusWeaning: 从断奶到保育结束
x-enum-descriptions:
- 从断奶到保育结束
- 生长育肥阶段
- 最后的育肥阶段
- 饲养中
- 达到出栏标准
- ""
- 批次结束（如全群淘汰等）
x-enum-varnames:
- BatchStatusWeaning
- BatchStatusGrowing
- BatchStatusFinishing
- BatchStatusActive
- BatchStatusForSale
- BatchStatusSold
- BatchStatusArchived
@@ -2129,7 +2121,6 @@ definitions:
- PlanTypeFilterSystem
zapcore.Level:
enum:
- 7
- -1
- 0
- 1
@@ -2140,10 +2131,10 @@ definitions:
- -1
- 5
- 6
- 7
format: int32
type: integer
x-enum-varnames:
- _numLevels
- DebugLevel
- InfoLevel
- WarnLevel
@@ -2154,6 +2145,7 @@ definitions:
- _minLevel
- _maxLevel
- InvalidLevel
- _numLevels
info:
contact:
email: divano@example.com
@@ -3270,7 +3262,6 @@ paths:
name: end_time
type: string
- enum:
- 7
- -1
- 0
- 1
@@ -3281,12 +3272,12 @@ paths:
- -1
- 5
- 6
- 7
format: int32
in: query
name: level
type: integer
x-enum-varnames:
- _numLevels
- DebugLevel
- InfoLevel
- WarnLevel
@@ -3297,6 +3288,7 @@ paths:
- _minLevel
- _maxLevel
- InvalidLevel
- _numLevels
- enum:
- 邮件
- 企业微信

236
internal/domain/recipe/recipe_service.go
View File

@@ -14,12 +14,15 @@ import (
// 定义领域特定的错误
var (
ErrNutrientNameConflict = fmt.Errorf("营养种类名称已存在")
ErrNutrientNotFound = fmt.Errorf("营养种类不存在")
ErrNutrientInUse = fmt.Errorf("营养种类正在被原料使用，无法删除")
ErrNutrientNameConflict = fmt.Errorf("营养种类名称已存在")
ErrNutrientNotFound = fmt.Errorf("营养种类不存在")
ErrRawMaterialNameConflict = fmt.Errorf("原料名称已存在")
ErrRawMaterialNotFound = fmt.Errorf("原料不存在")
ErrPigBreedInUse = fmt.Errorf("猪品种正在被猪类型使用，无法删除")
ErrPigBreedNotFound = fmt.Errorf("猪品种不存在")
ErrPigAgeStageInUse = fmt.Errorf("猪年龄阶段正在被猪类型使用，无法删除")
ErrPigAgeStageNotFound = fmt.Errorf("猪年龄阶段不存在")
ErrPigTypeNotFound = fmt.Errorf("猪类型不存在")
)
// Service 定义了配方与原料领域的核心业务服务接口
@@ -37,6 +40,27 @@ type Service interface {
DeleteRawMaterial(ctx context.Context, id uint32) error
GetRawMaterial(ctx context.Context, id uint32) (*models.RawMaterial, error)
ListRawMaterials(ctx context.Context, page, pageSize int) ([]models.RawMaterial, int64, error)
// 猪品种相关接口
CreatePigBreed(ctx context.Context, breed *models.PigBreed) error
GetPigBreedByID(ctx context.Context, id uint32) (*models.PigBreed, error)
UpdatePigBreed(ctx context.Context, breed *models.PigBreed) error
DeletePigBreed(ctx context.Context, id uint32) error
ListPigBreeds(ctx context.Context, opts repository.PigBreedListOptions, page, pageSize int) ([]models.PigBreed, int64, error)
// 猪年龄阶段相关接口
CreatePigAgeStage(ctx context.Context, ageStage *models.PigAgeStage) error
GetPigAgeStageByID(ctx context.Context, id uint32) (*models.PigAgeStage, error)
UpdatePigAgeStage(ctx context.Context, ageStage *models.PigAgeStage) error
DeletePigAgeStage(ctx context.Context, id uint32) error
ListPigAgeStages(ctx context.Context, opts repository.PigAgeStageListOptions, page, pageSize int) ([]models.PigAgeStage, int64, error)
// 猪类型相关接口
CreatePigType(ctx context.Context, pigType *models.PigType) error
GetPigTypeByID(ctx context.Context, id uint32) (*models.PigType, error)
UpdatePigType(ctx context.Context, pigType *models.PigType) error
DeletePigType(ctx context.Context, id uint32) error
ListPigTypes(ctx context.Context, opts repository.PigTypeListOptions, page, pageSize int) ([]models.PigType, int64, error)
}
// recipeServiceImpl 是 RecipeService 的实现
@@ -44,14 +68,16 @@ type recipeServiceImpl struct {
ctx context.Context
nutrientRepo repository.NutrientRepository
rawMaterialRepo repository.RawMaterialRepository
pigTypeRepo repository.PigTypeRepository
}
// NewRecipeService 创建一个新的 RecipeService 实例
func NewRecipeService(ctx context.Context, nutrientRepo repository.NutrientRepository, rawMaterialRepo repository.RawMaterialRepository) Service {
func NewRecipeService(ctx context.Context, nutrientRepo repository.NutrientRepository, rawMaterialRepo repository.RawMaterialRepository, pigTypeRepo repository.PigTypeRepository) Service {
return &recipeServiceImpl{
ctx: ctx,
nutrientRepo: nutrientRepo,
rawMaterialRepo: rawMaterialRepo,
pigTypeRepo: pigTypeRepo,
}
}
@@ -262,3 +288,203 @@ func (s *recipeServiceImpl) ListRawMaterials(ctx context.Context, page, pageSize
}
return rawMaterials, total, nil
}
// CreatePigBreed 实现了创建猪品种的核心业务逻辑
func (s *recipeServiceImpl) CreatePigBreed(ctx context.Context, breed *models.PigBreed) error {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "CreatePigBreed")
if err := s.pigTypeRepo.CreatePigBreed(serviceCtx, breed); err != nil {
return fmt.Errorf("创建猪品种失败: %w", err)
}
return nil
}
// GetPigBreedByID 实现了获取单个猪品种的逻辑
func (s *recipeServiceImpl) GetPigBreedByID(ctx context.Context, id uint32) (*models.PigBreed, error) {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "GetPigBreedByID")
breed, err := s.pigTypeRepo.GetPigBreedByID(serviceCtx, id)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, ErrPigBreedNotFound
}
return nil, fmt.Errorf("获取猪品种失败: %w", err)
}
return breed, nil
}
// UpdatePigBreed 实现了更新猪品种的核心业务逻辑
func (s *recipeServiceImpl) UpdatePigBreed(ctx context.Context, breed *models.PigBreed) error {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "UpdatePigBreed")
if err := s.pigTypeRepo.UpdatePigBreed(serviceCtx, breed); err != nil {
return fmt.Errorf("更新猪品种失败: %w", err)
}
return nil
}
// DeletePigBreed 实现了删除猪品种的核心业务逻辑
func (s *recipeServiceImpl) DeletePigBreed(ctx context.Context, id uint32) error {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "DeletePigBreed")
// 检查是否有猪类型关联到该品种
opts := repository.PigTypeListOptions{BreedID: &id}
pigTypes, _, err := s.pigTypeRepo.ListPigTypes(serviceCtx, opts, 1, 1) // 只需检查是否存在，所以取1条
if err != nil {
return fmt.Errorf("检查猪品种关联失败: %w", err)
}
if len(pigTypes) > 0 {
return ErrPigBreedInUse
}
// 检查实体是否存在
_, err = s.pigTypeRepo.GetPigBreedByID(serviceCtx, id)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return ErrPigBreedNotFound
}
return fmt.Errorf("获取待删除的猪品种失败: %w", err)
}
if err := s.pigTypeRepo.DeletePigBreed(serviceCtx, id); err != nil {
return fmt.Errorf("删除猪品种失败: %w", err)
}
return nil
}
// ListPigBreeds 实现了列出猪品种的逻辑
func (s *recipeServiceImpl) ListPigBreeds(ctx context.Context, opts repository.PigBreedListOptions, page, pageSize int) ([]models.PigBreed, int64, error) {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "ListPigBreeds")
breeds, total, err := s.pigTypeRepo.ListPigBreeds(serviceCtx, opts, page, pageSize)
if err != nil {
return nil, 0, fmt.Errorf("获取猪品种列表失败: %w", err)
}
return breeds, total, nil
}
// CreatePigAgeStage 实现了创建猪年龄阶段的核心业务逻辑
func (s *recipeServiceImpl) CreatePigAgeStage(ctx context.Context, ageStage *models.PigAgeStage) error {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "CreatePigAgeStage")
if err := s.pigTypeRepo.CreatePigAgeStage(serviceCtx, ageStage); err != nil {
return fmt.Errorf("创建猪年龄阶段失败: %w", err)
}
return nil
}
// GetPigAgeStageByID 实现了获取单个猪年龄阶段的逻辑
func (s *recipeServiceImpl) GetPigAgeStageByID(ctx context.Context, id uint32) (*models.PigAgeStage, error) {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "GetPigAgeStageByID")
ageStage, err := s.pigTypeRepo.GetPigAgeStageByID(serviceCtx, id)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, ErrPigAgeStageNotFound
}
return nil, fmt.Errorf("获取猪年龄阶段失败: %w", err)
}
return ageStage, nil
}
// UpdatePigAgeStage 实现了更新猪年龄阶段的核心业务逻辑
func (s *recipeServiceImpl) UpdatePigAgeStage(ctx context.Context, ageStage *models.PigAgeStage) error {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "UpdatePigAgeStage")
if err := s.pigTypeRepo.UpdatePigAgeStage(serviceCtx, ageStage); err != nil {
return fmt.Errorf("更新猪年龄阶段失败: %w", err)
}
return nil
}
// DeletePigAgeStage 实现了删除猪年龄阶段的核心业务逻辑
func (s *recipeServiceImpl) DeletePigAgeStage(ctx context.Context, id uint32) error {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "DeletePigAgeStage")
// 检查是否有猪类型关联到该年龄阶段
opts := repository.PigTypeListOptions{AgeStageID: &id}
pigTypes, _, err := s.pigTypeRepo.ListPigTypes(serviceCtx, opts, 1, 1) // 只需检查是否存在，所以取1条
if err != nil {
return fmt.Errorf("检查猪年龄阶段关联失败: %w", err)
}
if len(pigTypes) > 0 {
return ErrPigAgeStageInUse
}
// 检查实体是否存在
_, err = s.pigTypeRepo.GetPigAgeStageByID(serviceCtx, id)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return ErrPigAgeStageNotFound
}
return fmt.Errorf("获取待删除的猪年龄阶段失败: %w", err)
}
if err := s.pigTypeRepo.DeletePigAgeStage(serviceCtx, id); err != nil {
return fmt.Errorf("删除猪年龄阶段失败: %w", err)
}
return nil
}
// ListPigAgeStages 实现了列出猪年龄阶段的逻辑
func (s *recipeServiceImpl) ListPigAgeStages(ctx context.Context, opts repository.PigAgeStageListOptions, page, pageSize int) ([]models.PigAgeStage, int64, error) {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "ListPigAgeStages")
ageStages, total, err := s.pigTypeRepo.ListPigAgeStages(serviceCtx, opts, page, pageSize)
if err != nil {
return nil, 0, fmt.Errorf("获取猪年龄阶段列表失败: %w", err)
}
return ageStages, total, nil
}
// CreatePigType 实现了创建猪类型的核心业务逻辑
func (s *recipeServiceImpl) CreatePigType(ctx context.Context, pigType *models.PigType) error {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "CreatePigType")
if err := s.pigTypeRepo.CreatePigType(serviceCtx, pigType); err != nil {
return fmt.Errorf("创建猪类型失败: %w", err)
}
return nil
}
// GetPigTypeByID 实现了获取单个猪类型的逻辑
func (s *recipeServiceImpl) GetPigTypeByID(ctx context.Context, id uint32) (*models.PigType, error) {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "GetPigTypeByID")
pigType, err := s.pigTypeRepo.GetPigTypeByID(serviceCtx, id)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, ErrPigTypeNotFound
}
return nil, fmt.Errorf("获取猪类型失败: %w", err)
}
return pigType, nil
}
// UpdatePigType 实现了更新猪类型的核心业务逻辑
func (s *recipeServiceImpl) UpdatePigType(ctx context.Context, pigType *models.PigType) error {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "UpdatePigType")
if err := s.pigTypeRepo.UpdatePigType(serviceCtx, pigType); err != nil {
return fmt.Errorf("更新猪类型失败: %m", err)
}
return nil
}
// DeletePigType 实现了删除猪类型的核心业务逻辑
func (s *recipeServiceImpl) DeletePigType(ctx context.Context, id uint32) error {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "DeletePigType")
// 检查实体是否存在
_, err := s.pigTypeRepo.GetPigTypeByID(serviceCtx, id)
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return ErrPigTypeNotFound
}
return fmt.Errorf("获取待删除的猪类型失败: %w", err)
}
if err := s.pigTypeRepo.DeletePigType(serviceCtx, id); err != nil {
return fmt.Errorf("删除猪类型失败: %w", err)
}
return nil
}
// ListPigTypes 实现了列出猪类型的逻辑
func (s *recipeServiceImpl) ListPigTypes(ctx context.Context, opts repository.PigTypeListOptions, page, pageSize int) ([]models.PigType, int64, error) {
serviceCtx := logs.AddFuncName(ctx, s.ctx, "ListPigTypes")
pigTypes, total, err := s.pigTypeRepo.ListPigTypes(serviceCtx, opts, page, pageSize)
if err != nil {
return nil, 0, fmt.Errorf("获取猪类型列表失败: %w", err)
}
return pigTypes, total, nil
}

25
internal/infra/models/pig.go
View File

@@ -4,7 +4,7 @@ package models
type PigBreed struct {
Model
Name string `gorm:"size:50;not null;comment:品种名称"`
Description string `gorm:"type:text" json:"description"` // 保留描述字段
Description string `gorm:"type:text" json:"description"` // 其他描述
ParentInfo string `gorm:"type:text" json:"parent_info"` // 父母信息
AppearanceFeatures string `gorm:"type:text" json:"appearance_features"` // 外貌特征
BreedAdvantages string `gorm:"type:text" json:"breed_advantages"` // 品种优点
@@ -29,17 +29,18 @@ func (PigAgeStage) TableName() string {
// PigType 猪类型模型，代表特定品种和年龄阶段的组合
type PigType struct {
Model
BreedID uint32 `gorm:"not null;index;comment:关联的猪品种ID"`
Breed PigBreed `gorm:"foreignKey:BreedID"`
AgeStageID uint32 `gorm:"not null;index;comment:关联的猪年龄阶段ID"`
AgeStage PigAgeStage `gorm:"foreignKey:AgeStageID"`
Description string `gorm:"size:255;comment:该猪类型的描述或特点"`
DailyFeedIntake float32 `gorm:"comment:理论日均食量 (g/天)"`
DailyGainWeight float32 `gorm:"comment:理论日增重 (g/天)"`
MinDays uint32 `gorm:"comment:该猪类型在该年龄阶段的最小日龄"`
MaxDays uint32 `gorm:"comment:该猪类型在该年龄阶段的最大日龄"`
MinWeight float32 `gorm:"comment:该猪类型在该年龄阶段的最小体重 (g)"`
MaxWeight float32 `gorm:"comment:该猪类型在该年龄阶段的最大体重 (g)"`
BreedID uint32 `gorm:"not null;index;comment:关联的猪品种ID"`
Breed PigBreed `gorm:"foreignKey:BreedID"`
AgeStageID uint32 `gorm:"not null;index;comment:关联的猪年龄阶段ID"`
AgeStage PigAgeStage `gorm:"foreignKey:AgeStageID"`
Description string `gorm:"size:255;comment:该猪类型的描述或特点"`
DailyFeedIntake float32 `gorm:"comment:理论日均食量 (g/天)"`
DailyGainWeight float32 `gorm:"comment:理论日增重 (g/天)"`
MinDays uint32 `gorm:"comment:该猪类型在该年龄阶段的最小日龄"`
MaxDays uint32 `gorm:"comment:该猪类型在该年龄阶段的最大日龄"`
MinWeight float32 `gorm:"comment:该猪类型在该年龄阶段的最小体重 (g)"`
MaxWeight float32 `gorm:"comment:该猪类型在该年龄阶段的最大体重 (g)"`
PigNutrientRequirements []PigNutrientRequirement `gorm:"foreignKey:PigTypeID"`
}
func (PigType) TableName() string {

281
internal/infra/repository/pig_type_repository.go Normal file
View File

@@ -0,0 +1,281 @@
package repository
import (
"context"
"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
"gorm.io/gorm"
)
// PigBreedListOptions 定义了查询猪品种记录时的可选参数
type PigBreedListOptions struct {
Name *string // 品种名称
OrderBy string // 例如 "name asc"
}
// PigAgeStageListOptions 定义了查询猪年龄阶段记录时的可选参数
type PigAgeStageListOptions struct {
Name *string // 年龄阶段名称
OrderBy string // 例如 "name asc"
}
// PigTypeListOptions 定义了查询猪类型记录时的可选参数
type PigTypeListOptions struct {
BreedID *uint32 // 关联的猪品种ID
AgeStageID *uint32 // 关联的猪年龄阶段ID
BreedName *string // 关联的猪品种名称 (用于模糊查询)
AgeStageName *string // 关联的猪年龄阶段名称 (用于模糊查询)
OrderBy string // 例如 "id desc"
}
// PigTypeRepository 定义了猪品种、猪年龄阶段和猪类型数据持久化的接口。
type PigTypeRepository interface {
// <-- 猪品种相关接口 -->
// CreatePigBreed 在数据库中创建一条猪品种记录。
CreatePigBreed(ctx context.Context, breed *models.PigBreed) error
// GetPigBreedByID 根据ID获取猪品种记录。
GetPigBreedByID(ctx context.Context, id uint32) (*models.PigBreed, error)
// UpdatePigBreed 更新猪品种记录。
UpdatePigBreed(ctx context.Context, breed *models.PigBreed) error
// DeletePigBreed 根据ID删除猪品种记录。
DeletePigBreed(ctx context.Context, id uint32) error
// ListPigBreeds 支持分页和过滤的猪品种记录列表查询。
ListPigBreeds(ctx context.Context, opts PigBreedListOptions, page, pageSize int) ([]models.PigBreed, int64, error)
// <-- 猪年龄阶段相关接口 -->
// CreatePigAgeStage 在数据库中创建一条猪年龄阶段记录。
CreatePigAgeStage(ctx context.Context, ageStage *models.PigAgeStage) error
// GetPigAgeStageByID 根据ID获取猪年龄阶段记录。
GetPigAgeStageByID(ctx context.Context, id uint32) (*models.PigAgeStage, error)
// UpdatePigAgeStage 更新猪年龄阶段记录。
UpdatePigAgeStage(ctx context.Context, ageStage *models.PigAgeStage) error
// DeletePigAgeStage 根据ID删除猪年龄阶段记录。
DeletePigAgeStage(ctx context.Context, id uint32) error
// ListPigAgeStages 支持分页和过滤的猪年龄阶段记录列表查询。
ListPigAgeStages(ctx context.Context, opts PigAgeStageListOptions, page, pageSize int) ([]models.PigAgeStage, int64, error)
// <-- 猪类型相关接口 -->
// CreatePigType 在数据库中创建一条猪类型记录。
CreatePigType(ctx context.Context, pigType *models.PigType) error
// GetPigTypeByID 根据ID获取猪类型记录。
GetPigTypeByID(ctx context.Context, id uint32) (*models.PigType, error)
// UpdatePigType 更新猪类型记录。
UpdatePigType(ctx context.Context, pigType *models.PigType) error
// DeletePigType 根据ID删除猪类型记录。
DeletePigType(ctx context.Context, id uint32) error
// ListPigTypes 支持分页和过滤的猪类型记录列表查询。
ListPigTypes(ctx context.Context, opts PigTypeListOptions, page, pageSize int) ([]models.PigType, int64, error)
}
// gormPigTypeRepository 是 PigTypeRepository 接口的 GORM 实现。
type gormPigTypeRepository struct {
ctx context.Context
db *gorm.DB
}
// NewGormPigTypeRepository 创建一个新的 PigTypeRepository GORM 实现实例。
func NewGormPigTypeRepository(ctx context.Context, db *gorm.DB) PigTypeRepository {
return &gormPigTypeRepository{ctx: ctx, db: db}
}
// CreatePigBreed 实现了在数据库中创建猪品种记录的逻辑。
func (r *gormPigTypeRepository) CreatePigBreed(ctx context.Context, breed *models.PigBreed) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "CreatePigBreed")
return r.db.WithContext(repoCtx).Create(breed).Error
}
// GetPigBreedByID 实现了根据ID获取猪品种记录的逻辑。
func (r *gormPigTypeRepository) GetPigBreedByID(ctx context.Context, id uint32) (*models.PigBreed, error) {
repoCtx := logs.AddFuncName(ctx, r.ctx, "GetPigBreedByID")
var breed models.PigBreed
err := r.db.WithContext(repoCtx).First(&breed, id).Error
if err != nil {
return nil, err
}
return &breed, nil
}
// UpdatePigBreed 实现了更新猪品种记录的逻辑。
func (r *gormPigTypeRepository) UpdatePigBreed(ctx context.Context, breed *models.PigBreed) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "UpdatePigBreed")
return r.db.WithContext(repoCtx).Save(breed).Error
}
// DeletePigBreed 实现了根据ID删除猪品种记录的逻辑。
func (r *gormPigTypeRepository) DeletePigBreed(ctx context.Context, id uint32) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "DeletePigBreed")
return r.db.WithContext(repoCtx).Delete(&models.PigBreed{}, id).Error
}
// ListPigBreeds 实现了分页和过滤查询猪品种记录的功能。
func (r *gormPigTypeRepository) ListPigBreeds(ctx context.Context, opts PigBreedListOptions, page, pageSize int) ([]models.PigBreed, int64, error) {
repoCtx := logs.AddFuncName(ctx, r.ctx, "ListPigBreeds")
if page <= 0 || pageSize <= 0 {
return nil, 0, ErrInvalidPagination
}
var results []models.PigBreed
var total int64
query := r.db.WithContext(repoCtx).Model(&models.PigBreed{})
if opts.Name != nil {
query = query.Where("name LIKE ?", "%"+*opts.Name+"%")
}
if err := query.Count(&total).Error; err != nil {
return nil, 0, err
}
orderBy := "id DESC" // 默认排序
if opts.OrderBy != "" {
orderBy = opts.OrderBy
}
query = query.Order(orderBy)
offset := (page - 1) * pageSize
err := query.Limit(pageSize).Offset(offset).Find(&results).Error
return results, total, err
}
// CreatePigAgeStage 实现了在数据库中创建猪年龄阶段记录的逻辑。
func (r *gormPigTypeRepository) CreatePigAgeStage(ctx context.Context, ageStage *models.PigAgeStage) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "CreatePigAgeStage")
return r.db.WithContext(repoCtx).Create(ageStage).Error
}
// GetPigAgeStageByID 实现了根据ID获取猪年龄阶段记录的逻辑。
func (r *gormPigTypeRepository) GetPigAgeStageByID(ctx context.Context, id uint32) (*models.PigAgeStage, error) {
repoCtx := logs.AddFuncName(ctx, r.ctx, "GetPigAgeStageByID")
var ageStage models.PigAgeStage
err := r.db.WithContext(repoCtx).First(&ageStage, id).Error
if err != nil {
return nil, err
}
return &ageStage, nil
}
// UpdatePigAgeStage 实现了更新猪年龄阶段记录的逻辑。
func (r *gormPigTypeRepository) UpdatePigAgeStage(ctx context.Context, ageStage *models.PigAgeStage) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "UpdatePigAgeStage")
return r.db.WithContext(repoCtx).Save(ageStage).Error
}
// DeletePigAgeStage 实现了根据ID删除猪年龄阶段记录的逻辑。
func (r *gormPigTypeRepository) DeletePigAgeStage(ctx context.Context, id uint32) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "DeletePigAgeStage")
return r.db.WithContext(repoCtx).Delete(&models.PigAgeStage{}, id).Error
}
// ListPigAgeStages 实现了分页和过滤查询猪年龄阶段记录的功能。
func (r *gormPigTypeRepository) ListPigAgeStages(ctx context.Context, opts PigAgeStageListOptions, page, pageSize int) ([]models.PigAgeStage, int64, error) {
repoCtx := logs.AddFuncName(ctx, r.ctx, "ListPigAgeStages")
if page <= 0 || pageSize <= 0 {
return nil, 0, ErrInvalidPagination
}
var results []models.PigAgeStage
var total int64
query := r.db.WithContext(repoCtx).Model(&models.PigAgeStage{})
if opts.Name != nil {
query = query.Where("name LIKE ?", "%"+*opts.Name+"%")
}
if err := query.Count(&total).Error; err != nil {
return nil, 0, err
}
orderBy := "id DESC" // 默认排序
if opts.OrderBy != "" {
orderBy = opts.OrderBy
}
query = query.Order(orderBy)
offset := (page - 1) * pageSize
err := query.Limit(pageSize).Offset(offset).Find(&results).Error
return results, total, err
}
// CreatePigType 实现了在数据库中创建猪类型记录的逻辑。
func (r *gormPigTypeRepository) CreatePigType(ctx context.Context, pigType *models.PigType) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "CreatePigType")
return r.db.WithContext(repoCtx).Create(pigType).Error
}
// GetPigTypeByID 实现了根据ID获取猪类型记录的逻辑。
func (r *gormPigTypeRepository) GetPigTypeByID(ctx context.Context, id uint32) (*models.PigType, error) {
repoCtx := logs.AddFuncName(ctx, r.ctx, "GetPigTypeByID")
var pigType models.PigType
err := r.db.WithContext(repoCtx).Preload("Breed").Preload("AgeStage").First(&pigType, id).Error
if err != nil {
return nil, err
}
return &pigType, nil
}
// UpdatePigType 实现了更新猪类型记录的逻辑。
func (r *gormPigTypeRepository) UpdatePigType(ctx context.Context, pigType *models.PigType) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "UpdatePigType")
return r.db.WithContext(repoCtx).Save(pigType).Error
}
// DeletePigType 实现了根据ID删除猪类型记录的逻辑。
func (r *gormPigTypeRepository) DeletePigType(ctx context.Context, id uint32) error {
repoCtx := logs.AddFuncName(ctx, r.ctx, "DeletePigType")
return r.db.WithContext(repoCtx).Delete(&models.PigType{}, id).Error
}
// ListPigTypes 实现了分页和过滤查询猪类型记录的功能。
func (r *gormPigTypeRepository) ListPigTypes(ctx context.Context, opts PigTypeListOptions, page, pageSize int) ([]models.PigType, int64, error) {
repoCtx := logs.AddFuncName(ctx, r.ctx, "ListPigTypes")
if page <= 0 || pageSize <= 0 {
return nil, 0, ErrInvalidPagination
}
var results []models.PigType
var total int64
query := r.db.WithContext(repoCtx).Model(&models.PigType{})
if opts.BreedID != nil {
query = query.Where("breed_id = ?", *opts.BreedID)
}
if opts.AgeStageID != nil {
query = query.Where("age_stage_id = ?", *opts.AgeStageID)
}
if opts.BreedName != nil {
query = query.Joins("left join pig_breeds on pig_types.breed_id = pig_breeds.id").
Where("pig_breeds.name LIKE ?", "%"+*opts.BreedName+"%")
}
if opts.AgeStageName != nil {
query = query.Joins("left join pig_age_stages on pig_types.age_stage_id = pig_age_stages.id").
Where("pig_age_stages.name LIKE ?", "%"+*opts.AgeStageName+"%")
}
if err := query.Count(&total).Error; err != nil {
return nil, 0, err
}
orderBy := "id DESC" // 默认排序
if opts.OrderBy != "" {
orderBy = opts.OrderBy
}
query = query.Order(orderBy)
offset := (page - 1) * pageSize
err := query.Limit(pageSize).Offset(offset).Preload("Breed").Preload("AgeStage").Find(&results).Error
return results, total, err
}

4
project_structure.txt
View File

@@ -1,4 +1,6 @@

.continue/mcpServers/new-mcp-server.yaml
.continue/rules/new-rule.md
.gitignore
AGENTS.md
Makefile
@@ -10,6 +12,7 @@ config/.golangci.yml
config/config.example.yml
config/config.yml
config/presets-data/nutrient.json
config/presets-data/pig_nutrient_requirement.json
config/presets-data/system_plans.json
design/archive/2025-11-03-verification-before-device-deletion/add_get_device_id_configs_to_task.md
design/archive/2025-11-03-verification-before-device-deletion/check_before_device_deletion.md
@@ -163,6 +166,7 @@ internal/infra/repository/pig_pen_repository.go
internal/infra/repository/pig_sick_repository.go
internal/infra/repository/pig_trade_repository.go
internal/infra/repository/pig_transfer_log_repository.go
internal/infra/repository/pig_type_repository.go
internal/infra/repository/plan_repository.go
internal/infra/repository/raw_material_repository.go
internal/infra/repository/repository.go