/*
 * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.controller.cluster.datastore.actors.client;

import akka.actor.ActorRef;
import com.google.common.base.Preconditions;
import java.util.concurrent.CompletionStage;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import javax.annotation.Nonnull;
import javax.annotation.concurrent.ThreadSafe;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Caching resolver which resolves a cookie to a leader {@link ActorRef}. This class needs to be specialized by the
 * client. It is used by {@link ClientActorBehavior} for request dispatch. Results are cached until they are invalidated
 * by either the client actor (when a message timeout is detected) and by the specific frontend (on explicit
 * invalidation or when updated information becomes available).
 *
 * @author Robert Varga
 */
@ThreadSafe
public abstract class BackendInfoResolver<T extends BackendInfo> {
    private static final Logger LOG = LoggerFactory.getLogger(BackendInfoResolver.class);
    private final ConcurrentMap<Long, CompletionStage<T>> backends = new ConcurrentHashMap<>();

    // This is what the client needs to start processing. For as long as we do not have this, we should not complete
    // this stage until we have this information
    public final CompletionStage<? extends T> getBackendInfo(final long cookie) {
        return backends.computeIfAbsent(cookie, this::resolveBackendInfo);
    }

    /**
     * Invalidate a particular instance of {@link BackendInfo}, typically as a response to a request timing out. If
     * the provided information is not the one currently cached this method does nothing.
     *
     * @param cookie Backend cookie
     * @param info Previous information to be invalidated
     */
    public final void invalidateBackend(final long cookie, final @Nonnull CompletionStage<? extends BackendInfo> info) {
        if (backends.remove(cookie, Preconditions.checkNotNull(info))) {
            LOG.trace("Invalidated cache %s -> %s", Long.toUnsignedString(cookie), info);
            invalidateBackendInfo(info);
        }
    }

    /**
     * Request new resolution of a particular backend identified by a cookie. This method is invoked when a client
     * requests information which is not currently cached.
     *
     * @param cookie Backend cookie
     * @return A {@link CompletionStage} resulting in information about the backend
     */
    protected abstract @Nonnull CompletionStage<T> resolveBackendInfo(final @Nonnull Long cookie);

    /**
     * Invalidate previously-resolved shard information. This method is invoked when a timeout is detected
     * and the information may need to be refreshed.
     *
     * @param info Previous promise of backend information
     */
    protected abstract void invalidateBackendInfo(@Nonnull CompletionStage<? extends BackendInfo> info);
}