Skip to content

Commit

Permalink
Ordered WebMvcConfigurer interceptor registrations
Browse files Browse the repository at this point in the history
Closes gh-22434
  • Loading branch information
rstoyanchev committed Mar 19, 2019
1 parent 58b17bf commit dba9c90
Show file tree
Hide file tree
Showing 3 changed files with 70 additions and 6 deletions.
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* Copyright 2002-2018 the original author or authors.
* Copyright 2002-2019 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
Expand All @@ -20,6 +20,7 @@
import java.util.Arrays;
import java.util.List;

import org.springframework.core.Ordered;
import org.springframework.util.Assert;
import org.springframework.util.PathMatcher;
import org.springframework.util.StringUtils;
Expand All @@ -43,6 +44,8 @@ public class InterceptorRegistration {

private PathMatcher pathMatcher;

private int order = 0;


/**
* Create an {@link InterceptorRegistration} instance.
Expand Down Expand Up @@ -80,6 +83,15 @@ public InterceptorRegistration pathMatcher(PathMatcher pathMatcher) {
return this;
}

/**
* Specify an order position to be used. Default is 0.
* @since 4.23
*/
public InterceptorRegistration order(int order){
this.order = order;
return this;
}

/**
* Build the underlying interceptor. If URL patterns are provided, the returned
* type is {@link MappedInterceptor}; otherwise {@link HandlerInterceptor}.
Expand All @@ -98,4 +110,13 @@ protected Object getInterceptor() {
return mappedInterceptor;
}

Ordered toOrdered() {
return new Ordered() {
@Override
public int getOrder() {
return order;
}
};
}

}
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* Copyright 2002-2017 the original author or authors.
* Copyright 2002-2019 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
Expand All @@ -17,8 +17,11 @@
package org.springframework.web.servlet.config.annotation;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;

import org.springframework.core.OrderComparator;
import org.springframework.web.context.request.WebRequestInterceptor;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.handler.WebRequestHandlerInterceptorAdapter;
Expand Down Expand Up @@ -64,11 +67,25 @@ public InterceptorRegistration addWebRequestInterceptor(WebRequestInterceptor in
* Return all registered interceptors.
*/
protected List<Object> getInterceptors() {
List<Object> interceptors = new ArrayList<Object>(this.registrations.size());
Collections.sort(this.registrations, INTERCEPTOR_ORDER_COMPARATOR);
List<Object> result = new ArrayList<Object>(this.registrations.size());
for (InterceptorRegistration registration : this.registrations) {
interceptors.add(registration.getInterceptor());
result.add(registration.getInterceptor());
}
return interceptors ;
return result;
}


private static final Comparator<Object> INTERCEPTOR_ORDER_COMPARATOR =
OrderComparator.INSTANCE.withSourceProvider(new OrderComparator.OrderSourceProvider() {
@Override
public Object getOrderSource(final Object object) {
if (object instanceof InterceptorRegistration) {
return ((InterceptorRegistration) object).toOrdered();
}
return null;
}
});


}
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* Copyright 2002-2014 the original author or authors.
* Copyright 2002-2019 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
Expand All @@ -25,6 +25,7 @@
import org.junit.Test;
import org.mockito.Mockito;

import org.springframework.core.Ordered;
import org.springframework.mock.web.test.MockHttpServletRequest;
import org.springframework.mock.web.test.MockHttpServletResponse;
import org.springframework.ui.ModelMap;
Expand Down Expand Up @@ -150,6 +151,30 @@ public void addInterceptorWithExcludePathPatternOnly() {
assertEquals(Collections.emptyList(), getInterceptorsForPath("/path1/secret"));
}

@Test
public void orderedInterceptors() {
this.registry.addInterceptor(this.interceptor1).order(Ordered.LOWEST_PRECEDENCE);
this.registry.addInterceptor(this.interceptor2).order(Ordered.HIGHEST_PRECEDENCE);

List<Object> interceptors = this.registry.getInterceptors();
assertEquals(2, interceptors.size());

assertSame(this.interceptor2, interceptors.get(0));
assertSame(this.interceptor1, interceptors.get(1));
}

@Test
public void nonOrderedInterceptors() {
this.registry.addInterceptor(this.interceptor1).order(0);
this.registry.addInterceptor(this.interceptor2).order(0);

List<Object> interceptors = this.registry.getInterceptors();
assertEquals(2, interceptors.size());

assertSame(this.interceptor1, interceptors.get(0));
assertSame(this.interceptor2, interceptors.get(1));
}


private List<HandlerInterceptor> getInterceptorsForPath(String lookupPath) {
PathMatcher pathMatcher = new AntPathMatcher();
Expand Down Expand Up @@ -177,6 +202,7 @@ private void verifyWebInterceptor(HandlerInterceptor interceptor, TestWebRequest
assertTrue(webInterceptor.preHandleInvoked);
}


private static class TestWebRequestInterceptor implements WebRequestInterceptor {

private boolean preHandleInvoked = false;
Expand Down

0 comments on commit dba9c90

Please sign in to comment.