CNAS/CnasSynchronusDAL/DAL/Helper/PostgreSQLHelper.cs

using System;
using System.Collections.Generic;
using System.Data;
using System.Linq;
using System.Text;
using CnasSynchronousCommon;
using Npgsql;
using NpgsqlTypes;

namespace CnasSynchronusDAL
{
    public static class PostgreSQLHelper
    {

        private static String mConnStr = null;

        public static void InitConnectionString(string strServerIP,string strServerPort,string strServerHost,string strServerUser,string strServerPwd)
        {
            mConnStr = $"Server={strServerIP};Port={strServerPort};Database={strServerHost};User Id={strServerUser};Password={strServerPwd};";
        }

        public static void InitConnectionString(string strConnectiong)
        {
            mConnStr = strConnectiong;
        }

        /// <summary>  
        /// 对SQLite数据库执行增删改操作，返回受影响的行数。  
        /// </summary>  
        /// <param name="sql">要执行的增删改的SQL语句</param>  
        /// <returns></returns>  
        public static int ExecuteNonQuery(String sql)
        {
            NpgsqlConnection connection = new NpgsqlConnection(mConnStr);
            try
            {
                using (connection)
                {
                    connection.Open();
                    NpgsqlTransaction transaction = connection.BeginTransaction();

                    using (NpgsqlCommand cmd = new NpgsqlCommand())
                    {
                        try
                        {
                            PrepareCommand(cmd, connection, transaction, CommandType.Text, sql, null);

                            int rows = cmd.ExecuteNonQuery();
                            transaction.Commit();

                            cmd.Parameters.Clear();
                            return rows;
                        }
                        catch (NpgsqlException e1)
                        {
                            try
                            {
                                transaction.Rollback();
                            }
                            catch (Exception e2)
                            {
                                AppLog.Error(e2.Message);
                                throw;
                            }
                            AppLog.Error(e1.Message);
                            throw;
                        }
                    }
                }
            }
            catch (Exception e)
            {
                AppLog.Error(e.Message);
                throw;
            }
        }

        /// <summary>  
        /// 对数据库执行增删改操作，返回受影响的行数。  
        /// </summary>  
        /// <param name="sql">要执行的增删改的SQL语句</param>  
        /// <returns></returns>  
        public static int ExecuteNonQuery(String sql, NpgsqlParameter[] cmdParams)
        {
            try
            {
                using (NpgsqlConnection connection = new NpgsqlConnection(mConnStr))
                {
                    connection.Open();
                    NpgsqlTransaction transaction = connection.BeginTransaction();

                    using (NpgsqlCommand cmd = new NpgsqlCommand())
                    {
                        try
                        {
                            PrepareCommand(cmd, connection, transaction, CommandType.Text, sql, cmdParams);

                            int rows = cmd.ExecuteNonQuery();
                            transaction.Commit();

                            cmd.Parameters.Clear();
                            return rows;
                        }
                        catch (NpgsqlException e1)
                        {
                            try
                            {
                                transaction.Rollback();
                            }
                            catch (Exception e2)
                            {
                                AppLog.Error(e2.Message);
                                throw;
                            }
                            AppLog.Error(e1.Message);
                            throw;
                        }
                    }
                }
            }
            catch (Exception e)
            {
                AppLog.Error(e.Message);
                throw;
            }
        }

        /// <summary>
        /// 对SQLite数据库执行操作，返回 返回第一行第一列数据
        /// </summary>
        /// <param name="sql"></param>
        /// <returns></returns>
        public static int ExecuteScalar(String sql)
        {
            try
            {
                using (NpgsqlConnection connection = new NpgsqlConnection(mConnStr))
                {
                    connection.Open();
                    NpgsqlTransaction transaction = connection.BeginTransaction();

                    using (NpgsqlCommand cmd = new NpgsqlCommand())
                    {
                        try
                        {
                            int line = 0;

                            PrepareCommand(cmd, connection, transaction, CommandType.Text, sql, null);

                            String str = cmd.ExecuteScalar().ToString();
                            transaction.Commit();

                            line = Convert.ToInt32(str);
                            cmd.Parameters.Clear();

                            return line;
                        }
                        catch (NpgsqlException e1)
                        {
                            try
                            {
                                transaction.Rollback();
                            }
                            catch (Exception e2)
                            {
                                AppLog.Error(e2.Message);
                                throw;
                            }

                            AppLog.Error(e1.Message);
                            throw;
                        }
                    }
                }
            }
            catch (Exception e)
            {
                AppLog.Error(e.Message);
                throw;
            }
        }

