初始化2

bj_power_mes/internal/logic/product.go

@@ -0,0 +1,46 @@
+package logic
+
+import (
+	"context"
+	"errors"
+
+	"bj_power_mes/ent"
+	"bj_power_mes/ent/producttype"
+)
+
+type ProductTypeReq struct {
+	Id       int    `json:"id"`
+	Name     string `json:"name"`
+	Code     string `json:"code"`
+	Category string `json:"category"`
+	Remark   string `json:"remark"`
+	IsActive bool   `json:"isActive"`
+}
+
+func (s *Service) CreateProductType(ctx context.Context, req ProductTypeReq) error {
+	active := req.IsActive
+	if !req.IsActive {
+		active = true
+	}
+	return s.ctx.EntClient.ProductType.Create().
+		SetName(req.Name).SetCode(req.Code).SetCategory(req.Category).
+		SetRemark(req.Remark).SetIsActive(active).Exec(ctx)
+}
+
+func (s *Service) UpdateProductType(ctx context.Context, req ProductTypeReq) error {
+	if req.Id == 0 {
+		return errors.New("缺少 id")
+	}
+	return s.ctx.EntClient.ProductType.UpdateOneID(req.Id).
+		SetName(req.Name).SetCode(req.Code).SetCategory(req.Category).
+		SetRemark(req.Remark).SetIsActive(req.IsActive).Exec(ctx)
+}
+
+func (s *Service) DeleteProductType(ctx context.Context, id int) error {
+	return s.ctx.EntClient.ProductType.DeleteOneID(id).Exec(ctx)
+}
+
+func (s *Service) ListProductTypes(ctx context.Context) ([]*ent.ProductType, error) {
+	return s.ctx.EntClient.ProductType.Query().
+		Order(ent.Desc(producttype.FieldID)).All(ctx)
+}