收集的一些数据库操作代码

Posted on 2009-07-03 17:53  sky-v  阅读(288)  评论(0)    收藏  举报

废话不说,上代码

 

public static class DataHelper
    {
        /// <summary>
        /// 返回从指定位置开始的普通或包含二进制数据的指定个数的记录(用DataTable包装)
        /// </summary>
        /// <param name="dataReader">一个DataReader</param>
        /// <param name="location">开始位置(从0开始的索引)</param>
        /// <param name="count">指定的记录个数</param>
        /// <returns>返回一个DataTable,包含了记录值</returns>
        public static DataTable PackTable(IDataReader dataReader, int location, int count)
        {
            if (dataReader == null)
            {
                return null;
            }
            if (!(location >= 0 && count >= 0))
            {
                return null;
            }

            DataTable tmpTable = new DataTable();
            for (int i = 0; i < dataReader.FieldCount; i++)
            {
                DataColumn tmpCol = tmpTable.Columns.Add(dataReader.GetName(i));
            }

            int n = 1;
            while (dataReader.Read())
            {
                if (n > location && n <= location + count)
                {
                    DataRow tmpRow = tmpTable.NewRow();
                    for (int i = 0; i < dataReader.FieldCount; i++)
                    {
                        tmpRow[i] = dataReader[i];
                    }
                    tmpTable.Rows.Add(tmpRow);
                }
                if (n == location + count)
                {
                    break;
                }
                n++;
            }
            dataReader.Close();
            dataReader.Dispose();
            return tmpTable;
        }

        public static object ExecuteScalar(string connString, CommandType cmdType, string cmdText, params SqlParameter[] cmdParms)
        {
            SqlConnection conn = new SqlConnection(connString);
            SqlCommand cmd = conn.CreateCommand();
            try
            {
                PrepareCommand(cmd, conn, cmdType, cmdText, cmdParms);
                object val = cmd.ExecuteScalar();
                cmd.Parameters.Clear();
                conn.Close();
                return val;
            }
            catch (Exception e)
            {
                if (conn.State == ConnectionState.Open)
                {
                    conn.Close();
                }
                throw new ApplicationException(e.Message + cmdText, e);
            }
        }

        public static int ExecuteNonQuery(string connString, CommandType cmdType, string cmdText, params SqlParameter[] cmdParms)
        {
            SqlConnection conn = new SqlConnection(connString);
            SqlCommand cmd = conn.CreateCommand();
            try
            {
                PrepareCommand(cmd, conn, cmdType, cmdText, cmdParms);
                int val = cmd.ExecuteNonQuery();
                cmd.Parameters.Clear();
                conn.Close();
                return val;
            }
            catch (Exception e)
            {
                if (conn.State == ConnectionState.Open)
                {
                    conn.Close();
                }
                throw new ApplicationException(e.Message + cmdText, e);
            }
        }

        public static DataTable ExecuteDataTable(string connString, CommandType cmdType, string cmdText, params SqlParameter[] cmdParms)
        {
            SqlConnection conn = new SqlConnection(connString);
            SqlDataAdapter da = new SqlDataAdapter();
            SqlCommand cmd = conn.CreateCommand();
            try
            {
                PrepareCommand(cmd, conn, cmdType, cmdText, cmdParms);
                da.SelectCommand = cmd;
                DataTable dt = new DataTable();
                da.Fill(dt);
                cmd.Parameters.Clear();
                conn.Close();
                return dt;
            }
            catch (Exception e)
            {
                if (conn.State == ConnectionState.Open)
                {
                    conn.Close();
                }
                throw new ApplicationException(e.Message + cmdText, e);
            }
        }

        public static IDataReader ExecuteDataReader(string connString, CommandType cmdType, string cmdText, params SqlParameter[] cmdParms)
        {
            SqlConnection conn = new SqlConnection(connString);
            SqlCommand cmd = conn.CreateCommand();
            try
            {
                PrepareCommand(cmd, conn, cmdType, cmdText, cmdParms);
                SqlDataReader dr = cmd.ExecuteReader(CommandBehavior.CloseConnection);
                cmd.Parameters.Clear();
                return dr;
            }
            catch (Exception e)
            {
                if (conn.State == ConnectionState.Open)
                {
                    conn.Close();
                }
                throw new ApplicationException(e.Message + cmdText, e);
            }
        }

        private static void PrepareCommand(SqlCommand cmd, SqlConnection conn, CommandType cmdType, string cmdText, SqlParameter[] cmdParms)
        {
            if (conn.State != ConnectionState.Open)
            {
                conn.Open();
            }
            cmd.Connection = conn;
            cmd.CommandText = cmdText;
            cmd.CommandType = cmdType;
            if (cmdParms != null)
            {
                foreach (SqlParameter parm in cmdParms)
                {
                    cmd.Parameters.Add(parm);
                }
            }
        }

        /// <summary>
        /// 执行存储过程,返回影响的行数  
        /// </summary>
        /// <param name="storedProcName">存储过程名</param>
        /// <param name="parameters">存储过程参数</param>
        /// <param name="rowsAffected">影响的行数</param>
        /// <returns></returns>
        public static int RunProcedure(string connectionString,string storedProcName, IDataParameter[] parameters, out int rowsAffected)
        {
            using (SqlConnection connection = new SqlConnection(connectionString))
            {
                int result;
                connection.Open();
                SqlCommand command = BuildIntCommand(connection, storedProcName, parameters);
                rowsAffected = command.ExecuteNonQuery();
                result = (int)command.Parameters["ReturnValue"].Value;
                //Connection.Close();
                return result;
            }
        }

        /// <summary>
        /// 执行有返回值的存储过程,返回结果、Message
        /// </summary>
        /// <param name="storedProcName">存储过程名</param>
        /// <param name="parameters">存储过程参数</param>
        /// <returns></returns>
        public static int RunProcedure(string connectionString, string storedProcName, IDataParameter[] parameters, out string Message)
        {
            using (SqlConnection connection = new SqlConnection(connectionString))
            {
                int result;
                connection.Open();


                SqlCommand command = BuildIntCommand(connection, storedProcName, parameters);

                command.ExecuteNonQuery();
                Message = command.Parameters["@cMessage"].Value.ToString();

                result = int.Parse(command.Parameters["ReturnValue"].Value.ToString());

                return result;
            }
        }

        public static void RunProcedure(string connectionString, string storedProcName, IDataParameter[] parameters)
        {
            using (SqlConnection connection = new SqlConnection(connectionString))
            {
                connection.Open();
                SqlCommand command = BuildIntCommand(connection, storedProcName, parameters);

                command.ExecuteNonQuery();
            }
        }

        /// <summary>
        /// 执行存储过程
        /// </summary>
        /// <param name="storedProcName">存储过程名</param>
        /// <param name="parameters">存储过程参数</param>
        /// <param name="tableName">DataSet结果中的表名</param>
        /// <returns>DataSet</returns>
        public static DataSet RunProcedure(string connectionString, string storedProcName, IDataParameter[] parameters, string tableName)
        {
            using (SqlConnection connection = new SqlConnection(connectionString))
            {
                DataSet dataSet = new DataSet();
                connection.Open();
                SqlDataAdapter sqlDA = new SqlDataAdapter();
                sqlDA.SelectCommand = BuildQueryCommand(connection, storedProcName, parameters);
                sqlDA.Fill(dataSet, tableName);
                connection.Close();
                return dataSet;
            }
        }

        /// <summary>
        /// 构建 SqlCommand 对象(用来返回一个结果集,而不是一个整数值)
        /// </summary>
        /// <param name="connection">数据库连接</param>
        /// <param name="storedProcName">存储过程名</param>
        /// <param name="parameters">存储过程参数</param>
        /// <returns>SqlCommand</returns>
        private static SqlCommand BuildQueryCommand(SqlConnection connection, string storedProcName, IDataParameter[] parameters)
        {
            SqlCommand command = new SqlCommand(storedProcName, connection);
            command.CommandType = CommandType.StoredProcedure;
            foreach (SqlParameter parameter in parameters)
            {
                if (parameter != null)
                {
                    // 检查未分配值的输出参数,将其分配以DBNull.Value.
                    if ((parameter.Direction == ParameterDirection.InputOutput || parameter.Direction == ParameterDirection.Input) &&
                        (parameter.Value == null))
                    {
                        parameter.Value = DBNull.Value;
                    }
                    command.Parameters.Add(parameter);
                }
            }

            return command;
        }

        /// <summary>
        /// 创建 SqlCommand 对象实例(用来返回一个整数值) 
        /// </summary>
        /// <param name="storedProcName">存储过程名</param>
        /// <param name="parameters">存储过程参数</param>
        /// <returns>SqlCommand 对象实例</returns>
        private static SqlCommand BuildIntCommand(SqlConnection connection, string storedProcName, IDataParameter[] parameters)
        {
            SqlCommand command = BuildQueryCommand(connection, storedProcName, parameters);
            command.Parameters.Add(new SqlParameter("ReturnValue",
                SqlDbType.Int, 4, ParameterDirection.ReturnValue,
                false, 0, 0, string.Empty, DataRowVersion.Default, null));
            return command;
        }

        public static DataTable GetErrInfo(string msg)
        {
            DataTable dt = new DataTable();
            dt.Columns.Add("err");
            DataRow dr = dt.NewRow();
            dr[0] = msg;
            dt.Rows.Add(dr);
            return dt;
        }

}

博客园  ©  2004-2026
浙公网安备 33010602011771号 浙ICP备2021040463号-3