/**
* 文件压缩
* @param newVo 图片地址集合
* @param filePath 文件生成地址
* @param zipPath 压缩包生成地址
* @return
* @throws Exception
*/
public List<String> fileZipSave(List<String> newVo, String filePath, String zipPath) throws Exception{
List<String> savePathList = new ArrayList<>();
try {
// 创建一个文件夹
File file = new File(filePath);
//如果文件夹存在
if(file.exists()){
// 先删除文件
FileUtil.delFolder(filePath);
}
// 再创建文件
file.mkdirs();
logger.info("成交确认单文件名称=======>>>> "+file.getName()+": "+file.getPath());
List<String> fileList = new ArrayList<>();
fileList.add(filePath);
long picLength = 0;
//多个图片下载地址
for(int i=0; i<newVo.size() ; i++) {
String dealPicUrl = newVo.get(i);
//new一个URL对象
StringBuffer sb = new StringBuffer();
URL url = new URL(sb.toString());
//打开链接
HttpURLConnection conn = (HttpURLConnection)url.openConnection();
//设置请求方式为"GET"
conn.setRequestMethod("GET");
//超时响应时间为5秒
conn.setConnectTimeout(5 * 1000);
//通过输入流获取图片数据
InputStream inStream = conn.getInputStream();
//得到图片的二进制数据，以二进制封装得到数据，具有通用性
byte[] data = FileUtil.readInputStream(inStream);
picLength+= data.length;
// 如果图片大于10M,就再创建一个文件夹
if (picLength > 10*1024*1024){
picLength = data.length;
filePath = filePath+i;
File files = new File(filePath);
//如果文件夹存在
if(files.exists()){
// 先删除文件
FileUtil.delFolder(filePath);
}
files.mkdirs();
fileList.add(filePath);
}
//new一个文件对象用来保存图片，默认保存当前工程根目录
String picName = filePath+"/"+"图片名称".png;
logger.info("成交确认单写入文件的图片名称：=======>>>> "+picName);
File imageFile = new File(picName);
//创建输出流
FileOutputStream outStream = new FileOutputStream(imageFile);
//写入数据
outStream.write(data);
//关闭输出流
outStream.close();
}
// 压缩法
for (int i = 0 ; i<fileList.size() ; i++){
zipPath = zipPath+i;
FileOutputStream fos1 = new FileOutputStream(new File(zipPath+".zip"));
FileUtil.toZip1(fileList.get(i), fos1,false);
File zipFile = new File(zipPath+".zip");
logger.info("成交确认单压缩文件名称=======>>>> "+zipFile.getName()+": "+zipFile.getPath());
FileInputStream fileInputStream = new FileInputStream(zipFile);
// 注意：这个是我保存到自己服务器代码
String savePath = mediaStorageService.storageMedia(fileInputStream, "zip");
if(org.apache.commons.lang.StringUtils.isNotBlank(savePath)){
savePathList.add(savePath);
}
logger.info("成交确认单zip=======>>>> "+Configure.getConfigFromZookeeper("uc.pai.file.real.url.prefix")+savePath);
if (org.apache.commons.lang.StringUtils.isNotEmpty(savePath)){
// 删除文件和压缩文件
FileUtil.delFolder(fileList.get(i));
FileUtil.delFolder(zipPath+".zip");
}
}
} catch (MalformedURLException e) {
e.printStackTrace();
} catch (IOException e) {
e.printStackTrace();
logger.error("方法fileZipSave发生IO异常："+e.getMessage());
}
return savePathList;
}
/**
* 得到图片的二进制数据，以二进制封装得到数据，具有通用性
* @param inStream
* @return
* @throws Exception
*/
public static byte[] readInputStream(InputStream inStream) throws Exception{
ByteArrayOutputStream outStream = new ByteArrayOutputStream();
//创建一个Buffer字符串
byte[] buffer = new byte[1024];
//每次读取的字符串长度，如果为-1，代表全部读取完毕
int len = 0;
//使用一个输入流从buffer里把数据读取出来
while( (len=inStream.read(buffer)) != -1 ){
//用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
outStream.write(buffer, 0, len);
}
//关闭输入流
inStream.close();
//把outStream里的数据写入内存
return outStream.toByteArray();
}
/**
* 压缩成ZIP 方法1
* @param srcDir 压缩文件夹路径
* @param out 压缩文件输出流
* @param KeepDirStructure 是否保留原来的目录结构,true:保留目录结构;
* false:所有文件跑到压缩包根目录下(注意：不保留目录结构可能会出现同名文件,会压缩失败)
* @throws RuntimeException 压缩失败会抛出运行时异常
*/
public static void toZip1(String srcDir, OutputStream out, boolean KeepDirStructure)
throws RuntimeException{
long start = System.currentTimeMillis();
ZipOutputStream zos = null ;
try {
zos = new ZipOutputStream(out);
File sourceFile = new File(srcDir);
compress(sourceFile,zos,sourceFile.getName(),KeepDirStructure);
long end = System.currentTimeMillis();
System.out.println("压缩完成，耗时：" + (end - start) +" ms");
} catch (Exception e) {
throw new RuntimeException("zip error from ZipUtils",e);
}finally{
if(zos != null){
try {
zos.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
/**
* 压缩成ZIP 方法2
* @param srcFiles 需要压缩的文件列表
* @param out 压缩文件输出流
* @throws RuntimeException 压缩失败会抛出运行时异常
*/
public static void toZip(List<File> srcFiles , OutputStream out)throws RuntimeException {
long start = System.currentTimeMillis();
ZipOutputStream zos = null ;
try {
zos = new ZipOutputStream(out);
for (File srcFile : srcFiles) {
byte[] buf = new byte[2 * 1024];
zos.putNextEntry(new ZipEntry(srcFile.getName()));
int len;
FileInputStream in = new FileInputStream(srcFile);
while ((len = in.read(buf)) != -1){
zos.write(buf, 0, len);
}
zos.closeEntry();
in.close();
}
long end = System.currentTimeMillis();
System.out.println("压缩完成，耗时：" + (end - start) +" ms");
} catch (Exception e) {
throw new RuntimeException("zip error from ZipUtils",e);
}finally{
if(zos != null){
try {
zos.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
/**
* 递归压缩方法
* @param sourceFile 源文件
* @param zos zip输出流
* @param name 压缩后的名称
* @param KeepDirStructure 是否保留原来的目录结构,true:保留目录结构;
* false:所有文件跑到压缩包根目录下(注意：不保留目录结构可能会出现同名文件,会压缩失败)
* @throws Exception
*/
private static void compress(File sourceFile, ZipOutputStream zos, String name,
boolean KeepDirStructure) throws Exception{
byte[] buf = new byte[2 * 1024];
if(sourceFile.isFile()){
// 向zip输出流中添加一个zip实体，构造器中name为zip实体的文件的名字
zos.putNextEntry(new ZipEntry(name));
// copy文件到zip输出流中
int len;
FileInputStream in = new FileInputStream(sourceFile);
while ((len = in.read(buf)) != -1){
zos.write(buf, 0, len);
}
// Complete the entry
zos.closeEntry();
in.close();
} else {
File[] listFiles = sourceFile.listFiles();
if(listFiles == null || listFiles.length == 0){
// 需要保留原来的文件结构时,需要对空文件夹进行处理
if(KeepDirStructure){
// 空文件夹的处理
zos.putNextEntry(new ZipEntry(name + "/"));
// 没有文件，不需要文件的copy
zos.closeEntry();
}
}else {
for (File file : listFiles) {
// 判断是否需要保留原来的文件结构
if (KeepDirStructure) {
// 注意：file.getName()前面需要带上父文件夹的名字加一斜杠,
// 不然最后压缩包中就不能保留原来的文件结构,即：所有文件都跑到压缩包根目录下了
compress(file, zos, name + "/" + file.getName(),KeepDirStructure);
} else {
compress(file, zos, file.getName(),KeepDirStructure);
}
}
}
}
}以上是接口，大家调用fileZipSave这个方法，然后传入相关参数即可
扫一扫
1万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
