108 lines
3.4 KiB
Java
108 lines
3.4 KiB
Java
package ba.steleks.storage;
|
|
|
|
import ba.steleks.storage.error.exception.StorageException;
|
|
import ba.steleks.storage.error.exception.StorageFileNotFoundException;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.core.io.InputStreamSource;
|
|
import org.springframework.core.io.Resource;
|
|
import org.springframework.core.io.UrlResource;
|
|
import org.springframework.stereotype.Service;
|
|
import org.springframework.util.FileSystemUtils;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
import java.io.IOException;
|
|
import java.net.MalformedURLException;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Path;
|
|
import java.nio.file.Paths;
|
|
import java.util.stream.Stream;
|
|
|
|
@Service
|
|
public class FileSystemStorageService implements StorageService {
|
|
|
|
private final Path rootLocation;
|
|
|
|
@Autowired
|
|
public FileSystemStorageService(StorageProperties properties) {
|
|
this.rootLocation = Paths.get(properties.getLocation());
|
|
}
|
|
|
|
@Override
|
|
public void store(MultipartFile file, String dest) {
|
|
try {
|
|
if (file.isEmpty()) {
|
|
throw new StorageException("Failed to store empty file " + file.getOriginalFilename());
|
|
}
|
|
|
|
store((InputStreamSource) file, dest);
|
|
} catch (StorageException e) {
|
|
throw new StorageException("Failed to store file " + file.getOriginalFilename(), e);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void store(InputStreamSource inputStreamSource, String dest) {
|
|
try {
|
|
String tempDest="";
|
|
String[] locations=this.rootLocation.resolve(dest).toString().split("/");
|
|
for(int i=0; i<locations.length-1;i++)
|
|
tempDest=tempDest+locations[i]+"/";
|
|
|
|
if(!Files.exists(Paths.get(tempDest)))
|
|
Files.createDirectory(Paths.get(tempDest));
|
|
|
|
Files.copy(inputStreamSource.getInputStream(),
|
|
this.rootLocation.resolve(dest));
|
|
} catch (IOException e) {
|
|
throw new StorageException("Failed to store file " + inputStreamSource.toString(), e);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public Stream<Path> loadAll() {
|
|
try {
|
|
return Files.walk(this.rootLocation, 1)
|
|
.filter(path -> !path.equals(this.rootLocation))
|
|
.map(path -> this.rootLocation.relativize(path));
|
|
} catch (IOException e) {
|
|
throw new StorageException("Failed to read stored files", e);
|
|
}
|
|
|
|
}
|
|
|
|
@Override
|
|
public Path load(String filename) {
|
|
return rootLocation.resolve(filename);
|
|
}
|
|
|
|
@Override
|
|
public Resource loadAsResource(String filename) {
|
|
try {
|
|
Path file = load(filename);
|
|
Resource resource = new UrlResource(file.toUri());
|
|
if(resource.exists() || resource.isReadable()) {
|
|
return resource;
|
|
}
|
|
else {
|
|
throw new StorageFileNotFoundException("Could not read file: " + filename);
|
|
|
|
}
|
|
} catch (MalformedURLException e) {
|
|
throw new StorageFileNotFoundException("Could not read file: " + filename, e);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void deleteAll() {
|
|
FileSystemUtils.deleteRecursively(rootLocation.toFile());
|
|
}
|
|
|
|
@Override
|
|
public void init() {
|
|
try {
|
|
Files.createDirectory(rootLocation);
|
|
} catch (IOException e) {
|
|
throw new StorageException("Could not initialize storage", e);
|
|
}
|
|
}
|
|
} |