add apply id support

FederatedAI · Oct 22, 2020 · c0bbd22 · c0bbd22
1 parent c6c44e1
commit c0bbd22
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/...c/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestBatchParamInterceptor.java b/...c/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestBatchParamInterceptor.java
@@ -50,6 +50,9 @@ public void doPreProcess(Context context, InboundPackage inboundPackage, Outboun
             }
             context.setCaseId(batchInferenceRequest.getCaseid());
             context.setServiceId(batchInferenceRequest.getServiceId());
+            if (batchInferenceRequest.getApplyId() != null) {
+                context.setApplyId(batchInferenceRequest.getApplyId());
+            }
         } catch (Exception e) {
             throw new GuestInvalidParamException(e.getMessage());
         }

diff --git a/.../main/java/com/webank/ai/fate/serving/guest/interceptors/GuestSingleParamInterceptor.java b/.../main/java/com/webank/ai/fate/serving/guest/interceptors/GuestSingleParamInterceptor.java
@@ -48,6 +48,9 @@ public void doPreProcess(Context context, InboundPackage inboundPackage, Outboun
             }
             context.setCaseId(inferenceRequest.getCaseid());
             context.setServiceId(inferenceRequest.getServiceId());
+            if (inferenceRequest.getApplyId() != null) {
+                context.setApplyId(inferenceRequest.getApplyId());
+            }
         } catch (Exception e) {
             throw new GuestInvalidParamException(e.getMessage());
         }