opendaylight/md-sal/sal-distributed-datastore/src/main/java/org/opendaylight/controller/cluster/databroker/actors/dds/BouncingReconnectForwarder.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.controller.cluster.databroker.actors.dds;
   9
  10 import com.google.common.base.Preconditions;
  11 import com.google.common.collect.Collections2;
  12 import com.google.common.collect.Maps;
  13 import java.util.Collection;
  14 import java.util.Map;
  15 import org.opendaylight.controller.cluster.access.client.ConnectedClientConnection;
  16 import org.opendaylight.controller.cluster.access.client.ConnectionEntry;
  17 import org.opendaylight.controller.cluster.access.client.ReconnectForwarder;
  18 import org.opendaylight.controller.cluster.access.commands.LocalHistoryRequest;
  19 import org.opendaylight.controller.cluster.access.commands.TransactionRequest;
  20 import org.opendaylight.controller.cluster.access.concepts.LocalHistoryIdentifier;
  21 import org.opendaylight.controller.cluster.access.concepts.Request;
  22 import org.opendaylight.controller.cluster.access.concepts.RequestException;
  23 import org.slf4j.Logger;
  24 import org.slf4j.LoggerFactory;
  25
  26 // Cohort aware forwarder, which forwards the request to the cohort, giving it a reference to the successor
  27 // connection
  28 final class BouncingReconnectForwarder extends ReconnectForwarder {
  29     private static final class CohortNotFoundException extends RequestException {
  30         private static final long serialVersionUID = 1L;
  31
  32         CohortNotFoundException(final LocalHistoryIdentifier historyId) {
  33             super("Cohort for " + historyId + " not found");
  34         }
  35
  36         @Override
  37         public boolean isRetriable() {
  38             return false;
  39         }
  40     }
  41
  42     private static final Logger LOG = LoggerFactory.getLogger(BouncingReconnectForwarder.class);
  43
  44     private final Map<LocalHistoryIdentifier, ProxyReconnectCohort> cohorts;
  45
  46     private BouncingReconnectForwarder(final ConnectedClientConnection<?> successor,
  47             final Map<LocalHistoryIdentifier, ProxyReconnectCohort> cohorts) {
  48         super(successor);
  49         this.cohorts = Preconditions.checkNotNull(cohorts);
  50     }
  51
  52     static ReconnectForwarder forCohorts(final ConnectedClientConnection<?> successor,
  53             final Collection<HistoryReconnectCohort> cohorts) {
  54         return new BouncingReconnectForwarder(successor, Maps.uniqueIndex(Collections2.transform(cohorts,
  55             HistoryReconnectCohort::getProxy), ProxyReconnectCohort::getIdentifier));
  56     }
  57
  58
  59     @Override
  60     protected void forwardEntry(final ConnectionEntry entry, final long now) {
  61         final Request<? , ?> request = entry.getRequest();
  62
  63         final LocalHistoryIdentifier historyId;
  64         if (request instanceof TransactionRequest) {
  65             historyId = ((TransactionRequest<?>) request).getTarget().getHistoryId();
  66         } else if (request instanceof LocalHistoryRequest) {
  67             historyId = ((LocalHistoryRequest<?>) request).getTarget();
  68         } else {
  69             throw new IllegalArgumentException("Unhandled request " + request);
  70         }
  71
  72         try {
  73             final ProxyReconnectCohort cohort = cohorts.get(historyId);
  74             if (cohort == null) {
  75                 LOG.warn("Cohort for request {} not found, aborting it", request);
  76                 throw new CohortNotFoundException(historyId);
  77             }
  78
  79             // FIXME: do not use sendRequest() once we have throttling in place, as we have already waited the
  80             //        period required to get into the queue.
  81             cohort.forwardRequest(request, entry.getCallback(), this::sendToSuccessor);
  82         } catch (RequestException e) {
  83             entry.complete(request.toRequestFailure(e));
  84         }
  85     }
  86 }