diff options
author | Jon Marius Venstad <venstad@gmail.com> | 2019-02-08 11:55:01 +0100 |
---|---|---|
committer | Jon Marius Venstad <venstad@gmail.com> | 2019-02-08 11:55:01 +0100 |
commit | 9a8780786ac09415dbcfec5d16b0585e3b4c533a (patch) | |
tree | 533ca5570090ad5c185e4dffb6c765cf08a82eb4 /orchestrator/src/main/java/com | |
parent | 89f9555b858152f7e565504613485d5f37bff6f8 (diff) |
Remove InMemoryStatusService, and clean up curator usage in ZookeeperStatusServiceTest
Diffstat (limited to 'orchestrator/src/main/java/com')
-rw-r--r-- | orchestrator/src/main/java/com/yahoo/vespa/orchestrator/status/InMemoryStatusService.java | 122 |
1 files changed, 0 insertions, 122 deletions
diff --git a/orchestrator/src/main/java/com/yahoo/vespa/orchestrator/status/InMemoryStatusService.java b/orchestrator/src/main/java/com/yahoo/vespa/orchestrator/status/InMemoryStatusService.java deleted file mode 100644 index 41044bae782..00000000000 --- a/orchestrator/src/main/java/com/yahoo/vespa/orchestrator/status/InMemoryStatusService.java +++ /dev/null @@ -1,122 +0,0 @@ -// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. -package com.yahoo.vespa.orchestrator.status; - -import com.yahoo.vespa.applicationmodel.ApplicationInstanceReference; -import com.yahoo.vespa.applicationmodel.HostName; -import com.yahoo.vespa.orchestrator.OrchestratorContext; - -import java.util.HashMap; -import java.util.HashSet; -import java.util.Map; -import java.util.Set; -import java.util.concurrent.locks.Lock; -import java.util.concurrent.locks.ReentrantLock; - -/** - * Implementation of the StatusService interface for testing. - * - * @author oyving - */ -public class InMemoryStatusService implements StatusService { - - private final Map<HostName, HostStatus> hostServiceStatus = new HashMap<>(); - private final Set<ApplicationInstanceReference> applicationStatus = new HashSet<>(); - private final LockService<ApplicationInstanceReference> instanceLockService = new LockService<>(); - - private void setHostStatus(HostName hostName, HostStatus status) { - hostServiceStatus.put(hostName, status); - } - - @Override - public ReadOnlyStatusRegistry forApplicationInstance(ApplicationInstanceReference applicationInstanceReference) { - return new ReadOnlyStatusRegistry() { - @Override - public HostStatus getHostStatus(HostName hostName) { - return hostServiceStatus.getOrDefault(hostName, HostStatus.NO_REMARKS); - } - - @Override - public ApplicationInstanceStatus getApplicationInstanceStatus() { - return applicationStatus.contains(applicationInstanceReference) ? - ApplicationInstanceStatus.ALLOWED_TO_BE_DOWN : ApplicationInstanceStatus.NO_REMARKS; - } - }; - } - - - @Override - public MutableStatusRegistry lockApplicationInstance_forCurrentThreadOnly( - OrchestratorContext context, - ApplicationInstanceReference applicationInstanceReference) { - return new InMemoryMutableStatusRegistry(applicationInstanceReference); - } - - @Override - public Set<ApplicationInstanceReference> getAllSuspendedApplications() { - return applicationStatus; - } - - private class InMemoryMutableStatusRegistry implements MutableStatusRegistry { - - private final Lock lockHandle; - private final ApplicationInstanceReference ref; - - private InMemoryMutableStatusRegistry(ApplicationInstanceReference ref) { - this.ref = ref; - - this.lockHandle = instanceLockService.get(ref); - this.lockHandle.lock(); - } - - @Override - public void setHostState(HostName hostName, HostStatus status) { - setHostStatus(hostName, status); - } - - @Override - public void setApplicationInstanceStatus(ApplicationInstanceStatus applicationInstanceStatus) { - if (applicationInstanceStatus == ApplicationInstanceStatus.NO_REMARKS) { - applicationStatus.remove(ref); - } else { - applicationStatus.add(ref); - } - } - - @Override - public HostStatus getHostStatus(HostName hostName) { - return hostServiceStatus.getOrDefault(hostName, HostStatus.NO_REMARKS); - } - - @Override - public ApplicationInstanceStatus getApplicationInstanceStatus() { - return applicationStatus.contains(ref) ? ApplicationInstanceStatus.ALLOWED_TO_BE_DOWN : - ApplicationInstanceStatus.NO_REMARKS; - } - - @Override - public void close() { - lockHandle.unlock(); - } - } - - private static class LockService<T> { - - private final Map<T, Lock> locks; - - public LockService() { - this.locks = new HashMap<>(); - } - - public Lock get(T lockId) { - synchronized (this) { - Lock lock = locks.computeIfAbsent( - lockId, - id -> new ReentrantLock() - ); - - return lock; - } - } - } - -} |