com.iana.config.IANAAPIApplication.java Source code

Introduction

Here is the source code for com.iana.config.IANAAPIApplication.java

Source

package com.iana.config;

import java.util.ArrayList;
import java.util.List;
import java.util.Locale;

import javax.sql.DataSource;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.ComponentScan;
import org.springframework.context.annotation.Primary;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.servlet.LocaleResolver;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurerAdapter;
import org.springframework.web.servlet.i18n.LocaleChangeInterceptor;
import org.springframework.web.servlet.i18n.SessionLocaleResolver;

import springfox.documentation.builders.PathSelectors;
import springfox.documentation.builders.RequestHandlerSelectors;
import springfox.documentation.builders.ResponseMessageBuilder;
import springfox.documentation.schema.ModelRef;
import springfox.documentation.service.ApiInfo;
import springfox.documentation.service.ResponseMessage;
import springfox.documentation.spi.DocumentationType;
import springfox.documentation.spring.web.plugins.Docket;
import springfox.documentation.swagger2.annotations.EnableSwagger2;

@SpringBootApplication
@EnableSwagger2
@ComponentScan(basePackages = { "com.iana" })
public class IANAAPIApplication extends WebMvcConfigurerAdapter {

    public static void main(String[] args) {
        SpringApplication.run(IANAAPIApplication.class, args);
    }

    @Bean(name = "equipReturnDataSource")
    @Primary
    @ConfigurationProperties(prefix = "datasource.primary")
    public DataSource getEquipReturnDataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "uiiaDataSource")
    @ConfigurationProperties(prefix = "datasource.uiia")
    public DataSource getUIIADataSource() {
        return DataSourceBuilder.create().build();
    }

    @Bean(name = "gierDataSource")
    @ConfigurationProperties(prefix = "datasource.gier")
    public DataSource getGierDataSource() {
        return DataSourceBuilder.create().build();
    }
    /* Internationalization Support Start */

    @Bean
    public LocaleResolver localeResolver() {
        SessionLocaleResolver slr = new SessionLocaleResolver();
        slr.setDefaultLocale(Locale.US);
        return slr;
    }

    @Bean
    public LocaleChangeInterceptor localeChangeInterceptor() {
        LocaleChangeInterceptor lci = new LocaleChangeInterceptor();
        lci.setParamName("lang");
        return lci;
    }

    // extends WebMvcConfigurerAdapter : only when need support of internationalization 
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(localeChangeInterceptor());
    }
    /* Internationalization Support End */

    @Bean
    public Docket api() {
        List<ResponseMessage> list = new ArrayList<>();
        list.add(new ResponseMessageBuilder().code(500).message("500 message").responseModel(new ModelRef("Error"))
                .build());

        list.add(new ResponseMessageBuilder().code(403).message("Forbidden!").build());
        return new Docket(DocumentationType.SWAGGER_2).select()
                /*.apis(RequestHandlerSelectors.any())              */
                .apis(RequestHandlerSelectors.basePackage("com.iana")).paths(PathSelectors.any()).build()
                .apiInfo(apiInfo()).useDefaultResponseMessages(true) //TRUE means gives first priority to default messages                                 
                .globalResponseMessage(RequestMethod.GET, list); //this will override or add new response messages. 
    }

    private ApiInfo apiInfo() {
        ApiInfo apiInfo = new ApiInfo("IANA REST API", "Some custom description of API.", "API TOS",
                "Terms of service", "chauhanvipul87@gmail.com", "License of API", "API license URL");
        return apiInfo;
    }

}