com.danielflower.apprunner.mgmt.FileBasedGitRepoLoader Maven / Gradle / Ivy

Go to download
package com.danielflower.apprunner.mgmt;

import org.apache.commons.io.FileUtils;

import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock;

public class FileBasedGitRepoLoader implements GitRepoLoader {

    public static GitRepoLoader getGitRepoLoader(File dataDir) throws IOException {
        File dataStore = new File(dataDir, "repos.properties");
        FileUtils.touch(dataStore);

        Properties properties = new Properties();
        try (FileReader reader = new FileReader(dataStore)) {
            properties.load(reader);
        }

        return new FileBasedGitRepoLoader(dataStore, properties);
    }

    private final File file;
    private final Properties properties;
    private final ReadWriteLock lock = new ReentrantReadWriteLock();

    private FileBasedGitRepoLoader(File file, Properties properties) {
        this.file = file;
        this.properties = properties;
    }

    @Override
    public Map loadAll() throws Exception {
        Lock l = lock.readLock();
        l.lock();
        try {
            Map all = new HashMap<>();
            for (String key : properties.stringPropertyNames()) {
                all.put(key, properties.getProperty(key));
            }
            return all;
        } finally {
            l.unlock();
        }
    }

    @Override
    public void save(String name, String gitUrl) throws IOException {
        Lock l = lock.writeLock();
        l.lock();
        try {
            properties.setProperty(name, gitUrl);
            try (FileWriter writer = new FileWriter(file)) {
                properties.store(writer, "Saved by " + getClass().getSimpleName());
            }
        } finally {
            l.unlock();
        }
    }
}