org.apache.activemq.artemis.core.messagecounter.impl.MessageCounterManagerImpl 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.
*/
package org.apache.activemq.artemis.core.messagecounter.impl;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.Executor;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import org.apache.activemq.artemis.api.config.ActiveMQDefaultConfiguration;
import org.apache.activemq.artemis.core.messagecounter.MessageCounter;
import org.apache.activemq.artemis.core.messagecounter.MessageCounterManager;
import org.apache.activemq.artemis.core.server.ActiveMQScheduledComponent;
/**
* A MessageCounterManager
*/
public class MessageCounterManagerImpl implements MessageCounterManager {
public static final long DEFAULT_SAMPLE_PERIOD = ActiveMQDefaultConfiguration.getDefaultMessageCounterSamplePeriod();
public static final long MIN_SAMPLE_PERIOD = 1000;
public static final int DEFAULT_MAX_DAY_COUNT = ActiveMQDefaultConfiguration.getDefaultMessageCounterMaxDayHistory();
private final Map messageCounters;
private final MessageCountersPinger messageCountersPinger;
private int maxDayCount = MessageCounterManagerImpl.DEFAULT_MAX_DAY_COUNT;
public MessageCounterManagerImpl(final ScheduledExecutorService scheduledThreadPool, Executor executor) {
messageCounters = new HashMap<>();
messageCountersPinger = new MessageCountersPinger(scheduledThreadPool, executor, MessageCounterManagerImpl.DEFAULT_SAMPLE_PERIOD, TimeUnit.MILLISECONDS, false);
}
@Override
public synchronized void start() {
messageCountersPinger.start();
}
@Override
public synchronized void stop() {
messageCountersPinger.stop();
}
@Override
public synchronized void clear() {
messageCounters.clear();
}
@Override
public synchronized void reschedule(final long newPeriod) {
messageCountersPinger.setPeriod(newPeriod);
}
@Override
public long getSamplePeriod() {
return messageCountersPinger.getPeriod();
}
@Override
public int getMaxDayCount() {
return maxDayCount;
}
@Override
public void setMaxDayCount(final int count) {
maxDayCount = count;
}
@Override
public void registerMessageCounter(final String name, final MessageCounter counter) {
synchronized (messageCounters) {
messageCounters.put(name, counter);
}
}
@Override
public MessageCounter unregisterMessageCounter(final String name) {
synchronized (messageCounters) {
return messageCounters.remove(name);
}
}
public MessageCounter getMessageCounter(String counter) {
synchronized (messageCounters) {
return messageCounters.get(counter);
}
}
public Set getMessageCounters() {
synchronized (messageCounters) {
return new HashSet<>(messageCounters.values());
}
}
@Override
public void resetAllCounters() {
synchronized (messageCounters) {
for (MessageCounter counter : messageCounters.values()) {
counter.resetCounter();
}
}
}
@Override
public void resetAllCounterHistories() {
synchronized (messageCounters) {
for (MessageCounter counter : messageCounters.values()) {
counter.resetHistory();
}
}
}
private class MessageCountersPinger extends ActiveMQScheduledComponent {
MessageCountersPinger(ScheduledExecutorService scheduledExecutorService,
Executor executor,
long checkPeriod,
TimeUnit timeUnit,
boolean onDemand) {
super(scheduledExecutorService, executor, checkPeriod, timeUnit, onDemand);
}
@Override
public void run() {
synchronized (messageCounters) {
for (MessageCounter counter : messageCounters.values()) {
counter.onTimer();
}
}
}
}
}