|
|
|
/*
|
|
|
|
* Licensed to the Apache Software Foundation (ASF) under one or more
|
|
|
|
* contributor license agreements. See the NOTICE file distributed with
|
|
|
|
* this work for additional information regarding copyright ownership.
|
|
|
|
* The ASF licenses this file to You under the Apache License, Version 2.0
|
|
|
|
* (the "License"); you may not use this file except in compliance with
|
|
|
|
* the License. You may obtain a copy of the License at
|
|
|
|
*
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
*
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
* limitations under the License.
|
|
|
|
*/
|
|
|
|
|
|
|
|
package org.apache.dolphinscheduler.api.configuration;
|
|
|
|
|
|
|
|
import org.apache.dolphinscheduler.api.interceptor.LocaleChangeInterceptor;
|
|
|
|
import org.apache.dolphinscheduler.api.interceptor.LoginHandlerInterceptor;
|
|
|
|
import org.apache.dolphinscheduler.api.interceptor.RateLimitInterceptor;
|
|
|
|
|
|
|
|
import java.util.Locale;
|
|
|
|
|
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
|
import org.springframework.context.annotation.Bean;
|
|
|
|
import org.springframework.context.annotation.Configuration;
|
|
|
|
import org.springframework.web.cors.CorsConfiguration;
|
|
|
|
import org.springframework.web.cors.UrlBasedCorsConfigurationSource;
|
|
|
|
import org.springframework.web.filter.CorsFilter;
|
|
|
|
import org.springframework.web.servlet.LocaleResolver;
|
|
|
|
import org.springframework.web.servlet.config.annotation.ContentNegotiationConfigurer;
|
|
|
|
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
|
|
|
|
import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
|
|
|
|
import org.springframework.web.servlet.config.annotation.ViewControllerRegistry;
|
|
|
|
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
|
|
|
|
import org.springframework.web.servlet.i18n.CookieLocaleResolver;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* application configuration
|
|
|
|
*/
|
|
|
|
@Configuration
|
|
|
|
public class AppConfiguration implements WebMvcConfigurer {
|
|
|
|
|
|
|
|
public static final String LOGIN_INTERCEPTOR_PATH_PATTERN = "/**/*";
|
|
|
|
public static final String LOGIN_PATH_PATTERN = "/login";
|
|
|
|
public static final String REGISTER_PATH_PATTERN = "/users/register";
|
|
|
|
public static final String PATH_PATTERN = "/**";
|
|
|
|
public static final String LOCALE_LANGUAGE_COOKIE = "language";
|
|
|
|
|
|
|
|
@Autowired
|
|
|
|
private TrafficConfiguration trafficConfiguration;
|
|
|
|
|
|
|
|
@Bean
|
|
|
|
public CorsFilter corsFilter() {
|
|
|
|
CorsConfiguration config = new CorsConfiguration();
|
|
|
|
config.addAllowedOrigin("*");
|
|
|
|
config.addAllowedMethod("*");
|
|
|
|
config.addAllowedHeader("*");
|
|
|
|
UrlBasedCorsConfigurationSource configSource = new UrlBasedCorsConfigurationSource();
|
|
|
|
configSource.registerCorsConfiguration(PATH_PATTERN, config);
|
|
|
|
return new CorsFilter(configSource);
|
|
|
|
}
|
|
|
|
|
|
|
|
@Bean
|
|
|
|
public LoginHandlerInterceptor loginInterceptor() {
|
|
|
|
return new LoginHandlerInterceptor();
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Cookie
|
|
|
|
* @return local resolver
|
|
|
|
*/
|
|
|
|
@Bean(name = "localeResolver")
|
|
|
|
public LocaleResolver localeResolver() {
|
|
|
|
CookieLocaleResolver localeResolver = new CookieLocaleResolver();
|
|
|
|
localeResolver.setCookieName(LOCALE_LANGUAGE_COOKIE);
|
|
|
|
// set default locale
|
|
|
|
localeResolver.setDefaultLocale(Locale.US);
|
|
|
|
// set language tag compliant
|
|
|
|
localeResolver.setLanguageTagCompliant(false);
|
|
|
|
return localeResolver;
|
|
|
|
}
|
|
|
|
|
|
|
|
@Bean
|
|
|
|
public LocaleChangeInterceptor localeChangeInterceptor() {
|
|
|
|
return new LocaleChangeInterceptor();
|
|
|
|
}
|
|
|
|
|
|
|
|
@Bean
|
|
|
|
public RateLimitInterceptor createRateLimitInterceptor() {
|
|
|
|
return new RateLimitInterceptor(trafficConfiguration);
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public void addInterceptors(InterceptorRegistry registry) {
|
|
|
|
// i18n
|
|
|
|
registry.addInterceptor(localeChangeInterceptor());
|
|
|
|
if (trafficConfiguration.isTrafficGlobalControlSwitch() || trafficConfiguration.isTrafficTenantControlSwitch()) {
|
|
|
|
registry.addInterceptor(createRateLimitInterceptor());
|
|
|
|
}
|
|
|
|
registry.addInterceptor(loginInterceptor())
|
|
|
|
.addPathPatterns(LOGIN_INTERCEPTOR_PATH_PATTERN)
|
|
|
|
.excludePathPatterns(LOGIN_PATH_PATTERN, REGISTER_PATH_PATTERN,
|
|
|
|
"/swagger-resources/**", "/webjars/**", "/v2/**",
|
|
|
|
"/doc.html", "/swagger-ui.html", "*.html", "/ui/**");
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public void addResourceHandlers(ResourceHandlerRegistry registry) {
|
|
|
|
registry.addResourceHandler("/static/**").addResourceLocations("classpath:/static/");
|
|
|
|
registry.addResourceHandler("doc.html").addResourceLocations("classpath:/META-INF/resources/");
|
|
|
|
registry.addResourceHandler("swagger-ui.html").addResourceLocations("classpath:/META-INF/resources/");
|
|
|
|
registry.addResourceHandler("/webjars/**").addResourceLocations("classpath:/META-INF/resources/webjars/");
|
|
|
|
registry.addResourceHandler("/ui/**").addResourceLocations("file:ui/");
|
|
|
|
}
|
|
|
|
|
|
|
|
@Override
|
|
|
|
public void addViewControllers(ViewControllerRegistry registry) {
|
|
|
|
registry.addViewController("/ui/").setViewName("forward:/ui/index.html");
|
|
|
|
registry.addViewController("/").setViewName("forward:/ui/index.html");
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Turn off suffix-based content negotiation
|
|
|
|
*
|
|
|
|
* @param configurer configurer
|
|
|
|
*/
|
|
|
|
@Override
|
|
|
|
public void configureContentNegotiation(final ContentNegotiationConfigurer configurer) {
|
|
|
|
configurer.favorPathExtension(false);
|
|
|
|
}
|
|
|
|
}
|