269 lines
9.5 KiB
Java
269 lines
9.5 KiB
Java
package io.dataease.service.sys;
|
||
|
||
import cn.hutool.core.io.FileUtil;
|
||
import com.google.gson.Gson;
|
||
import io.dataease.base.domain.MyPlugin;
|
||
import io.dataease.base.mapper.MyPluginMapper;
|
||
import io.dataease.base.mapper.ext.ExtSysPluginMapper;
|
||
import io.dataease.base.mapper.ext.query.GridExample;
|
||
import io.dataease.commons.constants.AuthConstants;
|
||
import io.dataease.commons.exception.DEException;
|
||
import io.dataease.commons.utils.CodingUtil;
|
||
import io.dataease.commons.utils.DeFileUtils;
|
||
import io.dataease.commons.utils.LogUtil;
|
||
import io.dataease.commons.utils.ZipUtils;
|
||
import io.dataease.controller.sys.base.BaseGridRequest;
|
||
import io.dataease.listener.util.CacheUtils;
|
||
import io.dataease.plugins.config.LoadjarUtil;
|
||
import org.apache.commons.lang3.ArrayUtils;
|
||
import org.apache.commons.lang3.ObjectUtils;
|
||
import org.apache.commons.lang3.StringUtils;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.beans.factory.annotation.Value;
|
||
import org.springframework.stereotype.Service;
|
||
import org.springframework.web.multipart.MultipartFile;
|
||
|
||
import javax.annotation.Resource;
|
||
import java.io.File;
|
||
import java.io.IOException;
|
||
import java.lang.reflect.InvocationTargetException;
|
||
import java.util.Arrays;
|
||
import java.util.List;
|
||
import java.util.Map;
|
||
import java.util.stream.Collectors;
|
||
|
||
|
||
@Service
|
||
public class PluginService {
|
||
|
||
@Value("${dataease.plugin.dir:/opt/dataease/plugins/}")
|
||
private String pluginDir;
|
||
|
||
private final static String pluginJsonName = "plugin.json";
|
||
|
||
@Resource
|
||
private ExtSysPluginMapper extSysPluginMapper;
|
||
|
||
@Resource
|
||
private MyPluginMapper myPluginMapper;
|
||
|
||
@Autowired
|
||
private LoadjarUtil loadjarUtil;
|
||
|
||
@Value("${version}")
|
||
private String version;
|
||
|
||
|
||
public List<MyPlugin> query(BaseGridRequest request) {
|
||
GridExample gridExample = request.convertExample();
|
||
return extSysPluginMapper.query(gridExample);
|
||
}
|
||
|
||
/**
|
||
* 从本地安装处插件
|
||
*
|
||
* @param file
|
||
* @return
|
||
*/
|
||
public Map<String, Object> localInstall(MultipartFile file) {
|
||
//1.上传文件到服务器pluginDir目录下
|
||
File dest = DeFileUtils.upload(file, pluginDir + "temp/");
|
||
//2.解压目标文件dest 得到plugin.json和jar
|
||
String folder = pluginDir + "folder/";
|
||
try {
|
||
ZipUtils.upZipFile(dest, folder);
|
||
} catch (IOException e) {
|
||
DeFileUtils.deleteFile(pluginDir + "temp/");
|
||
DeFileUtils.deleteFile(folder);
|
||
// 需要删除文件
|
||
// e.printStackTrace();
|
||
LogUtil.error(e.getMessage(), e);
|
||
DEException.throwException(e);
|
||
}
|
||
//3.解析plugin.json 失败则 直接返回错误 删除文件
|
||
File folderFile = new File(folder);
|
||
File[] jsonFiles = folderFile.listFiles(this::isPluginJson);
|
||
if (ArrayUtils.isEmpty(jsonFiles)) {
|
||
DeFileUtils.deleteFile(pluginDir + "temp/");
|
||
DeFileUtils.deleteFile(folder);
|
||
String msg = "缺少插件描述文件【plugin.json】";
|
||
LogUtil.error(msg);
|
||
DEException.throwException(new RuntimeException(msg));
|
||
}
|
||
MyPlugin myPlugin = formatJsonFile(jsonFiles[0]);
|
||
|
||
if (!versionMatch(myPlugin.getRequire())) {
|
||
String msg = "当前插件要求系统版本最低为：" + myPlugin.getRequire();
|
||
LogUtil.error(msg);
|
||
DEException.throwException(new RuntimeException(msg));
|
||
}
|
||
//4.加载jar包 失败则 直接返回错误 删除文件
|
||
File[] jarFiles = folderFile.listFiles(this::isPluginJar);
|
||
if (ArrayUtils.isEmpty(jarFiles)) {
|
||
DeFileUtils.deleteFile(pluginDir + "temp/");
|
||
DeFileUtils.deleteFile(folder);
|
||
String msg = "缺少插件jar文件";
|
||
LogUtil.error(msg);
|
||
DEException.throwException(new RuntimeException(msg));
|
||
}
|
||
|
||
if (pluginExist(myPlugin)) {
|
||
String msg = "插件【"+myPlugin.getName()+"】已存在，请先卸载";
|
||
LogUtil.error(msg);
|
||
DEException.throwException(new RuntimeException(msg));
|
||
}
|
||
String targetDir = null;
|
||
try {
|
||
File jarFile = jarFiles[0];
|
||
targetDir = makeTargetDir(myPlugin);
|
||
String jarPath;
|
||
jarPath = DeFileUtils.copy(jarFile, targetDir);
|
||
loadJar(jarPath, myPlugin);
|
||
myPluginMapper.insert(myPlugin);
|
||
|
||
CacheUtils.removeAll(AuthConstants.USER_CACHE_NAME);
|
||
CacheUtils.removeAll(AuthConstants.USER_ROLE_CACHE_NAME);
|
||
CacheUtils.removeAll(AuthConstants.USER_PERMISSION_CACHE_NAME);
|
||
} catch (Exception e) {
|
||
if (StringUtils.isNotEmpty(targetDir)) {
|
||
DeFileUtils.deleteFile(targetDir);
|
||
}
|
||
LogUtil.error(e.getMessage(), e);
|
||
DEException.throwException(e);
|
||
} finally {
|
||
DeFileUtils.deleteFile(pluginDir + "temp/");
|
||
DeFileUtils.deleteFile(folder);
|
||
}
|
||
return null;
|
||
}
|
||
|
||
public void loadJar(String jarPath, MyPlugin myPlugin) throws Exception {
|
||
loadjarUtil.loadJar(jarPath, myPlugin);
|
||
}
|
||
|
||
private String makeTargetDir(MyPlugin myPlugin) {
|
||
String store = myPlugin.getStore();
|
||
String dir = pluginDir + store + "/";
|
||
File fileDir = new File(dir);
|
||
if (!fileDir.exists()) {
|
||
fileDir.mkdirs();
|
||
}
|
||
return dir;
|
||
}
|
||
|
||
/**
|
||
* 检测插件是否已存在
|
||
* @param myPlugin
|
||
* @return
|
||
*/
|
||
public boolean pluginExist(MyPlugin myPlugin) {
|
||
GridExample gridExample = new GridExample();
|
||
List<MyPlugin> plugins = extSysPluginMapper.query(gridExample);
|
||
return plugins.stream().anyMatch(plugin -> {
|
||
return StringUtils.equals(myPlugin.getName(), plugin.getName()) || StringUtils.equals(myPlugin.getModuleName(), plugin.getModuleName());
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 卸载插件
|
||
*
|
||
* @param pluginId
|
||
* @return
|
||
*/
|
||
public Boolean uninstall(Long pluginId) {
|
||
MyPlugin myPlugin = myPluginMapper.selectByPrimaryKey(pluginId);
|
||
if (ObjectUtils.isEmpty(myPlugin)) {
|
||
String msg = "当前插件不存在";
|
||
LogUtil.error(msg);
|
||
DEException.throwException(new RuntimeException(msg));
|
||
}
|
||
deleteJarFile(myPlugin);
|
||
CacheUtils.removeAll(AuthConstants.USER_CACHE_NAME);
|
||
CacheUtils.removeAll(AuthConstants.USER_ROLE_CACHE_NAME);
|
||
CacheUtils.removeAll(AuthConstants.USER_PERMISSION_CACHE_NAME);
|
||
myPluginMapper.deleteByPrimaryKey(pluginId);
|
||
return true;
|
||
}
|
||
|
||
private void deleteJarFile(MyPlugin plugin) {
|
||
String version = plugin.getVersion();
|
||
String moduleName = plugin.getModuleName();
|
||
String fileName = moduleName + "-" + version + ".jar";
|
||
String path = pluginDir + plugin.getStore() + "/" + fileName;
|
||
File jarFile = new File(path);
|
||
FileUtil.del(jarFile);
|
||
}
|
||
|
||
/**
|
||
* 改变插件状态
|
||
*
|
||
* @param pluginId
|
||
* @param status true ？ 使用状态 : 禁用状态
|
||
* @return
|
||
*/
|
||
public Boolean changeStatus(Long pluginId, Boolean status) {
|
||
CacheUtils.removeAll(AuthConstants.USER_CACHE_NAME);
|
||
CacheUtils.removeAll(AuthConstants.USER_ROLE_CACHE_NAME);
|
||
CacheUtils.removeAll(AuthConstants.USER_PERMISSION_CACHE_NAME);
|
||
return false;
|
||
}
|
||
|
||
//判断当前文件是否实插件描述文件
|
||
//文件名称必须plugin.json
|
||
private boolean isPluginJson(File file) {
|
||
return StringUtils.equals(file.getName(), pluginJsonName);
|
||
}
|
||
|
||
private boolean isPluginJar(File file) {
|
||
String name = file.getName();
|
||
return StringUtils.equals(DeFileUtils.getExtensionName(name), "jar");
|
||
}
|
||
|
||
/**
|
||
* 从plugin.json文件反序列化为MyPlugin实例对象
|
||
*
|
||
* @return
|
||
*/
|
||
private MyPlugin formatJsonFile(File file) {
|
||
String str = DeFileUtils.readJson(file);
|
||
Gson gson = new Gson();
|
||
Map<String, Object> myPlugin = gson.fromJson(str, Map.class);
|
||
myPlugin.put("free", (Double) myPlugin.get("free") > 0.0);
|
||
myPlugin.put("loadMybatis", (Double) myPlugin.get("loadMybatis") > 0.0);
|
||
MyPlugin result = new MyPlugin();
|
||
try {
|
||
org.apache.commons.beanutils.BeanUtils.populate(result, myPlugin);
|
||
result.setInstallTime(System.currentTimeMillis());
|
||
} catch (IllegalAccessException e) {
|
||
e.printStackTrace();
|
||
} catch (InvocationTargetException e) {
|
||
e.printStackTrace();
|
||
}
|
||
//BeanUtils.copyBean(result, myPlugin);
|
||
return result;
|
||
}
|
||
|
||
/**
|
||
* 从插件商城远程安装插件
|
||
* 2.0版本实现
|
||
*
|
||
* @param params
|
||
* @return
|
||
*/
|
||
public Map<String, Object> remoteInstall(Map<String, Object> params) {
|
||
return null;
|
||
}
|
||
|
||
public boolean versionMatch(String pluginVersion) {
|
||
List<Integer> versionLists = Arrays.stream(version.split(".")).map(CodingUtil::string2Integer).collect(Collectors.toList());
|
||
List<Integer> requireVersionLists = Arrays.stream(pluginVersion.split(".")).map(CodingUtil::string2Integer).collect(Collectors.toList());
|
||
int maxSize = Math.max(versionLists.size(), requireVersionLists.size());
|
||
for (int i = 0; i < maxSize; i++) {
|
||
Integer currentV = versionLists.size() == i ? 0 : versionLists.get(i);
|
||
Integer requireV = requireVersionLists.size() == i ? 0 : requireVersionLists.get(i);
|
||
if (requireV > currentV) return false;
|
||
}
|
||
return false;
|
||
}
|
||
}
|