com.seven.cow.servlet.download.service.impl.DefaultDownloadFileServiceImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of servlet-download Show documentation

The newest version!

package com.seven.cow.servlet.download.service.impl;

import com.seven.cow.servlet.download.service.DownloadFileService;
import com.seven.cow.spring.boot.autoconfigure.entity.ResponseCmd;
import com.seven.cow.spring.boot.autoconfigure.entity.file.FileInfo;
import com.seven.cow.spring.boot.autoconfigure.util.JSONUtil;
import com.seven.cow.spring.boot.autoconfigure.util.LoggerUtils;
import org.springframework.util.StreamUtils;

import java.io.File;
import java.io.FileInputStream;

/**
 * @description: TODO
 * @author：EDY
 * @date: 2021/11/26 10:26
 * @version: 1.0
 */
public class DefaultDownloadFileServiceImpl implements DownloadFileService {

    @Override
    public byte[] download(String fileKey, String storeAddress) {
        String filePath = (storeAddress + "/" + fileKey);
        File file = new File(filePath);
        if (file.isFile() && file.exists()) {
            try (FileInputStream inputStream = new FileInputStream(file)) {
                byte[] bytes = StreamUtils.copyToByteArray(inputStream);
                FileInfo fileInfo = new FileInfo(bytes.length) {
                    @Override
                    public String key() {
                        return String.format("%02X", System.nanoTime());
                    }
                };
                String fInfo = JSONUtil.toJson(fileInfo);
                LoggerUtils.info("Download a file successful:" + fInfo);
                return bytes;
            } catch (Exception ex) {
                LoggerUtils.error("Download a file failure:", ex);
            }
        } else {
            return JSONUtil.toJsonAsBytes(ResponseCmd.fail().message("fileKey:" + fileKey + " is missing!"));
        }
        return new byte[0];
    }
}