Please wait. This can take some minutes ...
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.scheduledexecutor.impl.ScheduledExecutorServiceProxy Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2018, 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.scheduledexecutor.impl;
import com.hazelcast.core.Member;
import com.hazelcast.core.PartitionAware;
import com.hazelcast.internal.cluster.ClusterService;
import com.hazelcast.logging.ILogger;
import com.hazelcast.nio.Address;
import com.hazelcast.quorum.QuorumException;
import com.hazelcast.scheduledexecutor.IScheduledExecutorService;
import com.hazelcast.scheduledexecutor.IScheduledFuture;
import com.hazelcast.scheduledexecutor.NamedTask;
import com.hazelcast.scheduledexecutor.ScheduledTaskHandler;
import com.hazelcast.scheduledexecutor.impl.operations.GetAllScheduledOnMemberOperation;
import com.hazelcast.scheduledexecutor.impl.operations.GetAllScheduledOnPartitionOperationFactory;
import com.hazelcast.scheduledexecutor.impl.operations.ScheduleTaskOperation;
import com.hazelcast.scheduledexecutor.impl.operations.ShutdownOperation;
import com.hazelcast.spi.AbstractDistributedObject;
import com.hazelcast.spi.NodeEngine;
import com.hazelcast.spi.Operation;
import com.hazelcast.spi.OperationService;
import com.hazelcast.spi.partition.IPartitionService;
import com.hazelcast.util.FutureUtil;
import com.hazelcast.util.UuidUtil;
import com.hazelcast.util.function.Supplier;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.Callable;
import java.util.concurrent.Future;
import java.util.concurrent.TimeUnit;
import java.util.logging.Level;
import static com.hazelcast.scheduledexecutor.impl.DistributedScheduledExecutorService.SERVICE_NAME;
import static com.hazelcast.util.ExceptionUtil.rethrow;
import static com.hazelcast.util.ExceptionUtil.sneakyThrow;
import static com.hazelcast.util.FutureUtil.waitWithDeadline;
import static com.hazelcast.util.MapUtil.createHashMap;
import static com.hazelcast.util.MapUtil.createHashMapAdapter;
import static com.hazelcast.util.Preconditions.checkNotNull;
@SuppressWarnings({"unchecked", "checkstyle:methodcount"})
public class ScheduledExecutorServiceProxy
extends AbstractDistributedObject
implements IScheduledExecutorService {
private static final int SHUTDOWN_TIMEOUT = 10;
private final FutureUtil.ExceptionHandler shutdownExceptionHandler = new FutureUtil.ExceptionHandler() {
@Override
public void handleException(Throwable throwable) {
if (throwable != null) {
if (throwable instanceof QuorumException) {
sneakyThrow(throwable);
}
if (throwable.getCause() instanceof QuorumException) {
sneakyThrow(throwable.getCause());
}
}
if (logger.isLoggable(Level.FINEST)) {
logger.log(Level.FINEST, "Exception while ExecutorService shutdown", throwable);
}
}
};
private final String name;
private final ILogger logger;
ScheduledExecutorServiceProxy(String name, NodeEngine nodeEngine, DistributedScheduledExecutorService service) {
super(nodeEngine, service);
this.name = name;
this.logger = nodeEngine.getLogger(ScheduledExecutorServiceProxy.class);
}
@Override
public String getName() {
return name;
}
@Override
public String getServiceName() {
return DistributedScheduledExecutorService.SERVICE_NAME;
}
@Override
public IScheduledFuture schedule(Runnable command, long delay, TimeUnit unit) {
checkNotNull(command, "Command is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
ScheduledRunnableAdapter callable = createScheduledRunnableAdapter(command);
return schedule(callable, delay, unit);
}
@Override
public IScheduledFuture schedule(Callable command, long delay, TimeUnit unit) {
checkNotNull(command, "Command is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
String name = extractNameOrGenerateOne(command);
int partitionId = getTaskOrKeyPartitionId(command, name);
TaskDefinition definition = new TaskDefinition(TaskDefinition.Type.SINGLE_RUN, name, command, delay, unit);
return submitOnPartitionSync(name, new ScheduleTaskOperation(getName(), definition), partitionId);
}
@Override
public IScheduledFuture scheduleAtFixedRate(Runnable command, long initialDelay, long period, TimeUnit unit) {
checkNotNull(command, "Command is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
String name = extractNameOrGenerateOne(command);
int partitionId = getTaskOrKeyPartitionId(command, name);
ScheduledRunnableAdapter adapter = createScheduledRunnableAdapter(command);
TaskDefinition definition =
new TaskDefinition(TaskDefinition.Type.AT_FIXED_RATE, name, adapter, initialDelay, period, unit);
return submitOnPartitionSync(name, new ScheduleTaskOperation(getName(), definition), partitionId);
}
@Override
public IScheduledFuture scheduleOnMember(Runnable command, Member member, long delay, TimeUnit unit) {
checkNotNull(member, "Member is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
return scheduleOnMembers(command, Collections.singleton(member), delay, unit).get(member);
}
@Override
public IScheduledFuture scheduleOnMember(Callable command, Member member, long delay, TimeUnit unit) {
checkNotNull(member, "Member is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
return scheduleOnMembers(command, Collections.singleton(member), delay, unit).get(member);
}
@Override
public IScheduledFuture scheduleOnMemberAtFixedRate(Runnable command, Member member, long initialDelay, long period,
TimeUnit unit) {
checkNotNull(member, "Member is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
return scheduleOnMembersAtFixedRate(command, Collections.singleton(member), initialDelay, period, unit).get(member);
}
@Override
public IScheduledFuture scheduleOnKeyOwner(Runnable command, Object key, long delay, TimeUnit unit) {
checkNotNull(command, "Command is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
ScheduledRunnableAdapter callable = createScheduledRunnableAdapter(command);
return scheduleOnKeyOwner(callable, key, delay, unit);
}
@Override
public IScheduledFuture scheduleOnKeyOwner(Callable command, Object key, long delay, TimeUnit unit) {
checkNotNull(command, "Command is null");
checkNotNull(key, "Key is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
String name = extractNameOrGenerateOne(command);
int partitionId = getKeyPartitionId(key);
TaskDefinition definition = new TaskDefinition(TaskDefinition.Type.SINGLE_RUN, name, command, delay, unit);
return submitOnPartitionSync(name, new ScheduleTaskOperation(getName(), definition), partitionId);
}
@Override
public IScheduledFuture scheduleOnKeyOwnerAtFixedRate(Runnable command, Object key, long initialDelay, long period,
TimeUnit unit) {
checkNotNull(command, "Command is null");
checkNotNull(key, "Key is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
String name = extractNameOrGenerateOne(command);
int partitionId = getKeyPartitionId(key);
ScheduledRunnableAdapter adapter = createScheduledRunnableAdapter(command);
TaskDefinition definition =
new TaskDefinition(TaskDefinition.Type.AT_FIXED_RATE, name, adapter, initialDelay, period, unit);
return submitOnPartitionSync(name, new ScheduleTaskOperation(getName(), definition), partitionId);
}
@Override
public Map> scheduleOnAllMembers(Runnable command, long delay, TimeUnit unit) {
checkNotNull(command, "Command is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
return scheduleOnMembers(command, getNodeEngine().getClusterService().getMembers(), delay, unit);
}
@Override
public Map> scheduleOnAllMembers(Callable command, long delay, TimeUnit unit) {
checkNotNull(command, "Command is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
return scheduleOnMembers(command, getNodeEngine().getClusterService().getMembers(), delay, unit);
}
@Override
public Map> scheduleOnAllMembersAtFixedRate(Runnable command, long initialDelay, long period,
TimeUnit unit) {
checkNotNull(command, "Command is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
return scheduleOnMembersAtFixedRate(command, getNodeEngine().getClusterService().getMembers(), initialDelay, period,
unit);
}
@Override
public Map> scheduleOnMembers(Runnable command, Collection members, long delay,
TimeUnit unit) {
checkNotNull(command, "Command is null");
checkNotNull(members, "Members is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
ScheduledRunnableAdapter callable = createScheduledRunnableAdapter(command);
return (Map>) scheduleOnMembers(callable, members, delay, unit);
}
@Override
public Map> scheduleOnMembers(Callable command, Collection members, long delay,
TimeUnit unit) {
checkNotNull(command, "Command is null");
checkNotNull(members, "Members is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
String name = extractNameOrGenerateOne(command);
Map> futures = createHashMap(members.size());
for (Member member : members) {
TaskDefinition definition = new TaskDefinition(TaskDefinition.Type.SINGLE_RUN, name, command, delay, unit);
futures.put(member,
(IScheduledFuture) submitOnMemberSync(name, new ScheduleTaskOperation(getName(), definition), member));
}
return futures;
}
@Override
public Map> scheduleOnMembersAtFixedRate(Runnable command, Collection members,
long initialDelay, long period, TimeUnit unit) {
checkNotNull(command, "Command is null");
checkNotNull(members, "Members is null");
checkNotNull(unit, "Unit is null");
initializeManagedContext(command);
String name = extractNameOrGenerateOne(command);
ScheduledRunnableAdapter adapter = createScheduledRunnableAdapter(command);
Map> futures = createHashMapAdapter(members.size());
for (Member member : members) {
TaskDefinition definition =
new TaskDefinition(TaskDefinition.Type.AT_FIXED_RATE, name, adapter, initialDelay, period, unit);
futures.put(member, submitOnMemberSync(name, new ScheduleTaskOperation(getName(), definition), member));
}
return futures;
}
@Override
public IScheduledFuture getScheduledFuture(ScheduledTaskHandler handler) {
checkNotNull(handler, "Handler is null");
ScheduledFutureProxy proxy = new ScheduledFutureProxy(handler, this);
initializeManagedContext(proxy);
return proxy;
}
@Override
public Map>> getAllScheduledFutures() {
Map>> accumulator = new LinkedHashMap>>();
retrieveAllPartitionOwnedScheduled(accumulator);
retrieveAllMemberOwnedScheduled(accumulator);
return accumulator;
}
@Override
public void shutdown() {
NodeEngine nodeEngine = getNodeEngine();
Collection members = nodeEngine.getClusterService().getMembers();
OperationService operationService = nodeEngine.getOperationService();
Collection calls = new LinkedList();
for (Member member : members) {
Operation op = new ShutdownOperation(name);
calls.add(operationService.invokeOnTarget(SERVICE_NAME, op, member.getAddress()));
}
waitWithDeadline(calls, SHUTDOWN_TIMEOUT, TimeUnit.SECONDS, shutdownExceptionHandler);
}
private void retrieveAllMemberOwnedScheduled(Map>> accumulator) {
try {
InvokeOnMembers invokeOnMembers =
new InvokeOnMembers(getNodeEngine(), getServiceName(), new GetAllScheduledOnMemberOperationFactory(name),
getNodeEngine().getClusterService().getMembers());
accumulateTaskHandlersAsScheduledFutures(accumulator, invokeOnMembers.invoke());
} catch (Exception e) {
throw rethrow(e);
}
}
private void retrieveAllPartitionOwnedScheduled(Map>> accumulator) {
try {
accumulateTaskHandlersAsScheduledFutures(accumulator,
getNodeEngine().getOperationService().invokeOnAllPartitions(getServiceName(),
new GetAllScheduledOnPartitionOperationFactory(name)));
} catch (Throwable t) {
throw rethrow(t);
}
}
@SuppressWarnings("unchecked")
private void accumulateTaskHandlersAsScheduledFutures(Map>> accumulator,
Map taskHandlersMap) {
ClusterService clusterService = getNodeEngine().getClusterService();
IPartitionService partitionService = getNodeEngine().getPartitionService();
for (Map.Entry entry : taskHandlersMap.entrySet()) {
Member owner;
Object key = entry.getKey();
if (key instanceof Number) {
owner = clusterService.getMember(partitionService.getPartitionOwner((Integer) key));
} else {
owner = (Member) key;
}
List handlers = (List) entry.getValue();
List> futures = new ArrayList>();
for (ScheduledTaskHandler handler : handlers) {
IScheduledFuture future = new ScheduledFutureProxy(handler, this);
initializeManagedContext(future);
futures.add(future);
}
if (accumulator.containsKey(owner)) {
List> memberFutures = accumulator.get(owner);
memberFutures.addAll(futures);
} else {
accumulator.put(owner, futures);
}
}
}
private ScheduledRunnableAdapter createScheduledRunnableAdapter(Runnable command) {
checkNotNull(command, "Command can't be null");
return new ScheduledRunnableAdapter(command);
}
private IScheduledFuture createFutureProxy(int partitionId, String taskName) {
ScheduledFutureProxy proxy =
new ScheduledFutureProxy(ScheduledTaskHandlerImpl.of(partitionId, getName(), taskName), this);
proxy.setHazelcastInstance(getNodeEngine().getHazelcastInstance());
return proxy;
}
private IScheduledFuture createFutureProxy(Address address, String taskName) {
ScheduledFutureProxy proxy = new ScheduledFutureProxy(ScheduledTaskHandlerImpl.of(address, getName(), taskName), this);
proxy.setHazelcastInstance(getNodeEngine().getHazelcastInstance());
return proxy;
}
private int getKeyPartitionId(Object key) {
return getNodeEngine().getPartitionService().getPartitionId(key);
}
private int getTaskOrKeyPartitionId(Callable task, Object key) {
if (task instanceof PartitionAware) {
Object newKey = ((PartitionAware) task).getPartitionKey();
if (newKey != null) {
key = newKey;
}
}
return getKeyPartitionId(key);
}
private int getTaskOrKeyPartitionId(Runnable task, Object key) {
if (task instanceof PartitionAware) {
Object newKey = ((PartitionAware) task).getPartitionKey();
if (newKey != null) {
key = newKey;
}
}
return getKeyPartitionId(key);
}
private String extractNameOrGenerateOne(Object command) {
String name = null;
if (command instanceof NamedTask) {
name = ((NamedTask) command).getName();
}
return name != null ? name : UuidUtil.newUnsecureUuidString();
}
private IScheduledFuture submitOnPartitionSync(String taskName, Operation op, int partitionId) {
op.setPartitionId(partitionId);
invokeOnPartition(op).join();
return createFutureProxy(partitionId, taskName);
}
private IScheduledFuture submitOnMemberSync(String taskName, Operation op, Member member) {
Address address = member.getAddress();
getOperationService().invokeOnTarget(getServiceName(), op, address).join();
return createFutureProxy(address, taskName);
}
private void initializeManagedContext(Object object) {
getNodeEngine().getSerializationService().getManagedContext().initialize(object);
}
private static class GetAllScheduledOnMemberOperationFactory
implements Supplier {
private final String schedulerName;
GetAllScheduledOnMemberOperationFactory(String schedulerName) {
this.schedulerName = schedulerName;
}
@Override
public Operation get() {
return new GetAllScheduledOnMemberOperation(schedulerName);
}
}
}