瀏覽代碼

ADD: FeatureClient

sunxy 1 年之前
父節點
當前提交
3ae4efac69

+ 39 - 0
recommend-feature-client/src/main/java/com/tzld/piaoquan/recommend/feature/client/FeatureClient.java

@@ -122,4 +122,43 @@ public class FeatureClient {
                 ? response.getAdItemFeatureList()
                 : Collections.emptyList();
     }
+
+    public List<GetUserFeatureInfo> getUserFeatureInfo(List<String> mids) {
+        GetUserFeatureInfoRequest request = GetUserFeatureInfoRequest.newBuilder()
+                .addAllMids(mids)
+                .build();
+        GetUserFeatureInfoResponse response = client.getUserFeatureInfo(request);
+        if (response == null || !response.hasResult()) {
+            log.info("getUserFeatureInfo grpc error");
+            return null;
+        }
+        if (response.getResult().getCode() != 1) {
+            log.info("getUserFeatureInfo grpc code={}, msg={}", response.getResult().getCode(),
+                    response.getResult().getMessage());
+            return null;
+        }
+        return response.getUserFeatureInfoCount() > 0
+                ? response.getUserFeatureInfoList()
+                : Collections.emptyList();
+    }
+
+    public List<GetVideoFeatureInfo> getVideoFeatureInfo(List<Long> videoIds) {
+        GetVideoFeatureInfoRequest request = GetVideoFeatureInfoRequest.newBuilder()
+                .addAllVideoIds(videoIds)
+                .build();
+        GetVideoFeatureInfoResponse response = client.getVideoFeatureInfo(request);
+        if (response == null || !response.hasResult()) {
+            log.info("getVideoFeatureInfo grpc error");
+            return null;
+        }
+        if (response.getResult().getCode() != 1) {
+            log.info("getVideoFeatureInfo grpc code={}, msg={}", response.getResult().getCode(),
+                    response.getResult().getMessage());
+            return null;
+        }
+        return response.getVideoFeatureInfoList().size() > 0
+                ? response.getVideoFeatureInfoList()
+                : Collections.emptyList();
+    }
+
 }