center/dataease-dm
center/dataease-dm
13
0
Fork 0
forked from github/dataease
Code Pull Requests Activity

refactor: 数据源插件完善

Browse Source
This commit is contained in:
junjun 2024-07-24 10:30:38 +08:00
parent bb60a8d9e6
commit c7d81c2dae
2 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
sdk/extensions/extensions-datasource/src/main/java/io/dataease/extensions/datasource/plugin/DataEaseDatasourcePlugin.java
View File

@ -5,7 +5,6 @@ import io.dataease.extensions.datasource.factory.ProviderFactory;
import io.dataease.extensions.datasource.provider.Provider; import io.dataease.extensions.datasource.provider.Provider;
import io.dataease.extensions.datasource.vo.XpackPluginsDatasourceVO; import io.dataease.extensions.datasource.vo.XpackPluginsDatasourceVO;
import io.dataease.license.utils.JsonUtil; import io.dataease.license.utils.JsonUtil;
import io.dataease.license.utils.LogUtil;
import io.dataease.plugins.template.DataEasePlugin; import io.dataease.plugins.template.DataEasePlugin;
import io.dataease.plugins.vo.DataEasePluginVO; import io.dataease.plugins.vo.DataEasePluginVO;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
@ -16,7 +15,6 @@ import java.io.InputStream;
import java.net.URL; import java.net.URL;
import java.net.URLClassLoader; import java.net.URLClassLoader;
import java.util.Enumeration; import java.util.Enumeration;
import java.util.Map;
import java.util.jar.JarEntry; import java.util.jar.JarEntry;
import java.util.jar.JarFile; import java.util.jar.JarFile;
@ -24,7 +22,7 @@ import java.util.jar.JarFile;
* @Author Junjun * @Author Junjun
*/ */
public abstract class DataEaseDatasourcePlugin extends Provider implements DataEasePlugin { public abstract class DataEaseDatasourcePlugin extends Provider implements DataEasePlugin {
private final String FILE_PATH = "/opt/dataease2.0/drivers"; private final String DEFAULT_FILE_PATH = "/opt/dataease2.0/drivers/plugin";
@Override @Override
public void loadPlugin() { public void loadPlugin() {
@ -38,6 +36,8 @@ public abstract class DataEaseDatasourcePlugin extends Provider implements DataE
} }
private void loadDriver() throws Exception { private void loadDriver() throws Exception {
XpackPluginsDatasourceVO config = getConfig();
String localPath = StringUtils.isEmpty(config.getDriverPath()) ? DEFAULT_FILE_PATH : config.getDriverPath();
ClassLoader classLoader = this.getClass().getClassLoader(); ClassLoader classLoader = this.getClass().getClassLoader();
URL[] urls = ((URLClassLoader) classLoader).getURLs(); URL[] urls = ((URLClassLoader) classLoader).getURLs();
String jarPath = urls[0].getPath(); String jarPath = urls[0].getPath();
@ -48,7 +48,10 @@ public abstract class DataEaseDatasourcePlugin extends Provider implements DataE
String name = entry.getName(); String name = entry.getName();
if (StringUtils.endsWith(name, ".jar")) { if (StringUtils.endsWith(name, ".jar")) {
InputStream inputStream = jarFile.getInputStream(entry); InputStream inputStream = jarFile.getInputStream(entry);
File file = new File(FILE_PATH, name.substring(name.indexOf("/") + 1)); File file = new File(localPath, name.substring(name.indexOf("/") + 1));
if (!file.getParentFile().exists()) {
file.getParentFile().mkdirs();
}
FileOutputStream outputStream = new FileOutputStream(file); FileOutputStream outputStream = new FileOutputStream(file);
byte[] bytes = new byte[1024]; byte[] bytes = new byte[1024];
int length; int length;
@ -72,8 +75,10 @@ public abstract class DataEaseDatasourcePlugin extends Provider implements DataE
return vo; return vo;
} }
/*@Override @Override
public void unloadPlugin() { public void unloadPlugin() {
XpackPluginsDatasourceVO config = getConfig();
String localPath = StringUtils.isEmpty(config.getDriverPath()) ? DEFAULT_FILE_PATH : config.getDriverPath();
try { try {
ClassLoader classLoader = this.getClass().getClassLoader(); ClassLoader classLoader = this.getClass().getClassLoader();
URL[] urls = ((URLClassLoader) classLoader).getURLs(); URL[] urls = ((URLClassLoader) classLoader).getURLs();
@ -84,12 +89,12 @@ public abstract class DataEaseDatasourcePlugin extends Provider implements DataE
JarEntry entry = (JarEntry) entries.nextElement(); JarEntry entry = (JarEntry) entries.nextElement();
String name = entry.getName(); String name = entry.getName();
if (StringUtils.endsWith(name, ".jar")) { if (StringUtils.endsWith(name, ".jar")) {
File file = new File(FILE_PATH, name.substring(name.indexOf("/") + 1)); File file = new File(localPath, name.substring(name.indexOf("/") + 1));
file.delete(); file.delete();
} }
} }
} catch (Exception e) { } catch (Exception e) {
DEException.throwException(e); DEException.throwException(e);
} }
}*/ }
} }

2
sdk/extensions/extensions-datasource/src/main/java/io/dataease/extensions/datasource/vo/XpackPluginsDatasourceVO.java
View File

@ -27,6 +27,8 @@ public class XpackPluginsDatasourceVO implements Serializable {
private String suffix; private String suffix;
private String driverPath;
private Map<String, String> staticMap; private Map<String, String> staticMap;
} }