乐趣区

C-数据操作系列-4-自己实现一个ORM

jiezi

0. 前言

在之前的几篇内容中，我们了解了如何通过 ADO.NET 访问数据库，如何修改、新增数据。如何通过 DataSet 和 DataAdapter 获取数据，我们将在这一篇试试自己实现一个简单的 ORM 框架或者说 ORM 工具类。

涉及到的知识点：

  • 反射（初级）
  • ADO.NET 已有知识

1. ORM

那么，问题来了，什么是 ORM？ORM 全称 Object Relational Mapping，翻译过来就是对象关系映射。是一种通过描述对象与数据库之间映射关系的数据，将对象保存到数据库中的技术。

在 C# 中，曾经 Entity Framework 光芒万丈，遮盖了其他 ORM 框架的光辉（甚至如今都是如此）。

后来慢慢涌现除了其他的一些 ORM 框架，进一步丰富了市场。所以现有比较流行的大概有以下几种：

  • Dapper 一个轻量的 ORM 框架
  • Entity Framework/Entity Framework Core 功能完备的框架
  • Nhibernate Java 平台上著名的 Hibernate 的.net 版
  • 等等

嗯，这是我最近找到的创作组还在更新的几个框架，当然还有其他的很多有趣好用的 ORM 框架。欢迎各位补充哈。

这一篇的主要目的不是介绍这些框架（这是以后的内容），而是通过我们自己实现一个类 ORM 框架来了解底层核心。

2. 设计

我们先分析一下，如果我们设计一个实体对象与数据库之间转换的工具类应该具有哪些功能？

  • 一个属性与数据库字段的映射关系
  • 增删改查的 SQL 模板
  • 查询结果与对象的转换

3. 实现

首先，声明一个类，因为不能仅支持一种类型，所以这个类的所有与数据库有关的方法都是泛型方法，或者这个类是泛型类，所以定义为泛型类：

public class OrmUtil<T>
{}

我们事先约定类名即表名，属性名即表的列名，所以我们可以快速得到以下内容：

/// <summary>
/// T 的类型实例
/// </summary>
private Type dType;
/// <summary>
/// T 的属性表
/// </summary>
private PropertyInfo[] properties;
public OrmUtil()
{dType = typeof(T);
    properties = dType.GetProperties();}

声明一个数据库连接：

public SqlConnection Connection {get; set;}

创建一个私有方法，检查连接是否可用：

/// <summary>
/// 检查连接是否可用
/// </summary>
/// <returns></returns>
private bool CheckConnection()
{return Connection?.State == ConnectionState.Open;}

准备工作完成，然后开始编写具体的业务方法：

Insert：

public int Insert(T entity)
{if (!CheckConnection()) return -1;// 检查状态
    var insert = $"insert into {dType.Name}({string.Join(",", properties.Select(t => t.Name))})";
    var values = properties.Select(p => p.GetValue(entity));
    var commandText = $"{insert} values('{string.Join("','", values)}')";

    var command = Connection.CreateCommand();
    command.CommandText = commandText;
    var result = command.ExecuteNonQuery();
    return result;
}

首先按照属性名与列名之间的映射拼接 SQL，然后执行 SQL 命令。

Update：

public int Update(T entity,string keyName,object keyValue)
{if (!CheckConnection()) return -1;
    var setValues = properties.ToDictionary(p => p.Name, p => $"'{p.GetValue(entity)}'");
    var setSql = string.Join(",", setValues.Select(pair=>$"{pair.Key}='{pair.Value}'"));
    var sql = $"update {dType.Name} set {setSql} where {keyName} ='{keyValue}'";
    var command = Connection.CreateCommand();
    command.CommandText = sql;
    return command.ExecuteNonQuery();}

Update 需要注意的就是如何正确拼接赋值 sql。

Delete：

删除满足条件的对象：

public int Delete(T entity)
{if (!CheckConnection()) return -1;
    var querySet = properties.Select(p => $"{p.Name} ='{p.GetValue(entity)}'");
    var sql = $"delete from {dType.Name} where {string.Join(" and ", querySet)}";
    var command = Connection.CreateCommand();
    command.CommandText = sql;
    return command.ExecuteNonQuery();}

这里写法有时候根据实际业务不同，大多数情况下删除主键对应的元素，或者满足某一个条件的所有元素。这里只是做了个演示，小伙伴们可以试试自己改造一下。

Search：

先创建一个从 DataTable 转成对象的工具方法：

private List<T> Convert(DataTable table)
{var list = new List<T>(table.Rows.Count);// 事先声明一下容量
    foreach(DataRow row in table.AsEnumerable())
    {T entity = Activator.CreateInstance<T>();
        foreach(var p in properties)
        {if (!table.Columns.Contains(p.Name)) continue;// 如果属性名不在表格中，则忽略
            p.SetValue(entity, row[p.Name]);
        }
        list.Add(entity);
    }
    return list;
}

好，我们写一个查询方法：

public List<T> SearchAll()
{var adapter = new SqlDataAdapter($"select * from {dType.Name}", Connection);
    var set = new DataSet();
    adapter.Fill(set);
    return Convert(set.Tables[0]);
}

这样一个简单的 ORM 框架就这样形成雏形了，当然实际上的 ORM 底层比这复杂，因为需要支持不同的数据库，所以 Connection 就不能简简单单的是一个 SqlConnection 了，或者底层不是像我们一样取巧使用 DataTable 了。

实际上的 DataTable 到类对象的转换要比我写的复杂一点，因为还要判断这个属性是否是可读、可写的。

4. 总结

在这里我做了个抛砖引玉，带领小伙伴们一起构思了一个简陋的 ORM 框架，也让大伙对此有了一定的印象。嗯，今天就到这了。同时 ADO.NET 也告一段落了，接下来就是上 Entity Framework 了。当然，DataSet、DataAdapter 这两个类并没有讲完。这部分内容可能会在后续的番外篇内补全。

更多内容烦请关注我的博客《高先生小屋》

退出移动版