From db32c37318dead6b841da374f9905dbe01286e22 Mon Sep 17 00:00:00 2001
From: huang <1724659546@qq.com>
Date: Wed, 22 Oct 2025 17:58:24 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AEbug?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../management/pig_farm_controller.go         | 58 ++++++++++---------
 1 file changed, 32 insertions(+), 26 deletions(-)

diff --git a/internal/app/controller/management/pig_farm_controller.go b/internal/app/controller/management/pig_farm_controller.go
index d31ef31..e787254 100644
--- a/internal/app/controller/management/pig_farm_controller.go
+++ b/internal/app/controller/management/pig_farm_controller.go
@@ -239,12 +239,11 @@ func (c *PigFarmController) CreatePen(ctx *gin.Context) {
 	}
 
 	resp := dto.PenResponse{
-		ID:         pen.ID,
-		PenNumber:  pen.PenNumber,
-		HouseID:    pen.HouseID,
-		Capacity:   pen.Capacity,
-		Status:     pen.Status,
-		PigBatchID: *pen.PigBatchID,
+		ID:        pen.ID,
+		PenNumber: pen.PenNumber,
+		HouseID:   pen.HouseID,
+		Capacity:  pen.Capacity,
+		Status:    pen.Status,
 	}
 	controller.SendSuccessWithAudit(ctx, controller.CodeCreated, "创建成功", resp, action, "创建成功", resp)
 }
@@ -278,12 +277,14 @@ func (c *PigFarmController) GetPen(ctx *gin.Context) {
 	}
 
 	resp := dto.PenResponse{
-		ID:         pen.ID,
-		PenNumber:  pen.PenNumber,
-		HouseID:    pen.HouseID,
-		Capacity:   pen.Capacity,
-		Status:     pen.Status,
-		PigBatchID: *pen.PigBatchID,
+		ID:        pen.ID,
+		PenNumber: pen.PenNumber,
+		HouseID:   pen.HouseID,
+		Capacity:  pen.Capacity,
+		Status:    pen.Status,
+	}
+	if pen.PigBatchID != nil {
+		resp.PigBatchID = *pen.PigBatchID
 	}
 	controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "获取成功", resp, action, "获取成功", resp)
 }
@@ -307,14 +308,17 @@ func (c *PigFarmController) ListPens(ctx *gin.Context) {
 
 	var resp []dto.PenResponse
 	for _, pen := range pens {
-		resp = append(resp, dto.PenResponse{
-			ID:         pen.ID,
-			PenNumber:  pen.PenNumber,
-			HouseID:    pen.HouseID,
-			Capacity:   pen.Capacity,
-			Status:     pen.Status,
-			PigBatchID: *pen.PigBatchID,
-		})
+		res := dto.PenResponse{
+			ID:        pen.ID,
+			PenNumber: pen.PenNumber,
+			HouseID:   pen.HouseID,
+			Capacity:  pen.Capacity,
+			Status:    pen.Status,
+		}
+		if pen.PigBatchID != nil {
+			res.PigBatchID = *pen.PigBatchID
+		}
+		resp = append(resp, res)
 	}
 
 	controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "获取成功", resp, action, "获取成功", resp)
@@ -358,12 +362,14 @@ func (c *PigFarmController) UpdatePen(ctx *gin.Context) {
 	}
 
 	resp := dto.PenResponse{
-		ID:         pen.ID,
-		PenNumber:  pen.PenNumber,
-		HouseID:    pen.HouseID,
-		Capacity:   pen.Capacity,
-		Status:     pen.Status,
-		PigBatchID: *pen.PigBatchID,
+		ID:        pen.ID,
+		PenNumber: pen.PenNumber,
+		HouseID:   pen.HouseID,
+		Capacity:  pen.Capacity,
+		Status:    pen.Status,
+	}
+	if pen.PigBatchID != nil {
+		resp.PigBatchID = *pen.PigBatchID
 	}
 	controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "更新成功", resp, action, "更新成功", resp)
 }