128 lines
4.3 KiB
C#
128 lines
4.3 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
using NewLife.Data;
|
||
using NewLife.Reflection;
|
||
using XCode.DataAccessLayer;
|
||
|
||
namespace AntJob.Extensions
|
||
{
|
||
/// <summary>Sql操作</summary>
|
||
public enum SqlActions
|
||
{
|
||
/// <summary>查询。返回数据集</summary>
|
||
Query,
|
||
|
||
/// <summary>执行Sql或存储过程。返回影响行数</summary>
|
||
Execute,
|
||
|
||
/// <summary>插入。批量插入数据，需要传入数据集</summary>
|
||
Insert,
|
||
}
|
||
|
||
/// <summary>Sql片段。解析sql语句集合</summary>
|
||
public class SqlSection
|
||
{
|
||
#region 属性
|
||
/// <summary>连接名</summary>
|
||
public String ConnName { get; set; }
|
||
|
||
/// <summary>操作。添删改查等</summary>
|
||
public SqlActions Action { get; set; }
|
||
|
||
/// <summary>Sql语句</summary>
|
||
public String Sql { get; set; }
|
||
#endregion
|
||
|
||
#region 解析
|
||
/// <summary>分析sql语句集合，得到片段集合，以双换行分隔</summary>
|
||
/// <param name="sqls"></param>
|
||
/// <returns></returns>
|
||
public static SqlSection[] ParseAll(String sqls)
|
||
{
|
||
var list = new List<SqlSection>();
|
||
if (sqls.IsNullOrEmpty()) return list.ToArray();
|
||
|
||
// 两个换行隔开片段
|
||
var ss = sqls.Split(new[] { "\r\n\r", "\r\r", "\n\n" }, StringSplitOptions.RemoveEmptyEntries);
|
||
var connName = "";
|
||
foreach (var item in ss)
|
||
{
|
||
var section = new SqlSection();
|
||
section.Parse(item);
|
||
|
||
// 如果当前片段未指定连接名，则使用上一个
|
||
if (section.ConnName.IsNullOrEmpty())
|
||
section.ConnName = connName;
|
||
else
|
||
connName = section.ConnName;
|
||
|
||
if (section.ConnName.IsNullOrEmpty()) throw new Exception("未指定连接名！");
|
||
|
||
list.Add(section);
|
||
}
|
||
|
||
return list.ToArray();
|
||
}
|
||
|
||
/// <summary>分析sql语句到片段</summary>
|
||
/// <param name="sql"></param>
|
||
public void Parse(String sql)
|
||
{
|
||
// 解析出来连接名
|
||
var str = sql.Substring("/*", "*/")?.Trim();
|
||
if (!str.IsNullOrEmpty()) ConnName = str.Substring("use ")?.Trim();
|
||
|
||
// 剩下的Sql
|
||
sql = sql.Substring("*/")?.Trim()?.Trim(';')?.Trim();
|
||
Sql = sql;
|
||
|
||
// 猜测操作
|
||
if (sql.StartsWithIgnoreCase("select "))
|
||
Action = SqlActions.Query;
|
||
else if (sql.StartsWithIgnoreCase("insert into ", "update ", "delete "))
|
||
Action = SqlActions.Execute;
|
||
// 批量插入
|
||
else if (sql.StartsWithIgnoreCase("insert "))
|
||
Action = SqlActions.Insert;
|
||
// 默认执行，可能是存储过程
|
||
else
|
||
Action = SqlActions.Execute;
|
||
}
|
||
#endregion
|
||
|
||
#region 执行处理
|
||
/// <summary>查询数据集</summary>
|
||
/// <returns></returns>
|
||
public DbTable Query() => DAL.Create(ConnName).Query(Sql);
|
||
|
||
/// <summary>执行</summary>
|
||
/// <returns></returns>
|
||
public Int32 Execute() => DAL.Create(ConnName).Execute(Sql);
|
||
|
||
/// <summary>批量插入</summary>
|
||
/// <param name="dt"></param>
|
||
/// <returns></returns>
|
||
public Int32 BatchInsert(DbTable dt)
|
||
{
|
||
var tableName = Sql.Substring(" ")?.Trim(';');
|
||
var dal = DAL.Create(ConnName);
|
||
var table = dal.Tables?.FirstOrDefault(e => e.TableName.EqualIgnoreCase(tableName));
|
||
if (table == null) throw new Exception($"在连接[{ConnName}]中无法找到数据表[{tableName}]");
|
||
|
||
// 执行反向工程，该建表就建表
|
||
dal.CheckDatabase();
|
||
|
||
// 选取目标表和数据集共有的字段
|
||
tableName = dal.Db.FormatTableName(tableName);
|
||
var columns = new List<IDataColumn>();
|
||
foreach (var dc in table.Columns)
|
||
{
|
||
if (dc.ColumnName.EqualIgnoreCase(dt.Columns)) columns.Add(dc);
|
||
}
|
||
|
||
return dal.Session.Insert(tableName, columns.ToArray(), dt.Cast<IIndexAccessor>());
|
||
}
|
||
#endregion
|
||
}
|
||
} |