SQLiteHelper

来源:互联网 发布:深圳网站seo 编辑:程序博客网 时间:2024/05/20 22:36
using System;using System.Data;using System.Text.RegularExpressions;using System.Xml;using System.IO;using System.Collections;using System.Data.SQLite;namespace Util{    /// <summary>    /// SQLiteHelper is a utility class similar to "SQLHelper" in MS    /// Data Access Application Block and follows similar pattern.    /// </summary>    public class SQLiteHelper    {        /// <summary>        /// Creates a new <see cref="SQLiteHelper"/> instance. The ctor is marked private since all members are static.        /// </summary>        private SQLiteHelper()        {        }        /// <summary>        /// Creates the command.        /// </summary>        /// <param name="connection">Connection.</param>        /// <param name="commandText">Command text.</param>        /// <param name="commandParameters">Command parameters.</param>        /// <returns>SQLite Command</returns>        public static SQLiteCommand CreateCommand(SQLiteConnection connection, string commandText, params SQLiteParameter[] commandParameters)        {            SQLiteCommand cmd = new SQLiteCommand(commandText, connection);            if (commandParameters.Length > 0)            {                foreach (SQLiteParameter parm in commandParameters)                    cmd.Parameters.Add(parm);            }            return cmd;        }        /// <summary>        /// Creates the command.        /// </summary>        /// <param name="connectionString">Connection string.</param>        /// <param name="commandText">Command text.</param>        /// <param name="commandParameters">Command parameters.</param>        /// <returns>SQLite Command</returns>        public static SQLiteCommand CreateCommand(string connectionString, string commandText, params SQLiteParameter[] commandParameters)        {            SQLiteConnection cn = new SQLiteConnection(connectionString);            SQLiteCommand cmd = new SQLiteCommand(commandText, cn);            if (commandParameters.Length > 0)            {                foreach (SQLiteParameter parm in commandParameters)                    cmd.Parameters.Add(parm);            }            return cmd;        }        /// <summary>        /// Creates the parameter.        /// </summary>        /// <param name="parameterName">Name of the parameter.</param>        /// <param name="parameterType">Parameter type.</param>        /// <param name="parameterValue">Parameter value.</param>        /// <returns>SQLiteParameter</returns>        public static SQLiteParameter CreateParameter(string parameterName, System.Data.DbType parameterType, object parameterValue)        {            SQLiteParameter parameter = new SQLiteParameter();            parameter.DbType = parameterType;            parameter.ParameterName = parameterName;            parameter.Value = parameterValue;            return parameter;        }        /// <summary>        /// Shortcut method to execute dataset from SQL Statement and object[] arrray of parameter values        /// </summary>        /// <param name="connectionString">SQLite Connection string</param>        /// <param name="commandText">SQL Statement with embedded "@param" style parameter names</param>        /// <param name="paramList">object[] array of parameter values</param>        /// <returns></returns>        public static DataSet ExecuteDataSet(string connectionString, string commandText, object[] paramList)        {            SQLiteConnection cn = new SQLiteConnection(connectionString);            SQLiteCommand cmd = cn.CreateCommand();            cmd.CommandText = commandText;            if (paramList != null)            {                AttachParameters(cmd, commandText, paramList);            }            DataSet ds = new DataSet();            if (cn.State == ConnectionState.Closed)                cn.Open();            SQLiteDataAdapter da = new SQLiteDataAdapter(cmd);            da.Fill(ds);            da.Dispose();            cmd.Dispose();            cn.Close();            return ds;        }        /// <summary>        /// Shortcut method to execute dataset from SQL Statement and object[] arrray of  parameter values        /// </summary>        /// <param name="cn">Connection.</param>        /// <param name="commandText">Command text.</param>        /// <param name="paramList">Param list.</param>        /// <returns></returns>        public static DataSet ExecuteDataSet(SQLiteConnection cn, string commandText, object[] paramList)        {            SQLiteCommand cmd = cn.CreateCommand();            cmd.CommandText = commandText;            if (paramList != null)            {                AttachParameters(cmd, commandText, paramList);            }            DataSet ds = new DataSet();            if (cn.State == ConnectionState.Closed)                cn.Open();            SQLiteDataAdapter da = new SQLiteDataAdapter(cmd);            da.Fill(ds);            da.Dispose();            cmd.Dispose();            cn.Close();            return ds;        }        /// <summary>        /// Executes the dataset from a populated Command object.        /// </summary>        /// <param name="cmd">Fully populated SQLiteCommand</param>        /// <returns>DataSet</returns>        public static DataSet ExecuteDataset(SQLiteCommand cmd)        {            if (cmd.Connection.State == ConnectionState.Closed)                cmd.Connection.Open();            DataSet ds = new DataSet();            SQLiteDataAdapter da = new SQLiteDataAdapter(cmd);            da.Fill(ds);            da.Dispose();            cmd.Connection.Close();            cmd.Dispose();            return ds;        }        /// <summary>        /// Executes the dataset in a SQLite Transaction        /// </summary>        /// <param name="transaction">SQLiteTransaction. Transaction consists of Connection, Transaction,  /// and Command, all of which must be created prior to making this method call. </param>        /// <param name="commandText">Command text.</param>        /// <param name="commandParameters">Sqlite Command parameters.</param>        /// <returns>DataSet</returns>        /// <remarks>user must examine Transaction Object and handle transaction.connection .Close, etc.</remarks>        public static DataSet ExecuteDataset(SQLiteTransaction transaction, string commandText, params SQLiteParameter[] commandParameters)        {            if (transaction == null) throw new ArgumentNullException("transaction");            if (transaction != null && transaction.Connection == null) throw new ArgumentException("The transaction was rolled back or committed, please provide an open transaction.", "transaction");            IDbCommand cmd = transaction.Connection.CreateCommand();            cmd.CommandText = commandText;            foreach (SQLiteParameter parm in commandParameters)            {                cmd.Parameters.Add(parm);            }            if (transaction.Connection.State == ConnectionState.Closed)                transaction.Connection.Open();            DataSet ds = ExecuteDataset((SQLiteCommand)cmd);            return ds;        }        /// <summary>        /// Executes the dataset with Transaction and object array of parameter values.        /// </summary>        /// <param name="transaction">SQLiteTransaction. Transaction consists of Connection, Transaction,    /// and Command, all of which must be created prior to making this method call. </param>        /// <param name="commandText">Command text.</param>        /// <param name="commandParameters">object[] array of parameter values.</param>        /// <returns>DataSet</returns>        /// <remarks>user must examine Transaction Object and handle transaction.connection .Close, etc.</remarks>        public static DataSet ExecuteDataset(SQLiteTransaction transaction, string commandText, object[] commandParameters)        {            if (transaction == null) throw new ArgumentNullException("transaction");            if (transaction != null && transaction.Connection == null) throw new ArgumentException("The transaction was rolled back or committed,                                                          please provide an open transaction.", "transaction");            IDbCommand cmd = transaction.Connection.CreateCommand();            cmd.CommandText = commandText;            AttachParameters((SQLiteCommand)cmd, cmd.CommandText, commandParameters);            if (transaction.Connection.State == ConnectionState.Closed)                transaction.Connection.Open();            DataSet ds = ExecuteDataset((SQLiteCommand)cmd);            return ds;        }        #region UpdateDataset        /// <summary>        /// Executes the respective command for each inserted, updated, or deleted row in the DataSet.        /// </summary>        /// <remarks>        /// e.g.:          ///  UpdateDataset(conn, insertCommand, deleteCommand, updateCommand, dataSet, "Order");        /// </remarks>        /// <param name="insertCommand">A valid SQL statement  to insert new records into the data source</param>        /// <param name="deleteCommand">A valid SQL statement to delete records from the data source</param>        /// <param name="updateCommand">A valid SQL statement used to update records in the data source</param>        /// <param name="dataSet">The DataSet used to update the data source</param>        /// <param name="tableName">The DataTable used to update the data source.</param>        public static void UpdateDataset(SQLiteCommand insertCommand, SQLiteCommand deleteCommand, SQLiteCommand updateCommand, DataSet dataSet, string tableName)        {            if (insertCommand == null) throw new ArgumentNullException("insertCommand");            if (deleteCommand == null) throw new ArgumentNullException("deleteCommand");            if (updateCommand == null) throw new ArgumentNullException("updateCommand");            if (tableName == null || tableName.Length == 0) throw new ArgumentNullException("tableName");            // Create a SQLiteDataAdapter, and dispose of it after we are done            using (SQLiteDataAdapter dataAdapter = new SQLiteDataAdapter())            {                // Set the data adapter commands                dataAdapter.UpdateCommand = updateCommand;                dataAdapter.InsertCommand = insertCommand;                dataAdapter.DeleteCommand = deleteCommand;                // Update the dataset changes in the data source                dataAdapter.Update(dataSet, tableName);                // Commit all the changes made to the DataSet                dataSet.AcceptChanges();            }        }        #endregion        /// <summary>        /// ShortCut method to return IDataReader        /// NOTE: You should explicitly close the Command.connection you passed in as        /// well as call Dispose on the Command  after reader is closed.        /// We do this because IDataReader has no underlying Connection Property.        /// </summary>        /// <param name="cmd">SQLiteCommand Object</param>        /// <param name="commandText">SQL Statement with optional embedded "@param" style parameters</param>        /// <param name="paramList">object[] array of parameter values</param>        /// <returns>IDataReader</returns>        public static IDataReader ExecuteReader(SQLiteCommand cmd, string commandText, object[] paramList)        {            if (cmd.Connection == null)                throw new ArgumentException("Command must have live connection attached.", "cmd");            cmd.CommandText = commandText;            AttachParameters(cmd, commandText, paramList);            if (cmd.Connection.State == ConnectionState.Closed)                cmd.Connection.Open();            IDataReader rdr = cmd.ExecuteReader(CommandBehavior.CloseConnection);            return rdr;        }        /// <summary>        /// Shortcut to ExecuteNonQuery with SqlStatement and object[] param values        /// </summary>        /// <param name="connectionString">SQLite Connection String</param>        /// <param name="commandText">Sql Statement with embedded "@param" style parameters</param>        /// <param name="paramList">object[] array of parameter values</param>        /// <returns></returns>        public static int ExecuteNonQuery(string connectionString, string commandText, params object[] paramList)        {            SQLiteConnection cn = new SQLiteConnection(connectionString);            SQLiteCommand cmd = cn.CreateCommand();            cmd.CommandText = commandText;            AttachParameters(cmd, commandText, paramList);            if (cn.State == ConnectionState.Closed)                cn.Open();            int result = cmd.ExecuteNonQuery();            cmd.Dispose();            cn.Close();            return result;        }        public static int ExecuteNonQuery(SQLiteConnection cn, string commandText, params  object[] paramList)        {            SQLiteCommand cmd = cn.CreateCommand();            cmd.CommandText = commandText;            AttachParameters(cmd, commandText, paramList);            if (cn.State == ConnectionState.Closed)                cn.Open();            int result = cmd.ExecuteNonQuery();            cmd.Dispose();            cn.Close();            return result;        }        /// <summary>        /// Executes  non-query sql Statment with Transaction        /// </summary>        /// <param name="transaction">SQLiteTransaction. Transaction consists of Connection, Transaction,   /// and Command, all of which must be created prior to making this method call. </param>        /// <param name="commandText">Command text.</param>        /// <param name="paramList">Param list.</param>        /// <returns>Integer</returns>        /// <remarks>user must examine Transaction Object and handle transaction.connection .Close, etc.</remarks>        public static int ExecuteNonQuery(SQLiteTransaction transaction, string commandText, params  object[] paramList)        {            if (transaction == null) throw new ArgumentNullException("transaction");            if (transaction != null && transaction.Connection == null) throw new ArgumentException("The transaction was rolled back or committed,                                                        please provide an open transaction.", "transaction");            IDbCommand cmd = transaction.Connection.CreateCommand();            cmd.CommandText = commandText;            AttachParameters((SQLiteCommand)cmd, cmd.CommandText, paramList);            if (transaction.Connection.State == ConnectionState.Closed)                transaction.Connection.Open();            int result = cmd.ExecuteNonQuery();            cmd.Dispose();            return result;        }        /// <summary>        /// Executes the non query.        /// </summary>        /// <param name="cmd">CMD.</param>        /// <returns></returns>        public static int ExecuteNonQuery(IDbCommand cmd)        {            if (cmd.Connection.State == ConnectionState.Closed)                cmd.Connection.Open();            int result = cmd.ExecuteNonQuery();            cmd.Connection.Close();            cmd.Dispose();            return result;        }        /// <summary>        /// Shortcut to ExecuteScalar with Sql Statement embedded params and object[] param values        /// </summary>        /// <param name="connectionString">SQLite Connection String</param>        /// <param name="commandText">SQL statment with embedded "@param" style parameters</param>        /// <param name="paramList">object[] array of param values</param>        /// <returns></returns>        public static object ExecuteScalar(string connectionString, string commandText, params  object[] paramList)        {            SQLiteConnection cn = new SQLiteConnection(connectionString);            SQLiteCommand cmd = cn.CreateCommand();            cmd.CommandText = commandText;            AttachParameters(cmd, commandText, paramList);            if (cn.State == ConnectionState.Closed)                cn.Open();            object result = cmd.ExecuteScalar();            cmd.Dispose();            cn.Close();            return result;        }        /// <summary>        /// Execute XmlReader with complete Command        /// </summary>        /// <param name="command">SQLite Command</param>        /// <returns>XmlReader</returns>        public static XmlReader ExecuteXmlReader(IDbCommand command)        { // open the connection if necessary, but make sure we             // know to close it when we�re done.            if (command.Connection.State != ConnectionState.Open)            {                command.Connection.Open();            }            // get a data adapter              SQLiteDataAdapter da = new SQLiteDataAdapter((SQLiteCommand)command);            DataSet ds = new DataSet();            // fill the data set, and return the schema information            da.MissingSchemaAction = MissingSchemaAction.AddWithKey;            da.Fill(ds);            // convert our dataset to XML            StringReader stream = new StringReader(ds.GetXml());            command.Connection.Close();            // convert our stream of text to an XmlReader            return new XmlTextReader(stream);        }        /// <summary>        /// Parses parameter names from SQL Statement, assigns values from object array ,   /// and returns fully populated ParameterCollection.        /// </summary>        /// <param name="commandText">Sql Statement with "@param" style embedded parameters</param>        /// <param name="paramList">object[] array of parameter values</param>        /// <returns>SQLiteParameterCollection</returns>        /// <remarks>Status experimental. Regex appears to be handling most issues. Note that parameter object array must be in same ///order as parameter names appear in SQL statement.</remarks>        private static SQLiteParameterCollection AttachParameters(SQLiteCommand cmd, string commandText, params  object[] paramList)        {            if (paramList == null || paramList.Length == 0) return null;            SQLiteParameterCollection coll = cmd.Parameters;            string parmString = commandText.Substring(commandText.IndexOf("@"));            // pre-process the string so always at least 1 space after a comma.            parmString = parmString.Replace(",", " ,");            // get the named parameters into a match collection            string pattern = @"(@)\S*(.*?)\b";            Regex ex = new Regex(pattern, RegexOptions.IgnoreCase);            MatchCollection mc = ex.Matches(parmString);            string[] paramNames = new string[mc.Count];            int i = 0;            foreach (Match m in mc)            {                paramNames[i] = m.Value;                i++;            }            // now let's type the parameters            int j = 0;            Type t = null;            foreach (object o in paramList)            {                t = o.GetType();                SQLiteParameter parm = new SQLiteParameter();                switch (t.ToString())                {                    case ("DBNull"):                    case ("Char"):                    case ("SByte"):                    case ("UInt16"):                    case ("UInt32"):                    case ("UInt64"):                        throw new SystemException("Invalid data type");                    case ("System.String"):                        parm.DbType = DbType.String;                        parm.ParameterName = paramNames[j];                        parm.Value = (string)paramList[j];                        coll.Add(parm);                        break;                    case ("System.Byte[]"):                        parm.DbType = DbType.Binary;                        parm.ParameterName = paramNames[j];                        parm.Value = (byte[])paramList[j];                        coll.Add(parm);                        break;                    case ("System.Int32"):                        parm.DbType = DbType.Int32;                        parm.ParameterName = paramNames[j];                        parm.Value = (int)paramList[j];                        coll.Add(parm);                        break;                    case ("System.Boolean"):                        parm.DbType = DbType.Boolean;                        parm.ParameterName = paramNames[j];                        parm.Value = (bool)paramList[j];                        coll.Add(parm);                        break;                    case ("System.DateTime"):                        parm.DbType = DbType.DateTime;                        parm.ParameterName = paramNames[j];                        parm.Value = Convert.ToDateTime(paramList[j]);                        coll.Add(parm);                        break;                    case ("System.Double"):                        parm.DbType = DbType.Double;                        parm.ParameterName = paramNames[j];                        parm.Value = Convert.ToDouble(paramList[j]);                        coll.Add(parm);                        break;                    case ("System.Decimal"):                        parm.DbType = DbType.Decimal;                        parm.ParameterName = paramNames[j];                        parm.Value = Convert.ToDecimal(paramList[j]);                        break;                    case ("System.Guid"):                        parm.DbType = DbType.Guid;                        parm.ParameterName = paramNames[j];                        parm.Value = (System.Guid)(paramList[j]);                        break;                    case ("System.Object"):                        parm.DbType = DbType.Object;                        parm.ParameterName = paramNames[j];                        parm.Value = paramList[j];                        coll.Add(parm);                        break;                    default:                        throw new SystemException("Value is of unknown data type");                } // end switch                j++;            }            return coll;        }        /// <summary>        /// Executes non query typed params from a DataRow        /// </summary>        /// <param name="command">Command.</param>        /// <param name="dataRow">Data row.</param>        /// <returns>Integer result code</returns>        public static int ExecuteNonQueryTypedParams(IDbCommand command, DataRow dataRow)        {            int retVal = 0;            // If the row has values, the store procedure parameters must be initialized            if (dataRow != null && dataRow.ItemArray.Length > 0)            {                // Set the parameters values                AssignParameterValues(command.Parameters, dataRow);                retVal = ExecuteNonQuery(command);            }            else            {                retVal = ExecuteNonQuery(command);            }            return retVal;        }        /// <summary>        /// This method assigns dataRow column values to an IDataParameterCollection        /// </summary>        /// <param name="commandParameters">The IDataParameterCollection to be assigned values</param>        /// <param name="dataRow">The dataRow used to hold the command's parameter values</param>        /// <exception cref="System.InvalidOperationException">Thrown if any of the parameter names are invalid.</exception>        protected internal static void AssignParameterValues(IDataParameterCollection commandParameters, DataRow dataRow)        {            if (commandParameters == null || dataRow == null)            {                // Do nothing if we get no data                return;            }            DataColumnCollection columns = dataRow.Table.Columns;            int i = 0;            // Set the parameters values            foreach (IDataParameter commandParameter in commandParameters)            {                // Check the parameter name                if (commandParameter.ParameterName == null ||                 commandParameter.ParameterName.Length <= 1)                    throw new InvalidOperationException(string.Format(                           "Please provide a valid parameter name on the parameter #{0},                            the ParameterName property has the following value: '{1}'.",                     i, commandParameter.ParameterName));                if (columns.Contains(commandParameter.ParameterName))                    commandParameter.Value = dataRow[commandParameter.ParameterName];                else if (columns.Contains(commandParameter.ParameterName.Substring(1)))                    commandParameter.Value = dataRow[commandParameter.ParameterName.Substring(1)];                i++;            }        }        /// <summary>        /// This method assigns dataRow column values to an array of IDataParameters        /// </summary>        /// <param name="commandParameters">Array of IDataParameters to be assigned values</param>        /// <param name="dataRow">The dataRow used to hold the stored procedure's parameter values</param>        /// <exception cref="System.InvalidOperationException">Thrown if any of the parameter names are invalid.</exception>        protected void AssignParameterValues(IDataParameter[] commandParameters, DataRow dataRow)        {            if ((commandParameters == null) || (dataRow == null))            {                // Do nothing if we get no data                return;            }            DataColumnCollection columns = dataRow.Table.Columns;            int i = 0;            // Set the parameters values            foreach (IDataParameter commandParameter in commandParameters)            {                // Check the parameter name                if (commandParameter.ParameterName == null ||                 commandParameter.ParameterName.Length <= 1)                    throw new InvalidOperationException(string.Format(                     "Please provide a valid parameter name on the parameter #{0}, the ParameterName property has the following value: '{1}'.",                     i, commandParameter.ParameterName));                if (columns.Contains(commandParameter.ParameterName))                    commandParameter.Value = dataRow[commandParameter.ParameterName];                else if (columns.Contains(commandParameter.ParameterName.Substring(1)))                    commandParameter.Value = dataRow[commandParameter.ParameterName.Substring(1)];                i++;            }        }        /// <summary>        /// This method assigns an array of values to an array of IDataParameters        /// </summary>        /// <param name="commandParameters">Array of IDataParameters to be assigned values</param>        /// <param name="parameterValues">Array of objects holding the values to be assigned</param>        /// <exception cref="System.ArgumentException">Thrown if an incorrect number of parameters are passed.</exception>        protected void AssignParameterValues(IDataParameter[] commandParameters, params  object[] parameterValues)        {            if ((commandParameters == null) || (parameterValues == null))            {                // Do nothing if we get no data                return;            }            // We must have the same number of values as we pave parameters to put them in            if (commandParameters.Length != parameterValues.Length)            {                throw new ArgumentException("Parameter count does not match Parameter Value count.");            }            // Iterate through the IDataParameters, assigning the values from the corresponding position in the             // value array            for (int i = 0, j = commandParameters.Length, k = 0; i < j; i++)            {                if (commandParameters[i].Direction != ParameterDirection.ReturnValue)                {                    // If the current array value derives from IDataParameter, then assign its Value property                    if (parameterValues[k] is IDataParameter)                    {                        IDataParameter paramInstance;                        paramInstance = (IDataParameter)parameterValues[k];                        if (paramInstance.Direction == ParameterDirection.ReturnValue)                        {                            paramInstance = (IDataParameter)parameterValues[++k];                        }                        if (paramInstance.Value == null)                        {                            commandParameters[i].Value = DBNull.Value;                        }                        else                        {                            commandParameters[i].Value = paramInstance.Value;                        }                    }                    else if (parameterValues[k] == null)                    {                        commandParameters[i].Value = DBNull.Value;                    }                    else                    {                        commandParameters[i].Value = parameterValues[k];                    }                    k++;                }            }        }    }}