--- /dev/null
+package net.pterodactylus.fcp.quelaton;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.Optional;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Future;
+import java.util.concurrent.atomic.AtomicBoolean;
+
+import net.pterodactylus.fcp.AllData;
+import net.pterodactylus.fcp.ClientGet;
+import net.pterodactylus.fcp.FcpUtils.TempInputStream;
+import net.pterodactylus.fcp.GetFailed;
+import net.pterodactylus.fcp.Priority;
+import net.pterodactylus.fcp.ReturnType;
+
+/**
+ * Implementation of the {@link ClientGetCommand}.
+ *
+ * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+ */
+class ClientGetCommandImpl implements ClientGetCommand {
+
+ private final ExecutorService threadPool;
+ private final ConnectionSupplier connectionSupplier;
+
+ private String identifier;
+ private boolean ignoreDataStore;
+ private boolean dataStoreOnly;
+ private Long maxSize;
+ private Priority priority;
+ private boolean realTime;
+ private boolean global;
+
+ public ClientGetCommandImpl(ExecutorService threadPool, ConnectionSupplier connectionSupplier) {
+ this.threadPool = threadPool;
+ this.connectionSupplier = connectionSupplier;
+ }
+
+ @Override
+ public ClientGetCommand identifier(String identifier) {
+ this.identifier = identifier;
+ return this;
+ }
+
+ @Override
+ public ClientGetCommand ignoreDataStore() {
+ ignoreDataStore = true;
+ return this;
+ }
+
+ @Override
+ public ClientGetCommand dataStoreOnly() {
+ dataStoreOnly = true;
+ return this;
+ }
+
+ @Override
+ public ClientGetCommand maxSize(long maxSize) {
+ this.maxSize = maxSize;
+ return this;
+ }
+
+ @Override
+ public ClientGetCommand priority(Priority priority) {
+ this.priority = priority;
+ return this;
+ }
+
+ @Override
+ public ClientGetCommand realTime() {
+ realTime = true;
+ return this;
+ }
+
+ @Override
+ public ClientGetCommand global() {
+ global = true;
+ return this;
+ }
+
+ @Override
+ public Future<Optional<Data>> uri(String uri) {
+ ClientGet clientGet = new ClientGet(uri, identifier, ReturnType.direct);
+ if (ignoreDataStore) {
+ clientGet.setIgnoreDataStore(true);
+ }
+ if (dataStoreOnly) {
+ clientGet.setDataStoreOnly(true);
+ }
+ if (maxSize != null) {
+ clientGet.setMaxSize(maxSize);
+ }
+ if (priority != null) {
+ clientGet.setPriority(priority);
+ }
+ if (realTime) {
+ clientGet.setRealTimeFlag(true);
+ }
+ if (global) {
+ clientGet.setGlobal(true);
+ }
+ return threadPool.submit(() -> {
+ FcpReplySequence<Optional<Data>> replySequence =
+ new ClientGetReplySequence();
+ return replySequence.send(clientGet).get();
+ });
+ }
+
+ private class ClientGetReplySequence extends FcpReplySequence<Optional<Data>> {
+
+ private final AtomicBoolean finished = new AtomicBoolean();
+ private final AtomicBoolean failed = new AtomicBoolean();
+
+ private final String identifier = ClientGetCommandImpl.this.identifier;
+
+ private String contentType;
+ private long dataLength;
+ private InputStream payload;
+
+ public ClientGetReplySequence() throws IOException {
+ super(ClientGetCommandImpl.this.threadPool, ClientGetCommandImpl.this.connectionSupplier.get());
+ }
+
+ @Override
+ protected boolean isFinished() {
+ return finished.get() || failed.get();
+ }
+
+ @Override
+ protected Optional<Data> getResult() {
+ return failed.get() ? Optional.empty() : Optional.of(new Data() {
+ @Override
+ public String getMimeType() {
+ return contentType;
+ }
+
+ @Override
+ public long size() {
+ return dataLength;
+ }
+
+ @Override
+ public InputStream getInputStream() {
+ return payload;
+ }
+ });
+ }
+
+ @Override
+ protected void consumeAllData(AllData allData) {
+ if (allData.getIdentifier().equals(identifier)) {
+ synchronized (this) {
+ contentType = allData.getContentType();
+ dataLength = allData.getDataLength();
+ try {
+ payload = new TempInputStream(allData.getPayloadInputStream(), dataLength);
+ finished.set(true);
+ } catch (IOException e) {
+ // TODO – logging
+ failed.set(true);
+ }
+ }
+ }
+ }
+
+ @Override
+ protected void consumeGetFailed(GetFailed getFailed) {
+ if (getFailed.getIdentifier().equals(identifier)) {
+ failed.set(true);
+ }
+ }
+
+ @Override
+ protected void consumeConnectionClosed(Throwable throwable) {
+ failed.set(true);
+ }
+
+ }
+
+}
package net.pterodactylus.fcp.quelaton;
import java.io.IOException;
-import java.io.InputStream;
-import java.util.Optional;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
-import java.util.concurrent.Future;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicReference;
import java.util.function.Supplier;
-import net.pterodactylus.fcp.AllData;
-import net.pterodactylus.fcp.ClientGet;
import net.pterodactylus.fcp.ClientHello;
import net.pterodactylus.fcp.CloseConnectionDuplicateClientName;
import net.pterodactylus.fcp.FcpConnection;
-import net.pterodactylus.fcp.FcpUtils.TempInputStream;
-import net.pterodactylus.fcp.GetFailed;
import net.pterodactylus.fcp.NodeHello;
-import net.pterodactylus.fcp.Priority;
-import net.pterodactylus.fcp.ReturnType;
/**
* Default {@link FcpClient} implementation.
@Override
public ClientGetCommand clientGet() {
- return new ClientGetCommandImpl();
- }
-
- private class ClientGetCommandImpl implements ClientGetCommand {
-
- private String identifier;
- private boolean ignoreDataStore;
- private boolean dataStoreOnly;
- private Long maxSize;
- private Priority priority;
- private boolean realTime;
- private boolean global;
-
- @Override
- public ClientGetCommand identifier(String identifier) {
- this.identifier = identifier;
- return this;
- }
-
- @Override
- public ClientGetCommand ignoreDataStore() {
- ignoreDataStore = true;
- return this;
- }
-
- @Override
- public ClientGetCommand dataStoreOnly() {
- dataStoreOnly = true;
- return this;
- }
-
- @Override
- public ClientGetCommand maxSize(long maxSize) {
- this.maxSize = maxSize;
- return this;
- }
-
- @Override
- public ClientGetCommand priority(Priority priority) {
- this.priority = priority;
- return this;
- }
-
- @Override
- public ClientGetCommand realTime() {
- realTime = true;
- return this;
- }
-
- @Override
- public ClientGetCommand global() {
- global = true;
- return this;
- }
-
- @Override
- public Future<Optional<Data>> uri(String uri) {
- ClientGet clientGet = new ClientGet(uri, identifier, ReturnType.direct);
- if (ignoreDataStore) {
- clientGet.setIgnoreDataStore(true);
- }
- if (dataStoreOnly) {
- clientGet.setDataStoreOnly(true);
- }
- if (maxSize != null) {
- clientGet.setMaxSize(maxSize);
- }
- if (priority != null) {
- clientGet.setPriority(priority);
- }
- if (realTime) {
- clientGet.setRealTimeFlag(true);
- }
- if (global) {
- clientGet.setGlobal(true);
- }
- return threadPool.submit(() -> {
- FcpReplySequence<Optional<Data>> replySequence = new FcpReplySequence<Optional<Data>>(threadPool, connect()) {
- private final AtomicBoolean finished = new AtomicBoolean();
- private final AtomicBoolean failed = new AtomicBoolean();
-
- private final String identifier = ClientGetCommandImpl.this.identifier;
-
- private String contentType;
- private long dataLength;
- private InputStream payload;
-
- @Override
- protected boolean isFinished() {
- return finished.get() || failed.get();
- }
-
- @Override
- protected Optional<Data> getResult() {
- return failed.get() ? Optional.empty() : Optional.of(new Data() {
- @Override
- public String getMimeType() {
- return contentType;
- }
-
- @Override
- public long size() {
- return dataLength;
- }
-
- @Override
- public InputStream getInputStream() {
- return payload;
- }
- });
- }
-
- @Override
- protected void consumeAllData(AllData allData) {
- if (allData.getIdentifier().equals(identifier)) {
- synchronized (this) {
- contentType = allData.getContentType();
- dataLength = allData.getDataLength();
- try {
- payload = new TempInputStream(allData.getPayloadInputStream(), dataLength);
- finished.set(true);
- } catch (IOException e) {
- // TODO – logging
- failed.set(true);
- }
- }
- }
- }
-
- @Override
- protected void consumeGetFailed(GetFailed getFailed) {
- if (getFailed.getIdentifier().equals(identifier)) {
- failed.set(true);
- }
- }
-
- @Override
- protected void consumeConnectionClosed(Throwable throwable) {
- failed.set(true);
- }
- };
- return replySequence.send(clientGet).get();
- });
- }
-
+ return new ClientGetCommandImpl(threadPool, this::connect);
}
}
projects / jFCPlib.git / commitdiff