package com.simon.springbootupload.controller;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.util.List;
import java.util.UUID;

/**
 * @author Simon
 */
@Controller
public class MultiUploadController {
    private static final Logger logger = LoggerFactory.getLogger(MultiUploadController.class);

    @Value("${upload.savepath}")
    private String savepath;

    /**
     * 多个输入框的多文件上传
     *
     * @param request
     * @return
     */
    @ResponseBody
    @PostMapping("/multiUpload")
    public String multiUpload(HttpServletRequest request) {
        List<MultipartFile> files = ((MultipartHttpServletRequest) request).getFiles("file");
        for (int i = 0; i < files.size(); i++) {
            MultipartFile file = files.get(i);
            if (file.isEmpty()) {
                return "上传第" + (i++) + "个文件失败";
            }
            String fileName = file.getOriginalFilename();
            File dest = new File(savepath + UUID.randomUUID() + "-" + fileName);

            try {
                file.transferTo(dest);
                logger.info("第" + (i + 1) + "个文件上传成功");
            } catch (Exception e) {
                logger.error("第" + (i++) + "个文件失败: {}", e.getMessage());
                return "第" + (i++) + "个文件失败";
            }

        }
        return "上传成功";
    }


    /**
     * 一个输入框的多文件上传
     *
     * @param files
     * @return
     */
    @ResponseBody
    @RequestMapping("/singleMultiUpload")
    public String SingleMultiUpload(@RequestParam("file") List<MultipartFile> files) {
        for (int i = 0; i < files.size(); i++) {
            MultipartFile file = files.get(i);
            if (file.isEmpty()) {
                return "上传第" + (i++) + "个文件失败";
            }
            String fileName = file.getOriginalFilename();
            File dest = new File(savepath + UUID.randomUUID() + "-" + fileName);
            try {
                file.transferTo(dest);
                logger.info("第" + (i + 1) + "个文件上传成功");
            } catch (Exception e) {
                logger.error("第" + (i++) + "个文件失败: {}", e.getMessage());
                return "第" + (i++) + "个文件失败";
            }
        }
        return "上传成功";
    }
}