提交 078ecb6e 编写于 作者: O o2null

Merge branch 'fix/loadCustom' into 'wrdp'

修复custom未装载到jndi中

See merge request o2oa/o2oa!1835
......@@ -100,6 +100,7 @@ public class Config {
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_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_DYNAMIC = "local/temp/dynamic";
public static final String DIR_LOCAL_TEMP_DYNAMIC_SRC = "local/temp/dynamic/src";
......@@ -320,6 +321,20 @@ public class Config {
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 {
return new File(base(), DIR_LOCAL_TEMP_DYNAMIC);
}
......
package com.x.server.console;
import java.io.PrintStream;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
......@@ -13,12 +15,13 @@ import java.util.concurrent.LinkedBlockingQueue;
import org.apache.commons.collections.MapUtils;
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.eclipse.jetty.plus.jndi.Resource;
import org.eclipse.jetty.util.RolloverFileOutputStream;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.pool.DruidDataSourceC3P0Adapter;
import com.google.gson.JsonElement;
import com.x.base.core.container.factory.SlicePropertiesBuilder;
......@@ -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.DataServer;
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.LoggerFactory;
import com.x.base.core.project.script.ScriptFactory;
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.JarTools;
import com.x.base.core.project.tools.ListTools;
import com.x.base.core.project.tools.PathTools;
import com.x.server.console.node.EventQueueExecutor;
import io.github.classgraph.ClassGraph;
......@@ -51,13 +55,12 @@ public class ResourceFactory {
}
public static void bind() throws Exception {
try (ScanResult sr = new ClassGraph()
.addClassLoader(ClassLoaderTools.urlClassLoader(true, false, true, true, true)).enableAnnotationInfo()
.scan()) {
node(sr);
containerEntities(sr);
containerEntityNames(sr);
stroageContainerEntityNames(sr);
ClassLoader cl = ClassLoaderTools.urlClassLoader(true, false, true, true, true, unzipCustomWar());
try (ScanResult sr = new ClassGraph().addClassLoader(cl).enableAnnotationInfo().scan()) {
node(cl, sr);
containerEntities(cl, sr);
containerEntityNames(cl, sr);
stroageContainerEntityNames(cl, sr);
}
if (BooleanUtils.isTrue(Config.logLevel().audit().enable())) {
auditLog();
......@@ -70,7 +73,24 @@ public class ResourceFactory {
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<>();
EventQueueExecutor eventQueueExecutor = new EventQueueExecutor(eventQueue);
eventQueueExecutor.start();
......@@ -84,6 +104,34 @@ public class ResourceFactory {
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 {
external_druid_c3p0();
}
......@@ -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 {
RolloverFileOutputStream rolloverFileOutputStream = new RolloverFileOutputStream(
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.
先完成此消息的编辑!
想要评论请 注册