快捷導(dǎo)航

SpringBoot根據(jù)目錄結(jié)構(gòu)自動生成路由前綴的實現(xiàn)代碼

更新時間：2021年08月11日 15:28:45 作者：Maggieq8324

本文介紹如何根據(jù)目錄結(jié)構(gòu)給RequestMapping添加路由前綴，具體實現(xiàn)方法，本文通過示例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友參考下吧

前言

本文介紹如何根據(jù)目錄結(jié)構(gòu)給RequestMapping添加路由前綴（覆蓋RequestMappingHandlerMapping中的getMappingForMethod方法，修改其中的Url），如下圖的實際訪問路徑為：/v1/test/test。

在這里插入圖片描述

具體實現(xiàn)

配置文件指定基礎(chǔ)包

application.properties

api-package = com.coisini.springbootlearn.controller

自動補全路由前綴處理類

AutoPrefixUrlMapping.java

import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.servlet.mvc.method.RequestMappingInfo;
import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;
import java.lang.reflect.Method;
import java.util.Objects;

/**
 * @Description 自動補全路由前綴處理類
 *      RequestMappingHandlerMapping 負(fù)責(zé)處理標(biāo)注了@RequestMapping的控制器
 * @author coisini
 * @date Aug 10, 2021
 * @Version 1.0
 */
public class AutoPrefixUrlMapping extends RequestMappingHandlerMapping {

    /**
     * 讀取基礎(chǔ)包配置
     */
    @Value("${api-package}")
    private String bathApiPackagePath;

    /**
     * 重寫方法路由獲取
     * @param method
     * @param handlerType
     * @return
     */
    @Override
    protected RequestMappingInfo getMappingForMethod(Method method, Class<?> handlerType) {
        RequestMappingInfo mappingInfo = super.getMappingForMethod(method, handlerType);
        if (Objects.nonNull(mappingInfo)) {
            String prefix = this.getPrefix(handlerType);
            /**
             * RequestMappingInfo.paths(prefix).build() 根據(jù)前綴生成mappingInfo
             * combine(mappingInfo) 拼接原來的mappingInfo
             */
            return RequestMappingInfo.paths(prefix).build().combine(mappingInfo);
        }

        return mappingInfo;
    }

    /**
     * 獲取方法路由前綴
     * @param handleType
     * @return
     */
    private String getPrefix(Class<?> handleType) {
        String packageName = handleType.getPackage().getName();
        String dotPath = packageName.replace(this.bathApiPackagePath, "").replace(".","/");
        return dotPath;
    }

}

自動補全路由前綴配置類

AutoPrefixConfiguration.java

/**
 * @Description 自動補全路由前綴配置類
 * 通過接口的形式主動發(fā)現(xiàn)
 * @author coisini
 * @date Aug 10, 2021
 * @Version 1.0
 */
@Component
public class AutoPrefixConfiguration implements WebMvcRegistrations {
    @Override
    public RequestMappingHandlerMapping getRequestMappingHandlerMapping() {
        return new AutoPrefixUrlMapping();
    }
}

測試類

@RestController
@RequestMapping("/test")
public class TestController {

    @GetMapping(value = "/test")
    public String test(){
        return "hello";
    }

}

測試

目錄結(jié)構(gòu)如下

在這里插入圖片描述

訪問結(jié)果

在這里插入圖片描述

目錄結(jié)構(gòu)變更

在這里插入圖片描述

訪問結(jié)果

在這里插入圖片描述

到此這篇關(guān)于SpringBoot - 根據(jù)目錄結(jié)構(gòu)自動生成路由前綴的文章就介紹到這了,更多相關(guān)SpringBoot目錄結(jié)構(gòu)路由前綴內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: