com.hazelcast.client.impl.clientside.LifecycleServiceImpl 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.clientside;
import com.hazelcast.client.HazelcastClient;
import com.hazelcast.config.ListenerConfig;
import com.hazelcast.core.LifecycleEvent;
import com.hazelcast.core.LifecycleListener;
import com.hazelcast.core.LifecycleService;
import com.hazelcast.instance.BuildInfo;
import com.hazelcast.instance.BuildInfoProvider;
import com.hazelcast.logging.ILogger;
import com.hazelcast.nio.ClassLoaderUtil;
import com.hazelcast.util.UuidUtil;
import com.hazelcast.util.executor.PoolExecutorThreadFactory;
import java.util.EventListener;
import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
import static com.hazelcast.core.LifecycleEvent.LifecycleState.SHUTDOWN;
import static com.hazelcast.core.LifecycleEvent.LifecycleState.SHUTTING_DOWN;
import static com.hazelcast.core.LifecycleEvent.LifecycleState.STARTED;
import static com.hazelcast.core.LifecycleEvent.LifecycleState.STARTING;
import static com.hazelcast.util.StringUtil.isNullOrEmpty;
/**
* Default {@link com.hazelcast.core.LifecycleService} implementation for the client.
*/
public final class LifecycleServiceImpl implements LifecycleService {
private static final long TERMINATE_TIMEOUT_SECONDS = 30;
private final HazelcastClientInstanceImpl client;
private final ConcurrentMap lifecycleListeners
= new ConcurrentHashMap();
private final AtomicBoolean active = new AtomicBoolean(false);
private final BuildInfo buildInfo;
private final ExecutorService executor;
public LifecycleServiceImpl(HazelcastClientInstanceImpl client) {
this.client = client;
ClassLoader classLoader = client.getClientConfig().getClassLoader();
executor = Executors.newSingleThreadExecutor(
new PoolExecutorThreadFactory(client.getName() + ".lifecycle-", classLoader));
final List listenerConfigs = client.getClientConfig().getListenerConfigs();
if (listenerConfigs != null && !listenerConfigs.isEmpty()) {
for (ListenerConfig listenerConfig : listenerConfigs) {
EventListener implementation = listenerConfig.getImplementation();
if (implementation == null) {
try {
implementation = ClassLoaderUtil.newInstance(classLoader, listenerConfig.getClassName());
} catch (Exception e) {
getLogger().severe(e);
}
}
if (implementation instanceof LifecycleListener) {
addLifecycleListener((LifecycleListener) implementation);
}
}
}
buildInfo = BuildInfoProvider.getBuildInfo();
}
private ILogger getLogger() {
return client.getLoggingService().getLogger(LifecycleService.class);
}
@Override
public String addLifecycleListener(LifecycleListener lifecycleListener) {
final String id = UuidUtil.newUnsecureUuidString();
lifecycleListeners.put(id, lifecycleListener);
return id;
}
@Override
public boolean removeLifecycleListener(String registrationId) {
return lifecycleListeners.remove(registrationId) != null;
}
public void fireLifecycleEvent(LifecycleEvent.LifecycleState lifecycleState) {
final LifecycleEvent lifecycleEvent = new LifecycleEvent(lifecycleState);
String revision = buildInfo.getRevision();
if (isNullOrEmpty(revision)) {
revision = "";
} else {
revision = " - " + revision;
BuildInfo upstreamInfo = buildInfo.getUpstreamBuildInfo();
if (upstreamInfo != null) {
String upstreamRevision = upstreamInfo.getRevision();
if (!isNullOrEmpty(upstreamRevision)) {
revision += ", " + upstreamRevision;
}
}
}
getLogger().info("HazelcastClient " + buildInfo.getVersion() + " ("
+ buildInfo.getBuild() + revision + ") is "
+ lifecycleEvent.getState());
executor.execute(new Runnable() {
@Override
public void run() {
for (LifecycleListener lifecycleListener : lifecycleListeners.values()) {
lifecycleListener.stateChanged(lifecycleEvent);
}
}
});
}
@Override
public boolean isRunning() {
return active.get();
}
@Override
public void shutdown() {
client.onGracefulShutdown();
doShutdown();
}
@Override
public void terminate() {
doShutdown();
}
public void start() {
fireLifecycleEvent(STARTING);
active.set(true);
fireLifecycleEvent(STARTED);
}
private void doShutdown() {
if (!active.compareAndSet(true, false)) {
return;
}
fireLifecycleEvent(SHUTTING_DOWN);
HazelcastClient.shutdown(client.getName());
client.doShutdown();
fireLifecycleEvent(SHUTDOWN);
shutdownExecutor();
}
private void shutdownExecutor() {
executor.shutdown();
try {
boolean success = executor.awaitTermination(TERMINATE_TIMEOUT_SECONDS, TimeUnit.SECONDS);
if (!success) {
getLogger().warning("LifecycleService executor awaitTermination could not completed gracefully in "
+ TERMINATE_TIMEOUT_SECONDS + " seconds. Terminating forcefully.");
executor.shutdownNow();
}
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
getLogger().warning("LifecycleService executor awaitTermination is interrupted. Terminating forcefully.", e);
executor.shutdownNow();
}
}
}