package com.mongodb.client.internal;
import com.mongodb.ReadConcern;
import com.mongodb.ReadPreference;
import com.mongodb.client.ClientSession;
import com.mongodb.client.ListDatabasesIterable;
import com.mongodb.internal.operation.SyncOperations;
import com.mongodb.lang.Nullable;
import com.mongodb.operation.BatchCursor;
import com.mongodb.operation.ReadOperation;
import org.bson.BsonDocument;
import org.bson.codecs.configuration.CodecRegistry;
import org.bson.conversions.Bson;
import java.util.concurrent.TimeUnit;
import static com.mongodb.assertions.Assertions.notNull;
import static java.util.concurrent.TimeUnit.MILLISECONDS;
class ListDatabasesIterableImpl<TResult> extends MongoIterableImpl<TResult> implements ListDatabasesIterable<TResult> {
private final SyncOperations<BsonDocument> operations;
private final Class<TResult> resultClass;
private long maxTimeMS;
private Bson filter;
private Boolean nameOnly;
ListDatabasesIterableImpl(@Nullable final ClientSession clientSession, final Class<TResult> resultClass,
final CodecRegistry codecRegistry, final ReadPreference readPreference,
final OperationExecutor executor) {
this(clientSession, resultClass, codecRegistry, readPreference, executor, true);
}
ListDatabasesIterableImpl(@Nullable final ClientSession clientSession, final Class<TResult> resultClass,
final CodecRegistry codecRegistry, final ReadPreference readPreference,
final OperationExecutor executor, final boolean retryReads) {
super(clientSession, executor, ReadConcern.DEFAULT, readPreference, retryReads);
this.operations = new SyncOperations<BsonDocument>(BsonDocument.class, readPreference, codecRegistry, retryReads);
this.resultClass = notNull("clazz", resultClass);
}
@Override
public ListDatabasesIterableImpl<TResult> maxTime(final long maxTime, final TimeUnit timeUnit) {
notNull("timeUnit", timeUnit);
this.maxTimeMS = MILLISECONDS.convert(maxTime, timeUnit);
return this;
}
@Override
public ListDatabasesIterable<TResult> batchSize(final int batchSize) {
super.batchSize(batchSize);
return this;
}
@Override
public ListDatabasesIterable<TResult> filter(@Nullable final Bson filter) {
this.filter = filter;
return this;
}
@Override
public ListDatabasesIterable<TResult> nameOnly(@Nullable final Boolean nameOnly) {
this.nameOnly = nameOnly;
return this;
}
@Override
public ReadOperation<BatchCursor<TResult>> asReadOperation() {
return operations.listDatabases(resultClass, filter, nameOnly, maxTimeMS);
}
}