Sqlite类库,基于framework 4.0框架

来源:互联网 发布:vue.js webpack 部署 编辑:程序博客网 时间:2024/05/16 18:50
//// Copyright (c) 2009-2010 Krueger Systems, Inc.// using System;using System.Runtime.InteropServices;using System.Collections.Generic;using System.Reflection;using System.Linq;using System.Linq.Expressions;namespace SQLite{public class SQLiteException : System.Exception{public SQLite3.Result Result { get; private set; }protected SQLiteException (SQLite3.Result r,string message) : base(message){Result = r;}public static SQLiteException New (SQLite3.Result r, string message){return new SQLiteException (r, message);}}/// <summary>/// Represents an open connection to a SQLite database./// </summary>public class SQLiteConnection : IDisposable{private bool _open;private TimeSpan _busyTimeout;private Dictionary<string, TableMapping> _mappings = null;private Dictionary<string, TableMapping> _tables = null;private System.Diagnostics.Stopwatch _sw;private long _elapsedMilliseconds = 0;/// <summary>/// 句柄/// </summary>/// <value>The handle.</value>public IntPtr Handle { get; private set; }/// <summary>/// Gets the Database path./// </summary>/// <value>The Database path.</value>public string DatabasePath { get; private set; }public bool TimeExecution { get; set; }public bool Trace { get; set; }/// <summary>/// Constructs a new SQLiteConnection and opens a SQLite database specified by databasePath./// </summary>/// <param name="databasePath">/// Specifies the path to the database file./// </param>public SQLiteConnection (string databasePath){DatabasePath = databasePath;IntPtr _handle;var r = SQLite3.Open (DatabasePath, out _handle);Handle = _handle;if (r != SQLite3.Result.OK) {throw SQLiteException.New (r, "Could not open database file: " + DatabasePath);}_open = true;BusyTimeout = TimeSpan.FromSeconds (0.1);}static SQLiteConnection (){if (_preserveDuringLinkMagic) {var ti = new TableInfo ();ti.name = "magic";}}/// <summary>/// Used to list some code that we want the MonoTouch linker/// to see, but that we never want to actually execute./// </summary>static bool _preserveDuringLinkMagic = false;/// <summary>/// Sets a busy handler to sleep the specified amount of time when a table is locked./// The handler will sleep multiple times until a total time of <see cref="BusyTimeout"/> has accumulated./// </summary>public TimeSpan BusyTimeout {get { return _busyTimeout; }set {_busyTimeout = value;if (Handle != IntPtr.Zero) {SQLite3.BusyTimeout (Handle, (int)_busyTimeout.TotalMilliseconds);}}}/// <summary>/// Returns the mappings from types to tables that the connection/// currently understands./// </summary>public IEnumerable<TableMapping> TableMappings {get {if (_tables == null) {return Enumerable.Empty<TableMapping> ();} else {return _tables.Values;}}}/// <summary>/// Retrieves the mapping that is automatically generated for the given type./// </summary>/// <param name="type">/// The type whose mapping to the database is returned./// </param>/// <returns>/// The mapping represents the schema of the columns of the database and contains /// methods to set and get properties of objects./// </returns>public TableMapping GetMapping (Type type){if (_mappings == null) {_mappings = new Dictionary<string, TableMapping> ();}TableMapping map;if (!_mappings.TryGetValue (type.FullName, out map)) {map = new TableMapping (type);_mappings [type.FullName] = map;}return map;}/// <summary>/// Executes a "create table if not exists" on the database. It also/// creates any specified indexes on the columns of the table. It uses/// a schema automatically generated from the specified type. You can/// later access this schema by calling GetMapping./// </summary>/// <returns>/// The number of entries added to the database schema./// </returns>public int CreateTable<T> (){var ty = typeof(T);if (_tables == null) {_tables = new Dictionary<string, TableMapping> ();}TableMapping map;if (!_tables.TryGetValue (ty.FullName, out map)) {map = GetMapping (ty);_tables.Add (ty.FullName, map);}var query = "create table \"" + map.TableName + "\"\n(\n";var decls = map.Columns.Select (p => Orm.SqlDecl (p));var decl = string.Join (",\n", decls.ToArray ());query += decl;query += "\n)";var count = 0;try {Execute (query);count = 1;}catch (SQLiteException) {}if (count == 0) {// Table already exists, migrate itMigrateTable (map);}foreach (var p in map.Columns.Where (x => x.IsIndexed)) {var indexName = map.TableName + "_" + p.Name;var q = string.Format ("create index if not exists \"{0}\" on \"{1}\"(\"{2}\")", indexName, map.TableName, p.Name);count += Execute (q);}return count;}class TableInfo{public int cid { get; set; }public string name { get; set; }public string type { get; set; }public int notnull { get; set; }public string dflt_value { get; set; }public int pk { get; set; }}void MigrateTable (TableMapping map){var query = "pragma table_info(\"" + map.TableName + "\")";var existingCols = Query<TableInfo> (query);var toBeAdded = new List<TableMapping.Column> ();foreach (var p in map.Columns) {var found = false;foreach (var c in existingCols) {found = p.Name == c.name;if (found)break;}if (!found) {toBeAdded.Add (p);}}foreach (var p in toBeAdded) {var addCol = "alter table \"" + map.TableName + "\" add column " + Orm.SqlDecl (p);Execute (addCol);}}/// <summary>/// Creates a new SQLiteCommand given the command text with arguments. Place a '?'/// in the command text for each of the arguments./// </summary>/// <param name="cmdText">/// The fully escaped SQL./// </param>/// <param name="args">/// Arguments to substitute for the occurences of '?' in the command text./// </param>/// <returns>/// A <see cref="SQLiteCommand"/>/// </returns>public SQLiteCommand CreateCommand (string cmdText, params object[] ps){if (!_open) {throw SQLiteException.New (SQLite3.Result.Error, "Cannot create commands from unopened database");} else {var cmd = new SQLiteCommand (this);cmd.CommandText = cmdText;foreach (var o in ps) {cmd.Bind (o);}return cmd;}}/// <summary>/// Creates a SQLiteCommand given the command text (SQL) with arguments. Place a '?'/// in the command text for each of the arguments and then executes that command./// Use this method instead of Query when you don't expect rows back. Such cases include/// INSERTs, UPDATEs, and DELETEs./// You can set the Trace or TimeExecution properties of the connection/// to profile execution./// </summary>/// <param name="query">/// The fully escaped SQL./// </param>/// <param name="args">/// Arguments to substitute for the occurences of '?' in the query./// </param>/// <returns>/// The number of rows modified in the database as a result of this execution./// </returns>public int Execute (string query, params object[] args){var cmd = CreateCommand (query, args);if (TimeExecution) {if (_sw == null) {_sw = new System.Diagnostics.Stopwatch ();}_sw.Reset ();_sw.Start ();}int r = cmd.ExecuteNonQuery ();if (TimeExecution) {_sw.Stop ();_elapsedMilliseconds += _sw.ElapsedMilliseconds;Console.WriteLine ("Finished in {0} ms ({1:0.0} s total)", _sw.ElapsedMilliseconds, _elapsedMilliseconds / 1000.0);}return r;}/// <summary>/// Creates a SQLiteCommand given the command text (SQL) with arguments. Place a '?'/// in the command text for each of the arguments and then executes that command./// It returns each row of the result using the mapping automatically generated for/// the given type./// </summary>/// <param name="query">/// The fully escaped SQL./// </param>/// <param name="args">/// Arguments to substitute for the occurences of '?' in the query./// </param>/// <returns>/// An enumerable with one result for each row returned by the query./// </returns>public List<T> Query<T> (string query, params object[] args) where T : new(){var cmd = CreateCommand (query, args);return cmd.ExecuteQuery<T> ();}/// <summary>/// Creates a SQLiteCommand given the command text (SQL) with arguments. Place a '?'/// in the command text for each of the arguments and then executes that command./// It returns each row of the result using the specified mapping. This function is/// only used by libraries in order to query the database via introspection. It is/// normally not used./// </summary>/// <param name="map">/// A <see cref="TableMapping"/> to use to convert the resulting rows/// into objects./// </param>/// <param name="query">/// The fully escaped SQL./// </param>/// <param name="args">/// Arguments to substitute for the occurences of '?' in the query./// </param>/// <returns>/// An enumerable with one result for each row returned by the query./// </returns>public List<object> Query (TableMapping map, string query, params object[] args){var cmd = CreateCommand (query, args);return cmd.ExecuteQuery<object> (map);}/// <summary>/// Returns a queryable interface to the table represented by the given type./// </summary>/// <returns>/// A queryable object that is able to translate Where, OrderBy, and Take/// queries into native SQL./// </returns>public TableQuery<T> Table<T> () where T : new(){return new TableQuery<T> (this);}/// <summary>/// Attempts to retrieve an object with the given primary key from the table/// associated with the specified type. Use of this method requires that/// the given type have a designated PrimaryKey (using the PrimaryKeyAttribute)./// </summary>/// <param name="pk">/// The primary key./// </param>/// <returns>/// The object with the given primary key. Throws a not found exception/// if the object is not found./// </returns>public T Get<T> (object pk) where T : new(){var map = GetMapping (typeof(T));string query = string.Format ("select * from \"{0}\" where \"{1}\" = ?", map.TableName, map.PK.Name);return Query<T> (query, pk).First ();}/// <summary>/// Whether <see cref="BeginTransaction"/> has been called and the database is waiting for a <see cref="Commit"/>./// </summary>public bool IsInTransaction { get; private set; }/// <summary>/// Begins a new transaction. Call <see cref="Commit"/> to end the transaction./// </summary>public void BeginTransaction (){if (!IsInTransaction) {Execute ("begin transaction");IsInTransaction = true;}}/// <summary>/// Rolls back the transaction that was begun by <see cref="BeginTransaction"/>./// </summary>public void Rollback (){if (IsInTransaction) {Execute ("rollback");IsInTransaction = false;}}/// <summary>/// Commits the transaction that was begun by <see cref="BeginTransaction"/>./// </summary>public void Commit (){if (IsInTransaction) {Execute ("commit");IsInTransaction = false;}}/// <summary>/// Executes <param name="action"> within a transaction and automatically rollsback the transaction/// if an exception occurs. The exception is rethrown./// </summary>/// <param name="action">/// The <see cref="Action"/> to perform within a transaction. <param name="action"> can contain any number/// of operations on the connection but should never call <see cref="BeginTransaction"/>,/// <see cref="Rollback"/>, or <see cref="Commit"/>./// </param>public void RunInTransaction (Action action){if (IsInTransaction) {throw new InvalidOperationException ("The connection must not already be in a transaction when RunInTransaction is called");}try {BeginTransaction ();action ();Commit ();} catch (Exception) {Rollback ();throw;}}/// <summary>/// Inserts all specified objects./// </summary>/// <param name="objects">/// An <see cref="IEnumerable"/> of the objects to insert./// </param>/// <returns>/// The number of rows added to the table./// </returns>public int InsertAll (System.Collections.IEnumerable objects){BeginTransaction ();var c = 0;foreach (var r in objects) {c += Insert (r);}Commit ();return c;}/// <summary>/// Inserts the given object and retrieves its/// auto incremented primary key if it has one./// </summary>/// <param name="obj">/// The object to insert./// </param>/// <returns>/// The number of rows added to the table./// </returns>public int Insert (object obj){if (obj == null) {return 0;}return Insert (obj, "", obj.GetType ());}public int Insert (object obj, Type objType){return Insert (obj, "", objType);}public int Insert (object obj, string extra){if (obj == null) {return 0;}return Insert (obj, extra, obj.GetType ());}/// <summary>/// Inserts the given object and retrieves its/// auto incremented primary key if it has one./// </summary>/// <param name="obj">/// The object to insert./// </param>/// <param name="extra">/// Literal SQL code that gets placed into the command. INSERT {extra} INTO .../// </param>/// <returns>/// The number of rows added to the table./// </returns>public int Insert (object obj, string extra, Type objType){if (obj == null || objType == null) {return 0;}var map = GetMapping (objType);var cols = map.InsertColumns;var vals = new object[cols.Length];for (var i = 0; i < vals.Length; i++) {vals [i] = cols [i].GetValue (obj);}var insertCmd = map.GetInsertCommand (this, extra);var count = insertCmd.ExecuteNonQuery (vals);if (map.HasAutoIncPK) {var id = SQLite3.LastInsertRowid (Handle);map.SetAutoIncPK (obj, id);}return count;}/// <summary>/// Updates all of the columns of a table using the specified object/// except for its primary key./// The object is required to have a primary key./// </summary>/// <param name="obj">/// The object to update. It must have a primary key designated using the PrimaryKeyAttribute./// </param>/// <returns>/// The number of rows updated./// </returns>public int Update (object obj){if (obj == null) {return 0;}return Update (obj, obj.GetType ());}public int Update (object obj, Type objType){if (obj == null || objType == null) {return 0;}var map = GetMapping (objType);var pk = map.PK;if (pk == null) {throw new NotSupportedException ("Cannot update " + map.TableName + ": it has no PK");}var cols = from p in map.Columnswhere p != pkselect p;var vals = from c in colsselect c.GetValue (obj);var ps = new List<object> (vals);ps.Add (pk.GetValue (obj));var q = string.Format ("update \"{0}\" set {1} where {2} = ? ", map.TableName, string.Join (",", (from c in colsselect "\"" + c.Name + "\" = ? ").ToArray ()), pk.Name);return Execute (q, ps.ToArray ());}/// <summary>/// Deletes the given object from the database using its primary key./// </summary>/// <param name="obj">/// The object to delete. It must have a primary key designated using the PrimaryKeyAttribute./// </param>/// <returns>/// The number of rows deleted./// </returns>public int Delete<T> (T obj){var map = GetMapping (obj.GetType ());var pk = map.PK;if (pk == null) {throw new NotSupportedException ("Cannot delete " + map.TableName + ": it has no PK");}var q = string.Format ("delete from \"{0}\" where \"{1}\" = ?", map.TableName, pk.Name);return Execute (q, pk.GetValue (obj));}public void Dispose (){Close ();}public void Close (){if (_open && Handle != IntPtr.Zero) {SQLite3.Close (Handle);Handle = IntPtr.Zero;_open = false;}}}/*继承于Attribute的类,表示特性类,它在使用时候,如果它的类名字是以Attribute结尾了,则被调用时候类名上面的Attribute可以不用谢*/public class PrimaryKeyAttribute : Attribute{}public class AutoIncrementAttribute : Attribute{}public class IndexedAttribute : Attribute{}public class IgnoreAttribute : Attribute{}public class MaxLengthAttribute : Attribute{public int Value { get; private set; }public MaxLengthAttribute (int length){Value = length;}}public class CollationAttribute: Attribute{public string Value { get; private set; }public CollationAttribute (string collation){Value = collation;}}public class TableMapping{public Type MappedType { get; private set; }public string TableName { get; private set; }public Column[] Columns { get; private set; }public Column PK { get; private set; }Column _autoPk = null;Column[] _insertColumns = null;string _insertSql = null;public TableMapping (Type type){MappedType = type;TableName = MappedType.Name;var props = MappedType.GetProperties (BindingFlags.Public | BindingFlags.Instance | BindingFlags.SetProperty);var cols = new List<Column> ();foreach (var p in props) {var ignore = p.GetCustomAttributes (typeof(IgnoreAttribute), true).Length > 0;if (p.CanWrite && !ignore) {cols.Add (new PropColumn (p));}}Columns = cols.ToArray ();foreach (var c in Columns) {if (c.IsAutoInc && c.IsPK) {_autoPk = c;}if (c.IsPK) {PK = c;}}HasAutoIncPK = _autoPk != null;}public bool HasAutoIncPK { get; private set; }public void SetAutoIncPK (object obj, long id){if (_autoPk != null) {_autoPk.SetValue (obj, Convert.ChangeType (id, _autoPk.ColumnType));}}public Column[] InsertColumns {get {if (_insertColumns == null) {_insertColumns = Columns.Where (c => !c.IsAutoInc).ToArray ();}return _insertColumns;}}public Column FindColumn (string name){var exact = Columns.Where (c => c.Name == name).FirstOrDefault ();return exact;}public string InsertSql (string extra){if (_insertSql == null) {var cols = InsertColumns;_insertSql = string.Format ("insert {3} into \"{0}\"({1}) values ({2})", TableName, string.Join (",", (from c in colsselect "\"" + c.Name + "\"").ToArray ()), string.Join (",", (from c in colsselect "?").ToArray ()), extra);}return _insertSql;}PreparedSqlLiteInsertCommand _insertCommand;string _insertCommandExtra = null;public PreparedSqlLiteInsertCommand GetInsertCommand (SQLiteConnection conn, string extra){if (_insertCommand == null || _insertCommandExtra != extra) {var insertSql = InsertSql (extra);_insertCommand = new PreparedSqlLiteInsertCommand (conn);_insertCommand.CommandText = insertSql;_insertCommandExtra = extra;}return _insertCommand;}public abstract class Column{public string Name { get; protected set; }public Type ColumnType { get; protected set; }public string Collation { get; protected set; }public bool IsAutoInc { get; protected set; }public bool IsPK { get; protected set; }public bool IsIndexed { get; protected set; }public bool IsNullable { get; protected set; }public int MaxStringLength { get; protected set; }public abstract void SetValue (object obj, object val);public abstract object GetValue (object obj);}public class PropColumn : Column{PropertyInfo _prop;public PropColumn (PropertyInfo prop){_prop = prop;Name = prop.Name;//If this type is Nullable<T> then Nullable.GetUnderlyingType returns the T, otherwise it returns null, so get the the actual type insteadColumnType = Nullable.GetUnderlyingType(prop.PropertyType) ?? prop.PropertyType;Collation = Orm.Collation (prop);IsAutoInc = Orm.IsAutoInc (prop);IsPK = Orm.IsPK (prop);IsIndexed = Orm.IsIndexed (prop);IsNullable = !IsPK;MaxStringLength = Orm.MaxStringLength (prop);}public override void SetValue (object obj, object val){_prop.SetValue (obj, val, null);}public override object GetValue (object obj){return _prop.GetValue (obj, null);}}}public static class Orm{public const int DefaultMaxStringLength = 140;public static string SqlDecl (TableMapping.Column p){string decl = "\"" + p.Name + "\" " + SqlType (p) + " ";if (p.IsPK) {decl += "primary key ";}if (p.IsAutoInc) {decl += "autoincrement ";}if (!p.IsNullable) {decl += "not null ";}if (!string.IsNullOrEmpty (p.Collation)) {decl += "collate " + p.Collation + " ";}return decl;}public static string SqlType (TableMapping.Column p){var clrType = p.ColumnType;if (clrType == typeof(Boolean) || clrType == typeof(Byte) || clrType == typeof(UInt16) || clrType == typeof(SByte) || clrType == typeof(Int16) || clrType == typeof(Int32)) {return "integer";} else if (clrType == typeof(UInt32) || clrType == typeof(Int64)) {return "bigint";} else if (clrType == typeof(Single) || clrType == typeof(Double) || clrType == typeof(Decimal)) {return "float";} else if (clrType == typeof(String)) {int len = p.MaxStringLength;return "varchar(" + len + ")";} else if (clrType == typeof(DateTime)) {return "datetime";} else if (clrType.IsEnum) {return "integer";} else if (clrType == typeof(byte[])) {return "blob";} else {throw new NotSupportedException ("Don't know about " + clrType);}}public static bool IsPK (MemberInfo p){var attrs = p.GetCustomAttributes (typeof(PrimaryKeyAttribute), true);return attrs.Length > 0;}public static string Collation (MemberInfo p){var attrs = p.GetCustomAttributes (typeof(CollationAttribute), true);if (attrs.Length > 0) {return ((CollationAttribute)attrs [0]).Value;} else {return string.Empty;}}public static bool IsAutoInc (MemberInfo p){var attrs = p.GetCustomAttributes (typeof(AutoIncrementAttribute), true);return attrs.Length > 0;}public static bool IsIndexed (MemberInfo p){var attrs = p.GetCustomAttributes (typeof(IndexedAttribute), true);return attrs.Length > 0;}public static int MaxStringLength (PropertyInfo p){var attrs = p.GetCustomAttributes (typeof(MaxLengthAttribute), true);if (attrs.Length > 0) {return ((MaxLengthAttribute)attrs [0]).Value;} else {return DefaultMaxStringLength;}}}public class SQLiteCommand{SQLiteConnection _conn;private List<Binding> _bindings;public string CommandText { get; set; }internal SQLiteCommand (SQLiteConnection conn){_conn = conn;_bindings = new List<Binding> ();CommandText = "";}public int ExecuteNonQuery (){if (_conn.Trace) {Console.WriteLine ("Executing: " + this);}var r = SQLite3.Result.OK;var stmt = Prepare ();r = SQLite3.Step (stmt);Finalize (stmt);if (r == SQLite3.Result.Done) {int rowsAffected = SQLite3.Changes (_conn.Handle);return rowsAffected;} else if (r == SQLite3.Result.Error) {string msg = SQLite3.GetErrmsg (_conn.Handle);throw SQLiteException.New (r, msg);} else {throw SQLiteException.New (r, r.ToString ());}}public List<T> ExecuteQuery<T> () where T : new(){return ExecuteQuery<T> (_conn.GetMapping (typeof(T)));}public List<T> ExecuteQuery<T> (TableMapping map){if (_conn.Trace) {Console.WriteLine ("Executing Query: " + this);}var r = new List<T> ();var stmt = Prepare ();var cols = new TableMapping.Column[SQLite3.ColumnCount (stmt)];for (int i = 0; i < cols.Length; i++) {var name = Marshal.PtrToStringUni (SQLite3.ColumnName16 (stmt, i));cols [i] = map.FindColumn (name);}while (SQLite3.Step (stmt) == SQLite3.Result.Row) {var obj = Activator.CreateInstance (map.MappedType);for (int i = 0; i < cols.Length; i++) {if (cols [i] == null)continue;var colType = SQLite3.ColumnType (stmt, i);var val = ReadCol (stmt, i, colType, cols [i].ColumnType);cols [i].SetValue (obj, val);}r.Add ((T)obj);}Finalize (stmt);return r;}public T ExecuteScalar<T> (){if (_conn.Trace) {Console.WriteLine ("Executing Query: " + this);}T val = default(T);var stmt = Prepare ();if (SQLite3.Step (stmt) == SQLite3.Result.Row) {var colType = SQLite3.ColumnType (stmt, 0);val = (T)ReadCol (stmt, 0, colType, typeof(T));}Finalize (stmt);return val;}public void Bind (string name, object val){_bindings.Add (new Binding {Name = name,Value = val});}public void Bind (object val){Bind (null, val);}public override string ToString (){var parts = new string[1 + _bindings.Count];parts [0] = CommandText;var i = 1;foreach (var b in _bindings) {parts [i] = string.Format ("  {0}: {1}", i - 1, b.Value);i++;}return string.Join (Environment.NewLine, parts);}IntPtr Prepare (){var stmt = SQLite3.Prepare2 (_conn.Handle, CommandText);BindAll (stmt);return stmt;}void Finalize (IntPtr stmt){SQLite3.Finalize (stmt);}void BindAll (IntPtr stmt){int nextIdx = 1;foreach (var b in _bindings) {if (b.Name != null) {b.Index = SQLite3.BindParameterIndex (stmt, b.Name);} else {b.Index = nextIdx++;}}foreach (var b in _bindings) {BindParameter (stmt, b.Index, b.Value);}}internal static IntPtr NegativePointer = new IntPtr (-1);internal static void BindParameter (IntPtr stmt, int index, object value){if (value == null) {SQLite3.BindNull (stmt, index);} else {if (value is Int32) {SQLite3.BindInt (stmt, index, (int)value);} else if (value is String) {SQLite3.BindText (stmt, index, (string)value, -1, NegativePointer);} else if (value is Byte || value is UInt16 || value is SByte || value is Int16) {SQLite3.BindInt (stmt, index, Convert.ToInt32 (value));} else if (value is Boolean) {SQLite3.BindInt (stmt, index, (bool)value ? 1 : 0);} else if (value is UInt32 || value is Int64) {SQLite3.BindInt64 (stmt, index, Convert.ToInt64 (value));} else if (value is Single || value is Double || value is Decimal) {SQLite3.BindDouble (stmt, index, Convert.ToDouble (value));} else if (value is DateTime) {SQLite3.BindText (stmt, index, ((DateTime)value).ToString ("yyyy-MM-dd HH:mm:ss"), -1, NegativePointer);} else if (value.GetType ().IsEnum) {SQLite3.BindInt (stmt, index, Convert.ToInt32 (value));} else if (value is byte[]) {SQLite3.BindBlob (stmt, index, (byte[])value, ((byte[])value).Length, NegativePointer);} else {throw new NotSupportedException ("Cannot store type: " + value.GetType ());}}}class Binding{public string Name { get; set; }public object Value { get; set; }public int Index { get; set; }}object ReadCol (IntPtr stmt, int index, SQLite3.ColType type, Type clrType){if (type == SQLite3.ColType.Null) {return null;} else {if (clrType == typeof(String)) {return SQLite3.ColumnString (stmt, index);} else if (clrType == typeof(Int32)) {return (int)SQLite3.ColumnInt (stmt, index);} else if (clrType == typeof(Boolean)) {return SQLite3.ColumnInt (stmt, index) == 1;} else if (clrType == typeof(double)) {return SQLite3.ColumnDouble (stmt, index);} else if (clrType == typeof(float)) {return (float)SQLite3.ColumnDouble (stmt, index);} else if (clrType == typeof(DateTime)) {var text = SQLite3.ColumnString (stmt, index);return DateTime.Parse (text);} else if (clrType.IsEnum) {return SQLite3.ColumnInt (stmt, index);} else if (clrType == typeof(Int64)) {return SQLite3.ColumnInt64 (stmt, index);} else if (clrType == typeof(UInt32)) {return (uint)SQLite3.ColumnInt64 (stmt, index);} else if (clrType == typeof(decimal)) {return (decimal)SQLite3.ColumnDouble (stmt, index);} else if (clrType == typeof(Byte)) {return (byte)SQLite3.ColumnInt (stmt, index);} else if (clrType == typeof(UInt16)) {return (ushort)SQLite3.ColumnInt (stmt, index);} else if (clrType == typeof(Int16)) {return (short)SQLite3.ColumnInt (stmt, index);} else if (clrType == typeof(sbyte)) {return (sbyte)SQLite3.ColumnInt (stmt, index);} else if (clrType == typeof(byte[])) {return SQLite3.ColumnByteArray (stmt, index);} else {throw new NotSupportedException ("Don't know how to read " + clrType);}}}}/// <summary>/// Since the insert never changed, we only need to prepare once./// </summary>public class PreparedSqlLiteInsertCommand : IDisposable{public bool Initialized { get; set; }protected SQLiteConnection Connection { get; set; }public string CommandText { get; set; }protected IntPtr Statement { get; set; }internal PreparedSqlLiteInsertCommand (SQLiteConnection conn){Connection = conn;}public int ExecuteNonQuery (object[] source){if (Connection.Trace) {Console.WriteLine ("Executing: " + CommandText);}var r = SQLite3.Result.OK;if (!Initialized) {Statement = Prepare ();Initialized = true;}//bind the values.if (source != null) {for (int i = 0; i < source.Length; i++) {SQLiteCommand.BindParameter (Statement, i + 1, source [i]);}}r = SQLite3.Step (Statement);if (r == SQLite3.Result.Done) {int rowsAffected = SQLite3.Changes (Connection.Handle);SQLite3.Reset (Statement);return rowsAffected;} else if (r == SQLite3.Result.Error) {string msg = SQLite3.GetErrmsg (Connection.Handle);SQLite3.Reset (Statement);throw SQLiteException.New (r, msg);} else {SQLite3.Reset (Statement);throw SQLiteException.New (r, r.ToString ());}}protected virtual IntPtr Prepare (){var stmt = SQLite3.Prepare2 (Connection.Handle, CommandText);return stmt;}public void Dispose (){Dispose (true);GC.SuppressFinalize (this);}private void Dispose (bool disposing){if (Statement != IntPtr.Zero) {try {SQLite3.Finalize (Statement);} finally {Statement = IntPtr.Zero;Connection = null;}}}~PreparedSqlLiteInsertCommand (){Dispose (false);}}public class TableQuery<T> : IEnumerable<T> where T : new(){public SQLiteConnection Connection { get; private set; }public TableMapping Table { get; private set; }Expression _where;List<Ordering> _orderBys;int? _limit;int? _offset;class Ordering{public string ColumnName { get; set; }public bool Ascending { get; set; }}TableQuery (SQLiteConnection conn, TableMapping table){Connection = conn;Table = table;}public TableQuery (SQLiteConnection conn){Connection = conn;Table = Connection.GetMapping (typeof(T));}public TableQuery<T> Clone (){var q = new TableQuery<T> (Connection, Table);q._where = _where;if (_orderBys != null) {q._orderBys = new List<Ordering> (_orderBys);}q._limit = _limit;q._offset = _offset;return q;}public TableQuery<T> Where (Expression<Func<T, bool>> predExpr){if (predExpr.NodeType == ExpressionType.Lambda) {var lambda = (LambdaExpression)predExpr;var pred = lambda.Body;var q = Clone ();q.AddWhere (pred);return q;} else {throw new NotSupportedException ("Must be a predicate");}}public TableQuery<T> Take (int n){var q = Clone ();q._limit = n;return q;}public TableQuery<T> Skip (int n){var q = Clone ();q._offset = n;return q;}public TableQuery<T> OrderBy<U> (Expression<Func<T, U>> orderExpr){return AddOrderBy<U> (orderExpr, true);}public TableQuery<T> OrderByDescending<U> (Expression<Func<T, U>> orderExpr){return AddOrderBy<U> (orderExpr, false);}private TableQuery<T> AddOrderBy<U> (Expression<Func<T, U>> orderExpr, bool asc){if (orderExpr.NodeType == ExpressionType.Lambda) {var lambda = (LambdaExpression)orderExpr;var mem = lambda.Body as MemberExpression;if (mem != null && (mem.Expression.NodeType == ExpressionType.Parameter)) {var q = Clone ();if (q._orderBys == null) {q._orderBys = new List<Ordering> ();}q._orderBys.Add (new Ordering {ColumnName = mem.Member.Name,Ascending = asc});return q;} else {throw new NotSupportedException ("Order By does not support: " + orderExpr);}} else {throw new NotSupportedException ("Must be a predicate");}}private void AddWhere (Expression pred){if (_where == null) {_where = pred;} else {_where = Expression.AndAlso (_where, pred);}}private SQLiteCommand GenerateCommand (string selectionList){var cmdText = "select " + selectionList + " from \"" + Table.TableName + "\"";var args = new List<object> ();if (_where != null) {var w = CompileExpr (_where, args);cmdText += " where " + w.CommandText;}if ((_orderBys != null) && (_orderBys.Count > 0)) {var t = string.Join (", ", _orderBys.Select (o => "\"" + o.ColumnName + "\"" + (o.Ascending ? "" : " desc")).ToArray ());cmdText += " order by " + t;}if (_limit.HasValue) {cmdText += " limit " + _limit.Value;}if (_offset.HasValue) {if (!_limit.HasValue) {cmdText += " limit -1 ";}cmdText += " offset " + _offset.Value;}return Connection.CreateCommand (cmdText, args.ToArray ());}class CompileResult{public string CommandText { get; set; }public object Value { get; set; }}private CompileResult CompileExpr (Expression expr, List<object> queryArgs){if (expr == null) {throw new NotSupportedException ("Expression is NULL");} else if (expr is BinaryExpression) {var bin = (BinaryExpression)expr;var leftr = CompileExpr (bin.Left, queryArgs);var rightr = CompileExpr (bin.Right, queryArgs);//If either side is a parameter and is null, then handle the other side specially (for "is null"/"is not null")string text;if (leftr.CommandText == "?" && leftr.Value == null)text = CompileNullBinaryExpression(bin, rightr);else if (rightr.CommandText == "?" && rightr.Value == null)text = CompileNullBinaryExpression(bin, leftr);elsetext = "(" + leftr.CommandText + " " + GetSqlName(bin) + " " + rightr.CommandText + ")";return new CompileResult { CommandText = text };} else if (expr.NodeType == ExpressionType.Call) {var call = (MethodCallExpression)expr;var args = new CompileResult[call.Arguments.Count];for (var i = 0; i < args.Length; i++) {args [i] = CompileExpr (call.Arguments [i], queryArgs);}var sqlCall = "";if (call.Method.Name == "Like" && args.Length == 2) {sqlCall = "(" + args [0].CommandText + " like " + args [1].CommandText + ")";} else if (call.Method.Name == "Contains" && args.Length == 2) {sqlCall = "(" + args [1].CommandText + " in " + args [0].CommandText + ")";} else {sqlCall = call.Method.Name.ToLower () + "(" + string.Join (",", args.Select (a => a.CommandText).ToArray ()) + ")";}return new CompileResult { CommandText = sqlCall };} else if (expr.NodeType == ExpressionType.Constant) {var c = (ConstantExpression)expr;queryArgs.Add (c.Value);return new CompileResult {CommandText = "?",Value = c.Value};} else if (expr.NodeType == ExpressionType.Convert) {var u = (UnaryExpression)expr;var ty = u.Type;var valr = CompileExpr (u.Operand, queryArgs);return new CompileResult {CommandText = valr.CommandText,Value = valr.Value != null ? Convert.ChangeType (valr.Value, ty) : null};} else if (expr.NodeType == ExpressionType.MemberAccess) {var mem = (MemberExpression)expr;if (mem.Expression.NodeType == ExpressionType.Parameter) {//// This is a column of our table, output just the column name//return new CompileResult { CommandText = "\"" + mem.Member.Name + "\"" };} else {object obj = null;if (mem.Expression != null) {var r = CompileExpr (mem.Expression, queryArgs);if (r.Value == null) {throw new NotSupportedException ("Member access failed to compile expression");}if (r.CommandText == "?") {queryArgs.RemoveAt (queryArgs.Count - 1);}obj = r.Value;}//// Get the member value//object val = null;if (mem.Member.MemberType == MemberTypes.Property) {var m = (PropertyInfo)mem.Member;val = m.GetValue (obj, null);} else if (mem.Member.MemberType == MemberTypes.Field) {var m = (FieldInfo)mem.Member;val = m.GetValue (obj);} else {throw new NotSupportedException ("MemberExpr: " + mem.Member.MemberType.ToString ());}//// Work special magic for enumerables//if (val != null && val is System.Collections.IEnumerable && !(val is string)) {var sb = new System.Text.StringBuilder();sb.Append("(");var head = "";foreach (var a in (System.Collections.IEnumerable)val) {queryArgs.Add(a);sb.Append(head);sb.Append("?");head = ",";}sb.Append(")");return new CompileResult {CommandText = sb.ToString(),Value = val};}else {queryArgs.Add (val);return new CompileResult {CommandText = "?",Value = val};}}}throw new NotSupportedException ("Cannot compile: " + expr.NodeType.ToString ());}/// <summary>/// Compiles a BinaryExpression where one of the parameters is null./// </summary>/// <param name="parameter">The non-null parameter</param>private string CompileNullBinaryExpression(BinaryExpression expression, CompileResult parameter){if (expression.NodeType == ExpressionType.Equal)return "(" + parameter.CommandText + " is ?)";else if (expression.NodeType == ExpressionType.NotEqual)return "(" + parameter.CommandText + " is not ?)";elsethrow new NotSupportedException("Cannot compile Null-BinaryExpression with type " + expression.NodeType.ToString());}string GetSqlName (Expression expr){var n = expr.NodeType;if (n == ExpressionType.GreaterThan) {return ">";} else if (n == ExpressionType.GreaterThanOrEqual) {return ">=";} else if (n == ExpressionType.LessThan) {return "<";} else if (n == ExpressionType.LessThanOrEqual) {return "<=";} else if (n == ExpressionType.And) {return "and";} else if (n == ExpressionType.AndAlso) {return "and";} else if (n == ExpressionType.Or) {return "or";} else if (n == ExpressionType.OrElse) {return "or";} else if (n == ExpressionType.Equal) {return "=";} else if (n == ExpressionType.NotEqual) {return "!=";} else {throw new System.NotSupportedException ("Cannot get SQL for: " + n.ToString ());}}public int Count (){return GenerateCommand("count(*)").ExecuteScalar<int> ();}public IEnumerator<T> GetEnumerator (){return GenerateCommand ("*").ExecuteQuery<T> ().GetEnumerator ();}System.Collections.IEnumerator System.Collections.IEnumerable.GetEnumerator (){return GetEnumerator ();}}public static class SQLite3{public enum Result : int{OK = 0,Error = 1,Internal = 2,Perm = 3,Abort = 4,Busy = 5,Locked = 6,NoMem = 7,ReadOnly = 8,Interrupt = 9,IOError = 10,Corrupt = 11,NotFound = 12,TooBig = 18,Constraint = 19,Row = 100,Done = 101}public enum ConfigOption : int{SingleThread = 1,MultiThread = 2,Serialized = 3}[DllImport("sqlite3", EntryPoint = "sqlite3_open")]public static extern Result Open (string filename, out IntPtr db);[DllImport("sqlite3", EntryPoint = "sqlite3_close")]public static extern Result Close (IntPtr db);[DllImport("sqlite3", EntryPoint = "sqlite3_config")]public static extern Result Config (ConfigOption option);[DllImport("sqlite3", EntryPoint = "sqlite3_busy_timeout")]public static extern Result BusyTimeout (IntPtr db, int milliseconds);[DllImport("sqlite3", EntryPoint = "sqlite3_changes")]public static extern int Changes (IntPtr db);[DllImport("sqlite3", EntryPoint = "sqlite3_prepare_v2")]public static extern Result Prepare2 (IntPtr db, string sql, int numBytes, out IntPtr stmt, IntPtr pzTail);public static IntPtr Prepare2 (IntPtr db, string query){IntPtr stmt;var r = Prepare2 (db, query, query.Length, out stmt, IntPtr.Zero);if (r != Result.OK) {throw SQLiteException.New (r, GetErrmsg (db));}return stmt;}[DllImport("sqlite3", EntryPoint = "sqlite3_step")]public static extern Result Step (IntPtr stmt);[DllImport("sqlite3", EntryPoint = "sqlite3_reset")]public static extern Result Reset (IntPtr stmt);[DllImport("sqlite3", EntryPoint = "sqlite3_finalize")]public static extern Result Finalize (IntPtr stmt);[DllImport("sqlite3", EntryPoint = "sqlite3_last_insert_rowid")]public static extern long LastInsertRowid (IntPtr db);[DllImport("sqlite3", EntryPoint = "sqlite3_errmsg16")]public static extern IntPtr Errmsg (IntPtr db);public static string GetErrmsg (IntPtr db){return Marshal.PtrToStringUni (Errmsg (db));}[DllImport("sqlite3", EntryPoint = "sqlite3_bind_parameter_index")]public static extern int BindParameterIndex (IntPtr stmt, string name);[DllImport("sqlite3", EntryPoint = "sqlite3_bind_null")]public static extern int BindNull (IntPtr stmt, int index);[DllImport("sqlite3", EntryPoint = "sqlite3_bind_int")]public static extern int BindInt (IntPtr stmt, int index, int val);[DllImport("sqlite3", EntryPoint = "sqlite3_bind_int64")]public static extern int BindInt64 (IntPtr stmt, int index, long val);[DllImport("sqlite3", EntryPoint = "sqlite3_bind_double")]public static extern int BindDouble (IntPtr stmt, int index, double val);[DllImport("sqlite3", EntryPoint = "sqlite3_bind_text")]public static extern int BindText (IntPtr stmt, int index, string val, int n, IntPtr free);[DllImport("sqlite3", EntryPoint = "sqlite3_bind_blob")]public static extern int BindBlob (IntPtr stmt, int index, byte[] val, int n, IntPtr free);[DllImport("sqlite3", EntryPoint = "sqlite3_column_count")]public static extern int ColumnCount (IntPtr stmt);[DllImport("sqlite3", EntryPoint = "sqlite3_column_name")]public static extern IntPtr ColumnName (IntPtr stmt, int index);[DllImport("sqlite3", EntryPoint = "sqlite3_column_name16")]public static extern IntPtr ColumnName16 (IntPtr stmt, int index);[DllImport("sqlite3", EntryPoint = "sqlite3_column_type")]public static extern ColType ColumnType (IntPtr stmt, int index);[DllImport("sqlite3", EntryPoint = "sqlite3_column_int")]public static extern int ColumnInt (IntPtr stmt, int index);[DllImport("sqlite3", EntryPoint = "sqlite3_column_int64")]public static extern long ColumnInt64 (IntPtr stmt, int index);[DllImport("sqlite3", EntryPoint = "sqlite3_column_double")]public static extern double ColumnDouble (IntPtr stmt, int index);[DllImport("sqlite3", EntryPoint = "sqlite3_column_text")]public static extern IntPtr ColumnText (IntPtr stmt, int index);[DllImport("sqlite3", EntryPoint = "sqlite3_column_text16")]public static extern IntPtr ColumnText16 (IntPtr stmt, int index);[DllImport("sqlite3", EntryPoint = "sqlite3_column_blob")]public static extern IntPtr ColumnBlob (IntPtr stmt, int index);[DllImport("sqlite3", EntryPoint = "sqlite3_column_bytes")]public static extern int ColumnBytes (IntPtr stmt, int index);public static string ColumnString (IntPtr stmt, int index){return Marshal.PtrToStringUni (SQLite3.ColumnText16 (stmt, index));}public static byte[] ColumnByteArray (IntPtr stmt, int index){int length = ColumnBytes (stmt, index);byte[] result = new byte[length];if (length > 0)Marshal.Copy (ColumnBlob (stmt, index), result, 0, length);return result;}public enum ColType : int{Integer = 1,Float = 2,Text = 3,Blob = 4,Null = 5}}}

