package io.vertx.ext.mongo.impl.config;
import com.mongodb.*;
import com.mongodb.MongoClientSettings;
import com.mongodb.connection.*;
import io.vertx.core.Vertx;
import io.vertx.core.json.JsonObject;
import io.vertx.ext.mongo.MongoClient;
import io.vertx.ext.mongo.impl.codec.json.JsonObjectCodec;
import org.bson.codecs.*;
import org.bson.codecs.configuration.CodecRegistries;
import org.bson.codecs.configuration.CodecRegistry;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
public class MongoClientOptionsParser {
private final static CodecRegistry commonCodecRegistry = CodecRegistries.fromCodecs(new StringCodec(), new IntegerCodec(),
new BooleanCodec(), new DoubleCodec(), new LongCodec(), new BsonDocumentCodec());
private final MongoClientSettings settings;
private final String database;
public MongoClientOptionsParser(Vertx vertx, JsonObject config) {
Objects.requireNonNull(config);
MongoClientSettings.Builder options = MongoClientSettings.builder();
options.codecRegistry(CodecRegistries.fromRegistries(commonCodecRegistry, CodecRegistries.fromCodecs(new JsonObjectCodec(config))));
String cs = config.getString("connection_string");
ConnectionString connectionString = (cs == null) ? null : new ConnectionString(cs);
String csDatabase = (connectionString != null) ? connectionString.getDatabase() : null;
this.database = csDatabase != null ? csDatabase : config.getString("db_name", MongoClient.DEFAULT_DB_NAME);
ClusterSettings clusterSettings = new ClusterSettingsParser(connectionString, config).settings();
options.applyToClusterSettings(builder -> builder.applySettings(clusterSettings));
ConnectionPoolSettings connectionPoolSettings = new ConnectionPoolSettingsParser(connectionString, config).settings();
options.applyToConnectionPoolSettings(builder -> builder.applySettings(connectionPoolSettings));
List<MongoCredential> credentials = new CredentialListParser(config).credentials();
if (!credentials.isEmpty())
options.credential(credentials.get(credentials.size() - 1));
SocketSettings socketSettings = new SocketSettingsParser(connectionString, config).settings();
options.applyToSocketSettings(builder -> builder.applySettings(socketSettings));
new StreamTypeParser(config).streamFactory().ifPresent(options::streamFactoryFactory);
SslSettings sslSettings = new SSLSettingsParser(connectionString, config).settings(vertx);
options.applyToSslSettings(builder -> builder.applySettings(sslSettings));
WriteConcern writeConcern = new WriteConcernParser(connectionString, config).writeConcern();
if (writeConcern != null) {
options.writeConcern(writeConcern);
}
maybeReadConcern(connectionString, config).ifPresent(options::readConcern);
ReadPreference readPreference = new ReadPreferenceParser(connectionString, config).readPreference();
if (readPreference != null) {
options.readPreference(readPreference);
}
ServerSettings serverSettings = new ServerSettingsParser(config).settings();
options.applyToServerSettings(builder -> builder.applySettings(serverSettings));
this.settings = options.build();
}
public MongoClientSettings settings() {
return settings;
}
public String database() {
return database;
}
private Optional<ReadConcern> maybeReadConcern(ConnectionString connectionString, JsonObject config) {
return new ReadConcernLevelParser(connectionString, config).readConcern();
}
}