repos init.
This commit is contained in:
162
src/main/java/com/serliunx/ddns/support/SystemInitializer.java
Normal file
162
src/main/java/com/serliunx/ddns/support/SystemInitializer.java
Normal file
@@ -0,0 +1,162 @@
|
||||
package com.serliunx.ddns.support;
|
||||
|
||||
import com.serliunx.ddns.config.Configuration;
|
||||
import com.serliunx.ddns.constant.SystemConstants;
|
||||
import com.serliunx.ddns.core.context.MultipleSourceInstanceContext;
|
||||
import com.serliunx.ddns.core.instance.Instance;
|
||||
import com.serliunx.ddns.support.feign.client.IPAddressClient;
|
||||
import com.serliunx.ddns.support.feign.client.entity.IPAddressResponse;
|
||||
import com.serliunx.ddns.thread.TaskThreadFactory;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
import org.slf4j.MDC;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.InputStream;
|
||||
import java.io.OutputStream;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Path;
|
||||
import java.nio.file.Paths;
|
||||
import java.util.Set;
|
||||
import java.util.concurrent.ScheduledThreadPoolExecutor;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
import static com.serliunx.ddns.config.ConfigurationKeys.KEY_TASK_REFRESH_INTERVAL_IP;
|
||||
import static com.serliunx.ddns.config.ConfigurationKeys.KEY_THREAD_POOL_CORE_SIZE;
|
||||
|
||||
/**
|
||||
* 系统初始化
|
||||
* @author SerLiunx
|
||||
* @since 1.0
|
||||
*/
|
||||
public final class SystemInitializer implements Refreshable{
|
||||
|
||||
private static final Logger log = LoggerFactory.getLogger(SystemInitializer.class);
|
||||
|
||||
private final Configuration configuration;
|
||||
private final MultipleSourceInstanceContext instanceContext;
|
||||
|
||||
private ScheduledThreadPoolExecutor scheduledThreadPoolExecutor;
|
||||
private Set<Instance> instances;
|
||||
|
||||
SystemInitializer(Configuration configuration, MultipleSourceInstanceContext instanceContext) {
|
||||
this.configuration = configuration;
|
||||
this.instanceContext = instanceContext;
|
||||
}
|
||||
|
||||
public static Configurer configurer(){
|
||||
return new Configurer();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void refresh() {
|
||||
log.info("程序正在初始化, 请稍候.");
|
||||
|
||||
// 释放配置文件
|
||||
releaseResource(SystemConstants.PROPERTIES_FILE);
|
||||
|
||||
// 刷新配置信息
|
||||
configuration.refresh();
|
||||
|
||||
// 获取核心线程数量, 默认为CPU核心数量
|
||||
int coreSize = configuration.getInteger(KEY_THREAD_POOL_CORE_SIZE, Runtime.getRuntime().availableProcessors());
|
||||
|
||||
// 初始化线程池
|
||||
initThreadPool(coreSize);
|
||||
|
||||
// 加载实例(不同的容器加载时机不同)
|
||||
loadInstances();
|
||||
|
||||
// 运行实例
|
||||
runInstances();
|
||||
log.info("初始化完成!");
|
||||
}
|
||||
|
||||
public MultipleSourceInstanceContext getInstanceContext() {
|
||||
return instanceContext;
|
||||
}
|
||||
|
||||
public Set<Instance> getInstances() {
|
||||
return instances;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void afterRefresh() {
|
||||
// TODO
|
||||
}
|
||||
|
||||
private void loadInstances() {
|
||||
instances = instanceContext.getInstances();
|
||||
log.info("载入 {} 个实例.", instances.size());
|
||||
}
|
||||
|
||||
private void releaseResource(String resourceName){
|
||||
ClassLoader classLoader = SystemConstants.class.getClassLoader();
|
||||
Path path = Paths.get(SystemConstants.USER_DIR + File.separator + resourceName);
|
||||
// 检查文件是否已存在
|
||||
if(Files.exists(path)){
|
||||
log.debug("文件 {} 已存在, 无需解压.", resourceName);
|
||||
return;
|
||||
}
|
||||
try (InputStream inputStream = classLoader.getResourceAsStream(resourceName)) {
|
||||
log.debug("正在解压文件 {} 至路径: {}", resourceName, SystemConstants.USER_DIR);
|
||||
// 创建输出流，写入文件到指定目录
|
||||
OutputStream outputStream = Files.newOutputStream(path);
|
||||
byte[] buffer = new byte[1024];
|
||||
int bytesRead;
|
||||
if(inputStream != null){
|
||||
while ((bytesRead = inputStream.read(buffer)) != -1) {
|
||||
outputStream.write(buffer, 0, bytesRead);
|
||||
}
|
||||
}
|
||||
outputStream.close();
|
||||
}catch (Exception e){
|
||||
log.error("文件 {} 解压失败!, 原因: {}", resourceName, e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
private void runInstances() {
|
||||
Assert.notNull(scheduledThreadPoolExecutor);
|
||||
Assert.notNull(instances);
|
||||
|
||||
for (Instance i : instances) {
|
||||
if(!i.validate()){
|
||||
log.error("实例{}({})参数校验不通过, 将不会被运行.", i.getName(), i.getType());
|
||||
continue;
|
||||
}
|
||||
// 初始化实例
|
||||
i.refresh();
|
||||
scheduledThreadPoolExecutor.scheduleWithFixedDelay(i, 0, i.getInterval(), TimeUnit.SECONDS);
|
||||
log.info("{}({})已启动, 运行周期 {} 秒.", i.getName(), i.getType(), i.getInterval());
|
||||
}
|
||||
}
|
||||
|
||||
private void initThreadPool(int coreSize){
|
||||
Assert.isLargerThan(coreSize, 1);
|
||||
scheduledThreadPoolExecutor = new ScheduledThreadPoolExecutor(coreSize, new TaskThreadFactory());
|
||||
|
||||
// 初始化一个线程保活
|
||||
scheduledThreadPoolExecutor.submit(() -> {});
|
||||
|
||||
// 提交定时获取网络IP的定时任务
|
||||
scheduledThreadPoolExecutor.scheduleAtFixedRate(() -> {
|
||||
log.info("正在尝试获取本机最新的IP地址.");
|
||||
IPAddressResponse response = IPAddressClient.instance.getIPAddress();
|
||||
String ip;
|
||||
if(response != null
|
||||
&& (ip = response.getQuery()) != null){
|
||||
NetworkContextHolder.setIpAddress(ip);
|
||||
log.info("本机最新公网IP地址 => {}", ip);
|
||||
}
|
||||
}, 0, configuration.getLong(KEY_TASK_REFRESH_INTERVAL_IP, 300L), TimeUnit.SECONDS);
|
||||
|
||||
// 添加进程结束钩子函数
|
||||
Runtime.getRuntime().addShutdownHook(new Thread(() -> {
|
||||
MDC.put("pid", SystemSupport.getPid());
|
||||
log.info("程序正在关闭中, 可能需要一定时间.");
|
||||
afterRefresh();
|
||||
scheduledThreadPoolExecutor.shutdown();
|
||||
log.info("已关闭.");
|
||||
}, "DDNS-ShutDownHook"));
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user