在正式开始实现ORM之前还有一点准备工作需要完成,第一是实现一个配置类,这个很简单的就是通过静态变量来保存数据库的一些连接信息,等同于.net项目中的web.config的功能;第二需要设计实现一个sql语句的生成类来帮助生成sql语句,当前如果不实现这个类也不会影响orm的制作,之所以要做这么一个类主要有几个目的,1.减少sql语句中拼写错误的发生。2.统一解决防sql注入的问题。

下面分别说明一下这两个类的实现方式:

1.配置类DbConfig

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks; namespace ORM
{
public class DbConfig
{ /// <summary>
/// 数据库连接信息
/// </summary>
public static string Host = "D:/C#/ORM/test.db"; /// <summary>
/// 数据库类型
/// </summary>
public static DbType Type = DbType.Sqlite; } public enum DbType
{
Sqlite,
Mysql,
SqlServer,
Oracle
}
}

2.sql语句的生成类Sql

using System;
using System.Collections;
using System.Text.RegularExpressions; namespace ORM
{
public class Sql
{
/// <summary>
/// sql语句
/// </summary>
private string sql; /// <summary>
/// 是否有where关键字
/// </summary>
private bool hasWhere; /// <summary>
/// 是否有order关键字
/// </summary>
private bool hasOrder; /// <summary>
/// 防sql注入
/// </summary>
/// <param name="value"></param>
/// <returns></returns>
public static bool InjectionDefend(string value)
{
//网上随便找的,不确定是否有效
string SqlStr = @"and|or|exec|execute|insert|select|delete|update|alter|create|drop|count|\/\*|\*\/|chr|char|asc|mid|substring|master|truncate|declare|xp_cmdshell|restore|backup|net +user|net +localgroup +administrators";
try
{
if ((value != null) && (value != String.Empty))
{
//string str_Regex = @"\b(" + SqlStr + @")\b";
Regex Regex = new Regex(SqlStr, RegexOptions.IgnoreCase);
if (true == Regex.IsMatch(value))
{
return false;
}
}
}
catch
{
return false;
}
return true;
} /// <summary>
/// 构造函数
/// </summary>
public Sql()
{
sql = string.Empty;
hasWhere = false;
hasOrder = false;
} /// <summary>
/// 添加select
/// </summary>
/// <param name="column"></param>
/// <returns></returns>
public Sql Select(string column)
{
sql += ("SELECT " + column + " ");
return this;
} /// <summary>
/// 添加from
/// </summary>
/// <param name="Table"></param>
/// <returns></returns>
public Sql From(string Table)
{
sql += ("FROM " + Table + " ");
return this;
} /// <summary>
/// 添加where
/// </summary>
/// <param name="query"></param>
/// <param name="values"></param>
/// <returns></returns>
public Sql Where(string query, params object[] values)
{
if (!hasWhere)
{
sql += "WHERE ";
hasWhere = true;
}
else
{
sql += " AND ";
}
for (int i = ; i < values.Length; i++)
{
Regex r = new Regex(@"@\d+");
//bool类型需要特殊处理,不能直接用tostring转换,因为直接转换的结果为"True"或"False",而不是1和0
if (values[i] is bool)
{
bool value = bool.Parse(values[i].ToString());
query = r.Replace(query, (value ? "" : ""), );
continue;
}
else if (values[i].GetType().IsPrimitive)
{
query = r.Replace(query, values[i].ToString(), );
continue;
}
else if (values[i].GetType().IsEnum)
{
int intValue = (int)values[i];
query = r.Replace(query, intValue.ToString(), );
continue;
}
else
{
if (InjectionDefend(values[i].ToString()))
{
query = r.Replace(query, "\"" + values[i].ToString() + "\"", );
}
}
}
sql += query;
return this;
} /// <summary>
/// 在sql尾部插入任意sql语句
/// </summary>
/// <param name="sql"></param>
/// <returns></returns>
public Sql Append(string sql, params object[] values)
{
for (int i = ; i < values.Length; i++)
{
Regex r = new Regex(@"@\d+");
if (values[i] is bool)
{
bool value = bool.Parse(values[i].ToString());
sql = r.Replace(sql, (value ? "" : ""), );
continue;
}
else if (values[i].GetType().IsPrimitive)
{
sql = r.Replace(sql, values[i].ToString(), );
continue;
}
else if (values[i].GetType().IsEnum)
{
int intValue = (int)values[i];
sql = r.Replace(sql, intValue.ToString(), );
continue;
}
else
{
if (InjectionDefend(values[i].ToString()))
{
sql = r.Replace(sql, "\"" + values[i].ToString() + "\"", );
}
}
}
this.sql += (" " + sql + " ");
return this;
} /// <summary>
/// 添加order
/// </summary>
/// <param name="column"></param>
/// <returns></returns>
public Sql OrderBy(string column)
{
if (!sql.EndsWith(" "))
{
sql += " ";
}
if (hasOrder)
{
sql += (", " + column);
}
else
{
sql += ("ORDER BY " + column);
}
return this;
} /// <summary>
/// 获取当前完整的sql语句
/// </summary>
/// <returns></returns>
public string GetSql()
{
return sql;
}
}
}

到这里ORM的前期准备工作就完成了,当然Sql实现的非常简单,像inner join、left join等sql语句的帮助生成函数都没有做,但是实现原理时相同的,如果需要可以自己仿照实现,当然也可以直接用Append函数添加sql语句,下篇开始正式介绍ORM的核心内容。

04-28 13:22