mirror of
https://github.com/LamGC/Oracle-Sentry.git
synced 2025-04-29 22:27:34 +00:00
refactor: 将启动初始化代码移到专门的类中.
将启动时对组件的初始化工作由 ApplicationMain 移动到 ApplicationInitiation.
This commit is contained in:
parent
9db2671fd3
commit
b5d0e0f6a0
GPG Key ID:
6C5AE2A913941E1D
143
src/main/java/net/lamgc/oracle/sentry/ApplicationInitiation.java
Normal file
143
src/main/java/net/lamgc/oracle/sentry/ApplicationInitiation.java
Normal file
@ -0,0 +1,143 @@
|
||||
package net.lamgc.oracle.sentry;
|
||||
|
||||
import com.oracle.bmc.auth.AuthenticationDetailsProvider;
|
||||
import net.lamgc.oracle.sentry.script.ScriptComponent;
|
||||
import net.lamgc.oracle.sentry.script.ScriptManager;
|
||||
import net.lamgc.oracle.sentry.script.tools.http.ScriptHttpClient;
|
||||
import org.apache.http.impl.client.HttpClientBuilder;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.context.ApplicationContextInitializer;
|
||||
import org.springframework.context.ConfigurableApplicationContext;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
import org.springframework.lang.NonNull;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
|
||||
/**
|
||||
* @author LamGC
|
||||
*/
|
||||
@SuppressWarnings("rawtypes")
|
||||
@Configuration
|
||||
public class ApplicationInitiation implements ApplicationContextInitializer {
|
||||
|
||||
private final static Logger log = LoggerFactory.getLogger(ApplicationInitiation.class);
|
||||
|
||||
@Value("${oracle.identity.location:./identity/}")
|
||||
@NonNull
|
||||
private String identityDirectory;
|
||||
|
||||
@Value("${oracle.identity.pattern:.*\\.oracle.ini$}")
|
||||
@NonNull
|
||||
private String identityFilePattern;
|
||||
|
||||
@Value("${sentry.script.location:./scripts/}")
|
||||
@NonNull
|
||||
private String scriptsLocation;
|
||||
|
||||
@Value("${oracle.script.ssh.identityPath:./config/ssh.config.json}")
|
||||
@NonNull
|
||||
private String sshIdentityPath;
|
||||
|
||||
@Bean("oracle.identity.manager")
|
||||
public OracleIdentityManager initialOracleIdentityManager() throws IOException {
|
||||
OracleIdentityManager oracleUserManager = new OracleIdentityManager();
|
||||
log.info("正在加载 Oracle API 身份配置...");
|
||||
log.debug("Oracle API 身份配置查找路径: \"{}\", 匹配表达式: {}", identityDirectory, identityFilePattern);
|
||||
File identityDir = new File(identityDirectory);
|
||||
if (!identityDir.exists()) {
|
||||
if (identityDir.mkdirs()) {
|
||||
log.warn("身份配置文件夹不存在, 已创建该文件夹, 请将身份配置放入该文件夹后重新启动程序.");
|
||||
} else {
|
||||
log.error("身份配置文件夹创建失败, 请手动在运行目录下创建 'identity' 文件夹, 并将身份配置放入该文件夹中, 重新启动程序.");
|
||||
}
|
||||
System.exit(1);
|
||||
}
|
||||
int loadedCount = oracleUserManager.loadFromDirectory(identityDir, identityFilePattern);
|
||||
log.info("已加载 {} 个身份配置.", loadedCount);
|
||||
return oracleUserManager;
|
||||
}
|
||||
|
||||
@Bean("oracle.compute.instance.manager")
|
||||
@Autowired
|
||||
public ComputeInstanceManager initialComputeInstanceManager(OracleIdentityManager identityManager) throws IOException {
|
||||
ComputeInstanceManager instanceManager = new ComputeInstanceManager();
|
||||
int addTotal = 0;
|
||||
for (AuthenticationDetailsProvider provider : identityManager.getProviders()) {
|
||||
String identityName = identityManager.getIdentityName(provider.getUserId());
|
||||
log.info("正在加载用户 {} 所拥有的所有实例...", identityName);
|
||||
int addCount;
|
||||
try {
|
||||
addCount = instanceManager.addComputeInstanceFromUser(provider);
|
||||
} catch (Exception e) {
|
||||
log.error("加载实例时发生异常.", e);
|
||||
continue;
|
||||
}
|
||||
log.info("用户 {} 已添加 {} 个计算实例.", identityName, addCount);
|
||||
addTotal += addCount;
|
||||
}
|
||||
|
||||
log.info("正在初始化 SSH 认证配置提供器...");
|
||||
instanceManager.initialSshIdentityProvider(new File(sshIdentityPath));
|
||||
log.info("已完成 ComputeInstanceManager 初始化, 共加载了 {} 个计算实例.", addTotal);
|
||||
return instanceManager;
|
||||
}
|
||||
|
||||
@Bean("sentry.script.manager")
|
||||
@Autowired
|
||||
public ScriptManager initialScriptManager(ComputeInstanceManager instanceManager) {
|
||||
ScriptComponent context = new ScriptComponent(new ScriptHttpClient(HttpClientBuilder.create()
|
||||
.build()),
|
||||
instanceManager);
|
||||
|
||||
ScriptManager manager = new ScriptManager(new File(scriptsLocation), context);
|
||||
manager.loadScripts();
|
||||
return manager;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void initialize(ConfigurableApplicationContext applicationContext) {
|
||||
try {
|
||||
boolean result = initialDirectory();
|
||||
if (result) {
|
||||
System.exit(1);
|
||||
}
|
||||
} catch (IOException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
private boolean initialDirectory() throws IOException {
|
||||
String[] directors = new String[] {
|
||||
identityDirectory,
|
||||
scriptsLocation
|
||||
};
|
||||
|
||||
String[] files = new String[] {
|
||||
sshIdentityPath
|
||||
};
|
||||
|
||||
boolean hasFailure = false;
|
||||
for (String directory : directors) {
|
||||
File dir = new File(directory);
|
||||
if (!dir.exists() && !dir.mkdirs()) {
|
||||
log.error("文件夹 {} 创建失败.", dir.getCanonicalPath());
|
||||
hasFailure = true;
|
||||
}
|
||||
}
|
||||
|
||||
for (String file : files) {
|
||||
File dir = new File(file);
|
||||
if (!dir.exists() && !dir.createNewFile()) {
|
||||
log.error("文件 {} 创建失败.", dir.getCanonicalPath());
|
||||
hasFailure = true;
|
||||
}
|
||||
}
|
||||
log.info("目录检查完成.");
|
||||
return hasFailure;
|
||||
}
|
||||
}
|
||||
@ -1,22 +1,11 @@
|
||||
package net.lamgc.oracle.sentry;
|
||||
|
||||
import com.oracle.bmc.auth.AuthenticationDetailsProvider;
|
||||
import net.lamgc.oracle.sentry.script.ScriptComponent;
|
||||
import net.lamgc.oracle.sentry.script.ScriptManager;
|
||||
import net.lamgc.oracle.sentry.script.tools.http.ScriptHttpClient;
|
||||
import org.apache.http.impl.client.HttpClientBuilder;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.boot.SpringApplication;
|
||||
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
|
||||
/**
|
||||
* <p> 程序入口
|
||||
* @author LamGC
|
||||
@ -30,15 +19,6 @@ public class ApplicationMain {
|
||||
@SuppressWarnings("AlibabaConstantFieldShouldBeUpperCase")
|
||||
private final static Object mainThreadWaiter = new Object();
|
||||
|
||||
@Value("${oracle.identity.location:./identity/}")
|
||||
private String identityDirectory;
|
||||
@Value("${oracle.identity.pattern:.*\\.oracle.ini$}")
|
||||
private String identityFilePattern;
|
||||
@Value("${sentry.script.location:./scripts/}")
|
||||
private String scriptsLocation;
|
||||
@Value("${oracle.script.ssh.identityPath:./ssh.config.json}")
|
||||
private String sshIdentityPath;
|
||||
|
||||
public static void main(String[] args) {
|
||||
SpringApplication.run(ApplicationMain.class, args);
|
||||
Runtime.getRuntime().addShutdownHook(new Thread(mainThreadWaiter::notifyAll, "ShutdownMainThread"));
|
||||
@ -51,51 +31,4 @@ public class ApplicationMain {
|
||||
}
|
||||
}
|
||||
|
||||
@Bean("oracle.identity.manager")
|
||||
public OracleIdentityManager initialOracleIdentityManager() throws IOException {
|
||||
OracleIdentityManager oracleUserManager = new OracleIdentityManager();
|
||||
log.info("正在加载 Oracle API 身份配置...");
|
||||
log.debug("Oracle API 身份配置查找路径: \"{}\", 匹配表达式: {}", identityDirectory, identityFilePattern);
|
||||
int loadedCount = oracleUserManager.loadFromDirectory(new File(identityDirectory), identityFilePattern);
|
||||
log.info("已加载 {} 个身份配置.", loadedCount);
|
||||
return oracleUserManager;
|
||||
}
|
||||
|
||||
@Bean("oracle.compute.instance.manager")
|
||||
@Autowired
|
||||
public ComputeInstanceManager initialComputeInstanceManager(OracleIdentityManager identityManager) throws IOException {
|
||||
ComputeInstanceManager instanceManager = new ComputeInstanceManager();
|
||||
int addTotal = 0;
|
||||
for (AuthenticationDetailsProvider provider : identityManager.getProviders()) {
|
||||
String identityName = identityManager.getIdentityName(provider.getUserId());
|
||||
log.info("正在加载用户 {} 所拥有的所有实例...", identityName);
|
||||
int addCount;
|
||||
try {
|
||||
addCount = instanceManager.addComputeInstanceFromUser(provider);
|
||||
} catch (Exception e) {
|
||||
log.error("加载实例时发生异常.", e);
|
||||
continue;
|
||||
}
|
||||
log.info("用户 {} 已添加 {} 个计算实例.", identityName, addCount);
|
||||
addTotal += addCount;
|
||||
}
|
||||
|
||||
log.info("正在初始化 SSH 认证配置提供器...");
|
||||
instanceManager.initialSshIdentityProvider(new File(sshIdentityPath));
|
||||
log.info("已完成 ComputeInstanceManager 初始化, 共加载了 {} 个计算实例.", addTotal);
|
||||
return instanceManager;
|
||||
}
|
||||
|
||||
@Bean("sentry.script.manager")
|
||||
@Autowired
|
||||
public ScriptManager initialScriptManager(ComputeInstanceManager instanceManager) {
|
||||
ScriptComponent context = new ScriptComponent(new ScriptHttpClient(HttpClientBuilder.create()
|
||||
.build()),
|
||||
instanceManager);
|
||||
|
||||
ScriptManager manager = new ScriptManager(new File(scriptsLocation), context);
|
||||
manager.loadScripts();
|
||||
return manager;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user