Merge pull request #91 from FederatedAI/develop-2.0.3

fix bug, data security issues
FederatedAI · Nov 4, 2020 · 76238c6 · 76238c6
2 parents ff78e7b + 6b168e5
commit 76238c6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...src/main/java/com/webank/ai/fate/serving/guest/provider/GuestSingleInferenceProvider.java b/...src/main/java/com/webank/ai/fate/serving/guest/provider/GuestSingleInferenceProvider.java
@@ -53,7 +53,11 @@ public ReturnResult doService(Context context, InboundPackage inboundPackage, Ou
         ModelProcessor modelProcessor = model.getModelProcessor();
         InferenceRequest inferenceRequest = (InferenceRequest) inboundPackage.getBody();
         Map<String, Future> futureMap = Maps.newHashMap();
-        List<FederatedRpcInvoker.RpcDataWraper> rpcList = this.buildRpcDataWraper(context, Dict.FEDERATED_INFERENCE, inferenceRequest);
+
+        InferenceRequest remoteInferenceRequest = new InferenceRequest();
+        remoteInferenceRequest.setSendToRemoteFeatureData(inferenceRequest.getSendToRemoteFeatureData());
+
+        List<FederatedRpcInvoker.RpcDataWraper> rpcList = this.buildRpcDataWraper(context, Dict.FEDERATED_INFERENCE, remoteInferenceRequest);
         rpcList.forEach((rpcDataWraper -> {
             ListenableFuture<ReturnResult> future = federatedRpcInvoker.singleInferenceRpcWithCache(context, rpcDataWraper, MetaInfo.PROPERTY_REMOTE_MODEL_INFERENCE_RESULT_CACHE_SWITCH);
             futureMap.put(rpcDataWraper.getHostModel().getPartId(), future);