Skip to content

Commit

Permalink
Merge pull request #91 from FederatedAI/develop-2.0.3
Browse files Browse the repository at this point in the history
fix bug, data security issues
  • Loading branch information
forgivedengkai authored Nov 4, 2020
2 parents 497331c + d44eda9 commit 750ec1f
Showing 1 changed file with 5 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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);
Expand Down

0 comments on commit 750ec1f

Please sign in to comment.