        /// <summary>
        /// 对SQLite数据库执行操作，返回 返回第一行第一列数据
        /// </summary>
        /// <param name="sql"></param>
        /// <returns></returns>
        public static int ExecuteScalar(String sql, NpgsqlParameter[] cmdParams)
        {
            try
            {
                using (NpgsqlConnection connection = new NpgsqlConnection(mConnStr))
                {
                    connection.Open();
                    NpgsqlTransaction transaction = connection.BeginTransaction();

                    using (NpgsqlCommand cmd = new NpgsqlCommand())
                    {
                        try
                        {
                            int line = 0;

                            PrepareCommand(cmd, connection, transaction, CommandType.Text, sql, cmdParams);

                            String str = cmd.ExecuteScalar().ToString();
                            transaction.Commit();

                            line = Convert.ToInt32(str);
                            cmd.Parameters.Clear();

                            return line;
                        }
                        catch (NpgsqlException e1)
                        {
                            try
                            {
                                transaction.Rollback();
                            }
                            catch (Exception e2)
                            {
                                AppLog.Error(e2.Message);
                                throw;
                            }

                            AppLog.Error(e1.Message);
                            throw;
                        }
                    }
                }
            }
            catch (Exception e)
            {
                AppLog.Error(e.Message);
                throw;
            }
        }

        /// <summary>
        ///  用执行的数据库连接执行一个返回数据集的sql命令
        /// </summary>
        /// <param name="sql"></param>
        /// <returns></returns>
        public static NpgsqlDataReader ExecuteReader(String sql)
        {
            try
            {
                //创建一个NpgsqlConnection对象
                using (NpgsqlConnection connection = new NpgsqlConnection(mConnStr))
                {
                    connection.Open();
                    NpgsqlTransaction transaction = connection.BeginTransaction();

                    //创建一个NpgsqlCommand对象
                    using (NpgsqlCommand cmd = new NpgsqlCommand())
                    {
                        try
                        {
                            PrepareCommand(cmd, connection, transaction, CommandType.Text, sql, null);

                            NpgsqlDataReader reader = cmd.ExecuteReader(CommandBehavior.CloseConnection);
                            transaction.Commit();

                            cmd.Parameters.Clear();
                            return reader;
                        }
                        catch (NpgsqlException e1)
                        {
                            try
                            {
                                transaction.Rollback();
                            }
                            catch (Exception e2)
                            {
                                AppLog.Error(e2.Message);
                                throw;
                            }

                            AppLog.Error(e1.Message);
                            throw;
                        }
                    }
                }
            }
            catch (Exception e)
            {
                AppLog.Error(e.Message);
                throw;
            }
        }

        /// <summary>
        /// 查询返回Dtaset
        /// </summary>
        /// <param name="sql"></param>
        /// <returns></returns>
        public static DataSet ExecuteDataSet(String sql)
        {
            NpgsqlConnection connection = new NpgsqlConnection(mConnStr);
            try
            {
                //创建一个NpgsqlConnection对象
                using (connection)
                {
                    connection.Open();
                    NpgsqlTransaction transaction = connection.BeginTransaction();

                    //创建一个NpgsqlCommand对象
                    using (NpgsqlCommand cmd = new NpgsqlCommand())
                    {
                        try
                        {
                            PrepareCommand(cmd, connection, transaction, CommandType.Text, sql, null);

                            NpgsqlDataAdapter adapter = new NpgsqlDataAdapter();
                            adapter.SelectCommand = cmd;
                            DataSet ds = new DataSet();
                            adapter.Fill(ds);
                            transaction.Commit();
                            //清除参数
                            cmd.Parameters.Clear();
                            return ds;
                        }
                        catch (NpgsqlException e1)
                        {
                            try
                            {
                                transaction.Rollback();
                            }
                            catch (Exception e2)
                            {
                                AppLog.Error(e2.Message);
                                throw;
                            }

                            AppLog.Error(e1.Message);
                            throw;
                        }
                    }
                }
            }
            catch (Exception e)
            {
                AppLog.Error(e.Message);
                throw;
            }finally
            {
                if (connection != null)
                {
                    //关闭数据库连接
                    connection.Close();
                }
            }
        }


