Browse Source

Merge branch 'refs/heads/dev-xym-update-branch' into pre-master

xueyiming 1 week ago
parent
commit
f43cf4f286

+ 1 - 1
ad-engine-commons/src/main/java/com/tzld/piaoquan/ad/engine/commons/score/ScorerPipeline.java

@@ -18,7 +18,7 @@ import java.util.concurrent.*;
 @Slf4j
 public class ScorerPipeline {
     public static final int corePoolSize = 128;
-    public static final int SCORE_TIME_OUT = 400;
+    public static final int SCORE_TIME_OUT = 500;
     public static final Logger LOGGER = LoggerFactory.getLogger(ScorerPipeline.class);
     public static final ExecutorService executorService = Executors.newFixedThreadPool(corePoolSize);
 

+ 2 - 2
ad-engine-service/src/main/java/com/tzld/piaoquan/ad/engine/service/score/scorer/PAIScorer.java

@@ -88,7 +88,7 @@ public class PAIScorer extends AbstractScorer {
         long startTime = System.currentTimeMillis();
         PAIModelV1 model = PAIModelV1.getModel();
 
-        final int batchSize = 500;
+        final int batchSize = 300;
         List<List<AdRankItem>> batches = new ArrayList<>();
         for (int i = 0; i < items.size(); i += batchSize) {
             batches.add(new ArrayList<>(items.subList(i, Math.min(i + batchSize, items.size()))));
@@ -112,7 +112,7 @@ public class PAIScorer extends AbstractScorer {
         List<AdRankItem> merged = new ArrayList<>();
         for (Future<List<AdRankItem>> future : futures) {
             try {
-                merged.addAll(future.get());
+                merged.addAll(future.get(400, TimeUnit.MILLISECONDS));
             } catch (Exception e) {
                 LOGGER.error("Execution error in batch", e);
             }

+ 3 - 2
ad-engine-service/src/main/java/com/tzld/piaoquan/ad/engine/service/score/scorer/PAIScorerV2.java

@@ -20,6 +20,7 @@ import java.util.Map;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Future;
+import java.util.concurrent.TimeUnit;
 
 public class PAIScorerV2 extends AbstractScorer {
 
@@ -87,7 +88,7 @@ public class PAIScorerV2 extends AbstractScorer {
         long startTime = System.currentTimeMillis();
         PAIModelV2 model = PAIModelV2.getModel();
 
-        final int batchSize = 500;
+        final int batchSize = 300;
         List<List<AdRankItem>> batches = new ArrayList<>();
         for (int i = 0; i < items.size(); i += batchSize) {
             batches.add(new ArrayList<>(items.subList(i, Math.min(i + batchSize, items.size()))));
@@ -111,7 +112,7 @@ public class PAIScorerV2 extends AbstractScorer {
         List<AdRankItem> merged = new ArrayList<>();
         for (Future<List<AdRankItem>> future : futures) {
             try {
-                merged.addAll(future.get());
+                merged.addAll(future.get(400, TimeUnit.MILLISECONDS));
             } catch (Exception e) {
                 LOGGER.error("Execution error in batch", e);
             }