Fix batch interceptor bugs.

jianghaolu · Oct 4, 2016 · a823d02 · a823d02
1 parent 603e1df
commit a823d02
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/...batch/src/main/java/com/microsoft/azure/batch/interceptor/ClientRequestIdInterceptor.java b/...batch/src/main/java/com/microsoft/azure/batch/interceptor/ClientRequestIdInterceptor.java
@@ -18,13 +18,13 @@ public void modify(Object request) {
                 Class<?> c = request.getClass();
 
                 try {
-                    Method clientRequestIdMethod = c.getMethod("setClientRequestId", new Class[]{String.class});
+                    Method clientRequestIdMethod = c.getMethod("withClientRequestId", new Class[]{String.class});
                     if (clientRequestIdMethod != null) {
                         String clientRequestId = UUID.randomUUID().toString();
                         clientRequestIdMethod.invoke(request, clientRequestId);
                     }
 
-                    Method returnClientRequestIdMethod = c.getMethod("setReturnClientRequestId", new Class[]{Boolean.class});
+                    Method returnClientRequestIdMethod = c.getMethod("withReturnClientRequestId", new Class[]{Boolean.class});
                     if (returnClientRequestIdMethod != null) {
                         returnClientRequestIdMethod.invoke(request, true);
                     }

diff --git a/azure-batch/src/main/java/com/microsoft/azure/batch/interceptor/DetailLevelInterceptor.java b/azure-batch/src/main/java/com/microsoft/azure/batch/interceptor/DetailLevelInterceptor.java
@@ -22,7 +22,7 @@ public void modify(Object request) {
                 if (detailLevel != null) {
                     Class<?> c = request.getClass();
                     try {
-                        Method selectMethod = c.getMethod("setSelect", new Class[]{String.class});
+                        Method selectMethod = c.getMethod("withSelect", new Class[]{String.class});
                         if (selectMethod != null) {
                             selectMethod.invoke(request, detailLevel.selectClause());
                         }
@@ -31,7 +31,7 @@ public void modify(Object request) {
                     }
 
                     try {
-                        Method filterMethod = c.getMethod("setFilter", new Class[]{String.class});
+                        Method filterMethod = c.getMethod("withFilter", new Class[]{String.class});
                         if (filterMethod != null) {
                             filterMethod.invoke(request, detailLevel.filterClause());
                         }
@@ -40,7 +40,7 @@ public void modify(Object request) {
                     }
 
                     try {
-                        Method expandMethod = c.getMethod("setExpand", new Class[]{String.class});
+                        Method expandMethod = c.getMethod("withExpand", new Class[]{String.class});
                         if (expandMethod != null) {
                             expandMethod.invoke(request, detailLevel.expandClause());
                         }

diff --git a/azure-batch/src/main/java/com/microsoft/azure/batch/interceptor/PageSizeInterceptor.java b/azure-batch/src/main/java/com/microsoft/azure/batch/interceptor/PageSizeInterceptor.java
@@ -20,7 +20,7 @@ public PageSizeInterceptor(int pageSize) {
             public void modify(Object request) {
                 Class<?> c = request.getClass();
                 try {
-                    Method maxResultsMethod = c.getMethod("setMaxResults", new Class[]{Integer.class});
+                    Method maxResultsMethod = c.getMethod("withMaxResults", new Class[]{Integer.class});
                     if (maxResultsMethod != null) {
                         maxResultsMethod.invoke(request, maxResults);
                     }

diff --git a/...e-batch/src/main/java/com/microsoft/azure/batch/interceptor/ServerTimeoutInterceptor.java b/...e-batch/src/main/java/com/microsoft/azure/batch/interceptor/ServerTimeoutInterceptor.java
@@ -20,7 +20,7 @@ public ServerTimeoutInterceptor(int timeout) {
             public void modify(Object request) {
                 Class<?> c = request.getClass();
                 try {
-                    Method timeoutMethod = c.getMethod("setTimeout", new Class[]{Integer.class});
+                    Method timeoutMethod = c.getMethod("withTimeout", new Class[]{Integer.class});
                     if (timeoutMethod != null) {
                         timeoutMethod.invoke(request, serverTimeout);
                     }