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.
/*
* 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.flink.runtime.heartbeat;
import org.apache.flink.runtime.clusterframework.types.ResourceID;
import org.apache.flink.runtime.concurrent.ScheduledExecutor;
import org.apache.flink.util.Preconditions;
import org.slf4j.Logger;
import javax.annotation.concurrent.ThreadSafe;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
/**
* Heartbeat manager implementation. The heartbeat manager maintains a map of heartbeat monitors
* and resource IDs. Each monitor will be updated when a new heartbeat of the associated machine has
* been received. If the monitor detects that a heartbeat has timed out, it will notify the
* {@link HeartbeatListener} about it. A heartbeat times out iff no heartbeat signal has been
* received within a given timeout interval.
*
* @param Type of the incoming heartbeat payload
* @param Type of the outgoing heartbeat payload
*/
@ThreadSafe
public class HeartbeatManagerImpl implements HeartbeatManager {
/** Heartbeat timeout interval in milli seconds. */
private final long heartbeatTimeoutIntervalMs;
/** Resource ID which is used to mark one own's heartbeat signals. */
private final ResourceID ownResourceID;
/** Heartbeat listener with which the heartbeat manager has been associated. */
private final HeartbeatListener heartbeatListener;
/** Executor service used to run heartbeat timeout notifications. */
private final ScheduledExecutor mainThreadExecutor;
protected final Logger log;
/** Map containing the heartbeat monitors associated with the respective resource ID. */
private final ConcurrentHashMap> heartbeatTargets;
private final HeartbeatMonitor.Factory heartbeatMonitorFactory;
/** Running state of the heartbeat manager. */
protected volatile boolean stopped;
public HeartbeatManagerImpl(
long heartbeatTimeoutIntervalMs,
ResourceID ownResourceID,
HeartbeatListener heartbeatListener,
ScheduledExecutor mainThreadExecutor,
Logger log) {
this(
heartbeatTimeoutIntervalMs,
ownResourceID,
heartbeatListener,
mainThreadExecutor,
log,
new HeartbeatMonitorImpl.Factory<>());
}
public HeartbeatManagerImpl(
long heartbeatTimeoutIntervalMs,
ResourceID ownResourceID,
HeartbeatListener heartbeatListener,
ScheduledExecutor mainThreadExecutor,
Logger log,
HeartbeatMonitor.Factory heartbeatMonitorFactory) {
Preconditions.checkArgument(heartbeatTimeoutIntervalMs > 0L, "The heartbeat timeout has to be larger than 0.");
this.heartbeatTimeoutIntervalMs = heartbeatTimeoutIntervalMs;
this.ownResourceID = Preconditions.checkNotNull(ownResourceID);
this.heartbeatListener = Preconditions.checkNotNull(heartbeatListener, "heartbeatListener");
this.mainThreadExecutor = Preconditions.checkNotNull(mainThreadExecutor);
this.log = Preconditions.checkNotNull(log);
this.heartbeatMonitorFactory = heartbeatMonitorFactory;
this.heartbeatTargets = new ConcurrentHashMap<>(16);
stopped = false;
}
//----------------------------------------------------------------------------------------------
// Getters
//----------------------------------------------------------------------------------------------
ResourceID getOwnResourceID() {
return ownResourceID;
}
HeartbeatListener getHeartbeatListener() {
return heartbeatListener;
}
Map> getHeartbeatTargets() {
return heartbeatTargets;
}
//----------------------------------------------------------------------------------------------
// HeartbeatManager methods
//----------------------------------------------------------------------------------------------
@Override
public void monitorTarget(ResourceID resourceID, HeartbeatTarget heartbeatTarget) {
if (!stopped) {
if (heartbeatTargets.containsKey(resourceID)) {
log.debug("The target with resource ID {} is already been monitored.", resourceID);
} else {
HeartbeatMonitor heartbeatMonitor =
heartbeatMonitorFactory.createHeartbeatMonitor(
resourceID,
heartbeatTarget,
mainThreadExecutor,
heartbeatListener,
heartbeatTimeoutIntervalMs);
heartbeatTargets.put(
resourceID,
heartbeatMonitor);
// check if we have stopped in the meantime (concurrent stop operation)
if (stopped) {
heartbeatMonitor.cancel();
heartbeatTargets.remove(resourceID);
}
}
}
}
@Override
public void unmonitorTarget(ResourceID resourceID) {
if (!stopped) {
HeartbeatMonitor heartbeatMonitor = heartbeatTargets.remove(resourceID);
if (heartbeatMonitor != null) {
heartbeatMonitor.cancel();
}
}
}
@Override
public void stop() {
stopped = true;
for (HeartbeatMonitor heartbeatMonitor : heartbeatTargets.values()) {
heartbeatMonitor.cancel();
}
heartbeatTargets.clear();
}
@Override
public long getLastHeartbeatFrom(ResourceID resourceId) {
HeartbeatMonitor heartbeatMonitor = heartbeatTargets.get(resourceId);
if (heartbeatMonitor != null) {
return heartbeatMonitor.getLastHeartbeat();
} else {
return -1L;
}
}
ScheduledExecutor getMainThreadExecutor() {
return mainThreadExecutor;
}
//----------------------------------------------------------------------------------------------
// HeartbeatTarget methods
//----------------------------------------------------------------------------------------------
@Override
public void receiveHeartbeat(ResourceID heartbeatOrigin, I heartbeatPayload) {
if (!stopped) {
log.debug("Received heartbeat from {}.", heartbeatOrigin);
reportHeartbeat(heartbeatOrigin);
if (heartbeatPayload != null) {
heartbeatListener.reportPayload(heartbeatOrigin, heartbeatPayload);
}
}
}
@Override
public void requestHeartbeat(final ResourceID requestOrigin, I heartbeatPayload) {
if (!stopped) {
log.debug("Received heartbeat request from {}.", requestOrigin);
final HeartbeatTarget heartbeatTarget = reportHeartbeat(requestOrigin);
if (heartbeatTarget != null) {
if (heartbeatPayload != null) {
heartbeatListener.reportPayload(requestOrigin, heartbeatPayload);
}
heartbeatTarget.receiveHeartbeat(getOwnResourceID(), heartbeatListener.retrievePayload(requestOrigin));
}
}
}
HeartbeatTarget reportHeartbeat(ResourceID resourceID) {
if (heartbeatTargets.containsKey(resourceID)) {
HeartbeatMonitor heartbeatMonitor = heartbeatTargets.get(resourceID);
heartbeatMonitor.reportHeartbeat();
return heartbeatMonitor.getHeartbeatTarget();
} else {
return null;
}
}
}
