OSSFileService.java

package com.tanpu.community.service;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.tanpu.common.constant.BizStatus;
import com.tanpu.common.exception.BizException;
import com.tanpu.common.util.AliyunOSSHelper;
import com.tanpu.common.util.JsonUtil;
import com.tanpu.common.uuid.UuidGenHelper;
import com.tanpu.community.api.CommunityConstant;
import com.tanpu.community.api.enums.FileTypeEnum;
import com.tanpu.community.api.enums.OssRelType;
import com.tanpu.community.dao.entity.community.FileRecordEntity;
import com.tanpu.community.dao.mapper.community.FileRecordMapper;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.util.HashMap;
import java.util.List;

@Slf4j
@Service
public class OSSFileService {

    @Value("${aliyun.oss.endpoint}")
    public String endpoint;

    @Value("${aliyun.oss.accessId}")
    public String accessId;

    @Value("${aliyun.oss.accessSK}")
    public String accessSK;

    @Value("${aliyun.oss.bucketName}")
    public String bucketName;

    private AliyunOSSHelper ossHelper;

    @Resource
    private FileRecordMapper fileRecordMapper;


    @Autowired
    private UuidGenHelper uuidGenHelper;

    @PostConstruct
    public void init() {
        ossHelper = AliyunOSSHelper.build(endpoint, accessId, accessSK);
    }


    @Transactional
    public FileRecordEntity uploadFile(byte[] data, String fileName, String dirPrefix) {
        String fileSuffix = fileName.substring(fileName.lastIndexOf('.') + 1);
        return uploadFile(data, fileName, fileSuffix, dirPrefix);
    }

    @Transactional
    public FileRecordEntity uploadFile(byte[] data, String fileName, String fileSuffix, String dirPrefix) {
        String[] arr = StringUtils.split(fileName, ".");
        String suffix = arr[arr.length - 1];
        if (FileTypeEnum.imageTypeSet.contains(suffix)) {
            //上传
            String id = uuidGenHelper.getUuidStr();
            String key = CommunityConstant.OSS_PREFIX_FOLDER + dirPrefix + id + "." + suffix;
            ossHelper.writeFile(bucketName, key, data, fileSuffix);

            //落库
            FileRecordEntity record = new FileRecordEntity();
            record.setFileId(uuidGenHelper.getUuidStr());
            record.setDeleteTag(BizStatus.DeleteTag.tag_init);
            record.setFileOssKey(key);
            record.setFileName(fileName);
            record.setFileId(id);
            record.setPreviewUrl(ossHelper.getPreSignedUrl(bucketName, key));
            record.setFileType(FileTypeEnum.IMAGE.getCode());
            HashMap<String, Integer> attr = getStringIntegerHashMap(data);
            record.setExtInfo(JsonUtil.toJson(attr));

            fileRecordMapper.insert(record);

            return record;
        }else {

            throw new BizException("文件格式暂不支持:"+suffix);
        }

    }

    private HashMap<String, Integer> getStringIntegerHashMap(byte[] data) {
        ByteArrayInputStream bins = new ByteArrayInputStream(data);
        BufferedImage bi = null;
        try {
            bi = ImageIO.read(bins);
        } catch (Exception e) {
            e.printStackTrace();
        }
        HashMap<String, Integer> attr = new HashMap<>();
        attr.put("width",bi.getWidth());
        attr.put("height",bi.getHeight());
        return attr;
    }

    public FileRecordEntity queryById(String fileId) {
        return fileRecordMapper.selectOne(new LambdaQueryWrapper<FileRecordEntity>()
                .eq(FileRecordEntity::getFileId,fileId));
    }

    public List<FileRecordEntity> queryByIds(List<String> fileIds) {
        return fileRecordMapper.selectList(new LambdaQueryWrapper<FileRecordEntity>()
                .in(FileRecordEntity::getFileId,fileIds));
    }
}