package com.smtaiserver.smtaiserver.javaai.qwen.extcall;
|
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
import org.dom4j.Document;
|
import org.dom4j.Element;
|
import org.dom4j.Node;
|
|
import com.smtaiserver.smtaiserver.core.SMTAIServerRequest;
|
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecord;
|
import com.smtaiserver.smtaiserver.javaai.SMTJavaAIError;
|
import com.smtaiserver.smtaiserver.javaai.SMTJavaAISession;
|
import com.smtservlet.util.Json;
|
import com.smtservlet.util.SMTJsonWriter;
|
import com.smtservlet.util.SMTStatic;
|
|
public abstract class SMTQwenExtCall
|
{
|
protected static abstract class ExtCallArg
|
{
|
protected String _name;
|
protected String _default;
|
|
public abstract void loadArgToMap(Json callExtJson, SMTAIServerRequest tranReq, Map<String, Object> r_mapId2Value) throws Exception;
|
|
public void initInstance(Element xmlArg) throws Exception
|
{
|
_name = SMTStatic.getXmlAttr(xmlArg, "name");
|
_default = SMTStatic.getXmlAttr(xmlArg, "default", null);
|
}
|
|
public String getArgId()
|
{
|
return _name;
|
}
|
}
|
|
protected static class ExtCallArgRequest extends ExtCallArg
|
{
|
@Override
|
public void loadArgToMap(Json callExtJson, SMTAIServerRequest tranReq, Map<String, Object> r_mapId2Value) throws Exception
|
{
|
String value = callExtJson.safeGetStr(_name, null);
|
if(value == null)
|
{
|
if(_default == null)
|
throw new Exception("argument is not seted : " + _name);
|
|
value = _default;
|
}
|
|
r_mapId2Value.put(_name, value);
|
}
|
|
}
|
|
protected static class ExtCallArgCalcBaseTime extends ExtCallArg
|
{
|
private String _base_start_time;
|
private String _base_end_time;
|
private String _comp_start_time;
|
|
public void initInstance(Element xmlArg) throws Exception
|
{
|
super.initInstance(xmlArg);
|
_base_start_time = SMTStatic.getXmlAttr(xmlArg, "base_start_time");
|
_base_end_time = SMTStatic.getXmlAttr(xmlArg, "base_end_time");
|
_comp_start_time = SMTStatic.getXmlAttr(xmlArg, "comp_start_time");
|
}
|
|
@Override
|
public void loadArgToMap(Json callExtJson, SMTAIServerRequest tranReq, Map<String, Object> r_mapId2Value)
|
{
|
Date dateBaseStartTime = SMTStatic.toDate(r_mapId2Value.get(_base_start_time));
|
Date dateBaseEndTime = SMTStatic.toDate(r_mapId2Value.get(_base_end_time));
|
Date dateCompStartTime = SMTStatic.toDate(r_mapId2Value.get(_comp_start_time));
|
|
long subSecond = (dateBaseEndTime.getTime() - dateBaseStartTime.getTime()) / 1000;
|
String value = SMTStatic.toString(SMTStatic.calculateTime(dateCompStartTime, SMTStatic.SMTCalcTime.ADD_SECOND, (int)subSecond));
|
|
r_mapId2Value.put(_name, value);
|
}
|
}
|
|
protected static class ExtCallArgTodayCalc extends ExtCallArg
|
{
|
private SMTStatic.SMTCalcTime _calcOP;
|
private int _calcValue;
|
|
@Override
|
public void initInstance(Element xmlArg) throws Exception
|
{
|
super.initInstance(xmlArg);
|
|
_calcOP = SMTStatic.SMTCalcTime.valueOf(SMTStatic.getXmlAttr(xmlArg, "op", "ADD_MINUTE"));
|
_calcValue = SMTStatic.toInt(SMTStatic.getXmlAttr(xmlArg, "value", "0"));
|
}
|
|
@Override
|
public void loadArgToMap(Json callExtJson, SMTAIServerRequest tranReq, Map<String, Object> r_mapId2Value)
|
{
|
String value = SMTStatic.toString(SMTStatic.calculateTime(new Date(), SMTStatic.SMTCalcTime.ZERO_TIME, 0, _calcOP, _calcValue));
|
r_mapId2Value.put(_name, value);
|
}
|
}
|
|
////////////////////////////////////////////////////////////////////////////
|
|
protected String _callId;
|
protected List<ExtCallArg> _listCallArg = new ArrayList<>();
|
protected String _callConfig;
|
|
protected abstract SMTJavaAIError invokeByArgs(Map<String, Object> mapId2Value, SMTJavaAISession session, Json callExtJson, SMTJsonWriter jsonWr, SMTAIServerRequest tranReq) throws Exception;
|
|
public SMTJavaAIError invoke(SMTJavaAISession session, Json callExtJson, SMTJsonWriter jsonWr, SMTAIServerRequest tranReq) throws Exception
|
{
|
// 获取参数
|
Map<String, Object> mapId2Value = loadArguments(callExtJson, tranReq);
|
SMTJavaAIError error = invokeByArgs(mapId2Value, session, callExtJson, jsonWr, tranReq);
|
return error;
|
}
|
|
protected Map<String, Object> loadArguments(Json callExtJson, SMTAIServerRequest tranReq) throws Exception
|
{
|
Map<String, Object> mapId2Value = new HashMap<>();
|
for(ExtCallArg callArg : _listCallArg)
|
{
|
callArg.loadArgToMap(callExtJson, tranReq, mapId2Value);
|
}
|
|
return mapId2Value;
|
}
|
|
public String getCallId()
|
{
|
return _callId;
|
}
|
|
public void initInstance(DBRecord rec) throws Exception
|
{
|
_callId = rec.getString("call_id");
|
_callConfig = rec.getString("call_config");
|
Document doc = SMTStatic.convStrToXmlDoc("<ROOT>" + rec.getString("call_argument") + "</ROOT>");
|
for(Node nodeArg : doc.selectNodes("ROOT/REQUEST_ARGS/ARG"))
|
{
|
ExtCallArg callArg = new ExtCallArgRequest();
|
callArg.initInstance((Element)nodeArg);
|
_listCallArg.add(callArg);
|
}
|
|
for(Node nodeArg : doc.selectNodes("ROOT/CALC_ARGS/ARG"))
|
{
|
ExtCallArg callArg;
|
String type = SMTStatic.getXmlAttr((Element)nodeArg, "type");
|
if("calc_base_time".equals(type))
|
{
|
callArg = new ExtCallArgCalcBaseTime();
|
}
|
else if("today_calc".equals(type))
|
{
|
callArg = new ExtCallArgTodayCalc();
|
}
|
else
|
{
|
throw new Exception("can't find call ext : " + type);
|
}
|
|
callArg.initInstance((Element)nodeArg);
|
_listCallArg.add(callArg);
|
}
|
|
}
|
}
|
