1.4.5.2

EdgeAPI/internal/rpc/services/service_price_plus.go

@@ -0,0 +1,34 @@
+// Copyright 2022 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
+//go:build plus
+
+package services
+
+import (
+	"context"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+)
+
+// PriceService 价格相关服务
+type PriceService struct {
+	BaseService
+}
+
+// CalculatePrice 计算费用
+func (this *PriceService) CalculatePrice(ctx context.Context, req *pb.CalculatePriceRequest) (*pb.CalculatePriceResponse, error) {
+	_, _, err := this.ValidateAdminAndUser(ctx, true)
+	if err != nil {
+		return nil, err
+	}
+
+	var tx = this.NullTx()
+	amount, hasRegionPrice, err := models.SharedUserBillDAO.CalculatePrice(tx, req.PriceType, req.TrafficGB, req.BandwidthMB, req.NodeRegionId)
+	if err != nil {
+		return nil, err
+	}
+
+	return &pb.CalculatePriceResponse{
+		Amount:             amount,
+		HasNodeRegionPrice: hasRegionPrice,
+	}, nil
+}