114 lines
4.2 KiB
C#
114 lines
4.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Data;
|
|
using System.Data.Common;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using Npgsql;
|
|
|
|
namespace SqlSugar
|
|
{
|
|
public partial class PostgreSQLProvider : AdoProvider
|
|
{
|
|
public PostgreSQLProvider() { }
|
|
public override IDbConnection Connection
|
|
{
|
|
get
|
|
{
|
|
if (base._DbConnection == null)
|
|
{
|
|
try
|
|
{
|
|
var npgsqlConnectionString = base.Context.CurrentConnectionConfig.ConnectionString;
|
|
base._DbConnection = new NpgsqlConnection(npgsqlConnectionString);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Check.Exception(true, ErrorMessage.ConnnectionOpen, ex.Message);
|
|
}
|
|
}
|
|
return base._DbConnection;
|
|
}
|
|
set
|
|
{
|
|
base._DbConnection = value;
|
|
}
|
|
}
|
|
|
|
public override void BeginTran(string transactionName)
|
|
{
|
|
base.BeginTran();
|
|
}
|
|
/// <summary>
|
|
/// Only SqlServer
|
|
/// </summary>
|
|
/// <param name="iso"></param>
|
|
/// <param name="transactionName"></param>
|
|
public override void BeginTran(IsolationLevel iso, string transactionName)
|
|
{
|
|
base.BeginTran(iso);
|
|
}
|
|
public override IDataAdapter GetAdapter()
|
|
{
|
|
return new NpgsqlDataAdapter();
|
|
}
|
|
public override DbCommand GetCommand(string sql, SugarParameter[] parameters)
|
|
{
|
|
NpgsqlCommand sqlCommand = new NpgsqlCommand(sql, (NpgsqlConnection)this.Connection);
|
|
sqlCommand.CommandType = this.CommandType;
|
|
sqlCommand.CommandTimeout = this.CommandTimeOut;
|
|
if (this.Transaction != null)
|
|
{
|
|
sqlCommand.Transaction = (NpgsqlTransaction)this.Transaction;
|
|
}
|
|
if (parameters.HasValue())
|
|
{
|
|
IDataParameter[] ipars = ToIDbDataParameter(parameters);
|
|
sqlCommand.Parameters.AddRange((NpgsqlParameter[])ipars);
|
|
}
|
|
CheckConnection();
|
|
return sqlCommand;
|
|
}
|
|
public override void SetCommandToAdapter(IDataAdapter dataAdapter, DbCommand command)
|
|
{
|
|
((NpgsqlDataAdapter)dataAdapter).SelectCommand = (NpgsqlCommand)command;
|
|
}
|
|
/// <summary>
|
|
/// if mysql return MySqlParameter[] pars
|
|
/// if sqlerver return SqlParameter[] pars ...
|
|
/// </summary>
|
|
/// <param name="parameters"></param>
|
|
/// <returns></returns>
|
|
public override IDataParameter[] ToIDbDataParameter(params SugarParameter[] parameters)
|
|
{
|
|
if (parameters == null || parameters.Length == 0) return null;
|
|
NpgsqlParameter[] result = new NpgsqlParameter[parameters.Length];
|
|
int index = 0;
|
|
foreach (var parameter in parameters)
|
|
{
|
|
if (parameter.Value == null) parameter.Value = DBNull.Value;
|
|
var sqlParameter = new NpgsqlParameter();
|
|
sqlParameter.ParameterName = parameter.ParameterName;
|
|
sqlParameter.Size = parameter.Size;
|
|
sqlParameter.Value = parameter.Value;
|
|
sqlParameter.DbType = parameter.DbType;
|
|
sqlParameter.Direction = parameter.Direction;
|
|
if (sqlParameter.Direction == 0)
|
|
{
|
|
sqlParameter.Direction = ParameterDirection.Input;
|
|
}
|
|
result[index] = sqlParameter;
|
|
if (sqlParameter.Direction.IsIn(ParameterDirection.Output, ParameterDirection.InputOutput, ParameterDirection.ReturnValue))
|
|
{
|
|
if (this.OutputParameters == null) this.OutputParameters = new List<IDataParameter>();
|
|
this.OutputParameters.RemoveAll(it => it.ParameterName == sqlParameter.ParameterName);
|
|
this.OutputParameters.Add(sqlParameter);
|
|
}
|
|
++index;
|
|
}
|
|
return result;
|
|
}
|
|
}
|
|
}
|