快捷導(dǎo)航

Java全面深入探究SpringBoot攔截器與文件上傳

更新時(shí)間：2022年05月13日 10:18:15 作者：鐵甲小寶同學(xué)

攔截器對(duì)使用SpringMvc、Struts的開(kāi)發(fā)人員來(lái)說(shuō)特別熟悉，因?yàn)槟阒灰肴プ龊靡粋€(gè)項(xiàng)目必然會(huì)用到它，文件上傳是一個(gè)很常見(jiàn)的功能。在項(xiàng)目開(kāi)發(fā)過(guò)程中，我們通常都會(huì)使用一些成熟的上傳組件來(lái)實(shí)現(xiàn)對(duì)應(yīng)的功能

攔截器

攔截器的概念

動(dòng)態(tài)攔截Actioon調(diào)用的對(duì)象，使開(kāi)發(fā)者在一個(gè)Actioon執(zhí)行的前后執(zhí)行一段代碼，也可以在Action執(zhí)行前阻止其執(zhí)行，同時(shí)也提供了一種可以提取Action中可重用部分代碼的方式。

作用：

動(dòng)態(tài)攔截Action調(diào)用的對(duì)象（也就是實(shí)際項(xiàng)目中的controller層的接口）

一般攔截器用于對(duì)用戶(hù)訪(fǎng)問(wèn)的限制。如當(dāng)用戶(hù)沒(méi)有登錄時(shí)訪(fǎng)問(wèn)主頁(yè)面，則可以使用攔截器進(jìn)行攔截并重定向到登錄頁(yè)面。

攔截器的配置

創(chuàng)建interceptor 文件夾并創(chuàng)建LoginInterceptor Java文件且實(shí)現(xiàn)HandlerInterceptor 這個(gè)接口。

//必須實(shí)現(xiàn)HandlerInterceptor這個(gè)接口
public class LoginInterceptor implements HandlerInterceptor {
    /*
    * 目標(biāo)方法執(zhí)行以前
    * */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        HttpSession session = request.getSession();
        Object loginUser = session.getAttribute("loginUser");
        if(loginUser != null) {
            return true;
        }
        request.setAttribute("msg" ,"請(qǐng)登錄！");
//        response.sendRedirect("/");
        request.getRequestDispatcher("/").forward(request,response);
        return  false;
    }
    //目標(biāo)方法執(zhí)行以后
    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
    }
    /**
     * 頁(yè)面渲染以后
     * @param request
     * @param response
     * @param handler
     * @param ex
     * @throws Exception
     */
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        HandlerInterceptor.super.afterCompletion(request, response, handler, ex);
    }
}

配置攔截器

創(chuàng)建 config 文件夾并創(chuàng)建 AdminWebConfig 文件并實(shí)現(xiàn)WebMvcConfigurer的addInterceptors。

@Configuration
public class AdminWebConfig implements WebMvcConfigurer {
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new LoginInterceptor())
                .addPathPatterns("/**") //靜態(tài)路徑也會(huì)被攔截
                .excludePathPatterns("/","/login","/css/**","/fonts/**" ,"/images/**","/js/**");
    }
}

攔截器的原理

1、根據(jù)當(dāng)前請(qǐng)求，找到**HandlerExecutionChain【**可以處理請(qǐng)求的handler以及handler的所有攔截器】

2、先來(lái)順序執(zhí)行所有攔截器的 preHandle方法

1、如果當(dāng)前攔截器prehandler返回為true。則執(zhí)行下一個(gè)攔截器的preHandle
2、如果當(dāng)前攔截器返回為false。直接倒序執(zhí)行所有已經(jīng)執(zhí)行了的攔截器的 afterCompletion；

3、如果任何一個(gè)攔截器返回false。直接跳出不執(zhí)行目標(biāo)方法

4、所有攔截器都返回True。執(zhí)行目標(biāo)方法

5、倒序執(zhí)行所有攔截器的postHandle方法。

6、前面的步驟有任何異常都會(huì)直接倒序觸發(fā) afterCompletion

7、頁(yè)面成功渲染完成以后，也會(huì)倒序觸發(fā) afterCompletion

文件上傳

在之前我們學(xué)習(xí)SSM的時(shí)候就知道了文件上傳這個(gè)功能，所以在 SpringBoot中我們就不在過(guò)多介紹，原理其實(shí)都差不多。

前端文件：

<form method="post" action="/upload" enctype="multipart/form-data">
    <input type="file" name="file"><br>
    <input type="submit" value="提交">
</form>

Controller層的配置：

@Controller
public class FileController {
    @GetMapping("/updateFile")
    public String FileUp(){
        return "FileUp";
    }  
@PostMapping("/upload") // //new annotation since 4.3
    public String singleFileUpload(@RequestParam("file") MultipartFile file,
                                   RedirectAttributes redirectAttributes) {
        if (file.isEmpty()) {
            redirectAttributes.addFlashAttribute("message", "Please select a file to upload");
            return "redirect:uploadStatus";
        }
        try {
            // Get the file and save it somewhere
            byte[] bytes = file.getBytes();
            Path path = Paths.get( file.getOriginalFilename());
            Files.write(path, bytes);
            redirectAttributes.addFlashAttribute("message",
                    "You successfully uploaded '" + file.getOriginalFilename() + "'");
        } catch (IOException e) {
            e.printStackTrace();
        }
        return "redirect:/main.html";
    }
}