io.windmill.net.TransferTask Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of windmill-core Show documentation

There is a newer version: 0.2

package io.windmill.net;

import java.nio.channels.ClosedChannelException;
import java.util.Optional;

import io.windmill.core.Future;

import io.netty.buffer.ByteBuf;

public abstract class TransferTask implements AutoCloseable
{
    protected final ByteBuf buffer;
    protected final Optional> onComplete;

    public TransferTask(ByteBuf buffer, Optional> onComplete)
    {
        this.buffer = buffer;
        this.onComplete = onComplete;
    }

    public abstract boolean compute(I channel);

    public Future getFuture()
    {
        return onComplete.isPresent() ? onComplete.get() : null;
    }

    @Override
    public void close()
    {
        if (buffer != null)
            buffer.release();
        onComplete.ifPresent((f) -> f.setFailure(new ClosedChannelException()));
    }
}