forked from kidgrow-microservices-platform

zhaoxiaohao
2020-07-10 1e0ed9d181d1c877d635f8f9b2b5a25a6b6f078f
kidgrow-business/kidgrow-filecenter/kidgrow-filecenter-biz/src/main/java/com/kidgrow/filecenter/service/impl/AbstractIFileService.java
@@ -1,21 +1,31 @@
package com.kidgrow.filecenter.service.impl;
import cn.hutool.core.util.IdUtil;
import com.aliyun.oss.model.ObjectMetadata;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.kidgrow.common.constant.CommonConstant;
import com.kidgrow.common.model.PageResult;
import com.kidgrow.common.model.ResultBody;
import com.kidgrow.common.utils.DateUtil;
import com.kidgrow.common.utils.MultipartFileUtils;
import com.kidgrow.common.utils.RandomValueUtils;
import com.kidgrow.common.utils.StringUtils;
import com.kidgrow.filecenter.mapper.FileMapper;
import com.kidgrow.filecenter.model.FileInfo;
import com.kidgrow.filecenter.service.IFileService;
import com.kidgrow.filecenter.util.FileUtil;
import lombok.extern.slf4j.Slf4j;
import net.coobird.thumbnailator.Thumbnails;
import org.apache.commons.collections4.MapUtils;
import org.springframework.web.multipart.MultipartFile;
import sun.misc.BASE64Decoder;
import sun.misc.BASE64Encoder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.List;
import java.util.Map;
import java.io.*;
import java.util.*;
/**
 * 石家庄喜高科技有限责任公司 版权所有 © Copyright 2020<br>
@@ -28,6 +38,7 @@
@Slf4j
public abstract class AbstractIFileService extends ServiceImpl<FileMapper, FileInfo> implements IFileService {
    private static final String FILE_SPLIT = ".";
    private static Logger logger = LoggerFactory.getLogger(AbstractIFileService.class);
    @Override
    public FileInfo upload(MultipartFile file) throws Exception {
@@ -39,6 +50,26 @@
        if (!fileInfo.getName().contains(FILE_SPLIT)) {
            throw new IllegalArgumentException("缺少后缀名");
        }
        uploadFile(file, fileInfo);
        // 设置文件来源
        fileInfo.setSource(fileType());
        // 将文件信息保存到数据库
        baseMapper.insert(fileInfo);
        return fileInfo;
    }
    @Override
    public FileInfo upload(MultipartFile file,String imgType) throws Exception {
        FileInfo fileInfo = FileUtil.getFileInfo(file);
        FileInfo oldFileInfo = baseMapper.selectById(fileInfo.getId());
        if (oldFileInfo != null) {
            return oldFileInfo;
        }
        if (!fileInfo.getName().contains(FILE_SPLIT)) {
            throw new IllegalArgumentException("缺少后缀名");
        }
        fileInfo.setImgType(imgType); // X光片或者Logo
        uploadFile(file, fileInfo);
        // 设置文件来源
        fileInfo.setSource(fileType());
@@ -62,6 +93,8 @@
     * @param fileInfo
     */
    protected abstract void uploadFile(MultipartFile file, FileInfo fileInfo) throws Exception;
    protected abstract void fileUpLoadOss(FileInfo fileInfo,String newFilePath, InputStream fileStream);
    protected abstract String FilePath(String imgType,String folderByDate);
    /**
     * 删除文件
@@ -90,5 +123,125 @@
        List<FileInfo> list = baseMapper.findList(page, params);
        return PageResult.<FileInfo>builder().data(list).code(0).count(page.getTotal()).build();
    }
    /**
     * 上传图片-并缩略,其实是给文件服务器存储了两个文件 一个原文件  一个缩略图文件
     * @param file form内的文件数据
     * @param imgType 业务类型 例如logo
     * @return 返回的实体中  path字段保存的是缩略图地址
     * @throws Exception
     */
    @Override
    public FileInfo uploadForThumbnails(MultipartFile file, String imgType) throws Exception {
        FileInfo fileInfo=this.upload(file,imgType);
        List<String> lassStr= Arrays.asList("png","bmp","dib","gif","jfif","jpe","jpeg","jpg","tif","tiff","ico");
        String suffix=file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".")+1);
        if(!lassStr.contains(suffix)){
            return null;
        }
        if (StringUtils.isNotBlank(fileInfo.getUrl())) {
            //生成缩略图上传
//            File nowFile= MultipartFileUtils.multipartFileToFile(file);
//            File nowFile= new File(file.getOriginalFilename());
            //暂存目录  发布后真实存在的磁盘目录
            String result = CommonConstant.TEMP_IMAGE_PATH;
            //文件扩展名
            String newFileName= RandomValueUtils.uuid()+"."+suffix;
            //生成缩略图
            Thumbnails.of(file.getInputStream()).size(CommonConstant.TH_IMG_WIDTH, CommonConstant.TH_IMG_HEIGHT).toFile(result+newFileName);
            File thuFile= new File(result+newFileName);
            if (thuFile.exists()) {
                //将生成的文件转换为流
                InputStream inputStream=new FileInputStream(thuFile);
                String folderByDate = DateUtil.formatDate(new Date());
                String filefloder=this.FilePath(imgType,folderByDate);
                FileInfo newfileInfo=new FileInfo();
                newfileInfo.setName(newFileName);
                newfileInfo.setId(IdUtil.fastSimpleUUID());
                newfileInfo.setSize(thuFile.length());
                newfileInfo.setIsImg(true);
                newfileInfo.setContentType(file.getContentType());
                newfileInfo.setSource(fileInfo.getSource());
                newfileInfo.setImgType(imgType);
                newfileInfo.setCreateTime(new Date());
                //文件上传到oss
                this.fileUpLoadOss(newfileInfo,filefloder+newFileName,inputStream);
                inputStream.close();
//                //删除暂存文件
//                MultipartFileUtils.delteTempFile(thuFile);
                boolean delete = thuFile.delete();
                // 将文件信息保存到数据库
                baseMapper.insert(newfileInfo);
                fileInfo.setPath(newfileInfo.getUrl());
            }
            return fileInfo;
        }
        else
        {
            return fileInfo;
        }
    }
    @Override
    public FileInfo baseUplaod(String file, String imgType) {
        if(file==null||"".equals(file.trim())){
            return null;
        }
        String name=file.split(",")[0];
        file = file.split(",")[1];
        BASE64Decoder decoder = new BASE64Decoder();
        try {
            byte[] bytes = decoder.decodeBuffer(file);
            ObjectMetadata objectMeta = new ObjectMetadata();
            objectMeta.setContentLength(file.length());
            //将字节码转换成流
            InputStream input = new ByteArrayInputStream(bytes);
            String fileName= UUID.randomUUID().toString().replaceAll("-","");
            //image/jpeg;base64
            String newFileName=fileName+"."+name.substring(name.lastIndexOf("/")+1,name.lastIndexOf(";"));
            FileInfo newfileInfo=new FileInfo();
            newfileInfo.setName(newFileName);
            newfileInfo.setId(IdUtil.fastSimpleUUID());
            newfileInfo.setSize(file.length());
            newfileInfo.setIsImg(true);
            newfileInfo.setContentType("image/jpeg");
            newfileInfo.setSource("aliyun");
            newfileInfo.setImgType(imgType);
            newfileInfo.setCreateTime(new Date());
            this.fileUpLoadOss(newfileInfo,"HospitalLogo/"+newFileName,input);
            input.close();
            baseMapper.insert(newfileInfo);
            return newfileInfo;
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }
    @Override
    public ResultBody baseDownLoad(Map<String, Object> map) {
        List<FileInfo> fileInfos = baseMapper.selectByMap(map);
        if(fileInfos.size()>0){
            FileInfo fileInfo = fileInfos.get(0);
            InputStream in = null;
            byte[] data = null;
            //读取图片字节数组
            try{
                in = new FileInputStream(fileInfo.getUrl());
                data = new byte[in.available()];
                in.read(data);
                in.close();
            }catch (IOException e){
                e.printStackTrace();
            }
            //对字节数组Base64编码
            BASE64Encoder encoder = new BASE64Encoder();
            //返回Base64编码过的字节数组字符串
            String encode = encoder.encode(data);
            return ResultBody.ok().data(encode);
        }else {
            return ResultBody.failed("获取数据失败");
        }
    }
}