org.apache.camel.impl.DefaultInflightRepository Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of camel-core Show documentation
Show all versions of camel-core Show documentation
The Core Camel Java DSL based router
/**
* 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.
*/
package org.apache.camel.impl;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.Date;
import java.util.LinkedList;
import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import org.apache.camel.Endpoint;
import org.apache.camel.Exchange;
import org.apache.camel.MessageHistory;
import org.apache.camel.spi.InflightRepository;
import org.apache.camel.support.ServiceSupport;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Default {@link org.apache.camel.spi.InflightRepository}.
*
* @version
*/
public class DefaultInflightRepository extends ServiceSupport implements InflightRepository {
private static final Logger LOG = LoggerFactory.getLogger(DefaultInflightRepository.class);
private final ConcurrentMap inflight = new ConcurrentHashMap<>();
private final ConcurrentMap routeCount = new ConcurrentHashMap<>();
public void add(Exchange exchange) {
inflight.put(exchange.getExchangeId(), exchange);
}
public void remove(Exchange exchange) {
inflight.remove(exchange.getExchangeId());
}
public void add(Exchange exchange, String routeId) {
AtomicInteger existing = routeCount.get(routeId);
if (existing != null) {
existing.incrementAndGet();
}
}
public void remove(Exchange exchange, String routeId) {
AtomicInteger existing = routeCount.get(routeId);
if (existing != null) {
existing.decrementAndGet();
}
}
public int size() {
return inflight.size();
}
@Deprecated
public int size(Endpoint endpoint) {
return 0;
}
@Override
public void addRoute(String routeId) {
routeCount.putIfAbsent(routeId, new AtomicInteger(0));
}
@Override
public void removeRoute(String routeId) {
routeCount.remove(routeId);
}
@Override
public int size(String routeId) {
AtomicInteger existing = routeCount.get(routeId);
return existing != null ? existing.get() : 0;
}
@Override
public Collection browse() {
return browse(null, -1, false);
}
@Override
public Collection browse(String fromRouteId) {
return browse(fromRouteId, -1, false);
}
@Override
public Collection browse(int limit, boolean sortByLongestDuration) {
return browse(null, limit, sortByLongestDuration);
}
@Override
public Collection browse(String fromRouteId, int limit, boolean sortByLongestDuration) {
Stream values;
if (fromRouteId == null) {
// all values
values = inflight.values().stream();
} else {
// only if route match
values = inflight.values().stream()
.filter(e -> fromRouteId.equals(e.getFromRouteId()));
}
if (sortByLongestDuration) {
// sort by duration and grab the first
values = values.sorted((e1, e2) -> {
long d1 = getExchangeDuration(e1);
long d2 = getExchangeDuration(e2);
// need the biggest number first
return -1 * Long.compare(d1, d2);
});
} else {
// else sort by exchange id
values = values.sorted(Comparator.comparing(Exchange::getExchangeId));
}
if (limit > 0) {
values = values.limit(limit);
}
List answer = values.map(InflightExchangeEntry::new).collect(Collectors.toList());
return Collections.unmodifiableCollection(answer);
}
@Override
public InflightExchange oldest(String fromRouteId) {
Stream values;
if (fromRouteId == null) {
// all values
values = inflight.values().stream();
} else {
// only if route match
values = inflight.values().stream()
.filter(e -> fromRouteId.equals(e.getFromRouteId()));
}
// sort by duration and grab the first
Exchange first = values.sorted((e1, e2) -> {
long d1 = getExchangeDuration(e1);
long d2 = getExchangeDuration(e2);
// need the biggest number first
return -1 * Long.compare(d1, d2);
}).findFirst().orElse(null);
if (first != null) {
return new InflightExchangeEntry(first);
} else {
return null;
}
}
@Override
protected void doStart() throws Exception {
}
@Override
protected void doStop() throws Exception {
int count = size();
if (count > 0) {
LOG.warn("Shutting down while there are still {} inflight exchanges.", count);
} else {
LOG.debug("Shutting down with no inflight exchanges.");
}
routeCount.clear();
}
private static long getExchangeDuration(Exchange exchange) {
long duration = 0;
Date created = exchange.getCreated();
if (created != null) {
duration = System.currentTimeMillis() - created.getTime();
}
return duration;
}
private static final class InflightExchangeEntry implements InflightExchange {
private final Exchange exchange;
private InflightExchangeEntry(Exchange exchange) {
this.exchange = exchange;
}
@Override
public Exchange getExchange() {
return exchange;
}
@Override
public long getDuration() {
return DefaultInflightRepository.getExchangeDuration(exchange);
}
@Override
@SuppressWarnings("unchecked")
public long getElapsed() {
LinkedList list = exchange.getProperty(Exchange.MESSAGE_HISTORY, LinkedList.class);
if (list == null || list.isEmpty()) {
return 0;
}
// get latest entry
MessageHistory history = list.getLast();
if (history != null) {
long elapsed = history.getElapsed();
if (elapsed == 0 && history.getTime() > 0) {
// still in progress, so lets compute it via the start time
elapsed = System.currentTimeMillis() - history.getTime();
}
return elapsed;
} else {
return 0;
}
}
@Override
@SuppressWarnings("unchecked")
public String getNodeId() {
LinkedList list = exchange.getProperty(Exchange.MESSAGE_HISTORY, LinkedList.class);
if (list == null || list.isEmpty()) {
return null;
}
// get latest entry
MessageHistory history = list.getLast();
if (history != null) {
return history.getNode().getId();
} else {
return null;
}
}
@Override
public String getFromRouteId() {
return exchange.getFromRouteId();
}
@Override
public String getRouteId() {
return getAtRouteId();
}
@Override
@SuppressWarnings("unchecked")
public String getAtRouteId() {
LinkedList list = exchange.getProperty(Exchange.MESSAGE_HISTORY, LinkedList.class);
if (list == null || list.isEmpty()) {
return null;
}
// get latest entry
MessageHistory history = list.getLast();
if (history != null) {
return history.getRouteId();
} else {
return null;
}
}
@Override
public String toString() {
return "InflightExchangeEntry[exchangeId=" + exchange.getExchangeId() + "]";
}
}
}