muyu-apiserver/rpc/purchase/internal/logic/updatepurchaseorderlogic.go
Chever John a25acdc5e4 chore: sync codebase with upstream
Pull latest changes from upstream/main (GitHub) including:
- Purchase RPC service (proto, server, logic, models)
- Gateway route updates for purchase endpoints
- SQL migration and config updates

Excludes deploy/bin/ pre-compiled arm64 binaries (builds use
multi-stage Dockerfiles targeting amd64).
2026-06-15 09:20:56 +08:00

88 lines
2.4 KiB
Go

package logic
import (
"context"
"database/sql"
"strconv"
"time"
"muyu-apiserver/model"
"muyu-apiserver/pkg/tenantctx"
"muyu-apiserver/pkg/uid"
"muyu-apiserver/rpc/purchase/internal/svc"
"muyu-apiserver/rpc/purchase/pb"
"github.com/zeromicro/go-zero/core/logx"
)
type UpdatePurchaseOrderLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewUpdatePurchaseOrderLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdatePurchaseOrderLogic {
return &UpdatePurchaseOrderLogic{ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx)}
}
func (l *UpdatePurchaseOrderLogic) UpdatePurchaseOrder(in *pb.UpdatePurchaseOrderReq) (*pb.Empty, error) {
tenantId := tenantctx.ExtractTenantId(l.ctx)
order, err := l.svcCtx.OrderModel.FindOneByOrderId(l.ctx, in.OrderId)
if err != nil {
return nil, ErrOrderNotFound
}
if order.Status != 0 {
return nil, ErrOrderNotDraft
}
orderDate, err := time.Parse("2006-01-02", in.OrderDate)
if err != nil {
orderDate = order.OrderDate
}
var contractAmount float64
for _, d := range in.Details {
qty, _ := strconv.ParseFloat(d.Quantity, 64)
price, _ := strconv.ParseFloat(d.UnitPrice, 64)
contractAmount += qty * price
}
order.SupplierId = in.SupplierId
order.OrderDate = orderDate
order.ContractAmount = contractAmount
order.PurchaseBy = in.PurchaseBy
order.Operator = tenantId
order.Remark = sql.NullString{String: in.Remark, Valid: in.Remark != ""}
if err := l.svcCtx.OrderModel.Update(l.ctx, order); err != nil {
return nil, err
}
// Replace details
if err := l.svcCtx.OrderDetailModel.DeleteByOrderId(l.ctx, in.OrderId); err != nil {
return nil, err
}
details := make([]*model.PurPurchaseOrderDetail, 0, len(in.Details))
for _, d := range in.Details {
qty, _ := strconv.ParseFloat(d.Quantity, 64)
price, _ := strconv.ParseFloat(d.UnitPrice, 64)
details = append(details, &model.PurPurchaseOrderDetail{
DetailId: uid.Generate(),
TenantId: tenantId,
OrderId: in.OrderId,
ProductId: d.ProductId,
ProductName: d.ProductName,
Spec: d.Spec,
Color: d.Color,
Quantity: qty,
UnitPrice: price,
Amount: qty * price,
Remark: sql.NullString{String: d.Remark, Valid: d.Remark != ""},
})
}
if err := l.svcCtx.OrderDetailModel.BulkInsert(l.ctx, details); err != nil {
return nil, err
}
return &pb.Empty{}, nil
}