使用spring的使用spring的RequestMappingHandlerMapping类自动扫描入库类自动扫描URL写入数据库

来源：互联网发布：sql注入与防御编辑：程序博客网时间：2024/05/21 06:22

有时候我们在做权限控制的时候，需要将URL放入数据库进行保存，以便于与相应的权限关联。

如果手工键入的话，容易出错和遗漏，spring提供了RequestMappingHandlerMapping类，可以帮助我们，来获得所有的URL

在MainApplication,java中

package security;import java.io.IOException;import java.util.HashMap;import java.util.List;import java.util.Map;import java.util.Set;import javax.annotation.PostConstruct;import org.slf4j.Logger;import org.slf4j.LoggerFactory;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.boot.SpringApplication;import org.springframework.boot.autoconfigure.EnableAutoConfiguration;import org.springframework.boot.autoconfigure.SpringBootApplication;import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;import org.springframework.web.method.HandlerMethod;import org.springframework.web.servlet.mvc.method.RequestMappingInfo;import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;@SpringBootApplication@EnableAutoConfiguration(exclude = MySecurityFilter.class)public class MainApplication{@Autowiredprivate RequestMappingHandlerConfig requestMappingHandlerConfig; public static void main(String[] args) {SpringApplication.run(MainApplication.class, args);}/** * 扫描URL，如果数据库中不存在，则保存入数据库 */@PostConstruct  //这个注解很重要，可以在每次启动的时候检查是否有URL更新，RequestMappingHandlerMapping只能在controller层用。这里我们放在主类中 public void detectHandlerMethods(){        final RequestMappingHandlerMapping requestMappingHandlerMapping = requestMappingHandlerConfig.requestMappingHandlerMapping ();        Map<RequestMappingInfo, HandlerMethod> map = requestMappingHandlerMapping.getHandlerMethods();        Set<RequestMappingInfo> mappings = map.keySet();        Map<String, String> reversedMap = new HashMap<String, String>();        for(RequestMappingInfo info : mappings) {            HandlerMethod method = map.get(info);            String methodstr = method.toString();            methodstr = methodstr.split("\\(")[0];            methodstr = methodstr.split(" ")[2];            int i=methodstr.lastIndexOf(".");            methodstr = methodstr.substring(0,i);            String urlparm = info.getPatternsCondition().toString();            String url = urlparm.substring(1, urlparm.length()-1);            List<SysResource> list = sresourceService.findByResourceString(url);            if(list==null || list.size()<=0){            int num = (int)(Math.random()*100+1);            String rand = String.valueOf(num);            String resourceId = "res"+System.currentTimeMillis()+rand;            SysResource sysresource = new SysResource();            sysresource.setResourceString(url);            sysresource.setRemark("0");            sysresource.setResourceId(resourceId);            sysresource.setMethodPath(methodstr);            sresourceService.save(sysresource);            System.out.println ("===>"+url);            }                    }    }}

接下来：RequestMappingHandlerConfig.java

package security;//注意和主类同一个目录import org.springframework.context.annotation.Bean;import org.springframework.context.annotation.Configuration;import org.springframework.web.servlet.handler.SimpleUrlHandlerMapping;import org.springframework.web.servlet.mvc.method.annotation.RequestMappingHandlerMapping;@Configurationpublic class RequestMappingHandlerConfig {@Bean    public RequestMappingHandlerMapping requestMappingHandlerMapping() {        RequestMappingHandlerMapping mapping = new RequestMappingHandlerMapping();        return mapping;    }}

2 0