/// <summary>
/// 返回生成模板所需entity对象
/// </summary>
/// <param name="tableName">表名称</param>
/// <returns></returns>
public Entity GenTemplateEntity(string tableName)
{
NameTransService nameTransService = new NameTransService();
Dictionary <string, string> commentMap = SqlInfo.CommentMap(tableName);
string tableNodes = SqlInfo.TableNodes(tableName);
string keyCol = SqlInfo.KeyCol(tableName);
DataTable dataTable = SqlInfo.Columns(tableName);
if (dataTable.Rows.Count == 0)
{
throw new Exception("找不到该数据表！");
}
string entityName = NameTrans.TableNameToEntityName(tableName);
Entity entity = new Entity
{
TableName = tableName,
EntityName = entityName,
EntityNames = nameTransService.SingleToComplex(entityName),
EntityNotes = tableNodes,
KeyCol = keyCol,
KeyName = NameTrans.ColNameToPropName(keyCol)
};
foreach (DataRow cl in dataTable.Rows)
{
string colName = (string)cl.ItemArray[3];
string dbType = (string)cl.ItemArray[SqlInfo.ColTypeInfoIndex()];
string srcPropName = NameTrans.ColNameToPropName(colName);
entity.PropList.Add(new Prop
{
IsKey = colName == keyCol,
ColName = colName,
SrcPropName = srcPropName,
PropName = srcPropName,
CapUpperPropName = nameTransService.HumpToBigHump(srcPropName),
PropType = DbTypeTrans.SqlType2EntityType(dbType),
PropNotes = commentMap.ContainsKey(colName)?commentMap[colName]:"",
ParamsType = "equal"
});
string paramsType = DbTypeTrans.SqlType2ParamsType(dbType);
if (paramsType == "range")
{
entity.PropList.Add(new Prop
{
IsKey = colName == keyCol,
ColName = colName,
SrcPropName = srcPropName,
CapUpperPropName = nameTransService.HumpToBigHump(srcPropName + "Start"),
PropName = srcPropName + "Start",
PropType = DbTypeTrans.SqlType2EntityType(dbType),
PropNotes = commentMap.ContainsKey(colName) ? commentMap[colName] : "",
ParamsType = "rangeStart"
});
entity.PropList.Add(new Prop
{
IsKey = colName == keyCol,
ColName = colName,
SrcPropName = srcPropName,
CapUpperPropName = nameTransService.HumpToBigHump(srcPropName + "End"),
PropName = srcPropName + "End",
PropType = DbTypeTrans.SqlType2EntityType(dbType),
PropNotes = commentMap.ContainsKey(colName) ? commentMap[colName] : "",
ParamsType = "rangeEnd"
});
}
else if (paramsType == "like")
{
entity.PropList.Add(new Prop
{
IsKey = colName == keyCol,
ColName = colName,
SrcPropName = srcPropName,
CapUpperPropName = nameTransService.HumpToBigHump(srcPropName + "Like"),
PropName = srcPropName + "Like",
PropType = DbTypeTrans.SqlType2EntityType(dbType),
PropNotes = commentMap.ContainsKey(colName) ? commentMap[colName] : "",
ParamsType = "like"
});
}
if (DbTypeTrans.SqlTypeIsChangeType(dbType))
{
entity.PropList.Add(new Prop
{
ColName = colName,
SrcPropName = srcPropName,
CapUpperPropName = nameTransService.HumpToBigHump(srcPropName + "Change"),
PropName = srcPropName + "Change",
PropType = DbTypeTrans.SqlType2EntityType(dbType),
PropNotes = commentMap.ContainsKey(colName) ? commentMap[colName] : "",
ParamsType = "change"
});
}
}
return(entity);
}