|
| 1 | +package org.prebid.server.hooks.modules.greenbids.real.time.data.core; |
| 2 | + |
| 3 | +import com.fasterxml.jackson.databind.JsonNode; |
| 4 | +import com.iab.openrtb.request.BidRequest; |
| 5 | +import com.iab.openrtb.request.Imp; |
| 6 | +import org.apache.commons.lang3.ObjectUtils; |
| 7 | +import org.apache.commons.lang3.StringUtils; |
| 8 | +import org.prebid.server.analytics.reporter.greenbids.model.ExplorationResult; |
| 9 | +import org.prebid.server.analytics.reporter.greenbids.model.Ortb2ImpExtResult; |
| 10 | +import org.prebid.server.hooks.modules.greenbids.real.time.data.model.data.GreenbidsConfig; |
| 11 | +import org.prebid.server.hooks.modules.greenbids.real.time.data.model.result.AnalyticsResult; |
| 12 | +import org.prebid.server.hooks.modules.greenbids.real.time.data.model.result.GreenbidsInvocationResult; |
| 13 | +import org.prebid.server.hooks.v1.InvocationAction; |
| 14 | + |
| 15 | +import java.util.Map; |
| 16 | +import java.util.Optional; |
| 17 | +import java.util.UUID; |
| 18 | +import java.util.stream.Collectors; |
| 19 | + |
| 20 | +public class GreenbidsInvocationResultCreator { |
| 21 | + |
| 22 | + private static final int RANGE_16_BIT_INTEGER_DIVISION_BASIS = 0x10000; |
| 23 | + private static final double DEFAULT_EXPLORATION_RATE = 1.0; |
| 24 | + |
| 25 | + private GreenbidsInvocationResultCreator() { |
| 26 | + |
| 27 | + } |
| 28 | + |
| 29 | + public static GreenbidsInvocationResult create(GreenbidsConfig greenbidsConfig, |
| 30 | + BidRequest bidRequest, |
| 31 | + Map<String, Map<String, Boolean>> impsBiddersFilterMap) { |
| 32 | + |
| 33 | + final String greenbidsId = UUID.randomUUID().toString(); |
| 34 | + final boolean isExploration = isExploration(greenbidsConfig, greenbidsId); |
| 35 | + |
| 36 | + final boolean allRejected = bidRequest.getImp().stream() |
| 37 | + .noneMatch(imp -> impsBiddersFilterMap.get(imp.getId()).values().stream().anyMatch(isKept -> isKept)); |
| 38 | + |
| 39 | + final InvocationAction invocationAction = isExploration |
| 40 | + ? InvocationAction.no_action |
| 41 | + : allRejected |
| 42 | + ? InvocationAction.reject |
| 43 | + : InvocationAction.update; |
| 44 | + |
| 45 | + final Map<String, Ortb2ImpExtResult> ort2ImpExtResultMap = createOrtb2ImpExtForImps( |
| 46 | + bidRequest, impsBiddersFilterMap, greenbidsId, isExploration); |
| 47 | + final AnalyticsResult analyticsResult = AnalyticsResult.of("success", ort2ImpExtResultMap); |
| 48 | + return GreenbidsInvocationResult.of(invocationAction, analyticsResult); |
| 49 | + } |
| 50 | + |
| 51 | + private static boolean isExploration(GreenbidsConfig greenbidsConfig, String greenbidsId) { |
| 52 | + final double explorationRate = ObjectUtils.defaultIfNull( |
| 53 | + greenbidsConfig.getExplorationRate(), |
| 54 | + DEFAULT_EXPLORATION_RATE); |
| 55 | + final int hashInt = Integer.parseInt(greenbidsId.substring(greenbidsId.length() - 4), 16); |
| 56 | + return hashInt < explorationRate * RANGE_16_BIT_INTEGER_DIVISION_BASIS; |
| 57 | + } |
| 58 | + |
| 59 | + private static Map<String, Ortb2ImpExtResult> createOrtb2ImpExtForImps( |
| 60 | + BidRequest bidRequest, |
| 61 | + Map<String, Map<String, Boolean>> impsBiddersFilterMap, |
| 62 | + String greenbidsId, |
| 63 | + boolean isExploration) { |
| 64 | + |
| 65 | + return bidRequest.getImp().stream() |
| 66 | + .collect(Collectors.toMap( |
| 67 | + Imp::getId, |
| 68 | + imp -> createOrtb2ImpExt(imp, impsBiddersFilterMap, greenbidsId, isExploration))); |
| 69 | + } |
| 70 | + |
| 71 | + private static Ortb2ImpExtResult createOrtb2ImpExt(Imp imp, |
| 72 | + Map<String, Map<String, Boolean>> impsBiddersFilterMap, |
| 73 | + String greenbidsId, |
| 74 | + boolean isExploration) { |
| 75 | + |
| 76 | + final String tid = Optional.ofNullable(imp) |
| 77 | + .map(Imp::getExt) |
| 78 | + .map(impExt -> impExt.get("tid")) |
| 79 | + .map(JsonNode::asText) |
| 80 | + .orElse(StringUtils.EMPTY); |
| 81 | + final Map<String, Boolean> impBiddersFilterMap = impsBiddersFilterMap.get(imp.getId()); |
| 82 | + final ExplorationResult explorationResult = ExplorationResult.of( |
| 83 | + greenbidsId, impBiddersFilterMap, isExploration); |
| 84 | + return Ortb2ImpExtResult.of(explorationResult, tid); |
| 85 | + } |
| 86 | +} |
0 commit comments