parent
e7c9fcbd07
commit
f4e78612f7
9 changed files with 221 additions and 7 deletions
@ -0,0 +1,5 @@ |
|||||||
|
package com.teaching.backend.utils; |
||||||
|
|
||||||
|
public class Constants { |
||||||
|
public static final String UPLOAD_URL = "E:\\workspace\\file\\"; |
||||||
|
} |
@ -0,0 +1,115 @@ |
|||||||
|
package com.teaching.backend.utils; |
||||||
|
|
||||||
|
import lombok.extern.slf4j.Slf4j; |
||||||
|
import org.apache.commons.io.FilenameUtils; |
||||||
|
import org.apache.commons.io.FileUtils; |
||||||
|
import org.springframework.util.DigestUtils; |
||||||
|
import org.springframework.web.multipart.MultipartFile; |
||||||
|
|
||||||
|
import java.io.File; |
||||||
|
import java.io.FileInputStream; |
||||||
|
import java.io.IOException; |
||||||
|
import java.io.RandomAccessFile; |
||||||
|
import java.util.*; |
||||||
|
|
||||||
|
import static com.teaching.backend.utils.Constants.UPLOAD_URL; |
||||||
|
|
||||||
|
@Slf4j |
||||||
|
//文件上传工具类
|
||||||
|
public class UploadUtils { |
||||||
|
|
||||||
|
public static String saveFileByDirectory (MultipartFile file) { |
||||||
|
String url = ""; |
||||||
|
try { |
||||||
|
// 将文件保存在服务器目录中
|
||||||
|
// 文件名称
|
||||||
|
String uuid = UUID.randomUUID().toString(); |
||||||
|
// 得到上传文件后缀
|
||||||
|
String originalName = file.getOriginalFilename(); |
||||||
|
String ext = "." + FilenameUtils.getExtension(originalName); |
||||||
|
// 新生成的文件名称
|
||||||
|
String fileName = uuid + ext; |
||||||
|
// 复制文件
|
||||||
|
File targetFile = new File(UPLOAD_URL, fileName); |
||||||
|
url = UPLOAD_URL + fileName; |
||||||
|
FileUtils.writeByteArrayToFile(targetFile, file.getBytes()); |
||||||
|
System.out.println("保存成功"); |
||||||
|
} catch (IOException e) { |
||||||
|
log.error("保存文件到服务器(本地)失败",e); |
||||||
|
} |
||||||
|
return url; |
||||||
|
} |
||||||
|
|
||||||
|
public static void Chunk(File sourseFile) throws IOException { |
||||||
|
// //源文件
|
||||||
|
// File sourseFile = new File("D:/Users/Desktop/image/剪映/7.18/7月20.mp4");
|
||||||
|
//分块文件存储路径
|
||||||
|
String chunkFilePath = "E:\\workspace\\chunk"; |
||||||
|
//分块文件大小
|
||||||
|
int chunkSize = 1024 * 1024 * 1; |
||||||
|
//分块文件个数
|
||||||
|
int chunkNum = (int) Math.ceil(sourseFile.length() * 1.0 / chunkSize); |
||||||
|
//使用流从源文件读数据,向分块文件中写数据
|
||||||
|
RandomAccessFile raf_r = new RandomAccessFile(sourseFile, "r"); |
||||||
|
//缓存区
|
||||||
|
byte[] bytes = new byte[1024]; |
||||||
|
for (int i = 0; i < chunkNum; i++) { |
||||||
|
File chunkFile = new File(chunkFilePath + i); |
||||||
|
//分块文件写入流
|
||||||
|
RandomAccessFile raf_rw = new RandomAccessFile(chunkFile, "rw"); |
||||||
|
int len = -1; |
||||||
|
while((len=raf_r.read(bytes)) != -1){ |
||||||
|
raf_rw.write(bytes, 0, len); |
||||||
|
if (chunkFile.length() >= chunkSize){ |
||||||
|
break; |
||||||
|
} |
||||||
|
} |
||||||
|
raf_rw.close(); |
||||||
|
} |
||||||
|
raf_r.close(); |
||||||
|
} |
||||||
|
|
||||||
|
public static void Merge(File sourseFile) throws IOException { |
||||||
|
//块文件目录
|
||||||
|
File chunkFolder = new File("E:\\workspace\\chunk"); |
||||||
|
// //源文件
|
||||||
|
// File sourseFile = new File("D:/Users/Desktop/image/剪映/7.18/7月20.mp4");
|
||||||
|
//合并后的文件
|
||||||
|
File mergeFile = new File("E:\\workspace\\file\\"+sourseFile.getName()); |
||||||
|
|
||||||
|
//取出所有分块文件
|
||||||
|
File[] files = chunkFolder.listFiles(); |
||||||
|
//将数组转成list
|
||||||
|
List<File> fileList = Arrays.asList(files); |
||||||
|
//对分块文件排序
|
||||||
|
Collections.sort(fileList, new Comparator<File>() { |
||||||
|
@Override |
||||||
|
public int compare(File o1, File o2) { |
||||||
|
return Integer.parseInt(o1.getName()) - Integer.parseInt(o2.getName()); //升序
|
||||||
|
} |
||||||
|
}); |
||||||
|
//向合并文件写的流
|
||||||
|
RandomAccessFile raf_rw = new RandomAccessFile(mergeFile, "rw"); |
||||||
|
//缓存区
|
||||||
|
byte[] bytes = new byte[1024]; |
||||||
|
//遍历分块文件,向合并的文件写
|
||||||
|
for (File file : fileList) { |
||||||
|
//读分块的流
|
||||||
|
RandomAccessFile raf_r = new RandomAccessFile(file, "r"); |
||||||
|
int len = -1; |
||||||
|
while((len = raf_r.read(bytes)) != -1){ |
||||||
|
raf_rw.write(bytes, 0, len); |
||||||
|
} |
||||||
|
raf_r.close(); |
||||||
|
} |
||||||
|
raf_rw.close(); |
||||||
|
//合并文件完成后对合并的文件校验
|
||||||
|
FileInputStream fileInputStream_merge = new FileInputStream(mergeFile); |
||||||
|
FileInputStream fileInputStream_source = new FileInputStream(sourseFile); |
||||||
|
String md5_merge = DigestUtils.md5DigestAsHex(fileInputStream_merge); |
||||||
|
String md5_source = DigestUtils.md5DigestAsHex(fileInputStream_source); |
||||||
|
if (md5_merge.equals(md5_source)){ |
||||||
|
System.out.println("文件合并成功!"); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,37 @@ |
|||||||
|
package com.teaching; |
||||||
|
|
||||||
|
|
||||||
|
import lombok.extern.slf4j.Slf4j; |
||||||
|
import org.junit.jupiter.api.Test; |
||||||
|
import org.springframework.boot.test.context.SpringBootTest; |
||||||
|
|
||||||
|
import javax.annotation.Resource; |
||||||
|
|
||||||
|
@Slf4j |
||||||
|
@SpringBootTest |
||||||
|
public class test { |
||||||
|
|
||||||
|
@Resource |
||||||
|
private KnowledgeLearningCountServiceImpl knowledgeLearningCountService; |
||||||
|
@Test |
||||||
|
public void countknowledge(){ |
||||||
|
String userId = "2"; |
||||||
|
// System.out.println(knowledgeLearningCountService.list());
|
||||||
|
// List<KnowledgeLearningCount> list = knowledgeLearningCountService.query().eq("user_id", userId).list();
|
||||||
|
// HashSet<KnowledgeLearningCount> set = new HashSet<>(list);
|
||||||
|
// System.out.println(set.size());
|
||||||
|
// Long count = knowledgeLearningCountService.query().eq("user_id", userId).count();
|
||||||
|
// if (count == 0){
|
||||||
|
// KnowledgeLearningCount knowledgeLearningCount = new KnowledgeLearningCount();
|
||||||
|
// knowledgeLearningCount.setUserId(userId);
|
||||||
|
// knowledgeLearningCount.setNumber(1);
|
||||||
|
// knowledgeLearningCountService.save(knowledgeLearningCount);
|
||||||
|
// }else {
|
||||||
|
// //该用户学习知识点数+1
|
||||||
|
// boolean is = knowledgeLearningCountService.update()
|
||||||
|
// .setSql("number = "+set.size())
|
||||||
|
// .eq("user_id", userId).update();
|
||||||
|
// System.out.println(is);
|
||||||
|
// }
|
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue