com.galenframework.storage.repository.LocalFileStorage.java Source code

Introduction

Here is the source code for com.galenframework.storage.repository.LocalFileStorage.java

Source

/*******************************************************************************
* Copyright 2016 Ivan Shubin http://galenframework.com
*
* 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.galenframework.storage.repository;

import com.galenframework.storage.model.FileInfo;
import org.apache.commons.codec.binary.Base64;
import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;

import java.io.*;
import java.security.DigestInputStream;
import java.security.MessageDigest;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.UUID;

public class LocalFileStorage implements FileStorage {

    private final File root;

    public LocalFileStorage(String storagePath) {
        root = new File(storagePath);
        if (!root.exists()) {
            root.mkdirs();
        }
    }

    @Override
    public byte[] readFile(String imagePath) {
        try {
            File file = new File(root.getPath() + File.separator + imagePath);
            if (file.exists()) {
                return FileUtils.readFileToByteArray(file);
            } else {
                throw new FileNotFoundException(file.getPath());
            }
        } catch (Exception ex) {
            throw new RuntimeException("Can't read file: " + imagePath, ex);
        }
    }

    @Override
    public FileInfo saveImageToStorage(InputStream inputStream) {
        try {
            MessageDigest md = MessageDigest.getInstance("MD5");

            String dirsPath = generateImageDirsPath();

            File dirs = new File(root.getPath() + File.separator + dirsPath);
            dirs.mkdirs();

            String fileName = UUID.randomUUID().toString();
            File file = new File(dirs.getPath() + File.separator + fileName);
            file.createNewFile();
            FileOutputStream fos = new FileOutputStream(file);

            DigestInputStream dis = new DigestInputStream(inputStream, md);

            IOUtils.copy(dis, fos);
            fos.flush();
            fos.close();

            byte[] digest = md.digest();
            String hash = Base64.encodeBase64String(digest);
            return new FileInfo(hash, dirsPath + File.separator + fileName);
        } catch (Exception ex) {
            throw new RuntimeException("Cannot save image to storage", ex);
        }
    }

    public String generateImageDirsPath() {
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy/MM/dd/HH");
        return sdf.format(new Date()).replace("/", File.separator);
    }
}