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

This commit is contained in:
2025-11-21 15:03:42 +08:00
parent a669bfda6c
commit 9996fcfd74
8 changed files with 565 additions and 76 deletions

View File

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

View File

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

View File

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

View File

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

View File

@@ -14,12 +14,15 @@ import (
// 定义领域特定的错误 // 定义领域特定的错误
var ( var (
ErrNutrientNameConflict = fmt.Errorf("营养种类名称已存在") ErrNutrientNameConflict = fmt.Errorf("营养种类名称已存在")
ErrNutrientNotFound = fmt.Errorf("营养种类不存在") ErrNutrientNotFound = fmt.Errorf("营养种类不存在")
ErrNutrientInUse = fmt.Errorf("营养种类正在被原料使用,无法删除")
ErrRawMaterialNameConflict = fmt.Errorf("原料名称已存在") ErrRawMaterialNameConflict = fmt.Errorf("原料名称已存在")
ErrRawMaterialNotFound = fmt.Errorf("原料不存在") ErrRawMaterialNotFound = fmt.Errorf("原料不存在")
ErrPigBreedInUse = fmt.Errorf("猪品种正在被猪类型使用,无法删除")
ErrPigBreedNotFound = fmt.Errorf("猪品种不存在")
ErrPigAgeStageInUse = fmt.Errorf("猪年龄阶段正在被猪类型使用,无法删除")
ErrPigAgeStageNotFound = fmt.Errorf("猪年龄阶段不存在")
ErrPigTypeNotFound = fmt.Errorf("猪类型不存在")
) )
// Service 定义了配方与原料领域的核心业务服务接口 // Service 定义了配方与原料领域的核心业务服务接口
@@ -37,6 +40,27 @@ type Service interface {
DeleteRawMaterial(ctx context.Context, id uint32) error DeleteRawMaterial(ctx context.Context, id uint32) error
GetRawMaterial(ctx context.Context, id uint32) (*models.RawMaterial, error) GetRawMaterial(ctx context.Context, id uint32) (*models.RawMaterial, error)
ListRawMaterials(ctx context.Context, page, pageSize int) ([]models.RawMaterial, int64, 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 的实现 // recipeServiceImpl 是 RecipeService 的实现
@@ -44,14 +68,16 @@ type recipeServiceImpl struct {
ctx context.Context ctx context.Context
nutrientRepo repository.NutrientRepository nutrientRepo repository.NutrientRepository
rawMaterialRepo repository.RawMaterialRepository rawMaterialRepo repository.RawMaterialRepository
pigTypeRepo repository.PigTypeRepository
} }
// NewRecipeService 创建一个新的 RecipeService 实例 // 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{ return &recipeServiceImpl{
ctx: ctx, ctx: ctx,
nutrientRepo: nutrientRepo, nutrientRepo: nutrientRepo,
rawMaterialRepo: rawMaterialRepo, rawMaterialRepo: rawMaterialRepo,
pigTypeRepo: pigTypeRepo,
} }
} }
@@ -262,3 +288,203 @@ func (s *recipeServiceImpl) ListRawMaterials(ctx context.Context, page, pageSize
} }
return rawMaterials, total, nil 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
}

View File

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

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
}

View File

@@ -1,4 +1,6 @@
 
.continue/mcpServers/new-mcp-server.yaml
.continue/rules/new-rule.md
.gitignore .gitignore
AGENTS.md AGENTS.md
Makefile Makefile
@@ -10,6 +12,7 @@ config/.golangci.yml
config/config.example.yml config/config.example.yml
config/config.yml config/config.yml
config/presets-data/nutrient.json config/presets-data/nutrient.json
config/presets-data/pig_nutrient_requirement.json
config/presets-data/system_plans.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/add_get_device_id_configs_to_task.md
design/archive/2025-11-03-verification-before-device-deletion/check_before_device_deletion.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_sick_repository.go
internal/infra/repository/pig_trade_repository.go internal/infra/repository/pig_trade_repository.go
internal/infra/repository/pig_transfer_log_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/plan_repository.go
internal/infra/repository/raw_material_repository.go internal/infra/repository/raw_material_repository.go
internal/infra/repository/repository.go internal/infra/repository/repository.go