com.aliyun.apache.hc.client5.http.impl.classic.AbstractBackoff Maven / Gradle / Ivy
/*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation. For more
* information on the Apache Software Foundation, please see
*
* This class provides common functionality for maintaining the route-wise backoff and probe
* timestamps, as well as the cool-down period for each backoff attempt.
*
* It also contains the basic structure of the backOff and probe methods, which use the route-wise
* timestamps to determine if the backoff or probe should be applied, and then call the specific
* algorithm implementation for calculating the new pool size.
*
* @since 5.3
*/
public abstract class AbstractBackoff implements BackoffManager {
private static final Logger LOG = LoggerFactory.getLogger(AbstractBackoff.class);
/**
* Connection pool control responsible for managing the maximum number of connections per HTTP route.
*/
private final ConnPoolControl connPerRoute;
/**
* A map that stores the last probe timestamp for each HTTP route.
*/
private final Map lastRouteProbes;
/**
* A map that stores the last backoff timestamp for each HTTP route.
*/
private final Map lastRouteBackoffs;
/**
* The cool-down period after which the backoff or probe process can be performed again.
*/
private final AtomicReference coolDown = new AtomicReference<>(TimeValue.ofSeconds(5L));
/**
* The growth rate used in the exponential backoff algorithm.
*/
private final AtomicReference backoffFactor = new AtomicReference<>(0.5);
/**
* The per-host connection cap, as defined in RFC 2616 sec 8.1.4.
*/
private final AtomicInteger cap = new AtomicInteger(2);
/**
* The number of time intervals used in the exponential backoff algorithm.
*/
private final AtomicInteger timeInterval = new AtomicInteger(0);
/**
* Constructs a new ExponentialBackoffManager with the specified connection pool control.
*
* @param connPerRoute the connection pool control to be used for managing connections
* @throws IllegalArgumentException if connPerRoute is null
*/
public AbstractBackoff(final ConnPoolControl connPerRoute) {
this.connPerRoute = Args.notNull(connPerRoute, "Connection pool control");
this.lastRouteProbes = new ConcurrentHashMap<>();
this.lastRouteBackoffs = new ConcurrentHashMap<>();
}
/**
* Reduces the number of maximum allowed connections for the specified route based on the exponential backoff algorithm.
*
* @param route the HttpRoute for which the backoff needs to be applied
*/
@Override
public void backOff(final HttpRoute route) {
final int curr = connPerRoute.getMaxPerRoute(route);
final Instant now = Instant.now();
lastRouteBackoffs.compute(route, (r, lastUpdate) -> {
if (lastUpdate == null || now.isAfter(lastUpdate.plus(coolDown.get().toMilliseconds(), ChronoUnit.MILLIS))) {
final int backedOffPoolSize = getBackedOffPoolSize(curr); // Exponential backoff
connPerRoute.setMaxPerRoute(route, backedOffPoolSize);
if (LOG.isDebugEnabled()) {
LOG.debug("Backoff applied for route: {}, new max connections: {}", route, connPerRoute.getMaxPerRoute(route));
}
return now;
}
return lastUpdate;
});
}
/**
* Calculates the new pool size after applying the exponential backoff algorithm.
* The new pool size is calculated using the formula: floor(curr / (1 + growthRate) ^ t),
* where curr is the current pool size, growthRate is the exponential growth rate, and t is the time interval.
*
* @param curr the current pool size
* @return the new pool size after applying the backoff
*/
protected abstract int getBackedOffPoolSize(int curr);
/**
* Increases the number of maximum allowed connections for the specified route after a successful connection has been established.
*
* @param route the HttpRoute for which the probe needs to be applied
*/
@Override
public void probe(final HttpRoute route) {
final int curr = connPerRoute.getMaxPerRoute(route);
final int max = (curr >= cap.get()) ? cap.get() : curr + 1;
final Instant now = Instant.now();
lastRouteProbes.compute(route, (r, lastProbe) -> {
if (lastProbe == null || now.isAfter(lastProbe.plus(coolDown.get().toMilliseconds(), ChronoUnit.MILLIS))) {
final Instant lastBackoff = lastRouteBackoffs.get(r);
if (lastBackoff == null || now.isAfter(lastBackoff.plus(coolDown.get().toMilliseconds(), ChronoUnit.MILLIS))) {
connPerRoute.setMaxPerRoute(route, max);
if (LOG.isDebugEnabled()) {
LOG.info("Probe applied for route: {}, new max connections: {}", route, connPerRoute.getMaxPerRoute(route));
}
timeInterval.set(0); // Reset the time interval
return now;
}
}
return lastProbe;
});
}
/**
* Retrieves the last update timestamp for the specified route from the provided updates map.
*
* @param updates the map containing update timestamps for HttpRoutes
* @param route the HttpRoute for which the last update timestamp is needed
* @return the last update timestamp for the specified route or 0L if not present in the map
*/
public long getLastUpdate(final Map updates, final HttpRoute route) {
return updates.getOrDefault(route, 0L);
}
/**
* Sets the per-host connection cap.
*
* @param cap the per-host connection cap to be set
* @throws IllegalArgumentException if the cap is not positive
*/
public void setPerHostConnectionCap(final int cap) {
Args.positive(cap, "Per host connection cap");
this.cap.set(cap);
}
/**
* Sets the backoff factor for the backoff algorithm.
* The backoff factor should be a value between 0.0 and 1.0.
* The specific implementation of how the backoff factor is used should be provided by subclasses.
*
* @param d the backoff factor to be set
*/
abstract void setBackoffFactor(final double d);
/**
* Sets the cool-down time value for adjustments in pool sizes for a given host. This time value
* allows enough time for the adjustments to take effect before further adjustments are made.
* The cool-down time value must be positive and not null.
*
* @param coolDown the TimeValue representing the cool-down period between adjustments
* @throws IllegalArgumentException if the provided cool-down time value is null or non-positive
*/
public void setCoolDown(final TimeValue coolDown) {
Args.notNull(coolDown, "Cool down time value cannot be null");
Args.positive(coolDown.getDuration(), "coolDown");
this.coolDown.set(coolDown);
}
/**
* Returns the connection pool control for managing the maximum number of connections per route.
*
* @return the connection pool control instance
*/
protected ConnPoolControl getConnPerRoute() {
return connPerRoute;
}
/**
* Returns the map containing the last probe times for each HttpRoute.
*
* @return the map of HttpRoute to Instant representing the last probe times
*/
protected Map getLastRouteProbes() {
return lastRouteProbes;
}
/**
* Returns the map containing the last backoff times for each HttpRoute.
*
* @return the map of HttpRoute to Instant representing the last backoff times
*/
protected Map getLastRouteBackoffs() {
return lastRouteBackoffs;
}
/**
* Returns the cool down period between backoff and probe operations as an AtomicReference of TimeValue.
*
* @return the AtomicReference containing the cool down period
*/
protected AtomicReference getCoolDown() {
return coolDown;
}
/**
* Returns the backoff factor as an AtomicReference of Double.
*
* @return the AtomicReference containing the backoff factor
*/
protected AtomicReference getBackoffFactor() {
return backoffFactor;
}
/**
* Returns the cap on the maximum number of connections per route as an AtomicInteger.
*
* @return the AtomicInteger containing the cap value
*/
protected AtomicInteger getCap() {
return cap;
}
/**
* Returns the time interval between backoff and probe operations as an AtomicInteger.
*
* @return the AtomicInteger containing the time interval
*/
protected AtomicInteger getTimeInterval() {
return timeInterval;
}
}