package com.smtaiserver.smtaiserver.javaai.jsonflow.node;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import com.smtaiserver.smtaiserver.javaai.SMTJavaAIError;
|
import com.smtaiserver.smtaiserver.javaai.jsonflow.core.SMTJsonFlowExecArg;
|
import com.smtaiserver.smtaiserver.javaai.jsonflow.core.SMTJsonFlowManager;
|
import com.smtaiserver.smtaiserver.javaai.jsonflow.core.SMTJsonFlowNode;
|
import com.smtservlet.util.Json;
|
import com.smtservlet.util.SMTJsonWriter;
|
import com.smtservlet.util.SMTStatic;
|
|
public class SMTJsonFlowNodeUserAsk extends SMTJsonFlowNode
|
{
|
private static abstract class SMTJsonFlowNodeUserAskItem
|
{
|
abstract public void initEdge(SMTJsonFlowNode tagNode, Json jsonEdge);
|
abstract public void afterInstance() throws Exception;
|
abstract public SMTJavaAIError executeFlowNode(String title, SMTJsonFlowExecArg execArg) throws Exception;
|
}
|
|
private static class SMTJsonFlowNodeUserAskItemInput extends SMTJsonFlowNodeUserAskItem
|
{
|
private String _key;
|
private List<SMTJsonFlowNode> _listDownFlowNodes = new ArrayList<>();
|
private List<String> _listOptions = new ArrayList<>();
|
|
public SMTJsonFlowNodeUserAskItemInput(Json jsonCond)
|
{
|
Json jsonValue = Json.read(jsonCond.getJsonPath("value|value", false).asString());
|
_key = jsonValue.getJson("key").asString();
|
Json jsonOptions = jsonValue.safeGetJson("options");
|
for(Json jsonOption : jsonOptions.asJsonList())
|
{
|
_listOptions.add(jsonOption.asString());
|
}
|
}
|
|
@Override
|
public void initEdge(SMTJsonFlowNode tagNode, Json jsonEdge)
|
{
|
_listDownFlowNodes.add(tagNode);
|
}
|
|
@Override
|
public void afterInstance() throws Exception
|
{
|
if(_listDownFlowNodes.size() == 0)
|
throw new Exception("can't find any options downNode in input mode");
|
}
|
|
@Override
|
public SMTJavaAIError executeFlowNode(String title, SMTJsonFlowExecArg execArg) throws Exception
|
{
|
// 上前端询问
|
String replyId = SMTStatic.newUUID();
|
SMTJsonWriter jsonWr = new SMTJsonWriter(false);
|
jsonWr.addKeyValue("title", "请输入对地图的操作");
|
jsonWr.addKeyValue("type", "input-select");
|
jsonWr.addKeyValue("reply_id", replyId);
|
jsonWr.beginArray("options");
|
{
|
for(String option : _listOptions)
|
{
|
jsonWr.addKeyValue(null, option);
|
}
|
}
|
jsonWr.endArray();
|
|
// 等待应答
|
Json jsonReply = execArg._tranReq.sendReplyChunkedBlock(replyId, jsonWr.getRootJson());
|
if(jsonReply == null)
|
return new SMTJavaAIError("流程被用户中断");
|
|
// 将用户输入对象设置为当前对象
|
String value = jsonReply.getJson("select").asString();
|
|
execArg._jsonArgs.set(_key, value);
|
|
|
// 将所有子节点添加作为下一轮节点
|
for(SMTJsonFlowNode flowNode : _listDownFlowNodes)
|
{
|
execArg._stackNodeExec.addLast(flowNode.createFlowNodeExec());
|
}
|
|
return null;
|
}
|
|
}
|
|
private static class SMTJsonFlowNodeUserAskItemSelect extends SMTJsonFlowNodeUserAskItem
|
{
|
private static class SMTJsonFlowNodeUserAskItemOption
|
{
|
public String _title;
|
public String _sourceHandle;
|
public List<SMTJsonFlowNode> _listDownNodes = new ArrayList<>();
|
|
public SMTJsonFlowNodeUserAskItemOption(Json jsonOption)
|
{
|
_title = jsonOption.getJson("label").asString();
|
_sourceHandle = jsonOption.getJson("id").asString();
|
}
|
|
public void initEdge(SMTJsonFlowNode tagNode, String sourceHandle)
|
{
|
if(_sourceHandle.equals(sourceHandle))
|
_listDownNodes.add(tagNode);
|
}
|
|
public void afterInstance() throws Exception
|
{
|
if(_listDownNodes.size() == 0)
|
throw new Exception("can't find any options downNode in select mode");
|
}
|
}
|
|
//////////////////////////////////////////////////////////////////////////////////////
|
|
private List<SMTJsonFlowNodeUserAskItemOption> _listSelectItem = new ArrayList<>();
|
|
public SMTJsonFlowNodeUserAskItemSelect(Json jsonCond)
|
{
|
for(Json jsonOption : jsonCond.getJson("options").asJsonList())
|
{
|
SMTJsonFlowNodeUserAskItemOption option = new SMTJsonFlowNodeUserAskItemOption(jsonOption);
|
_listSelectItem.add(option);
|
}
|
}
|
|
@Override
|
public void initEdge(SMTJsonFlowNode tagNode, Json jsonEdge)
|
{
|
String sourceHandle = jsonEdge.safeGetStr("sourceHandle", "");
|
|
for(SMTJsonFlowNodeUserAskItemOption option : _listSelectItem)
|
{
|
option.initEdge(tagNode, sourceHandle);
|
}
|
}
|
|
@Override
|
public void afterInstance() throws Exception
|
{
|
if(_listSelectItem.size() == 0)
|
throw new Exception("can't find any options in select mode");
|
|
for(SMTJsonFlowNodeUserAskItemOption option : _listSelectItem)
|
{
|
option.afterInstance();
|
}
|
}
|
|
@Override
|
public SMTJavaAIError executeFlowNode(String title, SMTJsonFlowExecArg execArg) throws Exception
|
{
|
// 初始化返回前端的参数
|
String replyId = SMTStatic.newUUID();
|
SMTJsonWriter jsonWr = new SMTJsonWriter(false);
|
jsonWr.addKeyValue("title", title);
|
jsonWr.addKeyValue("type", "select");
|
jsonWr.addKeyValue("reply_id", replyId);
|
jsonWr.beginArray("options");
|
for(SMTJsonFlowNodeUserAskItemOption option : _listSelectItem)
|
{
|
jsonWr.addKeyValue(null, option._title);
|
}
|
jsonWr.endArray();
|
|
// 等待前端响应
|
Json jsonReply = execArg._tranReq.sendReplyChunkedBlock(replyId, jsonWr.getRootJson());
|
if(jsonReply == null)
|
return new SMTJavaAIError("流程被用户中断");
|
|
// 根据前端响应做出分流选择
|
String selectOpt = jsonReply.safeGetStr("select", "");
|
for(SMTJsonFlowNodeUserAskItemOption option : _listSelectItem)
|
{
|
if(option._title.equals(selectOpt))
|
{
|
for(SMTJsonFlowNode flowNode : option._listDownNodes)
|
{
|
execArg._stackNodeExec.addLast(flowNode.createFlowNodeExec());
|
}
|
|
return null;
|
}
|
}
|
|
|
return new SMTJavaAIError("流程无法判定");
|
}
|
}
|
|
/////////////////////////////////////////////////////////////////////////////////////
|
|
private String _title;
|
private SMTJsonFlowNodeUserAskItem _askItem;
|
|
@Override
|
public void initInstane(SMTJsonFlowManager manager, Json jsonNode) throws Exception
|
{
|
super.initInstane(manager, jsonNode);
|
|
Json jsonConds = jsonNode.getJsonPath("data|group_params|0|params|", false);
|
|
for(Json jsonCond : jsonConds.asJsonList())
|
{
|
String key = jsonCond.getJson("key").asString();
|
|
// 获取全局提示信息
|
if("output_msg".equals(key))
|
{
|
Json jsonTitle = jsonCond.getJsonPath("value|msg", false);
|
_title = jsonTitle.asString();
|
}
|
// 获取列表类型配置
|
else if("output_result".equals(key))
|
{
|
String resultType = jsonCond.getJsonPath("value|type", false).asString();
|
if("select".equals(resultType))
|
{
|
_askItem = new SMTJsonFlowNodeUserAskItemSelect(jsonCond);
|
}
|
else if("input".equals(resultType))
|
{
|
_askItem = new SMTJsonFlowNodeUserAskItemInput(jsonCond);
|
}
|
else
|
{
|
throw new Exception("unkonwn output_result type : " + resultType);
|
}
|
}
|
else
|
{
|
throw new Exception("unkonwn output_msg key : " + key);
|
}
|
}
|
|
}
|
|
@Override
|
public void initEdge(SMTJsonFlowNode tagNode, Json jsonEdge)
|
{
|
super.initEdge(tagNode, jsonEdge);
|
|
_askItem.initEdge(tagNode, jsonEdge);
|
}
|
|
public void afterInstance() throws Exception
|
{
|
super.afterInstance();
|
|
_askItem.afterInstance();
|
|
}
|
|
@Override
|
public SMTJavaAIError executeFlowNode(SMTJsonFlowExecArg execArg) throws Exception
|
{
|
SMTJavaAIError error = _askItem.executeFlowNode(_title, execArg);
|
return error;
|
}
|
|
}
|
