org.apache.solr.cluster.events.ClusterEventProducerBase Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of solr-core Show documentation
Show all versions of solr-core Show documentation
Apache Solr (module: core)
/*
* 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.solr.cluster.events;
import java.io.IOException;
import java.lang.invoke.MethodHandles;
import java.util.Collections;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import org.apache.solr.common.util.IOUtils;
import org.apache.solr.core.CoreContainer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/** Base class for implementing {@link ClusterEventProducer}. */
public abstract class ClusterEventProducerBase implements ClusterEventProducer {
private static final Logger log = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
protected final Map> listeners =
new ConcurrentHashMap<>();
protected volatile State state = State.STOPPED;
protected final CoreContainer cc;
protected ClusterEventProducerBase(CoreContainer cc) {
this.cc = cc;
}
@Override
public void registerListener(
ClusterEventListener listener, ClusterEvent.EventType... eventTypes) {
if (eventTypes == null || eventTypes.length == 0) {
eventTypes = ClusterEvent.EventType.values();
}
for (ClusterEvent.EventType type : eventTypes) {
if (!getSupportedEventTypes().contains(type)) {
log.warn("event type {} not supported yet.", type);
continue;
}
// to avoid removing no-longer empty set on race in unregister
synchronized (listeners) {
listeners.computeIfAbsent(type, t -> ConcurrentHashMap.newKeySet()).add(listener);
}
}
}
@Override
public void unregisterListener(
ClusterEventListener listener, ClusterEvent.EventType... eventTypes) {
if (eventTypes == null || eventTypes.length == 0) {
eventTypes = ClusterEvent.EventType.values();
}
synchronized (listeners) {
for (ClusterEvent.EventType type : eventTypes) {
Set perType = listeners.get(type);
if (perType != null) {
perType.remove(listener);
if (perType.isEmpty()) {
listeners.remove(type);
}
}
}
}
}
@Override
public State getState() {
return state;
}
@Override
public void close() throws IOException {
synchronized (listeners) {
listeners
.values()
.forEach(listenerSet -> listenerSet.forEach(listener -> IOUtils.closeQuietly(listener)));
}
}
public abstract Set getSupportedEventTypes();
protected void fireEvent(ClusterEvent event) {
synchronized (listeners) {
listeners
.getOrDefault(event.getType(), Collections.emptySet())
.forEach(
listener -> {
if (log.isDebugEnabled()) {
log.debug("--- firing event {} to {}", event, listener);
}
listener.onEvent(event);
});
}
}
}