Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.hazelcast.client.impl.querycache.subscriber.ClientQueryCacheEventService Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2020, Hazelcast, Inc. All Rights Reserved.
*
* Licensed 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 com.hazelcast.client.impl.querycache.subscriber;
import com.hazelcast.client.impl.clientside.HazelcastClientInstanceImpl;
import com.hazelcast.client.impl.protocol.ClientMessage;
import com.hazelcast.client.impl.protocol.codec.ContinuousQueryAddListenerCodec;
import com.hazelcast.client.impl.protocol.codec.MapRemoveEntryListenerCodec;
import com.hazelcast.client.spi.ClientListenerService;
import com.hazelcast.client.spi.EventHandler;
import com.hazelcast.client.spi.impl.ListenerMessageCodec;
import com.hazelcast.client.spi.impl.listener.AbstractClientListenerService;
import com.hazelcast.core.IMapEvent;
import com.hazelcast.internal.serialization.InternalSerializationService;
import com.hazelcast.logging.ILogger;
import com.hazelcast.logging.Logger;
import com.hazelcast.map.EventLostEvent;
import com.hazelcast.map.impl.ListenerAdapter;
import com.hazelcast.map.impl.event.EventData;
import com.hazelcast.map.impl.querycache.QueryCacheEventService;
import com.hazelcast.map.impl.querycache.event.BatchEventData;
import com.hazelcast.map.impl.querycache.event.BatchIMapEvent;
import com.hazelcast.map.impl.querycache.event.LocalEntryEventData;
import com.hazelcast.map.impl.querycache.event.QueryCacheEventData;
import com.hazelcast.map.impl.querycache.event.SingleIMapEvent;
import com.hazelcast.map.listener.MapListener;
import com.hazelcast.nio.serialization.Data;
import com.hazelcast.query.impl.QueryEntry;
import com.hazelcast.query.impl.getters.Extractors;
import com.hazelcast.spi.EventFilter;
import com.hazelcast.spi.impl.eventservice.impl.TrueEventFilter;
import com.hazelcast.spi.serialization.SerializationService;
import com.hazelcast.util.ConstructorFunction;
import com.hazelcast.util.executor.StripedExecutor;
import com.hazelcast.util.executor.StripedRunnable;
import com.hazelcast.util.executor.TimeoutRunnable;
import java.util.Collection;
import java.util.Collections;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.RejectedExecutionException;
import java.util.concurrent.TimeUnit;
import static com.hazelcast.map.impl.querycache.ListenerRegistrationHelper.generateListenerName;
import static com.hazelcast.map.impl.querycache.subscriber.EventPublisherHelper.createIMapEvent;
import static com.hazelcast.map.impl.querycache.subscriber.QueryCacheEventListenerAdapters.createQueryCacheListenerAdaptor;
import static com.hazelcast.util.ConcurrencyUtil.getOrPutIfAbsent;
import static com.hazelcast.util.Preconditions.checkHasText;
import static com.hazelcast.util.Preconditions.checkNotNull;
/**
* Client side event service implementation for query cache.
*
* @see QueryCacheEventService
*/
public class ClientQueryCacheEventService implements QueryCacheEventService {
private static final int EVENT_QUEUE_TIMEOUT_MILLIS = 500;
private static final ConstructorFunction REGISTRY_CONSTRUCTOR =
new ConstructorFunction() {
@Override
public QueryCacheToListenerMapper createNew(String arg) {
return new QueryCacheToListenerMapper();
}
};
private final StripedExecutor executor;
private final ClientListenerService listenerService;
private final InternalSerializationService serializationService;
private final ILogger logger = Logger.getLogger(getClass());
private final ConcurrentMap registrations;
public ClientQueryCacheEventService(HazelcastClientInstanceImpl client) {
AbstractClientListenerService listenerService = (AbstractClientListenerService) client.getListenerService();
this.listenerService = listenerService;
this.serializationService = client.getSerializationService();
this.executor = listenerService.getEventExecutor();
this.registrations = new ConcurrentHashMap();
}
@Override
public boolean hasListener(String mapName, String cacheId) {
QueryCacheToListenerMapper queryCacheToListenerMapper = registrations.get(mapName);
if (queryCacheToListenerMapper == null) {
return false;
}
return queryCacheToListenerMapper.hasListener(cacheId);
}
// used for testing purposes
public ConcurrentMap getRegistrations() {
return registrations;
}
@Override
public void sendEventToSubscriber(String name, Object eventData, int orderKey) {
// this is already subscriber side. So no need to implement it for subscriber side.
throw new UnsupportedOperationException();
}
@Override
public void publish(String mapName, String cacheId, Object event,
int orderKey, Extractors extractors) {
checkHasText(mapName, "mapName");
checkHasText(cacheId, "cacheId");
checkNotNull(event, "event cannot be null");
Collection listeners = getListeners(mapName, cacheId);
for (ListenerInfo info : listeners) {
if (!canPassFilter(event, info.getFilter(), extractors)) {
continue;
}
try {
executor.execute(new EventDispatcher(event, info, orderKey,
serializationService, EVENT_QUEUE_TIMEOUT_MILLIS));
} catch (RejectedExecutionException e) {
// TODO Should we notify user when we overloaded?
logger.warning("EventQueue overloaded! Can not process IMap=[" + mapName + "]"
+ ", QueryCache=[ " + cacheId + "]" + ", Event=[" + event + "]");
}
}
}
private boolean canPassFilter(Object eventData,
EventFilter filter, Extractors extractors) {
if (filter == null || filter instanceof TrueEventFilter) {
return true;
}
if (!(eventData instanceof LocalEntryEventData)) {
return true;
}
LocalEntryEventData localEntryEventData = (LocalEntryEventData) eventData;
if (localEntryEventData.getEventType() != EventLostEvent.EVENT_TYPE) {
Object value = getValueOrOldValue(localEntryEventData);
Data keyData = localEntryEventData.getKeyData();
QueryEntry entry = new QueryEntry(serializationService, keyData, value, extractors);
return filter.eval(entry);
}
return true;
}
private Object getValueOrOldValue(LocalEntryEventData localEntryEventData) {
Object value = localEntryEventData.getValue();
return value != null ? value : localEntryEventData.getOldValue();
}
@Override
public String addPublisherListener(String mapName, String cacheId, ListenerAdapter adapter) {
final String listenerName = generateListenerName(mapName, cacheId);
EventHandler handler = new QueryCacheHandler(adapter);
return listenerService.registerListener(createPublisherListenerCodec(listenerName), handler);
}
@Override
public boolean removePublisherListener(String mapName, String cacheId, String listenerId) {
return listenerService.deregisterListener(listenerId);
}
private ListenerMessageCodec createPublisherListenerCodec(final String listenerName) {
return new ListenerMessageCodec() {
@Override
public ClientMessage encodeAddRequest(boolean localOnly) {
return ContinuousQueryAddListenerCodec.encodeRequest(listenerName, localOnly);
}
@Override
public String decodeAddResponse(ClientMessage clientMessage) {
return ContinuousQueryAddListenerCodec.decodeResponse(clientMessage).response;
}
@Override
public ClientMessage encodeRemoveRequest(String realRegistrationId) {
return MapRemoveEntryListenerCodec.encodeRequest(listenerName, realRegistrationId);
}
@Override
public boolean decodeRemoveResponse(ClientMessage clientMessage) {
return MapRemoveEntryListenerCodec.decodeResponse(clientMessage).response;
}
};
}
@Override
public String addListener(String mapName, String cacheId, MapListener listener) {
return addListener(mapName, cacheId, listener, null);
}
@Override
public String addListener(String mapName, String cacheId, MapListener listener, EventFilter filter) {
checkHasText(mapName, "mapName");
checkHasText(cacheId, "cacheId");
checkNotNull(listener, "listener cannot be null");
QueryCacheToListenerMapper queryCacheToListenerMapper = getOrPutIfAbsent(registrations, mapName, REGISTRY_CONSTRUCTOR);
ListenerAdapter listenerAdaptor = createQueryCacheListenerAdaptor(listener);
return queryCacheToListenerMapper.addListener(cacheId, listenerAdaptor, filter);
}
@Override
public boolean removeListener(String mapName, String cacheId, String listenerId) {
checkHasText(mapName, "mapName");
checkHasText(cacheId, "cacheId");
checkHasText(listenerId, "listenerId");
QueryCacheToListenerMapper queryCacheToListenerMapper = getOrPutIfAbsent(registrations, mapName, REGISTRY_CONSTRUCTOR);
return queryCacheToListenerMapper.removeListener(cacheId, listenerId);
}
@Override
public void removeAllListeners(String mapName, String cacheId) {
checkHasText(mapName, "mapName");
checkHasText(cacheId, "cacheId");
QueryCacheToListenerMapper queryCacheToListenerMap = registrations.get(mapName);
if (queryCacheToListenerMap != null) {
queryCacheToListenerMap.removeAllListeners(cacheId);
}
}
/**
* Query cache event handler.
*/
private final class QueryCacheHandler extends ContinuousQueryAddListenerCodec.AbstractEventHandler
implements EventHandler {
private final ListenerAdapter adapter;
private QueryCacheHandler(ListenerAdapter adapter) {
this.adapter = adapter;
}
@Override
public void beforeListenerRegister() {
// NOP
}
@Override
public void onListenerRegister() {
// NOP
}
@Override
public void handleQueryCacheSingleEventV10(QueryCacheEventData data) {
adapter.onEvent(new SingleIMapEvent(data));
}
@Override
public void handleQueryCacheBatchEventV10(Collection events, String source, int partitionId) {
adapter.onEvent(new BatchIMapEvent(new BatchEventData(events, source, partitionId)));
}
}
private Collection getListeners(String mapName, String cacheName) {
QueryCacheToListenerMapper queryCacheToListenerMapper = registrations.get(mapName);
if (queryCacheToListenerMapper == null) {
return Collections.emptySet();
}
return queryCacheToListenerMapper.getListenerInfos(cacheName);
}
/**
* Dispatches an event to a listener.
*/
private static class EventDispatcher implements StripedRunnable, TimeoutRunnable {
private final Object event;
private final ListenerInfo listenerInfo;
private final int orderKey;
private final long timeoutMs;
private final SerializationService serializationService;
public EventDispatcher(Object event, ListenerInfo listenerInfo, int orderKey,
SerializationService serializationService, long timeoutMs) {
this.event = event;
this.listenerInfo = listenerInfo;
this.orderKey = orderKey;
this.timeoutMs = timeoutMs;
this.serializationService = serializationService;
}
@Override
public int getKey() {
return orderKey;
}
@Override
public void run() {
EventData eventData = (EventData) event;
EventFilter filter = listenerInfo.getFilter();
IMapEvent event = createIMapEvent(eventData, filter, null, serializationService);
ListenerAdapter listenerAdapter = listenerInfo.getListenerAdapter();
listenerAdapter.onEvent(event);
}
@Override
public long getTimeout() {
return timeoutMs;
}
@Override
public TimeUnit getTimeUnit() {
return TimeUnit.MILLISECONDS;
}
}
}
