阶段性提交

internal/rpc/services/sevice_http_gzip.go

@@ -0,0 +1,109 @@
+package services
+
+import (
+	"context"
+	"encoding/json"
+	"github.com/TeaOSLab/EdgeAPI/internal/db/models"
+	rpcutils "github.com/TeaOSLab/EdgeAPI/internal/rpc/utils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs/shared"
+)
+
+type HTTPGzipService struct {
+}
+
+// 创建Gzip配置
+func (this *HTTPGzipService) CreateHTTPGzip(ctx context.Context, req *pb.CreateHTTPGzipRequest) (*pb.CreateHTTPGzipResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	minLengthJSON := []byte{}
+	if req.MinLength != nil {
+		minLengthJSON, err = (&shared.SizeCapacity{
+			Count: req.MinLength.Count,
+			Unit:  req.MinLength.Unit,
+		}).AsJSON()
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	maxLengthJSON := []byte{}
+	if req.MaxLength != nil {
+		maxLengthJSON, err = (&shared.SizeCapacity{
+			Count: req.MaxLength.Count,
+			Unit:  req.MaxLength.Unit,
+		}).AsJSON()
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	gzipId, err := models.SharedHTTPGzipDAO.CreateGzip(int(req.Level), minLengthJSON, maxLengthJSON)
+	if err != nil {
+		return nil, err
+	}
+
+	return &pb.CreateHTTPGzipResponse{GzipId: gzipId}, nil
+}
+
+// 查找Gzip
+func (this *HTTPGzipService) FindEnabledHTTPGzipConfig(ctx context.Context, req *pb.FindEnabledGzipConfigRequest) (*pb.FindEnabledGzipConfigResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	config, err := models.SharedHTTPGzipDAO.ComposeGzipConfig(req.GzipId)
+	if err != nil {
+		return nil, err
+	}
+
+	configData, err := json.Marshal(config)
+	if err != nil {
+		return nil, err
+	}
+	return &pb.FindEnabledGzipConfigResponse{Config: configData}, nil
+}
+
+// 修改Gzip配置
+func (this *HTTPGzipService) UpdateHTTPGzip(ctx context.Context, req *pb.UpdateHTTPGzipRequest) (*pb.UpdateHTTPGzipResponse, error) {
+	// 校验请求
+	_, _, err := rpcutils.ValidateRequest(ctx, rpcutils.UserTypeAdmin)
+	if err != nil {
+		return nil, err
+	}
+
+	minLengthJSON := []byte{}
+	if req.MinLength != nil {
+		minLengthJSON, err = (&shared.SizeCapacity{
+			Count: req.MinLength.Count,
+			Unit:  req.MinLength.Unit,
+		}).AsJSON()
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	maxLengthJSON := []byte{}
+	if req.MaxLength != nil {
+		maxLengthJSON, err = (&shared.SizeCapacity{
+			Count: req.MaxLength.Count,
+			Unit:  req.MaxLength.Unit,
+		}).AsJSON()
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	err = models.SharedHTTPGzipDAO.UpdateGzip(req.GzipId, int(req.Level), minLengthJSON, maxLengthJSON)
+	if err != nil {
+		return nil, err
+	}
+
+	return &pb.UpdateHTTPGzipResponse{}, nil
+}