|
|
@@ -1,147 +0,0 @@
|
|
|
-package org.linlinjava.litemall.os.web;
|
|
|
-
|
|
|
-import org.linlinjava.litemall.core.storage.StorageService;
|
|
|
-import org.linlinjava.litemall.core.util.CharUtil;
|
|
|
-import org.linlinjava.litemall.core.util.ResponseUtil;
|
|
|
-import org.linlinjava.litemall.db.domain.LitemallStorage;
|
|
|
-import org.linlinjava.litemall.db.service.LitemallStorageService;
|
|
|
-import org.springframework.beans.factory.annotation.Autowired;
|
|
|
-import org.springframework.core.io.Resource;
|
|
|
-import org.springframework.http.HttpHeaders;
|
|
|
-import org.springframework.http.MediaType;
|
|
|
-import org.springframework.http.ResponseEntity;
|
|
|
-import org.springframework.web.bind.annotation.*;
|
|
|
-import org.springframework.web.multipart.MultipartFile;
|
|
|
-
|
|
|
-import java.io.IOException;
|
|
|
-import java.io.InputStream;
|
|
|
-import java.time.LocalDateTime;
|
|
|
-import java.util.HashMap;
|
|
|
-import java.util.List;
|
|
|
-import java.util.Map;
|
|
|
-
|
|
|
-@RestController
|
|
|
-@RequestMapping("/os/storage")
|
|
|
-public class OsStorageController {
|
|
|
-
|
|
|
- @Autowired
|
|
|
- private StorageService storageService;
|
|
|
- @Autowired
|
|
|
- private LitemallStorageService litemallStorageService;
|
|
|
-
|
|
|
- private String generateKey(String originalFilename){
|
|
|
- int index = originalFilename.lastIndexOf('.');
|
|
|
- String suffix = originalFilename.substring(index);
|
|
|
-
|
|
|
- String key = null;
|
|
|
- LitemallStorage storageInfo = null;
|
|
|
-
|
|
|
- do{
|
|
|
- key = CharUtil.getRandomString(20) + suffix;
|
|
|
- storageInfo = litemallStorageService.findByKey(key);
|
|
|
- }
|
|
|
- while(storageInfo != null);
|
|
|
-
|
|
|
- return key;
|
|
|
- }
|
|
|
-
|
|
|
- @GetMapping("/list")
|
|
|
- public Object list(String key, String name,
|
|
|
- @RequestParam(value = "page", defaultValue = "1") Integer page,
|
|
|
- @RequestParam(value = "limit", defaultValue = "10") Integer limit,
|
|
|
- String sort, String order){
|
|
|
- List<LitemallStorage> storageList = litemallStorageService.querySelective(key, name, page, limit, sort, order);
|
|
|
- int total = litemallStorageService.countSelective(key, name, page, limit, sort, order);
|
|
|
- Map<String, Object> data = new HashMap<>();
|
|
|
- data.put("total", total);
|
|
|
- data.put("items", storageList);
|
|
|
-
|
|
|
- return ResponseUtil.ok(data);
|
|
|
- }
|
|
|
-
|
|
|
- @PostMapping("/create")
|
|
|
- public Object create(@RequestParam("file") MultipartFile file) {
|
|
|
- String originalFilename = file.getOriginalFilename();
|
|
|
- InputStream inputStream = null;
|
|
|
- try {
|
|
|
- inputStream = file.getInputStream();
|
|
|
- } catch (IOException e) {
|
|
|
- e.printStackTrace();
|
|
|
- return ResponseUtil.badArgumentValue();
|
|
|
- }
|
|
|
- String key = generateKey(originalFilename);
|
|
|
- storageService.store(file, key);
|
|
|
-
|
|
|
- String url = storageService.generateUrl(key);
|
|
|
- LitemallStorage storageInfo = new LitemallStorage();
|
|
|
- storageInfo.setName(originalFilename);
|
|
|
- storageInfo.setSize((int)file.getSize());
|
|
|
- storageInfo.setType(file.getContentType());
|
|
|
- storageInfo.setAddTime(LocalDateTime.now());
|
|
|
- storageInfo.setModified(LocalDateTime.now());
|
|
|
- storageInfo.setKey(key);
|
|
|
- storageInfo.setUrl(url);
|
|
|
- litemallStorageService.add(storageInfo);
|
|
|
- return ResponseUtil.ok(storageInfo);
|
|
|
- }
|
|
|
-
|
|
|
- @PostMapping("/read")
|
|
|
- public Object read(Integer id) {
|
|
|
- if(id == null){
|
|
|
- return ResponseUtil.badArgument();
|
|
|
- }
|
|
|
- LitemallStorage storageInfo = litemallStorageService.findById(id);
|
|
|
- if(storageInfo == null){
|
|
|
- return ResponseUtil.badArgumentValue();
|
|
|
- }
|
|
|
- return ResponseUtil.ok(storageInfo);
|
|
|
- }
|
|
|
-
|
|
|
- @PostMapping("/update")
|
|
|
- public Object update(@RequestBody LitemallStorage litemallStorage) {
|
|
|
-
|
|
|
- litemallStorageService.update(litemallStorage);
|
|
|
- return ResponseUtil.ok(litemallStorage);
|
|
|
- }
|
|
|
-
|
|
|
- @PostMapping("/delete")
|
|
|
- public Object delete(@RequestBody LitemallStorage litemallStorage) {
|
|
|
- litemallStorageService.deleteByKey(litemallStorage.getKey());
|
|
|
- storageService.delete(litemallStorage.getKey());
|
|
|
- return ResponseUtil.ok();
|
|
|
- }
|
|
|
-
|
|
|
- @GetMapping("/fetch/{key:.+}")
|
|
|
- public ResponseEntity<Resource> fetch(@PathVariable String key) {
|
|
|
- LitemallStorage litemallStorage = litemallStorageService.findByKey(key);
|
|
|
- if(key == null){
|
|
|
- ResponseEntity.notFound();
|
|
|
- }
|
|
|
- String type = litemallStorage.getType();
|
|
|
- MediaType mediaType = MediaType.parseMediaType(type);
|
|
|
-
|
|
|
- Resource file = storageService.loadAsResource(key);
|
|
|
- if(file == null) {
|
|
|
- ResponseEntity.notFound();
|
|
|
- }
|
|
|
- return ResponseEntity.ok().contentType(mediaType).body(file);
|
|
|
- }
|
|
|
-
|
|
|
- @GetMapping("/download/{key:.+}")
|
|
|
- public ResponseEntity<Resource> download(@PathVariable String key) {
|
|
|
- LitemallStorage litemallStorage = litemallStorageService.findByKey(key);
|
|
|
- if(key == null){
|
|
|
- ResponseEntity.notFound();
|
|
|
- }
|
|
|
- String type = litemallStorage.getType();
|
|
|
- MediaType mediaType = MediaType.parseMediaType(type);
|
|
|
-
|
|
|
- Resource file = storageService.loadAsResource(key);
|
|
|
- if(file == null) {
|
|
|
- ResponseEntity.notFound();
|
|
|
- }
|
|
|
- return ResponseEntity.ok().contentType(mediaType).header(HttpHeaders.CONTENT_DISPOSITION,
|
|
|
- "attachment; filename=\"" + file.getFilename() + "\"").body(file);
|
|
|
- }
|
|
|
-
|
|
|
-}
|
Junling Bu