io.yawp.driver.postgresql.PGHelpersDriver Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of yawp-postgresql Show documentation

YAWP! postgresql

There is a newer version: 2.08alpha

package io.yawp.driver.postgresql;

import io.yawp.commons.utils.Environment;
import io.yawp.driver.api.HelpersDriver;
import io.yawp.driver.postgresql.configuration.InitialContextSetup;
import io.yawp.driver.postgresql.configuration.WebConfiguration;
import io.yawp.driver.postgresql.tools.DatabaseSynchronizer;
import io.yawp.repository.Yawp;
import io.yawp.repository.scanner.RepositoryScanner;

import java.io.File;
import java.util.Set;

import static io.yawp.repository.Yawp.*;

public class PGHelpersDriver implements HelpersDriver {

    private DatabaseSynchronizer dbSynchronizer = new DatabaseSynchronizer();

    @Override
    public void deleteAll() {
        dbSynchronizer.truncateAll();
    }

    @Override
    public void sync() {
        InitialContextSetup.configure(getJettyConfigFile());
        dbSynchronizer.sync(scanEndpointClazzes());
    }

    private Set> scanEndpointClazzes() {
        WebConfiguration webConfiguration = new WebConfiguration(getWebConfigFile());
        if (webConfiguration.getPackagePrefix() != null) {
            Yawp.init(webConfiguration.getPackagePrefix());
        }
        return yawp.getFeatures().getEndpointClazzes();
    }

    private File getJettyConfigFile() {
        return new File(String.format("%s/WEB-INF/jetty-env.xml", Environment.getAppDir()));
    }

    private String getWebConfigFile() {
        return String.format("%s/WEB-INF/web.xml", Environment.getAppDir());
    }

}