LamGC/Oracle-Sentry
1
0
Fork 0
mirror of https://github.com/LamGC/Oracle-Sentry.git synced 2025-07-01 04:47:27 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor: 设置自动关闭钩子.

Browse Source 
设置自动关闭钩子以关闭自动保存线程池.
This commit is contained in:
LamGC
2021-08-19 19:14:57 +08:00
parent d5d25dfa42
commit 499c3d283c
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/main/java/net/lamgc/oracle/sentry/oci/compute/ssh/SshAuthIdentityProvider.java
View File

@ -15,6 +15,7 @@ import java.io.IOException;
import java.nio.charset.StandardCharsets; import java.nio.charset.StandardCharsets;
import java.nio.file.Files; import java.nio.file.Files;
import java.nio.file.StandardOpenOption; import java.nio.file.StandardOpenOption;
import java.util.Collections;
import java.util.Map; import java.util.Map;
import java.util.NoSuchElementException; import java.util.NoSuchElementException;
import java.util.Set; import java.util.Set;
@ -29,7 +30,7 @@ import java.util.stream.Collectors;
* @author LamGC * @author LamGC
*/ */
@SuppressWarnings("UnstableApiUsage") @SuppressWarnings("UnstableApiUsage")
public final class SshAuthIdentityProvider { public final class SshAuthIdentityProvider implements AutoCloseable {
private final static String DEFAULT_AUTH_KEY = "@default"; private final static String DEFAULT_AUTH_KEY = "@default";
private final static Logger log = LoggerFactory.getLogger(SshAuthIdentityProvider.class); private final static Logger log = LoggerFactory.getLogger(SshAuthIdentityProvider.class);
@ -71,6 +72,7 @@ public final class SshAuthIdentityProvider {
log.warn("本次 SSH 认证配置保存失败.", e); log.warn("本次 SSH 认证配置保存失败.", e);
} }
}, 60, 10, TimeUnit.SECONDS); }, 60, 10, TimeUnit.SECONDS);
Runtime.getRuntime().addShutdownHook(new Thread(this::close, "Thread-ProviderAutoSave-Close"));
} }
/** /**
@ -169,6 +171,10 @@ public final class SshAuthIdentityProvider {
* @return 返回所有不存在对应 SSH 认证配置的实例 Id. * @return 返回所有不存在对应 SSH 认证配置的实例 Id.
*/ */
private Set<String> checkForMissingInstances() { private Set<String> checkForMissingInstances() {
if (instanceManager == null) {
log.info("实例管理器未设置, 跳过检查.");
return Collections.emptySet();
}
Set<String> instanceIdSet = instanceManager.getComputeInstances().stream() Set<String> instanceIdSet = instanceManager.getComputeInstances().stream()
.map(ComputeInstance::getInstanceId) .map(ComputeInstance::getInstanceId)
.collect(Collectors.toSet()); .collect(Collectors.toSet());
@ -178,4 +184,8 @@ public final class SshAuthIdentityProvider {
return instanceIdSet; return instanceIdSet;
} }
@Override
public void close() {
scheduledExec.shutdown();
}
} }