spring 对上传图片处理

package com.maizuo.web.controller;  

import com.maizuo.domain.Result;  
import com.maizuo.util.ConstantsConfig;  
import com.maizuo.util.Upload;  
import hyxlog.Log;  
import net.sf.json.JSONObject;  
import org.apache.commons.lang.StringUtils;  
import org.springframework.stereotype.Controller;  
import org.springframework.web.bind.annotation.*;  
import org.springframework.web.multipart.MultipartFile;  
import javax.imageio.ImageIO;  
import javax.servlet.http.HttpServletRequest;  
import java.awt.image.BufferedImage;  
import java.io.File;  
import java.io.FileInputStream;  
import java.io.FileNotFoundException;  
import java.io.IOException;  
import java.util.Date;  
import java.util.List;  

/** 
 * Created by qiyang on 2015/6/15 and improved by heisenberg on 2016/04/18 
 */  
@Controller  
@RequestMapping("/api/upload")  
public class UploadController {  
    @RequestMapping(value = "/img", method = RequestMethod.POST)  
    @ResponseBody  
    public Result uploadImg(@RequestParam(value = "file", required = false) MultipartFile file, String pathName,  
            Integer sizeRule, Integer isDeviation, HttpServletRequest request)  
                    throws FileNotFoundException, IOException {  
        String loghead = "通用接口-上传图片:";  

        JSONObject json = new JSONObject();  
        if (file == null) {  
            Log.info(loghead + "上传失败:文件为空");  
            return new Result(900001, "", "上传失败:文件为空");  
        }  
        if (StringUtils.isBlank(pathName)) {  
            pathName = ConstantsConfig.getString("DEFALTUPLOADPATH");  
        }  
        List<Object> uploadPathNames = ConstantsConfig.getList("UPLOADPATHNAMES");  
        boolean flag = false;  
        for (int i = 0; i < uploadPathNames.size(); i++) {  
            if (pathName.equals(uploadPathNames.get(i))) {  
                flag = true;  
            }  
        }  
        if (!flag) {  
            Log.info(loghead + "上传失败:上传路径无效");  
            return new Result(900001, "", "上传失败:上传路径无效");  
        }  
        String fileName = file.getOriginalFilename();  
        // 获取上传文件扩展名  
        String fileExt = fileName.substring(fileName.lastIndexOf(".") + 1, fileName.length());  
        // 对扩展名进行小写转换  
        fileExt = fileExt.toLowerCase();  
        // 图片文件大小过滤  
        if (!"jpg".equals(fileExt) && !"jpeg".equals(fileExt) && !"png".equals(fileExt) && !"bmp".equals(fileExt)  
                && !"gif".equals(fileExt)) {  
            Log.info(loghead + "上传失败:无效图片文件类型");  
            return new Result(900001, "", "上传失败:无效图片文件类型");  
        }  
        long fileSize = file.getSize();  
        Log.info(loghead + "fileInfo:fileName=" + fileName + "&fileSize=" + fileSize);  
        if (fileSize <= 0) {  
            Log.info(loghead + "上传失败:文件为空");  
            return new Result(900001, "", "上传失败:文件为空");  
        } else if (fileSize > (500 * 1024)) {  
            Log.info(loghead + "上传失败:文件大小不能超过500K");  
            return new Result(900001, "", "上传失败:文件大小不能超过500K");  
        }  
        File tmpFile = null;  
        // 判断文件是否为空  
        if (!file.isEmpty()) {  
            String uploadPath = request.getSession().getServletContext().getRealPath("/") + "/upload/";  
            File uploadDir = new File(uploadPath);  
            if (uploadDir.exists() && uploadDir.isDirectory()) {  
                String[] childFileNameList = uploadDir.list();  
                if (childFileNameList != null) {  
                    File temp;  
                    for (int i = 0; i < childFileNameList.length; i++) {  
                        temp = new File(uploadPath + childFileNameList[i]);  
                        if (temp.isFile()) {  
                            temp.delete();  
                        }  
                    }  
                }  
            } else {  
                uploadDir.mkdir();  
            }  

            try {  
                Date now = new Date();  
                String tmpFileName = String.valueOf(now.getTime()) + Math.round(Math.random() * 1000) + "." + fileExt;  
                // 文件保存路径  
                String tmpFilePath = uploadPath + tmpFileName;  
                tmpFile = new File(tmpFilePath);  
                file.transferTo(tmpFile);  
                BufferedImage sourceImg = ImageIO.read(new FileInputStream(tmpFile));  
                int imgWidth = sourceImg.getWidth();  
                int imgHeight = sourceImg.getHeight();  
                System.out.println("上传的图片宽:" + imgWidth);  
                System.out.println("上传的图片高:" + imgHeight);  
                // 图片文件尺寸过滤  
                if (sizeRule == null) {  
                    // 上传到图片服务器  
                    String imgUrl = Upload.upload(tmpFile, "/" + pathName + "/");  
                    json.put("fileName", fileName);  
                    json.put("url", imgUrl);  
                    json.put("imgWidth", imgWidth);  
                    json.put("imgHeight", imgHeight);  
                    return new Result(0, json, "success");  
                } else {  
                    System.out.println("前端选择图片尺寸规则" + sizeRule);  
                    int ruleWidth = 0;  
                    int ruleHeight = 0;  
                    try {  
                        String imgSizeRule = ConstantsConfig.getString("UPLOADIMG_RULE" + sizeRule);  
                        String imgSizeRule_width_height[] = imgSizeRule.split(",");  
                        String imgSizeRule_width = imgSizeRule_width_height[0];  
                        String imgSizeRule_height = imgSizeRule_width_height[1];  
                        ruleWidth = Integer.parseInt(imgSizeRule_width);  
                        ruleHeight = Integer.parseInt(imgSizeRule_height);  
                    } catch (Exception e) {  
                        System.out.println("没有配置尺寸规则" + sizeRule);  
                        json.put("fileName", fileName);  
                        json.put("imgWidth", imgWidth);  
                        json.put("imgHeight", imgHeight);  
                        return new Result(-1, json, "配置系统没有配置上传图片尺寸规则" + sizeRule  
                                + ",请前端修改参数sizeRule值或管理员在配置系统配置sizeRule" + sizeRule + "规则对应的配置项");  
                    }  
                    if (isDeviation == null) {  
                        System.out.println("严格限制图片尺寸");  
                        if (ruleWidth == imgWidth && ruleHeight == imgHeight) {  
                            String imgUrl = Upload.upload(tmpFile, "/" + pathName + "/");  
                            json.put("fileName", fileName);  
                            json.put("url", imgUrl);  
                            json.put("imgWidth", imgWidth);  
                            json.put("imgHeight", imgHeight);  
                            return new Result(0, json, "success");  
                        } else {  
                            json.put("fileName", fileName);  
                            json.put("imgWidth", imgWidth);  
                            json.put("imgHeight", imgHeight);  
                            json.put("ruleWidth", ruleWidth);  
                            json.put("ruleHeight", ruleHeight);  
                            return new Result(-1, json, "请上传" + ruleWidth + "*" + ruleHeight + "px的图片");  
                        }  
                    } else {  
                        int deviation = Integer.parseInt(ConstantsConfig.getString("UPLOADIMG_DEVIATION"));  
                        System.out.println("允许尺寸误差在" + deviation + "以内");  
                        if (Math.abs(ruleWidth - imgWidth) <= deviation  
                                && Math.abs(ruleHeight - imgHeight) <= deviation) {  
                            String imgUrl = Upload.upload(tmpFile, "/" + pathName + "/");  
                            json.put("fileName", fileName);  
                            json.put("url", imgUrl);  
                            json.put("imgWidth", imgWidth);  
                            json.put("imgHeight", imgHeight);  
                            return new Result(0, json, "success");  
                        } else {  
                            json.put("fileName", fileName);  
                            json.put("imgWidth", imgWidth);  
                            json.put("imgHeight", imgHeight);  
                            json.put("ruleWidth", ruleWidth);  
                            json.put("ruleHeight", ruleHeight);  
                            return new Result(-1, json, "请上传" + ruleWidth + "*" + ruleHeight + "px的图片");  
                        }  
                    }  
                }  
            } catch (Exception e) {  
                e.printStackTrace();  
                Log.info(loghead + "上传失败:文件上传失败");  
                return new Result(900001, "", "上传失败:文件上传异常");  
            } finally {  
                // 删除临时文件  
                if (tmpFile.exists()) {  
                    tmpFile.deleteOnExit();  
                    Log.info(loghead + "删除临时文件" + tmpFile.getAbsolutePath());  
                }  
            }  
        }  
        return new Result(-1, json, "后台校验上传图片流程异常");  
    }  
}  
阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页