nl.knaw.huygens.alexandria.text.FileSystemTextService.java Source code

Introduction

Here is the source code for nl.knaw.huygens.alexandria.text.FileSystemTextService.java

Source

package nl.knaw.huygens.alexandria.text;

/*
 * #%L
 * alexandria-service
 * =======
 * Copyright (C) 2015 - 2016 Huygens ING (KNAW)
 * =======
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/gpl-3.0.html>.
 * #L%
 */

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Optional;
import java.util.UUID;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;

import com.google.inject.Singleton;

@Singleton
public class FileSystemTextService implements TextService {
    File textDir;
    // LoadingCache<UUID, Optional<String>> textCache = CacheBuilder.newBuilder().maximumSize(1000).build(new CacheLoader<UUID, Optional<String>>() {
    // @Override
    // public Optional<String> load(UUID uuid) {
    // return loadText(uuid);
    // }
    // });

    public FileSystemTextService(String textsDirectoryName) {
        this.textDir = new File(textsDirectoryName);
        if (!textDir.isDirectory()) {
            throw new RuntimeException("directory not found: " + textsDirectoryName);
        }
    }

    // @Override
    // public void set(UUID resourceUUID, String text) {
    // textCache.put(resourceUUID, Optional.of(text));
    // File textFile = textFile(resourceUUID);
    // try {
    // FileUtils.writeStringToFile(textFile, text);
    //
    // } catch (IOException e) {
    // throw new RuntimeException(e);
    // }
    // }

    @Override
    public void setFromStream(UUID resourceUUID, InputStream inputStream) {
        File textFile = textFile(resourceUUID);
        try {
            FileOutputStream outputStream = FileUtils.openOutputStream(textFile);
            IOUtils.copy(inputStream, outputStream);
            inputStream.close();
            outputStream.close();

        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    // @Override
    // public Optional<String> get(UUID resourceUUID) {
    // try {
    // return textCache.get(resourceUUID);
    //
    // } catch (ExecutionException e) {
    // e.printStackTrace();
    // throw new RuntimeException(e);
    // }
    // }

    @Override
    public Optional<InputStream> getAsStream(UUID resourceUUID) {
        File textFile = textFile(resourceUUID);
        if (!textFile.isFile()) {
            return Optional.empty();
        }
        try {
            return Optional.of(FileUtils.openInputStream(textFile));
        } catch (IOException e) {
            e.printStackTrace();
            throw new RuntimeException(e);
        }
    }

    private Optional<String> loadText(UUID uuid) {
        File textFile = textFile(uuid);
        if (!textFile.isFile()) {
            return Optional.empty();
        }
        try {
            String text = FileUtils.readFileToString(textFile);
            return Optional.ofNullable(text);
        } catch (IOException e) {
            e.printStackTrace();
            throw new RuntimeException(e);
        }
    }

    private File textFile(UUID uuid) {
        String fileName = uuid + ".txt";
        return new File(textDir, fileName);
    }

}