From c0bbd22426d66c97e881bddaff8e0494dc2e86e8 Mon Sep 17 00:00:00 2001
From: v_dylanxu <136539068@qq.com>
Date: Thu, 22 Oct 2020 18:15:25 +0800
Subject: [PATCH] add apply id support

---
 .../serving/guest/interceptors/GuestBatchParamInterceptor.java | 3 +++
 .../guest/interceptors/GuestSingleParamInterceptor.java        | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestBatchParamInterceptor.java b/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestBatchParamInterceptor.java
index 25ff52fb..1e076a2c 100644
--- a/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestBatchParamInterceptor.java
+++ b/fate-serving-server/src/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/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestSingleParamInterceptor.java b/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestSingleParamInterceptor.java
index 4de9babd..975697d6 100644
--- a/fate-serving-server/src/main/java/com/webank/ai/fate/serving/guest/interceptors/GuestSingleParamInterceptor.java
+++ b/fate-serving-server/src/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());
         }