package com.smtaiserver.smtaiserver.javaai.metrics.base;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map.Entry;
|
|
import org.apache.commons.text.similarity.JaccardSimilarity;
|
import org.apache.logging.log4j.LogManager;
|
import org.apache.logging.log4j.Logger;
|
|
import com.smtaiserver.smtaiserver.control.SMTAIServerControl;
|
import com.smtaiserver.smtaiserver.core.SMTAIServerApp;
|
import com.smtaiserver.smtaiserver.database.SMTDatabase;
|
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecord;
|
import com.smtaiserver.smtaiserver.database.SMTDatabase.DBRecords;
|
import com.smtservlet.util.Json;
|
import com.smtservlet.util.SMTStatic;
|
|
public class SMTDimensionDef
|
{
|
private String _id;
|
private String _name;
|
private String _prompt;
|
private char _type;
|
private String _unitName;
|
private List<String[]> _listMatches = new ArrayList<>();
|
private List<String> _listEnums = null;
|
|
private static Logger _logger = LogManager.getLogger(SMTAIServerControl.class);
|
|
public SMTDimensionDef(DBRecord rec) throws Exception
|
{
|
_id = rec.getString("dim_id");
|
_name = rec.getString("dim_name");
|
_prompt = rec.getString("dim_title");
|
_unitName = rec.getString("dim_unit");
|
if(SMTStatic.isNullOrEmpty(_unitName))
|
_unitName = "";
|
if(SMTStatic.isNullOrEmpty(_prompt))
|
_prompt = _name;
|
else
|
_prompt = _name + "，" + _prompt;
|
_type = rec.getString("dim_type").charAt(0);
|
|
String sAlias = rec.getString("dim_alias");
|
if(!SMTStatic.isNullOrEmpty(sAlias))
|
{
|
Json jsonMatch = Json.read(sAlias);
|
for(Entry<String, Json> entry : jsonMatch.asJsonMap().entrySet())
|
{
|
String key = entry.getKey();
|
for(Json jsonValue : entry.getValue().asJsonList())
|
{
|
_listMatches.add(new String[] {jsonValue.asString(), key});
|
}
|
}
|
}
|
|
// 如果枚举值存在，则读取枚举值
|
String sJsonValueList = rec.getString("dim_value_list");
|
if(!SMTStatic.isNullOrEmpty(sJsonValueList))
|
{
|
Json jsonValueList = Json.read(sJsonValueList);
|
|
// 如果能取到直接值，则用直接值
|
Json jsonValues = jsonValueList.safeGetJson("values");
|
if(jsonValues != null && jsonValues.asJsonList().size() > 0)
|
{
|
if(_listEnums == null)
|
_listEnums = new ArrayList<>();
|
for(Json jsonValue : jsonValues.asJsonList())
|
{
|
_listEnums.add(jsonValue.asString());
|
}
|
}
|
|
// 如果配置了SQL, 则查询sql
|
Json jsonSqlConfig = jsonValueList.safeGetJson("sql");
|
if(jsonSqlConfig != null)
|
{
|
try
|
{
|
String dsId = jsonSqlConfig.getJson("ds_id").asString();
|
String sql = jsonSqlConfig.getJson("sql").asString();
|
SMTDatabase db = SMTAIServerApp.getApp().getDataSource(dsId).allocDatabase();
|
try
|
{
|
DBRecords recs = db.querySQL(sql, null);
|
if(recs.getRowCount() > 0)
|
{
|
if(_listEnums == null)
|
_listEnums = new ArrayList<>();
|
|
for(DBRecord recValue : recs.getRecords())
|
{
|
_listEnums.add(recValue.getString(0));
|
}
|
}
|
}
|
finally
|
{
|
db.close();
|
}
|
}
|
catch(Exception ex)
|
{
|
_logger.fatal("load dim value list from db error : " + _id + ", error=" + ex.getMessage());
|
}
|
}
|
|
}
|
}
|
|
public List<String> getValueList()
|
{
|
return _listEnums;
|
}
|
|
public boolean hasValueList()
|
{
|
return _listEnums != null && _listEnums.size() > 0;
|
}
|
|
public String getUnitName(boolean hasBound)
|
{
|
if(SMTStatic.isNullOrEmpty(_unitName))
|
return "";
|
if(hasBound)
|
return "(" + _unitName + ")";
|
|
return _unitName;
|
}
|
|
public String matchDimension(String value, JaccardSimilarity jaccardSimilarity)
|
{
|
double maxSim = 0;
|
String maxName = null;
|
for(String[] match : _listMatches)
|
{
|
double curSim = jaccardSimilarity.apply(match[0], value);
|
if(curSim == 1)
|
return match[1];
|
if(curSim < 0.8)
|
continue;
|
if(curSim > maxSim)
|
{
|
maxSim = curSim;
|
maxName = match[1];
|
}
|
}
|
|
if(maxName != null)
|
return maxName;
|
|
return value;
|
}
|
|
public char getType()
|
{
|
return _type;
|
}
|
|
public String getName()
|
{
|
return _name;
|
}
|
|
public String getPrompt()
|
{
|
return _prompt;
|
}
|
|
public String getId()
|
{
|
return _id;
|
}
|
}
|
