com.digitalpetri.opcua.stack.client.fsm.states.ReconnectDelay Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of stack-client Show documentation
There is a newer version: 1.1.1
/*
 * Copyright 2015 Kevin Herron
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.digitalpetri.opcua.stack.client.fsm.states;

import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;

import com.digitalpetri.opcua.stack.client.fsm.ConnectionEvent;
import com.digitalpetri.opcua.stack.client.fsm.ConnectionState;
import com.digitalpetri.opcua.stack.client.fsm.ConnectionStateFsm;
import com.digitalpetri.opcua.stack.core.Stack;
import com.digitalpetri.opcua.stack.core.channel.ClientSecureChannel;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ReconnectDelay implements ConnectionState {

    private final Logger logger = LoggerFactory.getLogger(getClass());

    private final CompletableFuture channelFuture = new CompletableFuture<>();

    private volatile ScheduledFuture scheduledFuture;

    private final long delaySeconds;
    private final ClientSecureChannel existingChannel;

    public ReconnectDelay(long delaySeconds, ClientSecureChannel existingChannel) {
        this.delaySeconds = delaySeconds;
        this.existingChannel = existingChannel;
    }

    @Override
    public CompletableFuture activate(ConnectionEvent event, ConnectionStateFsm fsm) {
        if (scheduledFuture == null || (scheduledFuture != null && scheduledFuture.cancel(false))) {
            logger.debug("Scheduling reconnect in {} seconds...", delaySeconds);

            scheduledFuture = Stack.sharedScheduledExecutor().schedule(() -> {
                logger.debug("{} seconds elapsed; requesting reconnect.", delaySeconds);

                fsm.handleEvent(ConnectionEvent.ReconnectRequested);
            }, delaySeconds, TimeUnit.SECONDS);
        }

        return CF_VOID_COMPLETED;
    }

    @Override
    public CompletableFuture deactivate(ConnectionEvent event, ConnectionStateFsm fsm) {
        if (scheduledFuture != null) {
            scheduledFuture.cancel(false);
            scheduledFuture = null;
        }

        return CF_VOID_COMPLETED;
    }

    @Override
    public ConnectionState transition(ConnectionEvent event, ConnectionStateFsm fsm) {
        switch (event) {
            case DisconnectRequested:
                return new Disconnecting(null);

            case ReconnectRequested:
                return new ReconnectExecute(channelFuture, existingChannel, delaySeconds);
        }

        return this;
    }

    @Override
    public CompletableFuture getSecureChannel() {
        return channelFuture;
    }

    @Override
    public String toString() {
        return "ReconnectDelay{" +
                "delaySeconds=" + delaySeconds +
                ", secureChannelId=" + existingChannel.getChannelId() +
                '}';
    }

}