com.arangodb.shaded.vertx.core.http.impl.HttpServerFileUploadImpl Maven / Gradle / Ivy
/*
* Copyright (c) 2011-2019 Contributors to the Eclipse Foundation
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License 2.0 which is available at
* http://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
* which is available at https://www.apache.org/licenses/LICENSE-2.0.
*
* SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
*/
package com.arangodb.shaded.vertx.core.http.impl;
import com.arangodb.shaded.vertx.core.AsyncResult;
import com.arangodb.shaded.vertx.core.Future;
import com.arangodb.shaded.vertx.core.Handler;
import com.arangodb.shaded.vertx.core.buffer.Buffer;
import com.arangodb.shaded.vertx.core.file.AsyncFile;
import com.arangodb.shaded.vertx.core.file.FileSystem;
import com.arangodb.shaded.vertx.core.file.OpenOptions;
import com.arangodb.shaded.vertx.core.http.HttpServerFileUpload;
import com.arangodb.shaded.vertx.core.impl.ContextInternal;
import com.arangodb.shaded.vertx.core.streams.Pipe;
import com.arangodb.shaded.vertx.core.streams.ReadStream;
import java.nio.charset.Charset;
/**
* This class is optimised for performance when used on the same event loop that is was passed to the handler with.
* However it can be used safely from other threads.
*
* The internal state is protected using the synchronized keyword. If always used on the same event loop, then
* we benefit from biased locking which makes the overhead of synchronized near zero.
*
* @author Norman Maurer
*/
class HttpServerFileUploadImpl implements HttpServerFileUpload {
private final ReadStream stream;
private final ContextInternal context;
private final String name;
private final String filename;
private final String contentType;
private final String contentTransferEncoding;
private final Charset charset;
private Handler dataHandler;
private Handler endHandler;
private Handler exceptionHandler;
private long size;
private boolean lazyCalculateSize;
private AsyncFile file;
private Pipe pipe;
private boolean cancelled;
HttpServerFileUploadImpl(ContextInternal context, ReadStream stream, String name, String filename, String contentType,
String contentTransferEncoding,
Charset charset, long size) {
this.context = context;
this.stream = stream;
this.name = name;
this.filename = filename;
this.contentType = contentType;
this.contentTransferEncoding = contentTransferEncoding;
this.charset = charset;
this.size = size;
this.lazyCalculateSize = size == 0;
stream.handler(this::handleData);
stream.exceptionHandler(this::handleException);
stream.endHandler(v -> this.handleEnd());
}
private void handleData(Buffer data) {
Handler handler;
synchronized (HttpServerFileUploadImpl.this) {
handler = dataHandler;
if (lazyCalculateSize) {
size += data.length();
}
}
if (handler != null) {
context.dispatch(data, handler);
}
}
private void handleException(Throwable cause) {
Handler handler;
synchronized (this) {
handler = exceptionHandler;
}
if (handler != null) {
context.dispatch(cause, handler);
}
}
private void handleEnd() {
Handler handler;
synchronized (this) {
lazyCalculateSize = false;
handler = endHandler;
}
if (handler != null) {
context.dispatch(handler);
}
}
@Override
public String filename() {
return filename;
}
@Override
public String name() {
return name;
}
@Override
public String contentType() {
return contentType;
}
@Override
public String contentTransferEncoding() {
return contentTransferEncoding;
}
@Override
public String charset() {
return charset.toString();
}
@Override
public synchronized long size() {
return size;
}
@Override
public synchronized HttpServerFileUpload handler(Handler handler) {
dataHandler = handler;
return this;
}
@Override
public HttpServerFileUpload pause() {
stream.pause();
return this;
}
@Override
public HttpServerFileUpload fetch(long amount) {
stream.fetch(amount);
return this;
}
@Override
public HttpServerFileUpload resume() {
stream.resume();
return this;
}
@Override
public synchronized HttpServerFileUpload exceptionHandler(Handler exceptionHandler) {
this.exceptionHandler = exceptionHandler;
return this;
}
@Override
public synchronized HttpServerFileUpload endHandler(Handler handler) {
this.endHandler = handler;
return this;
}
@Override
public void streamToFileSystem(String filename, Handler> handler) {
Future fut = streamToFileSystem(filename);
if (handler != null) {
fut.onComplete(handler);
}
}
@Override
public Future streamToFileSystem(String filename) {
synchronized (this) {
if (pipe != null) {
return context.failedFuture("Already streaming");
}
pipe = pipe().endOnComplete(true);
}
FileSystem fs = context.owner().fileSystem();
Future fut = fs.open(filename, new OpenOptions());
fut.onFailure(err -> {
pipe.close();
});
return fut.compose(f -> {
Future to = pipe.to(f);
return to.compose(v -> {
synchronized (HttpServerFileUploadImpl.this) {
if (!cancelled) {
file = f;
return context.succeededFuture();
}
fs.delete(filename);
return context.failedFuture("Streaming aborted");
}
}, err -> {
fs.delete(filename);
return context.failedFuture(err);
});
});
}
@Override
public boolean cancelStreamToFileSystem() {
synchronized (this) {
if (pipe == null) {
throw new IllegalStateException("Not a streaming upload");
}
if (file != null) {
return false;
}
cancelled = true;
}
pipe.close();
return true;
}
@Override
public synchronized boolean isSizeAvailable() {
return !lazyCalculateSize;
}
@Override
public synchronized AsyncFile file() {
return file;
}
}