打印请求耗时

ad-engine-commons/src/main/java/com/tzld/piaoquan/ad/engine/commons/score/model/PAIModelV1.java

@@ -121,8 +121,8 @@ public class PAIModelV1 {
     public List<Float> score(final List<AdRankItem> items,
                              final Map<String, String> userFeatureMap,
                              final Map<String, String> sceneFeatureMap) {
-        LOGGER.info("userFeatureMap={}",userFeatureMap);
-        LOGGER.info("sceneFeatureMap={}",sceneFeatureMap);
+        LOGGER.info("userFeatureMap={}", userFeatureMap);
+        LOGGER.info("sceneFeatureMap={}", sceneFeatureMap);
         try {
             TFRequest request = new TFRequest();
 
@@ -200,7 +200,10 @@ public class PAIModelV1 {
             }
             request.addFetch("probs");
             LOGGER.info("PAIModelV1 request={}", request.getRequest().toString());
+            long statTime = System.currentTimeMillis();
             TFResponse response = client.predict(request);
+            long endTime = System.currentTimeMillis();
+            LOGGER.info("PAIModelV1 cost={}", endTime - statTime);
             List<Float> result = response.getFloatVals("probs");
             if (!CollectionUtils.isEmpty(result)) {
                 return result;

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

@@ -82,7 +82,10 @@ public class PAIScorer extends AbstractScorer {
                                   final Map<String, String> sceneFeatureMap,
                                   final PAIModelV1 model) {
 
+        long startTime = System.currentTimeMillis();
         List<Float> score = model.score(items, userFeatureMap, sceneFeatureMap);
+        long endTime = System.currentTimeMillis();
+        LOGGER.info("PAIScorer cost={}", endTime - startTime);
         LOGGER.info("PAIScorer score={}", score);
         for (int i = 0; i < items.size(); i++) {
             Double pro = Double.valueOf(score.get(i));