com.hazelcast.cp.internal.CPSubsystemImpl Maven / Gradle / Ivy
/*
* Copyright (c) 2008-2021, 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.cp.internal;
import com.hazelcast.core.DistributedObject;
import com.hazelcast.core.HazelcastException;
import com.hazelcast.cp.CPGroup;
import com.hazelcast.cp.CPGroupId;
import com.hazelcast.cp.CPMember;
import com.hazelcast.cp.event.CPGroupAvailabilityListener;
import com.hazelcast.cp.event.CPMembershipListener;
import com.hazelcast.cp.CPSubsystem;
import com.hazelcast.cp.CPSubsystemManagementService;
import com.hazelcast.cp.IAtomicLong;
import com.hazelcast.cp.IAtomicReference;
import com.hazelcast.cp.ICountDownLatch;
import com.hazelcast.cp.ISemaphore;
import com.hazelcast.cp.internal.datastructures.atomiclong.AtomicLongService;
import com.hazelcast.cp.internal.datastructures.atomicref.AtomicRefService;
import com.hazelcast.cp.internal.datastructures.countdownlatch.CountDownLatchService;
import com.hazelcast.cp.internal.datastructures.lock.LockService;
import com.hazelcast.cp.internal.datastructures.semaphore.SemaphoreService;
import com.hazelcast.cp.internal.datastructures.spi.RaftRemoteService;
import com.hazelcast.cp.internal.session.RaftSessionService;
import com.hazelcast.cp.lock.FencedLock;
import com.hazelcast.cp.session.CPSessionManagementService;
import com.hazelcast.instance.impl.HazelcastInstanceImpl;
import com.hazelcast.logging.ILogger;
import com.hazelcast.spi.impl.InternalCompletableFuture;
import javax.annotation.Nonnull;
import java.util.Collection;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
import static com.hazelcast.internal.util.Preconditions.checkNotNull;
/**
* Provides access to CP Subsystem utilities
*/
public class CPSubsystemImpl implements CPSubsystem {
private final HazelcastInstanceImpl instance;
private final boolean cpSubsystemEnabled;
private volatile CPSubsystemManagementService cpSubsystemManagementService;
public CPSubsystemImpl(HazelcastInstanceImpl instance) {
this.instance = instance;
int cpMemberCount = instance.getConfig().getCPSubsystemConfig().getCPMemberCount();
this.cpSubsystemEnabled = cpMemberCount > 0;
ILogger logger = instance.node.getLogger(CPSubsystem.class);
if (cpSubsystemEnabled) {
logger.info("CP Subsystem is enabled with " + cpMemberCount + " members.");
} else {
logger.warning("CP Subsystem is not enabled. CP data structures will operate in UNSAFE mode! "
+ "Please note that UNSAFE mode will not provide strong consistency guarantees.");
}
}
@Nonnull
@Override
public IAtomicLong getAtomicLong(@Nonnull String name) {
checkNotNull(name, "Retrieving an atomic long instance with a null name is not allowed!");
return createProxy(AtomicLongService.SERVICE_NAME, name);
}
@Nonnull
@Override
public IAtomicReference getAtomicReference(@Nonnull String name) {
checkNotNull(name, "Retrieving an atomic reference instance with a null name is not allowed!");
return createProxy(AtomicRefService.SERVICE_NAME, name);
}
@Nonnull
@Override
public ICountDownLatch getCountDownLatch(@Nonnull String name) {
checkNotNull(name, "Retrieving a count down latch instance with a null name is not allowed!");
return createProxy(CountDownLatchService.SERVICE_NAME, name);
}
@Nonnull
@Override
public FencedLock getLock(@Nonnull String name) {
checkNotNull(name, "Retrieving an fenced lock instance with a null name is not allowed!");
return createProxy(LockService.SERVICE_NAME, name);
}
@Nonnull
@Override
public ISemaphore getSemaphore(@Nonnull String name) {
checkNotNull(name, "Retrieving a semaphore instance with a null name is not allowed!");
return createProxy(SemaphoreService.SERVICE_NAME, name);
}
@Override
public CPMember getLocalCPMember() {
return getCPSubsystemManagementService().getLocalCPMember();
}
@Override
public CPSubsystemManagementService getCPSubsystemManagementService() {
if (!cpSubsystemEnabled) {
throw new HazelcastException("CP Subsystem is not enabled!");
}
if (cpSubsystemManagementService != null) {
return cpSubsystemManagementService;
}
RaftService raftService = getService(RaftService.SERVICE_NAME);
cpSubsystemManagementService = new CPSubsystemManagementServiceImpl(raftService);
return cpSubsystemManagementService;
}
@Override
public CPSessionManagementService getCPSessionManagementService() {
if (!cpSubsystemEnabled) {
throw new HazelcastException("CP Subsystem is not enabled!");
}
return getService(RaftSessionService.SERVICE_NAME);
}
private T getService(@Nonnull String serviceName) {
return instance.node.getNodeEngine().getService(serviceName);
}
private T createProxy(String serviceName, String name) {
RaftRemoteService service = getService(serviceName);
return service.createProxy(name);
}
@Override
public UUID addMembershipListener(CPMembershipListener listener) {
RaftService raftService = getService(RaftService.SERVICE_NAME);
return raftService.registerMembershipListener(listener);
}
@Override
public boolean removeMembershipListener(UUID id) {
RaftService raftService = getService(RaftService.SERVICE_NAME);
return raftService.removeMembershipListener(id);
}
@Override
public UUID addGroupAvailabilityListener(CPGroupAvailabilityListener listener) {
RaftService raftService = getService(RaftService.SERVICE_NAME);
return raftService.registerAvailabilityListener(listener);
}
@Override
public boolean removeGroupAvailabilityListener(UUID id) {
RaftService raftService = getService(RaftService.SERVICE_NAME);
return raftService.removeAvailabilityListener(id);
}
private static class CPSubsystemManagementServiceImpl implements CPSubsystemManagementService {
private final RaftService raftService;
CPSubsystemManagementServiceImpl(RaftService raftService) {
this.raftService = raftService;
}
@Override
public CPMember getLocalCPMember() {
return raftService.getLocalCPMember();
}
@Override
public InternalCompletableFuture> getCPGroupIds() {
return raftService.getCPGroupIds();
}
@Override
public InternalCompletableFuture getCPGroup(String name) {
return raftService.getCPGroup(name);
}
@Override
public InternalCompletableFuture forceDestroyCPGroup(String groupName) {
return raftService.forceDestroyCPGroup(groupName);
}
@Override
public InternalCompletableFuture> getCPMembers() {
return raftService.getCPMembers();
}
@Override
public InternalCompletableFuture promoteToCPMember() {
return raftService.promoteToCPMember();
}
@Override
public InternalCompletableFuture removeCPMember(UUID cpMemberUuid) {
return raftService.removeCPMember(cpMemberUuid);
}
@Override
public InternalCompletableFuture reset() {
return raftService.resetCPSubsystem();
}
@Override
public boolean isDiscoveryCompleted() {
return raftService.isDiscoveryCompleted();
}
@Override
public boolean awaitUntilDiscoveryCompleted(long timeout, TimeUnit timeUnit) throws InterruptedException {
return raftService.awaitUntilDiscoveryCompleted(timeout, timeUnit);
}
}
}