MediaFileTest.java 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. import cn.reghao.dfs.store.DfsStoreApplication;
  2. import cn.reghao.dfs.store.db.mapper.*;
  3. import cn.reghao.dfs.store.model.po.*;
  4. import lombok.extern.slf4j.Slf4j;
  5. import org.junit.Test;
  6. import org.junit.runner.RunWith;
  7. import org.springframework.beans.factory.annotation.Autowired;
  8. import org.springframework.boot.test.context.SpringBootTest;
  9. import org.springframework.test.context.ActiveProfiles;
  10. import org.springframework.test.context.junit4.SpringRunner;
  11. import java.util.List;
  12. /**
  13. * @author reghao
  14. * @date 2023-03-22 16:19:12
  15. */
  16. @Slf4j
  17. @ActiveProfiles("dev")
  18. @SpringBootTest(classes = DfsStoreApplication.class)
  19. @RunWith(SpringRunner.class)
  20. public class MediaFileTest {
  21. @Autowired
  22. VideoFileMapper videoFileMapper;
  23. @Autowired
  24. VideoUrlMapper videoUrlMapper;
  25. @Autowired
  26. FileMetaMapper fileMetaMapper;
  27. public void mediaTest() {
  28. videoUrlTest();
  29. log.info("------------------------------------------------------------");
  30. imageFileTest();
  31. }
  32. @Test
  33. public void videoUrlTest() {
  34. List<VideoUrl> list = videoUrlMapper.findAll();
  35. for (VideoUrl videoUrl : list) {
  36. process(videoUrl);
  37. }
  38. }
  39. private void process(VideoUrl videoUrl) {
  40. String videoFileId = videoUrl.getVideoFileId();
  41. VideoFile videoFile = videoFileMapper.findByVideoFileId(videoFileId);
  42. if (videoFile == null) {
  43. log.error("videoFileId {} not exist in video_file", videoFileId);
  44. return;
  45. }
  46. String objectId = videoUrl.getObjectId();
  47. FileMeta fileMeta = fileMetaMapper.findByObjectId(objectId);
  48. if (fileMeta == null) {
  49. log.error("objectId {} not exist in file_meta", objectId);
  50. }
  51. }
  52. @Autowired
  53. ImageFileMapper imageFileMapper;
  54. @Test
  55. public void imageFileTest() {
  56. List<ImageFile> list = imageFileMapper.findAll();
  57. for (ImageFile imageFile : list) {
  58. process(imageFile);
  59. }
  60. }
  61. private void process(ImageFile imageFile) {
  62. String jpegObjectId = imageFile.getJpegObjectId();
  63. FileMeta fileMeta = fileMetaMapper.findByObjectId(jpegObjectId);
  64. if (fileMeta == null) {
  65. log.error("jpegObjectId {} not exist in file_meta", jpegObjectId);
  66. }
  67. String webpObjectId = imageFile.getWebpObjectId();
  68. if (webpObjectId != null) {
  69. FileMeta fileMeta1 = fileMetaMapper.findByObjectId(webpObjectId);
  70. if (fileMeta1 == null) {
  71. log.error("webpObjectId {} not exist in file_meta", webpObjectId);
  72. }
  73. }
  74. }
  75. @Test
  76. public void test11() {
  77. int scope = 4;
  78. List<String> list = List.of("eadf4146d67846f7abdf8ce63e354440", "98d20cf30c27417c8e5bcdb83e18562b");
  79. fileMetaMapper.updateScopeByObjectIds(scope, list);
  80. }
  81. }