300 lines
11 KiB
C#
300 lines
11 KiB
C#
using Microsoft.Data.SqlClient;
|
||
using System.Data;
|
||
using System.Data.Common;
|
||
using RB_MES_API.Context;
|
||
|
||
namespace RB_MES_API.Controllers
|
||
{
|
||
/// <summary>
|
||
/// SQL帮助类
|
||
/// </summary>
|
||
public class KDSqlHelper : IKDSqlHelper
|
||
{
|
||
private readonly RBContext _context;
|
||
/// <summary>
|
||
/// 依赖性注入
|
||
/// </summary>
|
||
/// <param name="context"></param>
|
||
/// <param name="configuration"></param>
|
||
public KDSqlHelper(RBContext context)
|
||
{
|
||
_context = context;
|
||
}
|
||
private string Myconnstr(bool KD)
|
||
{
|
||
return KD ? ConnectionString.GetConfig("ConnectionStrings", "ERP_Conn") : ConnectionString.GetConfig("ConnectionStrings", "RBConn");
|
||
}
|
||
/// <inheritdoc/>
|
||
public async Task<string> ExecuteNonQueryAsync(string cmdText, CommandType cmdType, SqlParameter[]? sqlParams, bool KD = false)
|
||
{
|
||
string mess = null;
|
||
if (cmdText.Trim() == null) return "-1";
|
||
string connstr = Myconnstr(KD);
|
||
using SqlConnection connection = new SqlConnection(connstr);
|
||
try
|
||
{
|
||
connection.Open();
|
||
using var cmd = connection.CreateCommand();
|
||
//cmd.CommandTimeout = 0;
|
||
PrepareCommand(cmd, cmdType, cmdText, sqlParams);
|
||
cmd.Transaction = connection.BeginTransaction();
|
||
try
|
||
{
|
||
await cmd.ExecuteNonQueryAsync();
|
||
cmd.Transaction.Commit();
|
||
mess = "OK";
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
cmd.Transaction.Rollback();
|
||
//将错误写到错误日志
|
||
mess = ex.Message;
|
||
LogHelper.WriteLog(string.Format("直接执行编译好的SQL语句的ExecuteNonQueryAsync方法发生错误在：{0}", mess + "\n" + cmdText));
|
||
//throw;
|
||
}
|
||
finally
|
||
{
|
||
cmd.Dispose();
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
//将错误写到错误日志，连不上数据库，只能写在磁盘文件中...
|
||
mess = ex.Message;
|
||
LogHelper.WriteLog(string.Format("直接执行编译好的SQL语句的ExecuteNonQueryAsync方法发生错误：{0}", mess));
|
||
}
|
||
finally
|
||
{
|
||
connection.Dispose();
|
||
}
|
||
return mess;
|
||
//throw new NotImplementedException();
|
||
}
|
||
/// <inheritdoc/>
|
||
|
||
public object? ExecuteScalar(CommandType cmdType, string cmdText, SqlParameter[]? sqlParams, ref string? mess, bool KD = false)
|
||
{
|
||
mess = null;
|
||
object? result = null;
|
||
if (string.IsNullOrEmpty(cmdText.Trim())) return null;
|
||
string connstr = Myconnstr(KD);
|
||
using SqlConnection connection = new SqlConnection(connstr);
|
||
try
|
||
{
|
||
connection.Open();
|
||
using var cmd = connection.CreateCommand();
|
||
//cmd.CommandTimeout = 0;
|
||
PrepareCommand(cmd, cmdType, cmdText, sqlParams);
|
||
//cmd.Transaction = connection.BeginTransaction();
|
||
try
|
||
{
|
||
result = cmd.ExecuteScalar();
|
||
//cmd.Transaction.Commit();
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
cmd.Transaction.Rollback();
|
||
//将错误写到错误日志
|
||
mess = ex.Message;
|
||
LogHelper.WriteLog(string.Format("并返回由查询返回的结果集中第一行的第一列的ExecuteScalar方法发生错误在：{0}", ex.Message + "\n" + cmdText));
|
||
}
|
||
finally
|
||
{
|
||
cmd.Dispose();
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
//将错误写到错误日志，连不上数据库，只能写在磁盘文件中...
|
||
mess = ex.Message;
|
||
LogHelper.WriteLog(string.Format("并返回由查询返回的结果集中第一行的第一列的ExecuteScalar方法发生错误：{0}", ex.Message));
|
||
}
|
||
finally
|
||
{
|
||
connection.Dispose();
|
||
}
|
||
//result = result is DBNull ? null : result;
|
||
return result;
|
||
//throw new NotImplementedException();
|
||
}
|
||
/// <inheritdoc/>
|
||
|
||
public DataSet? GetDataSet(CommandType cmdType, string? cmdText, SqlParameter[]? sqlParams, ref string? mess, bool KD = false)
|
||
{
|
||
mess = null;
|
||
if (cmdText == null || cmdText == "")
|
||
{
|
||
mess = "不执行空的查询脚本";
|
||
return null;
|
||
}
|
||
var data = new DataSet();
|
||
string connstr = Myconnstr(KD);
|
||
using SqlConnection connection = new SqlConnection(connstr);
|
||
try
|
||
{
|
||
connection.Open();
|
||
using var cmd = connection.CreateCommand();
|
||
//cmd.CommandTimeout = 0;
|
||
PrepareCommand(cmd, cmdType, cmdText, sqlParams);
|
||
cmd.Transaction = connection.BeginTransaction();
|
||
SqlDataAdapter adapter = new(cmd);
|
||
try
|
||
{
|
||
adapter.Fill(data);
|
||
cmd.Transaction.Commit();
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
cmd.Transaction.Rollback();
|
||
//将错误写到错误日志
|
||
mess = ex.Message;
|
||
LogHelper.WriteLog(string.Format("获取多个结果集的GetDataSet方法发生错误在：{0}", ex.Message + "\n" + cmdText));
|
||
|
||
}
|
||
finally
|
||
{
|
||
adapter.Dispose();
|
||
cmd.Dispose();
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
//将错误写到错误日志，连不上数据库，只能写在磁盘文件中...
|
||
mess = ex.Message;
|
||
LogHelper.WriteLog(string.Format("获取多个结果集的GetDataSet方法发生错误：{0}", ex.Message));
|
||
}
|
||
finally
|
||
{
|
||
connection.Dispose();
|
||
}
|
||
return data;
|
||
//throw new NotImplementedException();
|
||
}
|
||
/// <inheritdoc/>
|
||
|
||
public Dictionary<object, object> GetProcedurePairs(string cmdText, SqlParameter[] sqlParams, ref string? mess, bool KD = false)
|
||
{
|
||
mess = null;
|
||
Dictionary<object, object> keyValues = new();
|
||
//检查是否包含返回参数，如果没有就不用执行了
|
||
if (sqlParams == null) return keyValues;
|
||
string connstr = Myconnstr(KD);
|
||
using SqlConnection connection = new SqlConnection(connstr);
|
||
try
|
||
{
|
||
connection.Open();
|
||
using var cmd = connection.CreateCommand();
|
||
//cmd.CommandTimeout = 0;
|
||
PrepareCommand(cmd, CommandType.StoredProcedure, cmdText, sqlParams);
|
||
cmd.Transaction = connection.BeginTransaction();
|
||
try
|
||
{
|
||
cmd.ExecuteNonQuery();
|
||
cmd.Transaction.Commit();
|
||
foreach (DbParameter param in sqlParams)
|
||
{
|
||
if (param.Direction == ParameterDirection.Output || param.Direction == ParameterDirection.ReturnValue)
|
||
{
|
||
if (param.Value != null)
|
||
{
|
||
keyValues.Add(param.ParameterName, param.Value);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
cmd.Transaction.Rollback();
|
||
mess = ex.Message;
|
||
LogHelper.WriteLog(string.Format("执行存储过程，并获得返回值的GetProcedurePairs方法发生错误在：{0}", ex.Message + "\n" + cmdText));
|
||
}
|
||
finally
|
||
{
|
||
cmd.Dispose();
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
mess = ex.Message;
|
||
LogHelper.WriteLog(string.Format("执行存储过程，并获得返回值的GetProcedurePairs方法发生错误：{0}", ex.Message));
|
||
}
|
||
finally
|
||
{
|
||
connection.Dispose();
|
||
}
|
||
return keyValues;
|
||
//throw new NotImplementedException();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 为查询类赋值
|
||
/// </summary>
|
||
/// <param name="cmd"></param>
|
||
/// <param name="cmdType"></param>
|
||
/// <param name="cmdText"></param>
|
||
/// <param name="cmdParms"></param>
|
||
private static void PrepareCommand(IDbCommand cmd, CommandType cmdType, string cmdText, SqlParameter[]? cmdParms)
|
||
{
|
||
cmd.CommandText = cmdText;
|
||
cmd.CommandType = cmdType;
|
||
if (cmdParms != null)
|
||
{
|
||
foreach (DbParameter parm in cmdParms)
|
||
cmd.Parameters.Add(parm);
|
||
}
|
||
}
|
||
/// <inheritdoc/>
|
||
|
||
public bool InsertBill(DataTable idata, string TableName, ref string? reason, bool KD = false)
|
||
{
|
||
bool isok = false;
|
||
reason = null;
|
||
if (TableName == null || TableName == "")
|
||
{
|
||
reason += "未提供数据表的表名";
|
||
}
|
||
if (idata == null)
|
||
{
|
||
reason += "目标表不能为空";
|
||
}
|
||
if (idata!.Rows.Count == 0)
|
||
{
|
||
reason += "目标表中没有可导入数据";
|
||
}
|
||
if (!string.IsNullOrWhiteSpace(reason))
|
||
{
|
||
return isok;
|
||
}
|
||
string connstr = Myconnstr(KD);
|
||
using SqlConnection connection = new SqlConnection(connstr);
|
||
try
|
||
{
|
||
connection.Open();
|
||
using (SqlBulkCopy sqlBulk = new(connection))
|
||
{
|
||
sqlBulk.DestinationTableName = TableName;
|
||
foreach (DataColumn column in idata.Columns)
|
||
{
|
||
string NewEntryColunm = column.ColumnName;
|
||
sqlBulk.ColumnMappings.Add(NewEntryColunm, NewEntryColunm); //由于提前已经准备好单据了，这两个参数保持一致就行了
|
||
}
|
||
sqlBulk.BatchSize = idata.Rows.Count;
|
||
sqlBulk.WriteToServer(idata);
|
||
sqlBulk.Close();
|
||
}
|
||
isok = true;
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
reason = ex.Message;
|
||
LogHelper.WriteLog(string.Format("将整理好的表单批量导入数据库的InsertBill方法在WriteToServer处发生错误：{0}", ex.Message));
|
||
}
|
||
finally
|
||
{
|
||
connection.Dispose();
|
||
}
|
||
return isok;
|
||
}
|
||
}
|
||
}
|