io.graphenee.util.storage.FileStorageFactory Maven / Gradle / Ivy

Go to download
/*******************************************************************************
 * Copyright (c) 2016, 2018 Farrukh Ijaz
 *
 * 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 io.graphenee.util.storage;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.net.URI;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;

import org.springframework.util.StreamUtils;

public class FileStorageFactory {

	public static FileStorage createLocalFileStorage(File rootFolder) {
		return new LocalFileStorage(rootFolder);
	}

	public static FileStorage createS3FileStorage(String awsKey, String awsSecret, String bucketName) {
		throw new UnsupportedOperationException("This feature is not yet implemented.");
	}

	static class LocalFileStorage implements FileStorage {

		private File rootFolder;

		public LocalFileStorage(File rootFolder) {
			this.rootFolder = rootFolder;
		}

		@Override
		public URI resolveToURI(String resourcePath) throws ResolveFailedException {
			File resource = new File(rootFolder, resourcePath.replace('/', File.separatorChar));
			try {
				return resource.toURI();
			} catch (Exception e) {
				throw new ResolveFailedException("Failed to resolve resource " + resourcePath);
			}
		}

		@Override
		public InputStream resolve(String resourcePath) throws ResolveFailedException {
			File resource = new File(rootFolder, resourcePath.replace('/', File.separatorChar));
			try {
				if (resource.exists() && resource.length() > 0) {
					return new FileInputStream(resource);
				} else {
					throw new ResolveFailedException("Failed to resolve resource " + resourcePath);
				}
			} catch (FileNotFoundException e) {
				throw new ResolveFailedException("Failed to resolve resource " + resourcePath, e);
			}
		}

		@Override
		public Future save(String folder, String fileName, InputStream inputStream) throws SaveFailedException {
			return Executors.newCachedThreadPool().submit(() -> {
				String resourcePath = resourcePath(folder, fileName);
				File resource = new File(rootFolder, resourcePath.replace('/', File.separatorChar));
				resource.getParentFile().mkdirs();
				try {
					FileOutputStream fout = new FileOutputStream(resource);
					StreamUtils.copy(inputStream, fout);
					long fileSize = resource.length();
					// checksum will be computed in future...
					String checksum = null;
					FileMetaData fileMetaData = new FileMetaData(resourcePath, fileName, Long.valueOf(fileSize).intValue(), checksum);
					return fileMetaData;
				} catch (Exception e) {
					throw new SaveFailedException(e);
				}
			});
		}

		@Override
		public boolean exists(String resourcePath) {
			if (resourcePath == null)
				return false;
			File file = new File(rootFolder, resourcePath);
			return file.exists();
		}

	}

}