        /// <summary>
        /// 执行sql 返回一个DataTable
        /// </summary>
        /// <param name="sqlText"></param>
        /// <param name="parameters"></param>
        /// <returns></returns>
        public static DataTable ExecuteDataTable(string sqlText, params NpgsqlParameter[] parameters)
        {
            DataTable dt = null;
            try
            {
                using (NpgsqlDataAdapter adapter = new NpgsqlDataAdapter(sqlText, mConnStr))
                {
                    dt = new DataTable();
                    adapter.SelectCommand.Parameters.AddRange(parameters);
                    adapter.Fill(dt);
                    return dt;
                }
            }
            catch (Exception ex)
            {
                AppLog.Error(ex.Message);
            }
            return null;
        }

        /// <summary>
        /// 查询返回Dtaset
        /// </summary>
        /// <param name="sql"></param>
        /// <returns></returns>
        public static DataSet ExecuteDataSet(String sql, NpgsqlParameter[] cmdParams)
        {
            NpgsqlConnection connection = new NpgsqlConnection(mConnStr);
            try
            {
                //创建一个NpgsqlConnection对象
                using (connection)
                {
                    connection.Open();
                    NpgsqlTransaction transaction = connection.BeginTransaction();

                    //创建一个NpgsqlCommand对象
                    using (NpgsqlCommand cmd = new NpgsqlCommand())
                    {
                        try
                        {
                            PrepareCommand(cmd, connection, transaction, CommandType.Text, sql, cmdParams);

                            NpgsqlDataAdapter adapter = new NpgsqlDataAdapter();
                            adapter.SelectCommand = cmd;
                            DataSet ds = new DataSet();

                            adapter.Fill(ds);

                            transaction.Commit();

                            //清除参数
                            cmd.Parameters.Clear();
                            return ds;

                        }
                        catch (NpgsqlException e1)
                        {
                            try
                            {
                                transaction.Rollback();
                            }
                            catch (Exception e2)
                            {
                                AppLog.Error(e2.Message);
                                throw;
                            }

                            AppLog.Error(e1.Message);
                            throw;
                        }
                    }
                }
            }
            catch (Exception e)
            {
                AppLog.Error(e.Message);
                throw;
            }finally
            {
                if (connection != null)
                {
                    //关闭数据库连接
                    connection.Close();
                }
            }
        }

        /// <summary>
        /// 准备执行一个命令
        /// </summary>
        /// <param name="cmd">sql命令</param>
        /// <param name="conn">OleDb连接</param>
        /// <param name="trans">OleDb事务</param>
        /// <param name="cmdType">命令类型例如 存储过程或者文本</param>
        /// <param name="cmdText">命令文本,例如:Select * from Products</param>
        /// <param name="cmdParms">执行命令的参数</param>
        private static void PrepareCommand(NpgsqlCommand cmd, NpgsqlConnection conn, NpgsqlTransaction trans, CommandType cmdType, string cmdText, NpgsqlParameter[] cmdParms)
        {
            if (conn.State != ConnectionState.Open)
                conn.Open();

            cmd.Connection = conn;
            cmd.CommandText = cmdText;

            if (trans != null)
                cmd.Transaction = trans;

            cmd.CommandType = cmdType;

            if (cmdParms != null)
            {
                foreach (NpgsqlParameter parm in cmdParms)
                    cmd.Parameters.Add(parm);
            }
        }


        public static bool TestConnectPostgreSQL()
        {
            bool bIfSuccess = false;
            try
            {
                using (NpgsqlConnection connection = new NpgsqlConnection(mConnStr))
                {
                    connection.Open();

                    if (connection.State == System.Data.ConnectionState.Open)
                        bIfSuccess = true;
                }
            }
            catch (Exception ex)
            {
                AppLog.Error(ex.Message);
                bIfSuccess = false;
            }

            return bIfSuccess;
        }
            
    }
}