com.baifendian.swordfish.webserver.service.storage.FileSystemStorageService.java Source code

Introduction

Here is the source code for com.baifendian.swordfish.webserver.service.storage.FileSystemStorageService.java

Source

/*
 * Copyright (C) 2017 Baifendian Corporation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *          http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.baifendian.swordfish.webserver.service.storage;

import java.io.File;
import java.io.IOException;
import java.net.MalformedURLException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import org.apache.commons.io.FileUtils;
import org.springframework.core.io.Resource;
import org.springframework.core.io.UrlResource;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

@Service
public class FileSystemStorageService implements StorageService {

    /**
     * ?, 
     *
     * @param file
     * @param destFilename
     */
    @Override
    public void store(MultipartFile file, String destFilename) {
        try {
            if (file.isEmpty()) {
                throw new StorageException("Failed to store empty file " + file.getOriginalFilename());
            }

            File destFile = new File(destFilename);
            File destDir = new File(destFile.getParent());

            if (!destDir.exists()) {
                FileUtils.forceMkdir(destDir);
            }

            Files.copy(file.getInputStream(), Paths.get(destFilename));
        } catch (IOException e) {
            throw new StorageException("Failed to store file " + file.getOriginalFilename(), e);
        }
    }

    @Override
    public Resource loadAsResource(String filename) {
        try {
            Path file = Paths.get(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);
        }
    }

    /**
     * , 
     *
     * @param dir
     */
    @Override
    public void deleteDir(String dir) throws IOException {
        FileUtils.deleteDirectory(new File(dir));
    }

    /**
     * ?
     *
     * @param filename
     */
    public String readFileToString(String filename) throws IOException {
        return FileUtils.readFileToString(new File(filename), "utf-8");
    }

    /**
     * 
     *
     * @param filename
     */
    @Override
    public void deleteFile(String filename) throws IOException {
        FileUtils.forceDelete(new File(filename));
    }
}