com.flash3388.flashlib.communication.runner.CommunicationRunner Maven / Gradle / Ivy

Go to download
package com.flash3388.flashlib.communication.runner;

import com.flash3388.flashlib.communication.connection.Connection;
import com.flash3388.flashlib.communication.message.Messenger;
import com.flash3388.flashlib.io.Closer;
import com.flash3388.flashlib.io.serialization.Serializer;
import com.flash3388.flashlib.util.concurrent.ExecutorCloser;
import com.flash3388.flashlib.communication.message.Message;
import com.flash3388.flashlib.util.flow.SingleUseParameterizedRunner;
import org.slf4j.Logger;

import java.io.IOException;
import java.util.Optional;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.atomic.AtomicReference;
import java.util.function.Consumer;
import java.util.function.Supplier;

public class CommunicationRunner extends SingleUseParameterizedRunner {

    private final ExecutorService mExecutorService;
    private final Serializer mSerializer;
    private final Logger mLogger;

    private final Consumer mNewMessagesConsumer;
    private final Supplier> mMessagesToSendSupplier;

    private final AtomicReference mConnectionReference;

    public CommunicationRunner(ExecutorService executorService, Serializer serializer, Logger logger, Consumer newMessageConsumer, Supplier> messagesToSendSupplier) {
        mExecutorService = executorService;
        mSerializer = serializer;
        mLogger = logger;

        mNewMessagesConsumer = newMessageConsumer;
        mMessagesToSendSupplier = messagesToSendSupplier;

        mConnectionReference = new AtomicReference<>();
    }

    @Override
    protected void startRunner(Connection connection) {
        mConnectionReference.set(connection);

        Messenger messenger = new Messenger(connection, mSerializer);

        mExecutorService.execute(new MessageReadTask(messenger, mNewMessagesConsumer, mLogger));
        mExecutorService.execute(new MessageWriteTask(messenger, mMessagesToSendSupplier, mLogger));
    }

    @Override
    protected void stopRunner() {
        try {
            Closer closer = Closer.empty();

            Connection connection = mConnectionReference.getAndSet(null);
            closer.add(connection);

            closer.add(new ExecutorCloser(mExecutorService));

            closer.close();
        } catch (IOException e) {
            mLogger.warn("error while closing resources", e);
        }
    }
}