LocalMongoFactory.java

package de.is24.infrastructure.gridfs.http.mongo.util;

import com.google.common.annotations.VisibleForTesting;
import de.flapdoodle.embed.mongo.MongodExecutable;
import de.flapdoodle.embed.mongo.MongodProcess;
import de.flapdoodle.embed.mongo.MongodStarter;
import de.flapdoodle.embed.mongo.config.ArtifactStoreBuilder;
import de.flapdoodle.embed.mongo.config.DownloadConfigBuilder;
import de.flapdoodle.embed.mongo.config.IMongodConfig;
import de.flapdoodle.embed.mongo.config.MongodConfigBuilder;
import de.flapdoodle.embed.mongo.config.RuntimeConfigBuilder;
import de.flapdoodle.embed.mongo.distribution.Version;
import de.flapdoodle.embed.process.config.IRuntimeConfig;
import de.flapdoodle.embed.process.config.store.HttpProxyFactory;
import de.flapdoodle.embed.process.config.store.IProxyFactory;
import de.flapdoodle.embed.process.config.store.NoProxyFactory;
import de.flapdoodle.embed.process.io.directories.FixedPath;
import de.flapdoodle.embed.process.io.directories.PlatformTempDir;
import de.flapdoodle.embed.process.io.progress.LoggingProgressListener;
import de.is24.infrastructure.gridfs.http.utils.retry.RetryUtils;

import java.io.File;
import java.util.logging.Level;
import java.util.logging.Logger;

import static de.flapdoodle.embed.mongo.Command.MongoD;
import static de.is24.infrastructure.gridfs.http.utils.retry.RetryUtils.execute;


public class LocalMongoFactory {
  private static final String TEMP_DIR = new PlatformTempDir().asFile().getAbsolutePath();
  @VisibleForTesting
  static final FixedPath MONGO_DOWNLOAD_FOLDER = new FixedPath(TEMP_DIR + File.separator + "embedded-mongo");
  private static final Logger LOGGER = Logger.getLogger(LocalMongoFactory.class.getCanonicalName());

  @VisibleForTesting
  static MongodStarter createMongoStarter() {
    de.flapdoodle.embed.process.config.store.DownloadConfigBuilder downloadConfigBuilder =
      new DownloadConfigBuilder() //
      .defaultsForCommand(MongoD) //
      .progressListener(new LoggingProgressListener(LOGGER, Level.INFO)) //
      .downloadPath("https://fastdl.mongodb.org/") //
      .proxyFactory(proxyForEnvironment()) //
      .artifactStorePath(MONGO_DOWNLOAD_FOLDER); //
    de.flapdoodle.embed.process.store.ArtifactStoreBuilder download =
      new ArtifactStoreBuilder() //
      .defaults(MongoD) //
      .download(downloadConfigBuilder);

    IRuntimeConfig runtimeConfig =
      new RuntimeConfigBuilder() //
      .defaultsWithLogger(MongoD, LOGGER) //
      .artifactStore(download).build();
    return MongodStarter.getInstance(runtimeConfig);
  }

  private static IProxyFactory proxyForEnvironment() {
    if (isHttpProxyEnabled()) {
       return new HttpProxyFactory(System.getProperty("http.proxyHost"), Integer.valueOf(System.getProperty("http.proxyPort")));
    }
    else {
      return new NoProxyFactory();
    }
  }

  private static boolean isHttpProxyEnabled() {
    return System.getProperty("http.proxyHost") != null && System.getProperty("http.proxyPort") != null;
  }

  public static MongoProcessHolder createMongoProcess() throws Throwable {
    final MongodStarter runtime = createMongoStarter();

    return execute().maxTries(3).wait(3).command(new RetryUtils.Retryable<MongoProcessHolder>() {
        @Override
        public MongoProcessHolder run() throws Throwable {
          IMongodConfig config = new MongodConfigBuilder().version(getVersion()).build();
          MongodExecutable mongodExecutable = runtime.prepare(config);
          MongodProcess mongoProcess = mongodExecutable.start();
          return new MongoProcessHolder(mongodExecutable, mongoProcess, config.net().getPort());
        }

        private Version getVersion() {
          String version = System.getProperty("embedded.mongodb.version", Version.Main.PRODUCTION.asInDownloadPath());
          return Version.valueOf("V" + version.replaceAll("\\.", "_"));
        }
    });

  }
}