de.flapdoodle.embed.mongo.distribution.IFeatureAwareVersion Java Examples

ReplSet(IFeatureAwareVersion serverVersion, String replicaSetName, boolean silent, Integer... serverPorts) {

			this.serverVersion = serverVersion;
			this.replicaSetName = replicaSetName;
			this.serverPorts = defaultPortsIfRequired(serverPorts);
			this.configServerPorts = defaultPortsIfRequired(null);
			this.configServerReplicaSetName = DEFAULT_CONFIG_SERVER_REPLICA_SET_NAME;
			this.mongosPort = randomOrDefaultServerPort();

			if (silent) {
				outputFunction = it -> new ProcessOutput(Processors.silent(),
						Processors.namedConsole("[ " + it.commandName() + " error]"), Processors.console());
			} else {
				outputFunction = it -> ProcessOutput.getDefaultInstance(it.commandName());
			}
		}
 

/**
 * Create a default {@code mongos} config.
 *
 * @param version
 * @param port
 * @param cmdOptions
 * @param configServerReplicaSet
 * @param configServerPort
 * @return
 */
private static IMongosConfig defaultMongosConfig(IFeatureAwareVersion version, int port, IMongoCmdOptions cmdOptions,
		String configServerReplicaSet, int configServerPort) {

	try {

		MongosConfigBuilder builder = new MongosConfigBuilder() //
				.version(version) //
				.withLaunchArgument("--quiet", null) //
				.net(new Net(LOCALHOST, port, Network.localhostIsIPv6())) //
				.cmdOptions(cmdOptions);

		if (StringUtils.hasText(configServerReplicaSet)) {
			builder = builder.replicaSet(configServerReplicaSet) //
					.configDB(LOCALHOST + ":" + configServerPort);
		}

		return builder.build();
	} catch (IOException e) {
		throw new RuntimeException(e);
	}
}
 

private IMongodConfig newMongodConfig(final IFeatureAwareVersion version) throws IOException {
  MongoCmdOptionsBuilder builder = new MongoCmdOptionsBuilder().useSmallFiles(false).useNoPrealloc(false);
  return new MongodConfigBuilder().version(version)
                                  .cmdOptions(builder.build())
                                  .net(new Net(LOCALHOST, Network.getFreeServerPort(),
                                               Network.localhostIsIPv6())).build();
}
 

private IMongodConfig newMongodConfig(final IFeatureAwareVersion version) throws IOException {
  MongoCmdOptionsBuilder builder = new MongoCmdOptionsBuilder().useSmallFiles(false).useNoPrealloc(false);
  return new MongodConfigBuilder().version(version)
                                  .cmdOptions(builder.build())
                                  .net(new Net(LOCALHOST, Network.getFreeServerPort(),
                                                                Network.localhostIsIPv6())).build();
}
 

private IMongodConfig newMongodConfig(final IFeatureAwareVersion version) throws IOException {
  MongoCmdOptionsBuilder builder = new MongoCmdOptionsBuilder().useSmallFiles(false).useNoPrealloc(false);
  return new MongodConfigBuilder().version(version)
                                  .cmdOptions(builder.build())
                                  .net(new Net(LOCALHOST, Network.getFreeServerPort(),
                                               Network.localhostIsIPv6())).build();
}
 

private String getBitSize(Distribution distribution) {
  String sbitSize;
  switch (distribution.getBitsize()) {
    case B32:
      if (distribution.getVersion() instanceof IFeatureAwareVersion) {
        IFeatureAwareVersion featuredVersion = (IFeatureAwareVersion) distribution.getVersion();
        if (featuredVersion.enabled(Feature.ONLY_64BIT)) {
          throw new IllegalArgumentException("this version does not support 32Bit: "+distribution);
        }
      }

      switch (distribution.getPlatform()) {
        case Linux:
          sbitSize = "i686";
          break;
        case Windows:
        case OS_X:
          sbitSize = "i386";
          break;
        default:
          throw new IllegalArgumentException("Platform " + distribution.getPlatform() + " not supported yet on 32Bit Platform");
      }
      break;
    case B64:
      sbitSize = "x86_64";
      break;
    default:
      throw new IllegalArgumentException("Unknown BitSize " + distribution.getBitsize());
  }
  return sbitSize;
}
 

/**
 * Create a default {@code mongod} config.
 *
 * @param version
 * @param port
 * @param cmdOptions
 * @param configServer
 * @param shardServer
 * @param replicaSet
 * @return
 */
private static IMongodConfig defaultMongodConfig(IFeatureAwareVersion version, int port, IMongoCmdOptions cmdOptions,
		boolean configServer, boolean shardServer, String replicaSet) {

	try {

		MongodConfigBuilder builder = new MongodConfigBuilder() //
				.version(version) //
				.withLaunchArgument("--quiet") //
				.net(new Net(LOCALHOST, port, Network.localhostIsIPv6())) //
				.configServer(configServer).cmdOptions(cmdOptions); //

		if (StringUtils.hasText(replicaSet)) {

			builder = builder //
					.replication(new Storage(null, replicaSet, 0));

			if (!configServer) {
				builder = builder.shardServer(shardServer);
			} else {
				builder = builder.shardServer(false);
			}
		}

		return builder.build();
	} catch (IOException e) {
		throw new RuntimeException(e);
	}
}
 

public SingleMongoDBResource(IFeatureAwareVersion version) {
  super(version);
}
 

protected AbstractMongoDBResource(IFeatureAwareVersion version) {
  this.version = version;
}
 

public IFeatureAwareVersion getVersion() {
  return version;
}
 

@SuppressWarnings("SameParameterValue")
private static boolean isFeatureEnabled(Distribution distribution, Feature feature) {
  return (distribution.getVersion() instanceof IFeatureAwareVersion
          &&  ((IFeatureAwareVersion) distribution.getVersion()).enabled(feature));
}
 

public static EmbeddedMongoFactory with(final IFeatureAwareVersion version) throws IOException {
    return new EmbeddedMongoFactory(version);
}
 

private IMongodConfig newMongodConfig(final IFeatureAwareVersion version) throws UnknownHostException, IOException {
    final Net net = new Net(setPortToDefaultOrRandomOpen(), false);
    return new MongodConfigBuilder().version(version).net(net).build();
}
 

/**
 * Create the testing utility using the specified version of MongoDB.
 *
 * @param version
 *            version of MongoDB.
 */
private EmbeddedMongoFactory(final IFeatureAwareVersion version) throws IOException {
    final MongodStarter runtime = MongodStarter.getInstance(new RuntimeConfigBuilder().defaultsWithLogger(Command.MongoD, logger).build());
    mongodExecutable = runtime.prepare(newMongodConfig(version));
    mongodProcess = mongodExecutable.start();
}
 

/**
 * Configure the MongoDB {@link IFeatureAwareVersion version}.
 *
 * @param version
 * @return
 */
public Builder withVersion(IFeatureAwareVersion version) {

	this.version = version;
	return this;
}