Add video cache TTL enforcement for /vtrack endpoint (#4136)

Author: AntoxaAntoxic

src/main/java/org/prebid/server/cache/CoreCacheService.java

@@ -212,11 +212,12 @@ public Future<BidCacheResponse> cachePutObjects(List<BidPutObject> bidPutObjects
                                                     Boolean isEventsEnabled,
                                                     Set<String> biddersAllowingVastUpdate,
                                                     String accountId,
+                                                    Integer accountTtl,
                                                     String integration,
                                                     Timeout timeout) {
 
-        final List<CachedCreative> cachedCreatives =
-                updatePutObjects(bidPutObjects, isEventsEnabled, biddersAllowingVastUpdate, accountId, integration);
+        final List<CachedCreative> cachedCreatives = updatePutObjects(
+                bidPutObjects, isEventsEnabled, biddersAllowingVastUpdate, accountId, accountTtl, integration);
 
         updateCreativeMetrics(
                 cachedCreatives,
@@ -230,6 +231,7 @@ private List<CachedCreative> updatePutObjects(List<BidPutObject> bidPutObjects,
                                                   Boolean isEventsEnabled,
                                                   Set<String> allowedBidders,
                                                   String accountId,
+                                                  Integer accountTtl,
                                                   String integration) {
 
         return bidPutObjects.stream()
@@ -244,11 +246,18 @@ private List<CachedCreative> updatePutObjects(List<BidPutObject> bidPutObjects,
                                 putObject,
                                 accountId,
                                 integration))
+                        .ttlseconds(resolveVtrackTtl(putObject.getTtlseconds(), accountTtl))
                         .build())
                 .map(payload -> CachedCreative.of(payload, creativeSizeFromTextNode(payload.getValue())))
                 .toList();
     }
 
+    private static Integer resolveVtrackTtl(Integer initialObjectTtl, Integer initialAccountTtl) {
+        final Integer accountTtl = initialAccountTtl != null && initialAccountTtl > 0 ? initialAccountTtl : null;
+        final Integer objectTtl = initialObjectTtl != null && initialObjectTtl > 0 ? initialObjectTtl : null;
+        return ObjectUtils.min(objectTtl, accountTtl);
+    }
+
     public Future<CacheServiceResult> cacheBidsOpenrtb(List<BidInfo> bidsToCache,
                                                        AuctionContext auctionContext,
                                                        CacheContext cacheContext,

src/main/java/org/prebid/server/handler/PostVtrackHandler.java

@@ -29,13 +29,15 @@
 import org.prebid.server.settings.model.Account;
 import org.prebid.server.settings.model.AccountAuctionConfig;
 import org.prebid.server.settings.model.AccountEventsConfig;
+import org.prebid.server.settings.model.AccountVtrackConfig;
 import org.prebid.server.util.HttpUtil;
 import org.prebid.server.vertx.verticles.server.HttpEndpoint;
 import org.prebid.server.vertx.verticles.server.application.ApplicationResource;
 
 import java.util.Collections;
 import java.util.List;
 import java.util.Objects;
+import java.util.Optional;
 import java.util.Set;
 import java.util.stream.Collectors;
 
@@ -180,10 +182,12 @@ private void handleAccountResult(AsyncResult<Account> asyncAccount,
             respondWithServerError(routingContext, "Error occurred while fetching account", asyncAccount.cause());
         } else {
             // insert impression tracking if account allows events and bidder allows VAST modification
-            final Boolean isEventEnabled = accountEventsEnabled(asyncAccount.result());
+            final Account account = asyncAccount.result();
+            final Boolean isEventEnabled = accountEventsEnabled(account);
+            final Integer accountTtl = accountVtrackTtl(account);
             final Set<String> allowedBidders = biddersAllowingVastUpdate(vtrackPuts);
             coreCacheService.cachePutObjects(
-                    vtrackPuts, isEventEnabled, allowedBidders, accountId, integration, timeout)
+                    vtrackPuts, isEventEnabled, allowedBidders, accountId, accountTtl, integration, timeout)
                     .onComplete(asyncCache -> handleCacheResult(asyncCache, routingContext));
         }
     }
@@ -196,6 +200,12 @@ private static Boolean accountEventsEnabled(Account account) {
         return accountEventsConfig != null ? accountEventsConfig.getEnabled() : null;
     }
 
+    private static Integer accountVtrackTtl(Account account) {
+        return Optional.ofNullable(account.getVtrack())
+                .map(AccountVtrackConfig::getTtl)
+                .orElse(null);
+    }
+
     /**
      * Returns list of bidders that allow VAST XML modification.
      */

src/main/java/org/prebid/server/settings/model/Account.java

@@ -30,6 +30,8 @@ public class Account {
     @JsonAlias("alternate-bidder-codes")
     AccountAlternateBidderCodes alternateBidderCodes;
 
+    AccountVtrackConfig vtrack;
+
     public static Account empty(String id) {
         return Account.builder().id(id).build();
     }

src/main/java/org/prebid/server/settings/model/AccountVtrackConfig.java

@@ -0,0 +1,9 @@
+package org.prebid.server.settings.model;
+
+import lombok.Value;
+
+@Value(staticConstructor = "of")
+public class AccountVtrackConfig {
+
+    Integer ttl;
+}

src/test/groovy/org/prebid/server/functional/model/config/AccountConfig.groovy

@@ -26,6 +26,7 @@ class AccountConfig {
     AlternateBidderCodes alternateBidderCodes
     @JsonProperty("alternate_bidder_codes")
     AlternateBidderCodes alternateBidderCodesSnakeCase
+    AccountVtrackConfig vtrack
 
     static getDefaultAccountConfig() {
         new AccountConfig(status: AccountStatus.ACTIVE)

src/test/groovy/org/prebid/server/functional/model/config/AccountVtrackConfig.groovy

@@ -0,0 +1,9 @@
+package org.prebid.server.functional.model.config
+
+import groovy.transform.ToString
+
+@ToString(includeNames = true, ignoreNulls = true)
+class AccountVtrackConfig {
+
+    Integer ttl
+}

src/test/groovy/org/prebid/server/functional/service/PrebidServerService.groovy

@@ -215,7 +215,7 @@ class PrebidServerService implements ObjectMapperWrapper {
     }
 
     PrebidCacheResponse sendPostVtrackRequest(VtrackRequest request, String account) {
-        def response = given(requestSpecification).queryParam("a", account)
+        def response = given(requestSpecification).queryParams(["a": account])
                 .body(request)
                 .post(VTRACK_ENDPOINT)