HDDS-201. Add name for LeaseManager. Contributed by Sandeep Nemuri.
This commit is contained in:
parent
9089790cab
commit
a19229594e
@ -42,6 +42,7 @@ public class LeaseManager<T> {
|
||||
private static final Logger LOG =
|
||||
LoggerFactory.getLogger(LeaseManager.class);
|
||||
|
||||
private final String name;
|
||||
private final long defaultTimeout;
|
||||
private Map<T, Lease<T>> activeLeases;
|
||||
private LeaseMonitor leaseMonitor;
|
||||
@ -51,10 +52,13 @@ public class LeaseManager<T> {
|
||||
/**
|
||||
* Creates an instance of lease manager.
|
||||
*
|
||||
* @param name
|
||||
* Name for the LeaseManager instance.
|
||||
* @param defaultTimeout
|
||||
* Default timeout in milliseconds to be used for lease creation.
|
||||
*/
|
||||
public LeaseManager(long defaultTimeout) {
|
||||
public LeaseManager(String name, long defaultTimeout) {
|
||||
this.name = name;
|
||||
this.defaultTimeout = defaultTimeout;
|
||||
}
|
||||
|
||||
@ -62,11 +66,11 @@ public LeaseManager(long defaultTimeout) {
|
||||
* Starts the lease manager service.
|
||||
*/
|
||||
public void start() {
|
||||
LOG.debug("Starting LeaseManager service");
|
||||
LOG.debug("Starting {} LeaseManager service", name);
|
||||
activeLeases = new ConcurrentHashMap<>();
|
||||
leaseMonitor = new LeaseMonitor();
|
||||
leaseMonitorThread = new Thread(leaseMonitor);
|
||||
leaseMonitorThread.setName("LeaseManager#LeaseMonitor");
|
||||
leaseMonitorThread.setName(name + "-LeaseManager#LeaseMonitor");
|
||||
leaseMonitorThread.setDaemon(true);
|
||||
leaseMonitorThread.setUncaughtExceptionHandler((thread, throwable) -> {
|
||||
// Let us just restart this thread after logging an error.
|
||||
@ -75,7 +79,7 @@ public void start() {
|
||||
thread.toString(), throwable);
|
||||
leaseMonitorThread.start();
|
||||
});
|
||||
LOG.debug("Starting LeaseManager#LeaseMonitor Thread");
|
||||
LOG.debug("Starting {}-LeaseManager#LeaseMonitor Thread", name);
|
||||
leaseMonitorThread.start();
|
||||
isRunning = true;
|
||||
}
|
||||
@ -203,7 +207,7 @@ private LeaseMonitor() {
|
||||
@Override
|
||||
public void run() {
|
||||
while(monitor) {
|
||||
LOG.debug("LeaseMonitor: checking for lease expiry");
|
||||
LOG.debug("{}-LeaseMonitor: checking for lease expiry", name);
|
||||
long sleepTime = Long.MAX_VALUE;
|
||||
|
||||
for (T resource : activeLeases.keySet()) {
|
||||
|
@ -67,7 +67,7 @@ public boolean equals(Object obj) {
|
||||
public void testLeaseAcquireAndRelease() throws LeaseException {
|
||||
//It is assumed that the test case execution won't take more than 5 seconds,
|
||||
//if it takes more time increase the defaultTimeout value of LeaseManager.
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>(5000);
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>("Test", 5000);
|
||||
manager.start();
|
||||
DummyResource resourceOne = new DummyResource("one");
|
||||
DummyResource resourceTwo = new DummyResource("two");
|
||||
@ -93,7 +93,7 @@ public void testLeaseAcquireAndRelease() throws LeaseException {
|
||||
|
||||
@Test
|
||||
public void testLeaseAlreadyExist() throws LeaseException {
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>(5000);
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>("Test", 5000);
|
||||
manager.start();
|
||||
DummyResource resourceOne = new DummyResource("one");
|
||||
DummyResource resourceTwo = new DummyResource("two");
|
||||
@ -113,7 +113,7 @@ public void testLeaseAlreadyExist() throws LeaseException {
|
||||
|
||||
@Test
|
||||
public void testLeaseNotFound() throws LeaseException, InterruptedException {
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>(5000);
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>("Test", 5000);
|
||||
manager.start();
|
||||
DummyResource resourceOne = new DummyResource("one");
|
||||
DummyResource resourceTwo = new DummyResource("two");
|
||||
@ -154,7 +154,7 @@ public void testLeaseNotFound() throws LeaseException, InterruptedException {
|
||||
|
||||
@Test
|
||||
public void testCustomLeaseTimeout() throws LeaseException {
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>(5000);
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>("Test", 5000);
|
||||
manager.start();
|
||||
DummyResource resourceOne = new DummyResource("one");
|
||||
DummyResource resourceTwo = new DummyResource("two");
|
||||
@ -179,7 +179,7 @@ public void testCustomLeaseTimeout() throws LeaseException {
|
||||
@Test
|
||||
public void testLeaseCallback() throws LeaseException, InterruptedException {
|
||||
Map<DummyResource, String> leaseStatus = new HashMap<>();
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>(5000);
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>("Test", 5000);
|
||||
manager.start();
|
||||
DummyResource resourceOne = new DummyResource("one");
|
||||
Lease<DummyResource> leaseOne = manager.acquire(resourceOne);
|
||||
@ -209,7 +209,7 @@ public void testCallbackExecutionInCaseOfLeaseRelease()
|
||||
throws LeaseException, InterruptedException {
|
||||
// Callbacks should not be executed in case of lease release
|
||||
Map<DummyResource, String> leaseStatus = new HashMap<>();
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>(5000);
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>("Test", 5000);
|
||||
manager.start();
|
||||
DummyResource resourceOne = new DummyResource("one");
|
||||
Lease<DummyResource> leaseOne = manager.acquire(resourceOne);
|
||||
@ -231,7 +231,7 @@ public void testCallbackExecutionInCaseOfLeaseRelease()
|
||||
public void testLeaseCallbackWithMultipleLeases()
|
||||
throws LeaseException, InterruptedException {
|
||||
Map<DummyResource, String> leaseStatus = new HashMap<>();
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>(5000);
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>("Test", 5000);
|
||||
manager.start();
|
||||
DummyResource resourceOne = new DummyResource("one");
|
||||
DummyResource resourceTwo = new DummyResource("two");
|
||||
@ -302,7 +302,7 @@ public void testLeaseCallbackWithMultipleLeases()
|
||||
|
||||
@Test
|
||||
public void testReuseReleasedLease() throws LeaseException {
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>(5000);
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>("Test", 5000);
|
||||
manager.start();
|
||||
DummyResource resourceOne = new DummyResource("one");
|
||||
Lease<DummyResource> leaseOne = manager.acquire(resourceOne);
|
||||
@ -324,13 +324,12 @@ public void testReuseReleasedLease() throws LeaseException {
|
||||
@Test
|
||||
public void testReuseTimedOutLease()
|
||||
throws LeaseException, InterruptedException {
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>(5000);
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>("Test", 5000);
|
||||
manager.start();
|
||||
DummyResource resourceOne = new DummyResource("one");
|
||||
Lease<DummyResource> leaseOne = manager.acquire(resourceOne);
|
||||
Assert.assertEquals(leaseOne, manager.get(resourceOne));
|
||||
Assert.assertFalse(leaseOne.hasExpired());
|
||||
|
||||
// wait for lease to expire
|
||||
long sleepTime = leaseOne.getRemainingTime() + 1000;
|
||||
try {
|
||||
@ -352,7 +351,7 @@ public void testReuseTimedOutLease()
|
||||
|
||||
@Test
|
||||
public void testRenewLease() throws LeaseException, InterruptedException {
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>(5000);
|
||||
LeaseManager<DummyResource> manager = new LeaseManager<>("Test", 5000);
|
||||
manager.start();
|
||||
DummyResource resourceOne = new DummyResource("one");
|
||||
Lease<DummyResource> leaseOne = manager.acquire(resourceOne);
|
||||
|
@ -46,7 +46,7 @@ public class TestEventWatcher {
|
||||
@Before
|
||||
public void startLeaseManager() {
|
||||
DefaultMetricsSystem.instance();
|
||||
leaseManager = new LeaseManager<>(2000l);
|
||||
leaseManager = new LeaseManager<>("Test", 2000L);
|
||||
leaseManager.start();
|
||||
}
|
||||
|
||||
|
@ -139,8 +139,8 @@ public ContainerMapping(
|
||||
ScmConfigKeys.OZONE_SCM_CONTAINER_CREATION_LEASE_TIMEOUT,
|
||||
ScmConfigKeys.OZONE_SCM_CONTAINER_CREATION_LEASE_TIMEOUT_DEFAULT,
|
||||
TimeUnit.MILLISECONDS);
|
||||
LOG.trace("Starting Container Lease Manager.");
|
||||
containerLeaseManager = new LeaseManager<>(containerCreationLeaseTimeout);
|
||||
containerLeaseManager = new LeaseManager<>("ContainerCreation",
|
||||
containerCreationLeaseTimeout);
|
||||
containerLeaseManager.start();
|
||||
}
|
||||
|
||||
|
@ -99,8 +99,8 @@ public PipelineSelector(NodeManager nodeManager, Configuration conf) {
|
||||
ScmConfigKeys.OZONE_SCM_PIPELINE_CREATION_LEASE_TIMEOUT,
|
||||
ScmConfigKeys.OZONE_SCM_PIPELINE_CREATION_LEASE_TIMEOUT_DEFAULT,
|
||||
TimeUnit.MILLISECONDS);
|
||||
LOG.trace("Starting Pipeline Lease Manager.");
|
||||
pipelineLeaseManager = new LeaseManager<>(pipelineCreationLeaseTimeout);
|
||||
pipelineLeaseManager = new LeaseManager<>("PipelineCreation",
|
||||
pipelineCreationLeaseTimeout);
|
||||
pipelineLeaseManager.start();
|
||||
|
||||
// These are the steady states of a container.
|
||||
|
@ -223,7 +223,8 @@ private StorageContainerManager(OzoneConfiguration conf) throws IOException {
|
||||
conf.getTimeDuration(ScmConfigKeys.HDDS_SCM_WATCHER_TIMEOUT,
|
||||
HDDS_SCM_WATCHER_TIMEOUT_DEFAULT, TimeUnit.MILLISECONDS);
|
||||
|
||||
commandWatcherLeaseManager = new LeaseManager<>(watcherTimeout);
|
||||
commandWatcherLeaseManager = new LeaseManager<>("CommandWatcher",
|
||||
watcherTimeout);
|
||||
|
||||
//TODO: support configurable containerPlacement policy
|
||||
ContainerPlacementPolicy containerPlacementPolicy =
|
||||
|
@ -112,7 +112,7 @@ public void testEventSending() throws InterruptedException, IOException {
|
||||
|
||||
//GIVEN
|
||||
|
||||
LeaseManager<Long> leaseManager = new LeaseManager<>(100000L);
|
||||
LeaseManager<Long> leaseManager = new LeaseManager<>("Test", 100000L);
|
||||
try {
|
||||
leaseManager.start();
|
||||
|
||||
@ -152,7 +152,7 @@ protected List<DatanodeDetails> getCurrentReplicas(
|
||||
public void testCommandWatcher() throws InterruptedException, IOException {
|
||||
|
||||
Logger.getRootLogger().setLevel(Level.DEBUG);
|
||||
LeaseManager<Long> leaseManager = new LeaseManager<>(1000L);
|
||||
LeaseManager<Long> leaseManager = new LeaseManager<>("Test", 1000L);
|
||||
|
||||
try {
|
||||
leaseManager.start();
|
||||
|
Loading…
Reference in New Issue
Block a user