using Dapper.Contrib.Extensions;
|
using System;
|
using System.Collections.Generic;
|
using System.Linq;
|
using System.Reflection;
|
using System.Text;
|
using System.Threading.Tasks;
|
|
namespace IStation.DataProvider
|
{
|
/// <summary>
|
/// 实体辅助类
|
/// </summary>
|
public class EntityHelper
|
{
|
//默认主键名称
|
private static readonly string _defaultPrimaryKey = "ID";
|
|
/// <summary>
|
/// 获取表名称
|
/// </summary>
|
/// <typeparam name="T"></typeparam>
|
/// <returns></returns>
|
public static string GetTableName<T>()
|
{
|
var objType = typeof(T);
|
var attrs = objType.GetCustomAttributes(typeof(TableAttribute), true);
|
if (attrs != null && attrs.Count() > 0)
|
return (attrs[0] as TableAttribute).Name;
|
return objType.Name;
|
}
|
|
/// <summary>
|
/// 获取主键属性信息
|
/// </summary>
|
/// <typeparam name="T"></typeparam>
|
/// <returns></returns>
|
public static string GetPrimaryKey<T>()
|
{
|
var objType = typeof(T);
|
var pros = objType.GetProperties();
|
var keyPro = pros.FirstOrDefault(x => x.GetCustomAttribute(typeof(KeyAttribute)) != null);
|
if (keyPro != null)
|
return keyPro.Name;
|
var explicitKeyPro = pros.FirstOrDefault(x => x.GetCustomAttribute(typeof(ExplicitKeyAttribute)) != null);
|
if (explicitKeyPro != null)
|
return explicitKeyPro.Name;
|
return _defaultPrimaryKey;
|
}
|
|
|
}
|
}
|
