Fix setting cache key policy

perfectsense · Oct 20, 2023 · fa1de0e · fa1de0e
1 parent 2f3ecfc
commit fa1de0e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/main/java/gyro/google/compute/BackendBucketCdnPolicy.java b/src/main/java/gyro/google/compute/BackendBucketCdnPolicy.java
@@ -20,7 +20,6 @@
 import java.util.List;
 import java.util.stream.Collectors;
 
-import com.google.cloud.compute.v1.BackendBucketCdnPolicyCacheKeyPolicy;
 import gyro.core.resource.Diffable;
 import gyro.core.resource.Output;
 import gyro.core.resource.Updatable;
@@ -273,7 +272,7 @@ com.google.cloud.compute.v1.BackendBucketCdnPolicy toBackendBucketCdnPolicy() {
         return com.google.cloud.compute.v1.BackendBucketCdnPolicy.newBuilder()
             .setSignedUrlCacheMaxAgeSec(getSignedUrlMaxAge())
             .setCacheMode(getCacheMode())
-            .setCacheKeyPolicy(BackendBucketCdnPolicyCacheKeyPolicy.newBuilder().build())
+            .setCacheKeyPolicy(getCacheKeyPolicy() != null ? getCacheKeyPolicy().toBackendBucketCdnPolicyCacheKeyPolicy() : new BackendBucketCdnCacheKeyPolicy().toBackendBucketCdnPolicyCacheKeyPolicy())
             .addAllBypassCacheOnRequestHeaders(getBypassCacheOnRequestHeader().stream()
                 .map(BackendBucketCdnBypassCacheOnRequestHeader::toBackendBucketCdnPolicyBypassCacheOnRequestHeader)
                 .collect(Collectors.toList()))