调用举例:

using System;using System.Collections.Generic;using System.Linq;using MonoTouch.Foundation;using MonoTouch.UIKit;using System.Data;using System.IO;using SQLite;namespace Xamarin.Screens.SQLiteNet{public partial class BasicOperations : UITableViewController{protected List<Person> people = new List<Person> ();protected TableSource tableSource;#region Constructorspublic BasicOperations (IntPtr handle) : base(handle) { Initialize (); }[Export("initWithCoder:")]public BasicOperations (NSCoder coder) : base(coder) { Initialize (); }public BasicOperations () : base("DataSample", null) { Initialize (); }protected void Initialize (){this.Title = "SQLite .NET";string dbName = "db_sqlite-net.db3";// check the database, if it doesn't exist, create itCheckAndCreateDatabase (dbName);// create a connection to the databaseusing (SQLiteConnection db = new SQLiteConnection (GetDBPath (dbName))) {// query a list of people from the dbpeople = new List<Person> (from p in db.Table<Person> () select p);// create a new table source from our people collectiontableSource = new BasicOperations.TableSource (people);// initialize the table view and set the sourcebase.TableView = new UITableView () {Source = tableSource};}}#endregionprotected string GetDBPath (string dbName){// get a reference to the documents foldervar documents = Environment.GetFolderPath (Environment.SpecialFolder.Personal);// create the db pathstring db = Path.Combine (documents, dbName);return db;}// This method checks to see if the database exists, and if it doesn't, it creates// it and inserts some dataprotected void CheckAndCreateDatabase (string dbName){// create a connection object. if the database doesn't exist, it will create // a blank databaseusing(SQLiteConnection db = new SQLiteConnection (GetDBPath (dbName))){// create the tablesdb.CreateTable<Person> ();// skip inserting data if it already existsif(db.Table<Person>().Count() > 0)return;// declare varsList<Person> people = new List<Person> ();// create a list of people that we're going to insertpeople.Add (new Person () { FirstName = "Peter", LastName = "Gabriel" });people.Add (new Person () { FirstName = "Thom", LastName = "Yorke" });people.Add (new Person () { FirstName = "J", LastName = "Spaceman" });people.Add (new Person () { FirstName = "Benjamin", LastName = "Gibbard" });// insert our peopledb.InsertAll (people);// close the connectiondb.Close ();}}// A simple data source for our tableprotected class TableSource : UITableViewSource{List<Person> items;public TableSource (List<Person> _items) : base() { this.items = _items; }public override int NumberOfSections (UITableView tableView) { return 1; }public override int RowsInSection (UITableView tableview, int section) { return this.items.Count; }public override UITableViewCell GetCell (UITableView tableView, NSIndexPath indexPath){UITableViewCell cell;cell = tableView.DequeueReusableCell ("item");if(cell == null) cell = new UITableViewCell(UITableViewCellStyle.Default, "item");cell.TextLabel.Text = this.items[indexPath.Row].FirstName + " " + this.items[indexPath.Row].LastName;return cell;}}}}


1 0
原创粉丝点击