提交 21efe630 编写于 作者: Z zhourui

修复custom未装载到jndi中

上级 3a6187dc
...@@ -100,6 +100,7 @@ public class Config { ...@@ -100,6 +100,7 @@ public class Config {
public static final String DIR_LOCAL_UPDATE = "local/update"; public static final String DIR_LOCAL_UPDATE = "local/update";
public static final String DIR_LOCAL_TEMP = "local/temp"; public static final String DIR_LOCAL_TEMP = "local/temp";
public static final String DIR_LOCAL_TEMP_CLASSES = "local/temp/classes"; public static final String DIR_LOCAL_TEMP_CLASSES = "local/temp/classes";
public static final String DIR_LOCAL_TEMP_CUSTOM = "local/temp/custom";
public static final String DIR_LOCAL_TEMP_SQL = "local/temp/sql"; public static final String DIR_LOCAL_TEMP_SQL = "local/temp/sql";
public static final String DIR_LOCAL_TEMP_DYNAMIC = "local/temp/dynamic"; public static final String DIR_LOCAL_TEMP_DYNAMIC = "local/temp/dynamic";
public static final String DIR_LOCAL_TEMP_DYNAMIC_SRC = "local/temp/dynamic/src"; public static final String DIR_LOCAL_TEMP_DYNAMIC_SRC = "local/temp/dynamic/src";
...@@ -320,6 +321,20 @@ public class Config { ...@@ -320,6 +321,20 @@ public class Config {
return new File(base(), DIR_LOCAL_TEMP_CLASSES); return new File(base(), DIR_LOCAL_TEMP_CLASSES);
} }
public static File dir_local_temp_custom() throws Exception {
return new File(base(), DIR_LOCAL_TEMP_CUSTOM);
}
public static File dir_local_temp_custom(Boolean force) throws Exception {
File dir = new File(base(), DIR_LOCAL_TEMP_CUSTOM);
if (force) {
if ((!dir.exists()) || dir.isFile()) {
FileUtils.forceMkdir(dir);
}
}
return dir;
}
public static File dir_local_temp_dynamic() throws Exception { public static File dir_local_temp_dynamic() throws Exception {
return new File(base(), DIR_LOCAL_TEMP_DYNAMIC); return new File(base(), DIR_LOCAL_TEMP_DYNAMIC);
} }
......
package com.x.server.console; package com.x.server.console;
import java.io.PrintStream; import java.io.PrintStream;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
...@@ -13,12 +15,13 @@ import java.util.concurrent.LinkedBlockingQueue; ...@@ -13,12 +15,13 @@ import java.util.concurrent.LinkedBlockingQueue;
import org.apache.commons.collections.MapUtils; import org.apache.commons.collections.MapUtils;
import org.apache.commons.collections4.ListUtils; import org.apache.commons.collections4.ListUtils;
import org.apache.commons.dbcp2.BasicDataSource; import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;
import org.apache.commons.io.filefilter.WildcardFileFilter;
import org.apache.commons.lang3.BooleanUtils; import org.apache.commons.lang3.BooleanUtils;
import org.eclipse.jetty.plus.jndi.Resource; import org.eclipse.jetty.plus.jndi.Resource;
import org.eclipse.jetty.util.RolloverFileOutputStream; import org.eclipse.jetty.util.RolloverFileOutputStream;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.pool.DruidDataSourceC3P0Adapter; import com.alibaba.druid.pool.DruidDataSourceC3P0Adapter;
import com.google.gson.JsonElement; import com.google.gson.JsonElement;
import com.x.base.core.container.factory.SlicePropertiesBuilder; import com.x.base.core.container.factory.SlicePropertiesBuilder;
...@@ -29,13 +32,14 @@ import com.x.base.core.project.config.CenterServer; ...@@ -29,13 +32,14 @@ import com.x.base.core.project.config.CenterServer;
import com.x.base.core.project.config.Config; import com.x.base.core.project.config.Config;
import com.x.base.core.project.config.DataServer; import com.x.base.core.project.config.DataServer;
import com.x.base.core.project.config.ExternalDataSource; import com.x.base.core.project.config.ExternalDataSource;
import com.x.base.core.project.gson.XGsonBuilder;
import com.x.base.core.project.logger.Logger; import com.x.base.core.project.logger.Logger;
import com.x.base.core.project.logger.LoggerFactory; import com.x.base.core.project.logger.LoggerFactory;
import com.x.base.core.project.script.ScriptFactory;
import com.x.base.core.project.tools.ClassLoaderTools; import com.x.base.core.project.tools.ClassLoaderTools;
import com.x.base.core.project.tools.Crypto;
import com.x.base.core.project.tools.DefaultCharset; import com.x.base.core.project.tools.DefaultCharset;
import com.x.base.core.project.tools.JarTools;
import com.x.base.core.project.tools.ListTools; import com.x.base.core.project.tools.ListTools;
import com.x.base.core.project.tools.PathTools;
import com.x.server.console.node.EventQueueExecutor; import com.x.server.console.node.EventQueueExecutor;
import io.github.classgraph.ClassGraph; import io.github.classgraph.ClassGraph;
...@@ -51,13 +55,12 @@ public class ResourceFactory { ...@@ -51,13 +55,12 @@ public class ResourceFactory {
} }
public static void bind() throws Exception { public static void bind() throws Exception {
try (ScanResult sr = new ClassGraph() ClassLoader cl = ClassLoaderTools.urlClassLoader(true, false, true, true, true, unzipCustomWar());
.addClassLoader(ClassLoaderTools.urlClassLoader(true, false, true, true, true)).enableAnnotationInfo() try (ScanResult sr = new ClassGraph().addClassLoader(cl).enableAnnotationInfo().scan()) {
.scan()) { node(cl, sr);
node(sr); containerEntities(cl, sr);
containerEntities(sr); containerEntityNames(cl, sr);
containerEntityNames(sr); stroageContainerEntityNames(cl, sr);
stroageContainerEntityNames(sr);
} }
if (BooleanUtils.isTrue(Config.logLevel().audit().enable())) { if (BooleanUtils.isTrue(Config.logLevel().audit().enable())) {
auditLog(); auditLog();
...@@ -70,7 +73,24 @@ public class ResourceFactory { ...@@ -70,7 +73,24 @@ public class ResourceFactory {
processPlatformExecutors(); processPlatformExecutors();
} }
private static void node(ScanResult sr) throws Exception { private static Path[] unzipCustomWar() throws Exception {
FileUtils.cleanDirectory(Config.dir_local_temp_custom(true));
List<String> list = new ArrayList<>();
for (String str : Config.dir_custom(true).list(new WildcardFileFilter("*" + PathTools.DOT_WAR))) {
list.add(FilenameUtils.getBaseName(str));
}
list = ListTools.includesExcludesWildcard(list, Config.currentNode().getApplication().getIncludes(),
Config.currentNode().getApplication().getExcludes());
List<Path> paths = new ArrayList<>();
for (String str : list) {
Path path = Paths.get(Config.dir_custom().toString(), str + PathTools.DOT_WAR);
JarTools.unjar(path, "", Config.dir_local_temp_custom().toPath().resolve(str), true);
paths.add(Config.dir_local_temp_custom().toPath().resolve(str).resolve(PathTools.WEB_INF_CLASSES));
}
return paths.toArray(new Path[paths.size()]);
}
private static void node(ClassLoader classLoader, ScanResult sr) throws Exception {
LinkedBlockingQueue<JsonElement> eventQueue = new LinkedBlockingQueue<>(); LinkedBlockingQueue<JsonElement> eventQueue = new LinkedBlockingQueue<>();
EventQueueExecutor eventQueueExecutor = new EventQueueExecutor(eventQueue); EventQueueExecutor eventQueueExecutor = new EventQueueExecutor(eventQueue);
eventQueueExecutor.start(); eventQueueExecutor.start();
...@@ -84,6 +104,34 @@ public class ResourceFactory { ...@@ -84,6 +104,34 @@ public class ResourceFactory {
new Resource(Config.RESOURCE_NODE_CENTERSPRIMARYSSLENABLE, entry.getValue().getSslEnable()); new Resource(Config.RESOURCE_NODE_CENTERSPRIMARYSSLENABLE, entry.getValue().getSslEnable());
} }
private static void containerEntityNames(ClassLoader classLoader, ScanResult sr) throws Exception {
List<String> list = new ArrayList<>();
for (ClassInfo info : sr.getClassesWithAnnotation(ContainerEntity.class.getName())) {
list.add(info.getName());
}
list = ListTools.trim(list, true, true);
new Resource(Config.RESOURCE_CONTAINERENTITYNAMES, ListUtils.unmodifiableList(list));
}
private static void stroageContainerEntityNames(ClassLoader classLoader, ScanResult sr) throws Exception {
List<String> list = new ArrayList<>();
for (ClassInfo info : sr.getClassesWithAnnotation(Storage.class.getName())) {
list.add(info.getName());
}
list = ListTools.trim(list, true, true);
new Resource(Config.RESOURCE_STORAGECONTAINERENTITYNAMES, ListUtils.unmodifiableList(list));
}
private static void containerEntities(ClassLoader classLoader, ScanResult sr) throws Exception {
Map<String, List<String>> map = new TreeMap<>();
for (ClassInfo info : sr.getClassesWithAnnotation(Module.class.getName())) {
Class<?> cls = classLoader.loadClass(info.getName());
List<String> os = ListTools.toList(cls.getAnnotation(Module.class).containerEntities());
map.put(info.getName(), ListUtils.unmodifiableList(os));
}
new Resource(Config.RESOURCE_CONTAINERENTITIES, MapUtils.unmodifiableMap(map));
}
private static void external() throws Exception { private static void external() throws Exception {
external_druid_c3p0(); external_druid_c3p0();
} }
...@@ -146,34 +194,6 @@ public class ResourceFactory { ...@@ -146,34 +194,6 @@ public class ResourceFactory {
} }
} }
private static void containerEntityNames(ScanResult sr) throws Exception {
List<String> list = new ArrayList<>();
for (ClassInfo info : sr.getClassesWithAnnotation(ContainerEntity.class.getName())) {
list.add(info.getName());
}
list = ListTools.trim(list, true, true);
new Resource(Config.RESOURCE_CONTAINERENTITYNAMES, ListUtils.unmodifiableList(list));
}
private static void stroageContainerEntityNames(ScanResult sr) throws Exception {
List<String> list = new ArrayList<>();
for (ClassInfo info : sr.getClassesWithAnnotation(Storage.class.getName())) {
list.add(info.getName());
}
list = ListTools.trim(list, true, true);
new Resource(Config.RESOURCE_STORAGECONTAINERENTITYNAMES, ListUtils.unmodifiableList(list));
}
private static void containerEntities(ScanResult sr) throws Exception {
Map<String, List<String>> map = new TreeMap<>();
for (ClassInfo info : sr.getClassesWithAnnotation(Module.class.getName())) {
Class<?> cls = Class.forName(info.getName());
List<String> os = ListTools.toList(cls.getAnnotation(Module.class).containerEntities());
map.put(info.getName(), ListUtils.unmodifiableList(os));
}
new Resource(Config.RESOURCE_CONTAINERENTITIES, MapUtils.unmodifiableMap(map));
}
private static void auditLog() throws Exception { private static void auditLog() throws Exception {
RolloverFileOutputStream rolloverFileOutputStream = new RolloverFileOutputStream( RolloverFileOutputStream rolloverFileOutputStream = new RolloverFileOutputStream(
Config.dir_logs(true).getAbsolutePath() + "/yyyy_mm_dd.audit.log", true, Config.dir_logs(true).getAbsolutePath() + "/yyyy_mm_dd.audit.log", true,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册