Skip to content

Commit 0e9cbc9

Browse files
committed
file open service and controller
be
1 parent bac2580 commit 0e9cbc9

File tree

4 files changed

+21
-2
lines changed

4 files changed

+21
-2
lines changed

rsa-backend/src/main/java/com/example/rsa/controller/FileController.java

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,15 +39,21 @@ public List<RsaFile> allFiles() {
3939
return fileService.getAllFiles();
4040
}
4141

42-
@Operation(summary = "Get public key")
42+
@Operation(summary = "Get decrypt file")
4343
@GetMapping("/get/{fileName}")
4444
public byte[] getFile(@PathVariable String fileName) throws IOException {
4545
return fileEncryptionService.decryptFileByName(fileName);
4646
}
4747

48-
@Operation(summary = "Get public key")
48+
@Operation(summary = "Get file on storage")
4949
@GetMapping("/get/normal/{fileName}")
5050
public byte[] getNormalFile(@PathVariable String fileName) throws IOException {
5151
return fileEncryptionService.fileByName(fileName);
5252
}
53+
54+
@Operation(summary = "Get file info by userId")
55+
@GetMapping("/info/{userId}")
56+
public List<RsaFile> getFileInfo(@PathVariable Integer userId){
57+
return fileService.fileByUserId(userId);
58+
}
5359
}

rsa-backend/src/main/java/com/example/rsa/model/RsaFile.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ public class RsaFile {
77

88
private Integer id;
99
private String name;
10+
private String originalName;
1011
private Integer ownerId;
1112
private Integer recipientId;
1213
private Long size;

rsa-backend/src/main/java/com/example/rsa/service/FileEncryptionService.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
package com.example.rsa.service;
22

3+
import com.example.rsa.model.RsaFile;
34
import org.springframework.beans.factory.annotation.Autowired;
45
import org.springframework.stereotype.Service;
56

@@ -11,6 +12,7 @@
1112
import java.nio.file.Path;
1213
import java.nio.file.Paths;
1314
import java.util.Arrays;
15+
import java.util.List;
1416

1517
@Service
1618
public class FileEncryptionService {

rsa-backend/src/main/java/com/example/rsa/service/FileService.java

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
import java.util.ArrayList;
1313
import java.util.Comparator;
1414
import java.util.List;
15+
import java.util.stream.Collectors;
1516
import java.util.stream.Stream;
1617

1718
@Service
@@ -39,6 +40,7 @@ public void saveFile(MultipartFile file,Integer userId,Integer recipientId) thro
3940
RsaFile newFile = new RsaFile();
4041
newFile.setId(files.size()+1);
4142
newFile.setName(filename);
43+
newFile.setOriginalName(file.getOriginalFilename());
4244
newFile.setOwnerId(userId);
4345
newFile.setSize(fileSize);
4446
newFile.setRecipientId(recipientId);
@@ -70,4 +72,12 @@ public void cleanAllFiles() {
7072
System.out.println(getAllFiles());
7173
}
7274

75+
76+
public List<RsaFile> fileByUserId(Integer userId) {
77+
return files.stream()
78+
.filter(rsaFile -> rsaFile.getOwnerId().equals(userId))
79+
.collect(Collectors.toList());
80+
}
81+
82+
7383
}

0 commit comments

Comments
 (0)