之前在做unity项目时发现只能用odbc连接数据库，感觉非常的麻烦，因为之前做web开发的时候用惯了ORM映射，所以我想在unity中也用一下ORM（虽然我知道出于性能的考虑这样做事不好的，不过自己的小项目吗管他的，自己爽就行了）。不过现在世面上的ORM映射基本都是为web项目设计的，与unity项目很难契合，所以我决定自己做一个简易的ORM映射。虽然想的很美但是实际做起来才发现还是挺复杂的，所以我在这里记录一下这次的开发过程，防止以后遗忘。
今天先记录一下如何通过自定义attribute类实现对类名、属性名和关系数据库中的表名、字段名等信息映射。关于attribute类网上资料很多，这里不详细介绍了，下面具体代码中用到的地方会有具体说明。
首先需要自定义三个attribute类，分别是TableAttribute、ColumnAttribute和PrimaryKeyAttribute，这三个类将分别描述表名、字段名和主键名。下面是具体的实现。
1.TableAttribute
using System;
namespace ORM
{
[AttributeUsage(AttributeTargets.Class)]
public class TableAttribute : Attribute
{
public TableAttribute(string tableName)
{
this.Value = tableName;
}
public string Value { get; protected set; }
}
}
2.ColumnAttribute
using System;
namespace ORM
{
[AttributeUsage(AttributeTargets.Field | AttributeTargets.Property, AllowMultiple = false, Inherited = false)]
public class ColumnAttribute : Attribute
{
public ColumnAttribute(string columnName)
{
this.Value = columnName;
}
public string Value { get; protected set; }
}
}
3.PrimaryKeyAttribute
using System;
namespace ORM
{
[AttributeUsage(AttributeTargets.Class)]
public class PrimaryKeyAttribute : Attribute
{
public PrimaryKeyAttribute(string primaryKey)
{
this.Value = primaryKey;
}
public string Value { get; protected set; }
public bool autoIncrement = false;
}
}
这里要注意的地方不多，主要有以下几点：
1.AttributeTargets是用来表名attribute类应该在何种程序实体前放置，class表示应该在类声明前放置，Field表示可以在字段前放置，Property表示可以在属性前放置。
2.AllowMultiple表示同一个程序体前能否放置多个相同的该自定义attribute类，这里我设为false，因为一个属性在数据表中只能对应一个字段。
3.Inherited表示在描述类属性时，这个attribute能否被子类继承，这里我也设为了false，因为orm映射的类不会涉及到继承的问题。
4.自定义的attribute在定义是类名都是以attribute结尾的，但是在使用时不需要将attribute也打出来，下面我举个例子来说明。我用sqlite定义了一个userinfo表（为什么是用sqlite原因很简单，因为sqlite比较简单粗暴），表结构如下。
这张表对应的类声明如下：
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace ORM
{
[Table("userinfo")]
[PrimaryKey("Id", autoIncrement = true)]
public class User
{
[Column("Id")]
public int Id { get; set; }
[Column("UserName")]
public string UserName { get; set; }
[Column("Password")]
public string Password { get; set; }
[Column("CreateTime")]
public DateTime CreateTime { get; set; }
[Column("Status")]
public bool Status { get; set; }
[Column("RoleType")]
public RoleType RoleType { get; set; }
}
public enum RoleType : int
{
Common = 1,
Admin = 2
}
}
为了在后面实现数据库访问，包括增删改查操作时更加的方便，我们在做一个帮助类AttributeProcess，这个类是一个静态类，里面的方法也是静态方法。设计这个类的目的是提供一个公共的方法来提取类所对应的表名、字段名、主键名的属性。代码如下：
using System;
using System.Collections;
using System.Collections.Generic;
using System.Text;
using System.Reflection;
namespace ORM
{
public static class AttributeProcess
{
/// <summary>
/// 获取表名
/// </summary>
/// <param name="type"></param>
/// <returns></returns>
public static string GetTableName(Type type)
{
string tableName = string.Empty;
object[] attributes = type.GetCustomAttributes(false);
foreach (var attr in attributes)
{
if (attr is TableAttribute)
{
TableAttribute tableAttribute = attr as TableAttribute;
tableName = tableAttribute.Value;
}
}
if (string.IsNullOrEmpty(tableName))
{
tableName = type.Name;
}
return tableName;
}
/// <summary>
/// 获取字段名
/// </summary>
/// <param name="property"></param>
/// <returns></returns>
public static string GetColumnName(PropertyInfo property)
{
string columnName = string.Empty;
object[] attributes = property.GetCustomAttributes(false);
foreach (var attr in attributes)
{
if (attr is ColumnAttribute)
{
ColumnAttribute columnAttr = attr as ColumnAttribute;
columnName = columnAttr.Value;
}
}
if (string.IsNullOrEmpty(columnName))
{
columnName = property.Name;
}
return columnName;
}
/// <summary>
/// 判断主键是否自增
/// </summary>
/// <param name="property"></param>
/// <returns></returns>
public static bool IsIncrement(Type type)
{
object[] attributes = type.GetCustomAttributes(false);
foreach (var attr in attributes)
{
if (attr is PrimaryKeyAttribute)
{
PrimaryKeyAttribute primaryKeyAttr = attr as PrimaryKeyAttribute;
return primaryKeyAttr.autoIncrement;
}
}
return false;
}
/// <summary>
/// 获取主键名
/// </summary>
/// <param name="type"></param>
/// <returns></returns>
public static string GetPrimary(Type type)
{
object[] attributes = type.GetCustomAttributes(false);
foreach (var attr in attributes)
{
if (attr is PrimaryKeyAttribute)
{
PrimaryKeyAttribute primaryKeyAttr = attr as PrimaryKeyAttribute;
return primaryKeyAttr.Value;
}
}
return null;
}
/// <summary>
/// 判断属性是否为主键
/// </summary>
/// <param name="type"></param>
/// <param name="property"></param>
/// <returns></returns>
public static bool IsPrimary(Type type, PropertyInfo property)
{
string primaryKeyName = GetPrimary(type);
string columnName = GetColumnName(property);
return (primaryKeyName == columnName);
}
}
}
其中获取自定义attribute和其中的属性值的方法不难，主要就是先通过GetCustomAttributes方法来获取程序体前放置的所有的自定义attribute，然后循环遍历找到需要的attribute并读取需要的属性值，这样就可以获取到需要的数据库相关信息了。另外，为了方便起见，在获取表名和字段名时，如果没有在类名或者属性名前放置TableAttribute类或者ColumnAttribute类，那么将自动的读取类名或者属性名做为表名和字段名返回。下面做一个简单的测试，将user类对应的表名和其中属性对应的字段名都打印出来，测试代码如下：
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Reflection;
namespace ORM
{
class Program
{
static void Main(string[] args)
{
Type type = typeof(User);
PropertyInfo[] properties = type.GetProperties();
Console.WriteLine(AttributeProcess.GetTableName(type));
foreach (var item in properties)
{
Console.WriteLine(AttributeProcess.GetColumnName(item));
}
}
}
}
注：GetProperties方法是Type类下的一个方法，用来获取类中的所有属性的信息。
测试结果如下：
好了，到这里自定义Attribute类的工作就基本完成了，下面就要正式开始正式的数据库操作了，我会在后续的文章中进行说明，今天就先到这里。