forked from kidgrow-microservices-platform

zhaoxiaohao
2020-06-16 bdc56ebc9534620766d787a1dacb4c962ea10797
kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-biz/src/main/java/com/kidgrow/filecenter/util/FileUtil.java
@@ -32,7 +32,21 @@
        fileInfo.setId(IdUtil.fastSimpleUUID());
        fileInfo.setName(file.getOriginalFilename());
        fileInfo.setContentType(file.getContentType());
        fileInfo.setIsImg(fileInfo.getContentType().startsWith("image/"));
        if (fileInfo.getContentType().startsWith("image/")) {
            fileInfo.setIsImg(true);
        }
        else if (fileInfo.getContentType().startsWith("multipart")){
            String extendName = file.getOriginalFilename().substring(file.getOriginalFilename().indexOf(".")+1,file.getOriginalFilename().length()).toLowerCase();
            if (extendName.equals("png") || extendName.equals("bmp") || extendName.equals("jpg") || extendName.equals("jpeg") || extendName.equals("gif")) {
                fileInfo.setIsImg(true);
            } else  {
                fileInfo.setIsImg(false);
            }
        } else  {
            fileInfo.setIsImg(false);
        }
        fileInfo.setSize(file.getSize());
        fileInfo.setCreateTime(new Date());
        return fileInfo;