Browse Source

feat:添加出广告按用户分群配置实验

zhaohaipeng 1 day ago
parent
commit
e4e419fa3f

+ 6 - 0
ad-engine-service/src/main/java/com/tzld/piaoquan/ad/engine/service/predict/impl/PredictModelServiceImpl.java

@@ -24,6 +24,7 @@ import com.tzld.piaoquan.ad.engine.service.predict.param.ThresholdPredictModelPa
 import com.tzld.piaoquan.ad.engine.service.predict.param.request.RoiPredictModelRequestParam;
 import com.tzld.piaoquan.ad.engine.service.predict.param.request.ThresholdPredictModelRequestParam;
 import com.tzld.piaoquan.ad.engine.service.predict.v2.*;
+import com.tzld.piaoquan.ad.engine.service.user.UserService;
 import org.apache.commons.collections4.CollectionUtils;
 import org.apache.commons.collections4.MapUtils;
 import org.apache.commons.lang3.StringUtils;
@@ -67,6 +68,9 @@ public class PredictModelServiceImpl implements PredictModelService {
     @Autowired
     private UserLayerRootSessionIdPredict userLayerRootSessionIdPredict;
 
+    @Autowired
+    private UserService userService;
+
     @Autowired
     private LogHubService logHubService;
 
@@ -271,6 +275,8 @@ public class PredictModelServiceImpl implements PredictModelService {
             setExtraParam(modelParam);
 
             requestParam.setShareType(shareType);
+            String userLayer = userService.getUserLayerByMid(requestParam.getMid());
+            requestParam.setUserLayer(userLayer);
 
             // 先走rootSessionId 实验
             Map<String, Object> predict = rootSessionIdPredict.predict(ConvertUtil.predictParam2Context(requestParam));

+ 1 - 1
ad-engine-service/src/main/java/com/tzld/piaoquan/ad/engine/service/user/UserService.java

@@ -1,5 +1,5 @@
 package com.tzld.piaoquan.ad.engine.service.user;
 
 public interface UserService {
-    String getUserLayerByMi(String mid);
+    String getUserLayerByMid(String mid);
 }

+ 1 - 1
ad-engine-service/src/main/java/com/tzld/piaoquan/ad/engine/service/user/impl/UserServiceImpl.java

@@ -19,7 +19,7 @@ public class UserServiceImpl implements UserService {
     protected AlgorithmRedisHelper algRedisHelper;
 
     @Override
-    public String getUserLayerByMi(String mid) {
+    public String getUserLayerByMid(String mid) {
         String key = String.format("ad:engine:mid:layer:%s", mid);
         String value = algRedisHelper.get(key);
         if (StringUtils.isEmpty(value)) {