import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.atomic.AtomicBoolean;
-import java.util.concurrent.atomic.AtomicReference;
+import java.util.function.Supplier;
import net.pterodactylus.fcp.ConfigData;
import net.pterodactylus.fcp.GetConfig;
private final ListeningExecutorService threadPool;
private final ConnectionSupplier connectionSupplier;
+ private final Supplier<String> identifierGenerator;
private final AtomicBoolean withCurrent = new AtomicBoolean();
private final AtomicBoolean withDefaults = new AtomicBoolean();
private final AtomicBoolean withSortOrder = new AtomicBoolean();
private final AtomicBoolean withLongDescription = new AtomicBoolean();
private final AtomicBoolean withDataTypes = new AtomicBoolean();
- public GetConfigCommandImpl(ExecutorService threadPool, ConnectionSupplier connectionSupplier) {
+ public GetConfigCommandImpl(ExecutorService threadPool, ConnectionSupplier connectionSupplier, Supplier<String> identifierGenerator) {
this.threadPool = MoreExecutors.listeningDecorator(threadPool);
this.connectionSupplier = connectionSupplier;
+ this.identifierGenerator = identifierGenerator;
}
@Override
}
private ConfigData executeDialog() throws IOException, ExecutionException, InterruptedException {
- GetConfig getConfig = new GetConfig(new RandomIdentifierGenerator().generate());
+ GetConfig getConfig = new GetConfig(identifierGenerator.get());
getConfig.setWithCurrent(withCurrent.get());
getConfig.setWithDefaults(withDefaults.get());
getConfig.setWithSortOrder(withSortOrder.get());
private class GetConfigDialog extends FcpDialog<ConfigData> {
- private final AtomicReference<ConfigData> configData = new AtomicReference<>();
-
public GetConfigDialog() throws IOException {
- super(threadPool, connectionSupplier.get());
- }
-
- @Override
- protected boolean isFinished() {
- return configData.get() != null;
- }
-
- @Override
- protected ConfigData getResult() {
- return configData.get();
+ super(threadPool, connectionSupplier.get(), null);
}
@Override
protected void consumeConfigData(ConfigData configData) {
- this.configData.set(configData);
+ setResult(configData);
}
}