Merge pull request #51 from G-Core/feature/CDI-468_add_origin_shielding

CDI-468: fix error with null value
G-Core · Mar 12, 2024 · 866d385 · 866d385
2 parents e256a19 + 09638f0
commit 866d385
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/originshielding/originshielding.go b/originshielding/originshielding.go
@@ -7,15 +7,15 @@ import (
 type OriginShieldingService interface {
 	Update(ctx context.Context, id int64, req *UpdateRequest) (*OriginShieldingData, error)
 	Get(ctx context.Context, id int64) (*OriginShieldingData, error)
-	GetLocations(ctx context.Context) ([]OriginShieldingLocations, error)
+	GetLocations(ctx context.Context) (*[]OriginShieldingLocations, error)
 }
 
 type OriginShieldingData struct {
-	ShieldingPop int `json:"shielding_pop"`
+	ShieldingPop *int `json:"shielding_pop"`
 }
 
 type UpdateRequest struct {
-	ShieldingPop int `json:"shielding_pop"`
+	ShieldingPop *int `json:"shielding_pop"`
 }
 
 type OriginShieldingLocations struct {

diff --git a/originshielding/service.go b/originshielding/service.go
@@ -35,11 +35,11 @@ func (s *Service) Get(ctx context.Context, id int64) (*OriginShieldingData, erro
 
 }
 
-func (s *Service) GetLocations(ctx context.Context) ([]OriginShieldingLocations, error) {
+func (s *Service) GetLocations(ctx context.Context) (*[]OriginShieldingLocations, error) {
 	var originShieldingLocations []OriginShieldingLocations
-	if err := s.r.Request(ctx, http.MethodGet, "/cdn/shieldingpop_v2", nil, originShieldingLocations); err != nil {
+	if err := s.r.Request(ctx, http.MethodGet, "/cdn/shieldingpop_v2", nil, &originShieldingLocations); err != nil {
 		return nil, fmt.Errorf("request: %w", err)
 	}
 
-	return originShieldingLocations, nil
+	return &originShieldingLocations, nil
 }