package com.smtaiserver.smtaiserver.javaai.querydetail;
|
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.Iterator;
|
import java.util.List;
|
|
import org.dom4j.Document;
|
import org.dom4j.Element;
|
import org.dom4j.Node;
|
import org.dom4j.tree.DefaultText;
|
import org.springframework.web.servlet.ModelAndView;
|
|
import com.smtaiserver.smtaiserver.core.SMTAIServerRequest;
|
import com.smtaiserver.smtaiserver.database.SMTDatabase;
|
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecord;
|
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecords;
|
import com.smtaiserver.smtaiserver.javaai.SMTJavaAIError;
|
import com.smtaiserver.smtaiserver.javaai.ast.ASTDBMap;
|
import com.smtservlet.util.Json;
|
import com.smtservlet.util.SMTJsonWriter;
|
import com.smtservlet.util.SMTStatic;
|
|
|
|
public class SMTAIQueryDetailSqlXml extends SMTAIQueryDetail
|
{
|
private static class SQLXMLExecArg
|
{
|
public StringBuilder _sbSQL = new StringBuilder();
|
public List<Object> _sqlParams = new ArrayList<>();
|
public SMTJsonWriter _jsonWr;
|
//public SMTAIServerRequest _tranReq;
|
public Json _jsonParams;
|
public ASTDBMap _dbMap;
|
|
public SQLXMLExecArg(SMTAIServerRequest tranReq, ASTDBMap dbMap, Json jsonParams) throws Exception
|
{
|
_dbMap = dbMap;
|
//_tranReq = tranReq;
|
_jsonParams = jsonParams;
|
_jsonWr = tranReq.newReturnJsonWriter(true, null, null);
|
}
|
|
public String getSqlArg(String key)
|
{
|
Json jsonValue = _jsonParams.safeGetJson(key);
|
if(jsonValue == null)
|
return null;
|
|
return jsonValue.asString();
|
}
|
}
|
|
///////////////////////////////////////////////////////////////////////////////////////
|
private static class SQLXMLNodeSQL_TIME_VALUE extends SQLXMLNode
|
{
|
private String _dsId;
|
private List<Object> _listChildren = new ArrayList<>();
|
|
public SQLXMLNodeSQL_TIME_VALUE(Element xmlRoot) throws Exception
|
{
|
_dsId = SMTStatic.getXmlAttr(xmlRoot, "ds_id");
|
|
for (Iterator<Node> iterInner = xmlRoot.nodeIterator(); iterInner.hasNext();)
|
{
|
Node nodeInner = iterInner.next();
|
if(nodeInner.getNodeType() == Node.TEXT_NODE)
|
{
|
String text = ((DefaultText)nodeInner).getText();
|
int lastPos = _listChildren.size() - 1;
|
if(lastPos >= 0 && _listChildren.get(lastPos) instanceof String)
|
{
|
_listChildren.set(lastPos, (String)_listChildren.get(lastPos) + text);
|
}
|
else
|
{
|
_listChildren.add(text);
|
}
|
}
|
else
|
{
|
_listChildren.add(createSQLXMLNode((Element)nodeInner));
|
}
|
}
|
}
|
|
@Override
|
public SMTJavaAIError execute(SQLXMLExecArg execArg) throws Exception
|
{
|
for(Object oxmlNode : _listChildren)
|
{
|
if(oxmlNode instanceof String)
|
{
|
execArg._sbSQL.append(oxmlNode);
|
}
|
else if(oxmlNode instanceof SQLXMLNode)
|
{
|
SMTJavaAIError error = ((SQLXMLNode)oxmlNode).execute(execArg);
|
if(error != null)
|
return error;
|
}
|
}
|
|
SMTDatabase db = execArg._dbMap.getDatabase(_dsId);
|
DBRecords recs = db.querySQL(execArg._sbSQL.toString(), execArg._sqlParams.toArray(new Object[execArg._sqlParams.size()]));
|
|
execArg._jsonWr.beginArray("values");
|
{
|
for(DBRecord rec : recs.getRecords())
|
{
|
execArg._jsonWr.beginArray(null);
|
{
|
execArg._jsonWr.addKeyValue(null, rec.getString("otime"));
|
execArg._jsonWr.addKeyValue(null, rec.getString("ovalue"));
|
}
|
execArg._jsonWr.endArray();
|
}
|
}
|
execArg._jsonWr.endArray();
|
|
return null;
|
}
|
}
|
|
///////////////////////////////////////////////////////////////////////////////////////
|
private static class SQLXMLNodePARAM extends SQLXMLNode
|
{
|
private String _key;
|
|
public SQLXMLNodePARAM(Element xmlRoot) throws Exception
|
{
|
_key = SMTStatic.getXmlAttr(xmlRoot, "key");
|
}
|
|
@Override
|
public SMTJavaAIError execute(SQLXMLExecArg execArg) throws Exception
|
{
|
String value = execArg.getSqlArg(_key);
|
if(value == null)
|
throw new Exception("请输入参数：" + _key);
|
|
execArg._sbSQL.append("?");
|
execArg._sqlParams.add(value);
|
|
return null;
|
}
|
}
|
|
///////////////////////////////////////////////////////////////////////////////////////
|
private static class SQLXMLNodePARAM_TIME_S extends SQLXMLNode
|
{
|
private String _key;
|
|
public SQLXMLNodePARAM_TIME_S(Element xmlRoot) throws Exception
|
{
|
_key = SMTStatic.getXmlAttr(xmlRoot, "key");
|
}
|
|
@Override
|
public SMTJavaAIError execute(SQLXMLExecArg execArg) throws Exception
|
{
|
String value = execArg.getSqlArg(_key);
|
if(value == null)
|
{
|
value = SMTStatic.toString(SMTStatic.calculateTime(new Date(), SMTStatic.SMTCalcTime.ZERO_TIME, 0));
|
}
|
|
execArg._sbSQL.append("?");
|
execArg._sqlParams.add(value);
|
|
return null;
|
}
|
}
|
|
///////////////////////////////////////////////////////////////////////////////////////
|
private static class SQLXMLNodePARAM_TIME_E extends SQLXMLNode
|
{
|
private String _key;
|
|
public SQLXMLNodePARAM_TIME_E(Element xmlRoot) throws Exception
|
{
|
_key = SMTStatic.getXmlAttr(xmlRoot, "key");
|
}
|
|
@Override
|
public SMTJavaAIError execute(SQLXMLExecArg execArg) throws Exception
|
{
|
String value = execArg.getSqlArg(_key);
|
if(value == null)
|
{
|
value = SMTStatic.toString(SMTStatic.calculateTime(new Date(), SMTStatic.SMTCalcTime.ZERO_TIME, 0, SMTStatic.SMTCalcTime.ADD_DATE, 1));
|
}
|
|
execArg._sbSQL.append("?");
|
execArg._sqlParams.add(value);
|
|
return null;
|
}
|
}
|
|
///////////////////////////////////////////////////////////////////////////////////////
|
private static abstract class SQLXMLNode
|
{
|
public abstract SMTJavaAIError execute(SQLXMLExecArg execArg) throws Exception;
|
}
|
|
private static SQLXMLNode createSQLXMLNode(Element xmlRoot) throws Exception
|
{
|
String name = xmlRoot.getName().toUpperCase();
|
if("SQL_TIME_VALUE".equals(name))
|
return new SQLXMLNodeSQL_TIME_VALUE(xmlRoot);
|
else if("PARAM".equals(name))
|
return new SQLXMLNodePARAM(xmlRoot);
|
else if("PARAM_TIME_S".equals(name))
|
return new SQLXMLNodePARAM_TIME_S(xmlRoot);
|
else if("PARAM_TIME_E".equals(name))
|
return new SQLXMLNodePARAM_TIME_E(xmlRoot);
|
else
|
throw new Exception("unknow SQLXML : " + name);
|
}
|
|
private SQLXMLNode _sqlXmlRoot;
|
|
////////////////////////////////////////////////////////////
|
|
public void initInstance(DBRecord rec) throws Exception
|
{
|
super.initInstance(rec);
|
Document doc = SMTStatic.convStrToXmlDoc("<ROOT>" + rec.getString("sqlxml_config") + "</ROOT>");
|
|
Element xmlSQL = (Element) doc.selectSingleNode("ROOT/SQL_TIME_VALUE");
|
if(xmlSQL != null)
|
_sqlXmlRoot = createSQLXMLNode(xmlSQL);
|
else
|
throw new Exception("can' find ROOT/SQL");
|
|
}
|
|
|
@Override
|
public ModelAndView executeQuery(SMTAIServerRequest tranReq, Json jsonParams) throws Exception
|
{
|
ASTDBMap dbMap = new ASTDBMap();
|
try
|
{
|
SQLXMLExecArg execArg = new SQLXMLExecArg(tranReq, dbMap, jsonParams);
|
_sqlXmlRoot.execute(execArg);
|
return tranReq.returnJson(execArg._jsonWr);
|
}
|
finally
|
{
|
dbMap.close();
|
}
|
|
}
|
|
|
}
|
