mirror of
https://github.com/LamGC/Oracle-Sentry.git
synced 2025-04-30 06:37:42 +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;
|
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.Logger;
|
||||||
import org.slf4j.LoggerFactory;
|
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.SpringApplication;
|
||||||
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
||||||
import org.springframework.context.annotation.Bean;
|
|
||||||
import org.springframework.context.annotation.Configuration;
|
import org.springframework.context.annotation.Configuration;
|
||||||
|
|
||||||
import java.io.File;
|
|
||||||
import java.io.IOException;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* <p> 程序入口
|
* <p> 程序入口
|
||||||
* @author LamGC
|
* @author LamGC
|
||||||
@ -30,15 +19,6 @@ public class ApplicationMain {
|
|||||||
@SuppressWarnings("AlibabaConstantFieldShouldBeUpperCase")
|
@SuppressWarnings("AlibabaConstantFieldShouldBeUpperCase")
|
||||||
private final static Object mainThreadWaiter = new Object();
|
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) {
|
public static void main(String[] args) {
|
||||||
SpringApplication.run(ApplicationMain.class, args);
|
SpringApplication.run(ApplicationMain.class, args);
|
||||||
Runtime.getRuntime().addShutdownHook(new Thread(mainThreadWaiter::notifyAll, "ShutdownMainThread"));
|
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