init

abtest-client/src/main/java/com/tzld/piaoquan/abtest/client/ABTestClient.java

@@ -0,0 +1,56 @@
+package com.tzld.piaoquan.abtest.client;
+
+import com.tzld.piaoquan.abtest.client.model.*;
+import net.devh.boot.grpc.client.inject.GrpcClient;
+import org.springframework.stereotype.Component;
+
+import java.util.Collections;
+import java.util.Map;
+
+import static com.google.common.base.Strings.nullToEmpty;
+
+/**
+ * @author dyp
+ */
+@Component
+//@Slf4j
+public class ABTestClient {
+    @GrpcClient("abtest")
+    private ExpServiceGrpc.ExpServiceBlockingStub expClient;
+    @GrpcClient("abtest")
+    private GroupServiceGrpc.GroupServiceBlockingStub groupClient;
+
+    public String getGroup(String mid, String uid, int appType, String group) {
+        GetGroupRequest request = GetGroupRequest.newBuilder()
+                .setMid(nullToEmpty(mid))
+                .setUid(nullToEmpty(uid))
+                .setAppType(appType)
+                .setGroup(group)
+                .build();
+        GetGroupResponse response = groupClient.getGroup(request);
+        if (response == null || !response.hasResult()) {
+            return "";
+        }
+        if (response.getResult().getCode() != 1) {
+            return "";
+        }
+        return response.getGroup();
+    }
+
+    public Map<String, String> getExp(String mid, String uid, int appType, String group) {
+        GetExpRequest request = GetExpRequest.newBuilder()
+                .setMid(nullToEmpty(mid))
+                .setUid(nullToEmpty(uid))
+                .setAppType(appType)
+                .setGroup(group)
+                .build();
+        GetExpResponse response = expClient.getExp(request);
+        if (response == null || !response.hasResult()) {
+            return Collections.emptyMap();
+        }
+        if (response.getResult().getCode() != 1) {
+            return Collections.emptyMap();
+        }
+        return response.getExpMap();
+    }
+}

abtest-client/src/main/java/com/tzld/piaoquan/abtest/client/ModelClient.java

@@ -1,31 +0,0 @@
-package com.tzld.piaoquan.abtest.client;
-
-import org.springframework.stereotype.Component;
-
-/**
- * @author dyp
- */
-@Component
-//@Slf4j
-public class ModelClient {
-//    @GrpcClient("recommend-server")
-//    private ModelServiceGrpc.ModelServiceBlockingStub client;
-//
-//    public Map<String, Double> getScore(long videoId, String mid, String uid, String city, String province) {
-//        ScoreRequest request = ScoreRequest.newBuilder()
-//                .setVideoId(videoId)
-//                .setMid(mid)
-//                .setUid(uid)
-//                .setCity(city)
-//                .setProvince(province)
-//                .build();
-//        ScoreResponse response = client.score(request);
-//        if (response == null || !response.hasResult()) {
-//            return null;
-//        }
-//        if (response.getResult().getCode() != 1) {
-//            return null;
-//        }
-//        return response.getScoreMap();
-//    }
-}

abtest-service/src/main/java/com/tzld/piaoquan/abtest/web/ControllerAspect.java

@@ -1,8 +1,8 @@
 package com.tzld.piaoquan.abtest.web;
 
 import com.google.common.base.Stopwatch;
-import com.tzld.piaoquan.recommend.server.util.JSONUtils;
-import com.tzld.piaoquan.recommend.server.util.TraceUtils;
+import com.tzld.piaoquan.abtest.util.JSONUtils;
+import com.tzld.piaoquan.abtest.util.TraceUtils;
 import lombok.extern.slf4j.Slf4j;
 import org.aspectj.lang.ProceedingJoinPoint;
 import org.aspectj.lang.annotation.Around;

abtest-service/src/main/resources/application.yml

@@ -19,7 +19,7 @@ spring:
 #            rule-type: flow
 grpc:
   server:
-    port: 9001
+    port: 9003
     reflection-service-enabled: true
 
 server: