org.dromara.jpom.socket.handler.NodeScriptHandler Maven / Gradle / Ivy
/*
* The MIT License (MIT)
*
* Copyright (c) 2019 Code Technology Studio
*
* Permission is hereby granted, free of charge, to any person obtaining a copy of
* this software and associated documentation files (the "Software"), to deal in
* the Software without restriction, including without limitation the rights to
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
* the Software, and to permit persons to whom the Software is furnished to do so,
* subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
* IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
package org.dromara.jpom.socket.handler;
import cn.hutool.extra.spring.SpringUtil;
import com.alibaba.fastjson2.JSONObject;
import org.dromara.jpom.common.BaseServerController;
import org.dromara.jpom.common.Const;
import org.dromara.jpom.common.forward.NodeUrl;
import org.dromara.jpom.model.data.NodeModel;
import org.dromara.jpom.model.node.NodeScriptCacheModel;
import org.dromara.jpom.model.node.NodeScriptExecuteLogCacheModel;
import org.dromara.jpom.model.user.UserModel;
import org.dromara.jpom.permission.ClassFeature;
import org.dromara.jpom.permission.Feature;
import org.dromara.jpom.permission.MethodFeature;
import org.dromara.jpom.service.node.script.NodeScriptExecuteLogServer;
import org.dromara.jpom.service.node.script.NodeScriptServer;
import org.dromara.jpom.socket.BaseProxyHandler;
import org.dromara.jpom.socket.ConsoleCommandOp;
import org.dromara.jpom.transport.IProxyWebSocket;
import java.io.IOException;
import java.util.Map;
/**
* 脚本模板消息控制器
*
* @author bwcx_jzy
* @since 2019/4/24
*/
@Feature(cls = ClassFeature.NODE_SCRIPT, method = MethodFeature.EXECUTE)
public class NodeScriptHandler extends BaseProxyHandler {
public NodeScriptHandler() {
super(NodeUrl.Script_Run);
}
@Override
protected Object[] getParameters(Map attributes) {
NodeScriptCacheModel scriptModel = (NodeScriptCacheModel) attributes.get("dataItem");
return new Object[]{"id", attributes.get("scriptId"), "workspaceId", scriptModel.getWorkspaceId()};
}
@Override
protected String handleTextMessage(Map attributes, IProxyWebSocket proxySession, JSONObject json, ConsoleCommandOp consoleCommandOp) throws IOException {
if (consoleCommandOp != ConsoleCommandOp.heart) {
super.logOpt(this.getClass(), attributes, json);
}
if (consoleCommandOp == ConsoleCommandOp.start) {
// 开始执行
String executeId = this.createLog(attributes);
json.put(Const.SOCKET_MSG_TAG, Const.SOCKET_MSG_TAG);
json.put("executeId", executeId);
}
proxySession.send(json.toString());
return null;
}
/**
* 创建执行日志
*
* @param attributes 参数属性
* @return 执行ID
*/
private String createLog(Map attributes) {
NodeModel nodeInfo = (NodeModel) attributes.get("nodeInfo");
UserModel userModel = (UserModel) attributes.get("userInfo");
NodeScriptCacheModel dataItem = (NodeScriptCacheModel) attributes.get("dataItem");
NodeScriptExecuteLogServer logServer = SpringUtil.getBean(NodeScriptExecuteLogServer.class);
NodeScriptServer nodeScriptServer = SpringUtil.getBean(NodeScriptServer.class);
//
try {
BaseServerController.resetInfo(userModel);
//
NodeScriptCacheModel nodeScriptCacheModel = new NodeScriptCacheModel();
nodeScriptCacheModel.setId(dataItem.getId());
nodeScriptCacheModel.setLastRunUser(userModel.getId());
nodeScriptServer.updateById(nodeScriptCacheModel);
//
NodeScriptExecuteLogCacheModel nodeScriptExecuteLogCacheModel = new NodeScriptExecuteLogCacheModel();
nodeScriptExecuteLogCacheModel.setScriptId(dataItem.getScriptId());
nodeScriptExecuteLogCacheModel.setNodeId(nodeInfo.getId());
nodeScriptExecuteLogCacheModel.setScriptName(dataItem.getName());
nodeScriptExecuteLogCacheModel.setTriggerExecType(0);
nodeScriptExecuteLogCacheModel.setWorkspaceId(nodeInfo.getWorkspaceId());
logServer.insert(nodeScriptExecuteLogCacheModel);
return nodeScriptExecuteLogCacheModel.getId();
} finally {
BaseServerController.removeAll();
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy