Skip to content

Commit

Permalink
fix : mosaic video test 수정
Browse files Browse the repository at this point in the history
  • Loading branch information
sycuuui committed May 6, 2024
1 parent 9d64ff4 commit d019012
Showing 1 changed file with 16 additions and 8 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -224,42 +224,50 @@ public ResponseEntity<Resource> sendToFlaskWithImagesAndVideoTest(String folderN
List<byte[]> imageContents = new ArrayList<>();
for (String imageName : videoReq.imageName()) {
// 이미지 파일 가져오기
S3ObjectInputStream imageFile = s3Client.get(folderName + "/" + imageName);
byte[] imageContent = IOUtils.toByteArray(imageFile);
imageContents.add(imageContent);
for(int i=0;i<3;i++){
String imagePath = folderName + "/" + imageName+"/"+(i+1)+".jpeg";
System.out.println(imagePath);
S3ObjectInputStream imageFile = s3Client.get(imagePath);
byte[] imageContent = IOUtils.toByteArray(imageFile);
imageContents.add(imageContent);
}
System.out.println("finish");
}

System.out.println("check");
// return videoReq.videoName();
// Multipart 요청 생성
MultiValueMap<String, Object> body = new LinkedMultiValueMap<>();
System.out.println("check2");
body.add("video", new ByteArrayResource(videoContent) {
@Override
public String getFilename() {
return videoReq.videoName(); // 동영상 파일 이름 가져오기
}
});

System.out.println("check3");
// 이미지 파일들을 요청에 추가
for (int i = 0; i < imageContents.size(); i++) {
byte[] content = imageContents.get(i);
final String imageName = videoReq.imageName().get(i);
final String imageName = i +".jpeg";
body.add("image" + (i + 1), new ByteArrayResource(content) {
@Override
public String getFilename() {
return imageName; // 이미지 파일 이름 가져오기
}
});
}
System.out.println("check4");

body.add("imageSize",imageContents.size());

System.out.println("check5");
// HTTP 요청 설정
HttpHeaders headers = new HttpHeaders();
headers.setContentType(MediaType.MULTIPART_FORM_DATA);
HttpEntity<MultiValueMap<String, Object>> requestEntity = new HttpEntity<>(body, headers);

// Flask 서버로 요청 보내기
RestTemplate restTemplate = new RestTemplate();
String url = "http://13.124.110.226:5000/video"; // Flask 서버 URL
String url = "http://127.0.0.1:5000/video"; // Flask 서버 URL
ResponseEntity<byte[]> response = restTemplate.postForEntity(url, requestEntity, byte[].class);

// Flask에서 반환된 파일을 다시 클라이언트에게 반환
Expand Down

0 comments on commit d019012

Please sign in to comment.