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-20 13:43:09 +08:00
parent fd0939fe0a
commit c697e668e3
5 changed files with 296 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
internal/infra/repository/nutrient_repository.go
View File

@@ -38,33 +38,29 @@ func (r *gormNutrientRepository) CreateNutrient(ctx context.Context, nutrient *m
return r.db.WithContext(repoCtx).Create(nutrient).Error
}
// GetNutrientByID 根据ID获取单个营养种类
// GetNutrientByID 根据ID获取单个营养种类，并预加载关联的原料信息
func (r *gormNutrientRepository) GetNutrientByID(ctx context.Context, id uint32) (*models.Nutrient, error) {
repoCtx := logs.AddFuncName(ctx, r.ctx, "GetNutrientByID")
var nutrient models.Nutrient
if err := r.db.WithContext(repoCtx).First(&nutrient, id).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, nil // 记录未找到不应视为错误
}
// 如果记录未找到，GORM 会返回 gorm.ErrRecordNotFound 错误
if err := r.db.WithContext(repoCtx).Preload("RawMaterialNutrients.RawMaterial").First(&nutrient, id).Error; err != nil {
return nil, err
}
return &nutrient, nil
}
// GetNutrientByName 根据名称获取单个营养种类
// GetNutrientByName 根据名称获取单个营养种类，并预加载关联的原料信息
func (r *gormNutrientRepository) GetNutrientByName(ctx context.Context, name string) (*models.Nutrient, error) {
repoCtx := logs.AddFuncName(ctx, r.ctx, "GetNutrientByName")
var nutrient models.Nutrient
if err := r.db.WithContext(repoCtx).Where("name = ?", name).First(&nutrient).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, nil // 记录未找到不应视为错误
}
// 如果记录未找到，GORM 会返回 gorm.ErrRecordNotFound 错误
if err := r.db.WithContext(repoCtx).Preload("RawMaterialNutrients.RawMaterial").Where("name = ?", name).First(&nutrient).Error; err != nil {
return nil, err
}
return &nutrient, nil
}
// ListNutrients 列出所有营养种类（分页）
// ListNutrients 列出所有营养种类（分页），并预加载关联的原料信息
func (r *gormNutrientRepository) ListNutrients(ctx context.Context, page, pageSize int) ([]models.Nutrient, int64, error) {
repoCtx := logs.AddFuncName(ctx, r.ctx, "ListNutrients")
var nutrients []models.Nutrient
@@ -79,7 +75,7 @@ func (r *gormNutrientRepository) ListNutrients(ctx context.Context, page, pageSi
// 然后应用分页并获取数据
offset := (page - 1) * pageSize
if err := db.Offset(offset).Limit(pageSize).Find(&nutrients).Error; err != nil {
if err := db.Preload("RawMaterialNutrients.RawMaterial").Offset(offset).Limit(pageSize).Find(&nutrients).Error; err != nil {
return nil, 0, err
}