fix changed instanceof by mistake

monitorjbl#55
zifnab87 · Jan 18, 2018 · f204abf · f204abf
1 parent 9122365
commit f204abf
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/spring-json-view/src/main/java/com/monitorjbl/json/JsonViewSupportFactoryBean.java b/spring-json-view/src/main/java/com/monitorjbl/json/JsonViewSupportFactoryBean.java
@@ -81,9 +81,8 @@ private void decorateHandlers(List<HandlerMethodReturnValueHandler> handlers) {
       int index = handlers.indexOf(handler);
       if(handler instanceof HttpEntityMethodProcessor) {
         handlers.set(index, new JsonViewHttpEntityMethodProcessor(converters));
-      } else if(handler instanceof HandlerMethodReturnValueHandler) {
-        HandlerMethodReturnValueHandler valueHandler = handlers.get(index);
-        List<Object> responseBodyAdvice = (List<Object>) get(valueHandler,"advice");
+      } else if(handler instanceof RequestResponseBodyMethodProcessor) {
+        List<Object> responseBodyAdvice = (List<Object>) get(handler,"advice");
         HandlerMethodReturnValueHandler delegate = new JsonViewResponseProcessor(converters, responseBodyAdvice);
         handlers.set(index, new JsonViewReturnValueHandler(delegate,  defaultView));
         break;