Skip to content

Instantly share code, notes, and snippets.

@slodge
Created October 9, 2012 08:23
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save slodge/3857333 to your computer and use it in GitHub Desktop.
Save slodge/3857333 to your computer and use it in GitHub Desktop.
Portable sqlite-net - please note license from "Krueger Systems, Inc."
using System;
using Community.CsharpSqlite;
namespace Tasky.DL.SQLite
{
public class CommunitySqlite3 : SQLite3
{
public static void EnsureInitialised()
{
if (SQLite3.Instance == null)
{
SQLite3.Instance = new CommunitySqlite3();
}
}
public class Sqlite3DatabaseHandle : ISqlite3DatabaseHandle
{
public Sqlite3DatabaseHandle(Sqlite3.sqlite3 handle)
{
Handle = handle;
}
public Community.CsharpSqlite.Sqlite3.sqlite3 Handle { get; private set; }
}
public class Sqlite3Statement : ISqlite3Statement
{
public Sqlite3Statement(Sqlite3.Vdbe handle)
{
Handle = handle;
}
public Sqlite3.Vdbe Handle { get; private set; }
}
private Sqlite3.sqlite3 RealHandle(ISqlite3DatabaseHandle handle)
{
return ((Sqlite3DatabaseHandle)handle).Handle;
}
private Sqlite3.Vdbe RealHandle(ISqlite3Statement handle)
{
return ((Sqlite3Statement)handle).Handle;
}
public override Result OpenImpl(string filename, out ISqlite3DatabaseHandle db)
{
var handle = new Sqlite3.sqlite3();
var toReturn = (Result)Sqlite3.sqlite3_open(filename, ref handle);
db = (handle == null) ? null : new Sqlite3DatabaseHandle(handle);
return toReturn;
}
public override Result CloseImpl(ISqlite3DatabaseHandle db)
{
return (Result)Sqlite3.sqlite3_close(RealHandle(db));
}
public override Result BusyTimeoutImpl(ISqlite3DatabaseHandle db, int milliseconds)
{
return (Result)Sqlite3.sqlite3_busy_timeout(RealHandle(db), milliseconds);
}
public override int ChangesImpl(ISqlite3DatabaseHandle db)
{
return Sqlite3.sqlite3_changes(RealHandle(db));
}
public override ISqlite3Statement Prepare2Impl(ISqlite3DatabaseHandle db, string query)
{
Sqlite3.Vdbe stmt = new Sqlite3.Vdbe();
var r = Sqlite3.sqlite3_prepare_v2(RealHandle(db), query, query.Length, ref stmt, 0);
if (r != 0) {
throw SQLiteException.New((Result)r, GetErrmsg(db));
}
return new Sqlite3Statement(stmt);
}
public override Result StepImpl(ISqlite3Statement stmt)
{
return (Result)Sqlite3.sqlite3_step(RealHandle(stmt));
}
public override Result ResetImpl(ISqlite3Statement stmt)
{
return (Result)Sqlite3.sqlite3_reset(RealHandle(stmt));
}
public override Result FinalizeImpl(ISqlite3Statement stmt)
{
var real = RealHandle(stmt);
return (Result)Sqlite3.sqlite3_finalize(ref real);
}
public override long LastInsertRowidImpl(ISqlite3DatabaseHandle db)
{
return Sqlite3.sqlite3_last_insert_rowid(RealHandle(db));
}
public override string GetErrmsgImpl(ISqlite3DatabaseHandle db)
{
return Sqlite3.sqlite3_errmsg(RealHandle(db));
}
public override int BindParameterIndexImpl(ISqlite3Statement stmt, string name)
{
return Sqlite3.sqlite3_bind_parameter_index(RealHandle(stmt), name);
}
public override int BindNullImpl(ISqlite3Statement stmt, int index)
{
return Sqlite3.sqlite3_bind_null(RealHandle(stmt), index);
}
public override int BindIntImpl(ISqlite3Statement stmt, int index, int val)
{
return Sqlite3.sqlite3_bind_int(RealHandle(stmt), index, val);
}
public override int BindInt64Impl(ISqlite3Statement stmt, int index, long val)
{
return Sqlite3.sqlite3_bind_int64(RealHandle(stmt), index, val);
}
public override int BindDoubleImpl(ISqlite3Statement stmt, int index, double val)
{
return Sqlite3.sqlite3_bind_double(RealHandle(stmt), index, val);
}
public override int BindTextImpl(ISqlite3Statement stmt, int index, string val, int n, IntPtr free)
{
return Sqlite3.sqlite3_bind_text(RealHandle(stmt), index, val, n, null);
}
public override int BindBlobImpl(ISqlite3Statement stmt, int index, byte[] val, int n, IntPtr free)
{
return Sqlite3.sqlite3_bind_blob(RealHandle(stmt), index, val, n, null);
}
public override int ColumnCountImpl(ISqlite3Statement stmt)
{
return Sqlite3.sqlite3_column_count(RealHandle(stmt));
}
public override string ColumnNameImpl(ISqlite3Statement stmt, int index)
{
return Sqlite3.sqlite3_column_name(RealHandle(stmt), index);
}
public override string ColumnName16Impl(ISqlite3Statement stmt, int index)
{
return Sqlite3.sqlite3_column_name(RealHandle(stmt), index);
}
public override ColType ColumnTypeImpl(ISqlite3Statement stmt, int index)
{
return (ColType)Sqlite3.sqlite3_column_type(RealHandle(stmt), index);
}
public override int ColumnIntImpl(ISqlite3Statement stmt, int index)
{
return Sqlite3.sqlite3_column_int(RealHandle(stmt), index);
}
public override long ColumnInt64Impl(ISqlite3Statement stmt, int index)
{
return Sqlite3.sqlite3_column_int64(RealHandle(stmt), index);
}
public override double ColumnDoubleImpl(ISqlite3Statement stmt, int index)
{
return Sqlite3.sqlite3_column_double(RealHandle(stmt), index);
}
public override string ColumnTextImpl(ISqlite3Statement stmt, int index)
{
return Sqlite3.sqlite3_column_text(RealHandle(stmt), index);
}
public override string ColumnText16Impl(ISqlite3Statement stmt, int index)
{
return Sqlite3.sqlite3_column_text(RealHandle(stmt), index);
}
public override byte[] ColumnBlobImpl(ISqlite3Statement stmt, int index)
{
return Sqlite3.sqlite3_column_blob(RealHandle(stmt), index);
}
public override int ColumnBytesImpl(ISqlite3Statement stmt, int index)
{
return Sqlite3.sqlite3_column_bytes(RealHandle(stmt), index);
}
public override string ColumnStringImpl(ISqlite3Statement stmt, int index)
{
return Sqlite3.sqlite3_column_text(RealHandle(stmt), index);
}
public override byte[] ColumnByteArrayImpl(ISqlite3Statement stmt, int index)
{
return ColumnBlob(stmt, index);
}
}
}
using System;
using System.Runtime.InteropServices;
namespace Tasky.DL.SQLite
{
public class PInvokeSQLite3 : SQLite3
{
public static void EnsureInitialised()
{
if (SQLite3.Instance == null)
{
SQLite3.Instance = new PInvokeSQLite3();
}
}
public class Sqlite3DatabaseHandle : ISqlite3DatabaseHandle
{
public Sqlite3DatabaseHandle(IntPtr handle)
{
Handle = handle;
}
public IntPtr Handle { get; private set; }
}
public class Sqlite3Statement : ISqlite3Statement
{
public Sqlite3Statement(IntPtr handle)
{
Handle = handle;
}
public IntPtr Handle { get; private set; }
}
private IntPtr RealHandle(ISqlite3DatabaseHandle handle)
{
return ((Sqlite3DatabaseHandle)handle).Handle;
}
private IntPtr RealHandle(ISqlite3Statement handle)
{
return ((Sqlite3Statement)handle).Handle;
}
[DllImport("sqlite3", EntryPoint = "sqlite3_open")]
private static extern Result Pinvoke_Open (string filename, out IntPtr db);
[DllImport("sqlite3", EntryPoint = "sqlite3_close")]
public static extern Result Pinvoke_Close (IntPtr db);
[DllImport("sqlite3", EntryPoint = "sqlite3_config")]
public static extern Result Pinvoke_Config (ConfigOption option);
[DllImport("sqlite3", EntryPoint = "sqlite3_busy_timeout")]
public static extern Result Pinvoke_BusyTimeout (IntPtr db, int milliseconds);
[DllImport("sqlite3", EntryPoint = "sqlite3_changes")]
public static extern int Pinvoke_Changes (IntPtr db);
[DllImport("sqlite3", EntryPoint = "sqlite3_prepare_v2")]
public static extern Result Pinvoke_Prepare2 (IntPtr db, string sql, int numBytes, out IntPtr stmt, IntPtr pzTail);
[DllImport("sqlite3", EntryPoint = "sqlite3_step")]
public static extern Result Pinvoke_Step (IntPtr stmt);
[DllImport("sqlite3", EntryPoint = "sqlite3_reset")]
public static extern Result Pinvoke_Reset (IntPtr stmt);
[DllImport("sqlite3", EntryPoint = "sqlite3_finalize")]
public static extern Result Pinvoke_Finalize (IntPtr stmt);
[DllImport("sqlite3", EntryPoint = "sqlite3_last_insert_rowid")]
public static extern long Pinvoke_LastInsertRowid (IntPtr db);
[DllImport("sqlite3", EntryPoint = "sqlite3_errmsg16")]
public static extern IntPtr Pinvoke_Errmsg (IntPtr db);
[DllImport("sqlite3", EntryPoint = "sqlite3_bind_parameter_index")]
public static extern int Pinvoke_BindParameterIndex (IntPtr stmt, string name);
[DllImport("sqlite3", EntryPoint = "sqlite3_bind_null")]
public static extern int Pinvoke_BindNull (IntPtr stmt, int index);
[DllImport("sqlite3", EntryPoint = "sqlite3_bind_int")]
public static extern int Pinvoke_BindInt (IntPtr stmt, int index, int val);
[DllImport("sqlite3", EntryPoint = "sqlite3_bind_int64")]
public static extern int Pinvoke_BindInt64 (IntPtr stmt, int index, long val);
[DllImport("sqlite3", EntryPoint = "sqlite3_bind_double")]
public static extern int Pinvoke_BindDouble (IntPtr stmt, int index, double val);
[DllImport("sqlite3", EntryPoint = "sqlite3_bind_text")]
public static extern int Pinvoke_BindText (IntPtr stmt, int index, string val, int n, IntPtr free);
[DllImport("sqlite3", EntryPoint = "sqlite3_bind_blob")]
public static extern int Pinvoke_BindBlob (IntPtr stmt, int index, byte[] val, int n, IntPtr free);
[DllImport("sqlite3", EntryPoint = "sqlite3_column_count")]
public static extern int Pinvoke_ColumnCount (IntPtr stmt);
[DllImport("sqlite3", EntryPoint = "sqlite3_column_name")]
public static extern IntPtr Pinvoke_ColumnName (IntPtr stmt, int index);
[DllImport("sqlite3", EntryPoint = "sqlite3_column_name16")]
private static extern IntPtr Pinvoke_ColumnName16Internal (IntPtr stmt, int index);
[DllImport("sqlite3", EntryPoint = "sqlite3_column_type")]
public static extern ColType Pinvoke_ColumnType (IntPtr stmt, int index);
[DllImport("sqlite3", EntryPoint = "sqlite3_column_int")]
public static extern int Pinvoke_ColumnInt (IntPtr stmt, int index);
[DllImport("sqlite3", EntryPoint = "sqlite3_column_int64")]
public static extern long Pinvoke_ColumnInt64 (IntPtr stmt, int index);
[DllImport("sqlite3", EntryPoint = "sqlite3_column_double")]
public static extern double Pinvoke_ColumnDouble (IntPtr stmt, int index);
[DllImport("sqlite3", EntryPoint = "sqlite3_column_text")]
public static extern IntPtr Pinvoke_ColumnText (IntPtr stmt, int index);
[DllImport("sqlite3", EntryPoint = "sqlite3_column_text16")]
public static extern IntPtr Pinvoke_ColumnText16 (IntPtr stmt, int index);
[DllImport("sqlite3", EntryPoint = "sqlite3_column_blob")]
public static extern IntPtr Pinvoke_ColumnBlob (IntPtr stmt, int index);
[DllImport("sqlite3", EntryPoint = "sqlite3_column_bytes")]
public static extern int Pinvoke_ColumnBytes (IntPtr stmt, int index);
public override Result OpenImpl(string filename, out ISqlite3DatabaseHandle db)
{
IntPtr value;
var toReturn = Pinvoke_Open(filename, out value);
db = value == IntPtr.Zero ? null : new Sqlite3DatabaseHandle(value);
return toReturn;
}
public override Result CloseImpl(ISqlite3DatabaseHandle db)
{
return Pinvoke_Close(RealHandle(db));
}
public override Result BusyTimeoutImpl(ISqlite3DatabaseHandle db, int milliseconds)
{
return Pinvoke_BusyTimeout(RealHandle(db), milliseconds);
}
public override int ChangesImpl(ISqlite3DatabaseHandle db)
{
return Pinvoke_Changes(RealHandle(db));
}
public override ISqlite3Statement Prepare2Impl(ISqlite3DatabaseHandle db, string query)
{
IntPtr stmt;
var r = Pinvoke_Prepare2 (RealHandle(db), query, query.Length, out stmt, IntPtr.Zero);
if (r != Result.OK) {
throw SQLiteException.New (r, GetErrmsg (db));
}
return new Sqlite3Statement(stmt);
}
public override Result StepImpl(ISqlite3Statement stmt)
{
return Pinvoke_Step(RealHandle(stmt));
}
public override Result ResetImpl(ISqlite3Statement stmt)
{
return Pinvoke_Reset(RealHandle(stmt));
}
public override Result FinalizeImpl(ISqlite3Statement stmt)
{
return Pinvoke_Finalize(RealHandle(stmt));
}
public override long LastInsertRowidImpl(ISqlite3DatabaseHandle db)
{
return Pinvoke_LastInsertRowid(RealHandle(db));
}
public override string GetErrmsgImpl(ISqlite3DatabaseHandle db)
{
return Marshal.PtrToStringUni (Pinvoke_Errmsg (RealHandle(db)));
}
public override int BindParameterIndexImpl(ISqlite3Statement stmt, string name)
{
return Pinvoke_BindParameterIndex(RealHandle(stmt), name);
}
public override int BindNullImpl(ISqlite3Statement stmt, int index)
{
return Pinvoke_BindNull(RealHandle(stmt), index);
}
public override int BindIntImpl(ISqlite3Statement stmt, int index, int val)
{
return Pinvoke_BindInt(RealHandle(stmt), index, val);
}
public override int BindInt64Impl(ISqlite3Statement stmt, int index, long val)
{
return Pinvoke_BindInt64(RealHandle(stmt), index, val);
}
public override int BindDoubleImpl(ISqlite3Statement stmt, int index, double val)
{
return Pinvoke_BindDouble(RealHandle(stmt), index, val);
}
public override int BindTextImpl(ISqlite3Statement stmt, int index, string val, int n, IntPtr free)
{
return Pinvoke_BindText(RealHandle(stmt), index, val, n, free);
}
public override int BindBlobImpl(ISqlite3Statement stmt, int index, byte[] val, int n, IntPtr free)
{
return Pinvoke_BindBlob(RealHandle(stmt), index, val, n, free);
}
public override int ColumnCountImpl(ISqlite3Statement stmt)
{
return Pinvoke_ColumnCount(RealHandle(stmt));
}
public override string ColumnNameImpl(ISqlite3Statement stmt, int index)
{
#warning Need to check this marshalling - original code just returns an int?!
return Marshal.PtrToStringUni(Pinvoke_ColumnName(RealHandle(stmt), index));
}
public override string ColumnName16Impl(ISqlite3Statement stmt, int index)
{
return Marshal.PtrToStringUni(Pinvoke_ColumnName16Internal(RealHandle(stmt), index));
}
public override ColType ColumnTypeImpl(ISqlite3Statement stmt, int index)
{
return Pinvoke_ColumnType(RealHandle(stmt), index);
}
public override int ColumnIntImpl(ISqlite3Statement stmt, int index)
{
return Pinvoke_ColumnInt(RealHandle(stmt), index);
}
public override long ColumnInt64Impl(ISqlite3Statement stmt, int index)
{
return Pinvoke_ColumnInt64(RealHandle(stmt), index);
}
public override double ColumnDoubleImpl(ISqlite3Statement stmt, int index)
{
return Pinvoke_ColumnDouble(RealHandle(stmt), index);
}
public override string ColumnTextImpl(ISqlite3Statement stmt, int index)
{
#warning Check this ColumnText string marshalling
return Marshal.PtrToStringUni(Pinvoke_ColumnText(RealHandle(stmt), index));
}
public override string ColumnText16Impl(ISqlite3Statement stmt, int index)
{
#warning Check this ColumnText string marshalling
return Marshal.PtrToStringUni(Pinvoke_ColumnText16(RealHandle(stmt), index));
}
public override byte[] ColumnBlobImpl(ISqlite3Statement stmt, int index)
{
int length = ColumnBytes (stmt, index);
byte[] result = new byte[length];
if (length > 0)
Marshal.Copy (Pinvoke_ColumnBlob(RealHandle(stmt), index), result, 0, length);
return result;
}
public override int ColumnBytesImpl(ISqlite3Statement stmt, int index)
{
return Pinvoke_ColumnBytes(RealHandle(stmt), index);
}
public override string ColumnStringImpl(ISqlite3Statement stmt, int index)
{
return Marshal.PtrToStringUni(Pinvoke_ColumnText16(RealHandle(stmt), index));
}
public override byte[] ColumnByteArrayImpl(ISqlite3Statement stmt, int index)
{
return ColumnBlobImpl(stmt, index);
}
}
}
//
// Copyright (c) 2009-2011 Krueger Systems, Inc.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
//
using System;
using System.Collections.Generic;
using System.Reflection;
using System.Linq;
using System.Linq.Expressions;
public interface ISqlite3DatabaseHandle
{
}
public interface ISqlite3Statement
{
}
namespace Tasky.DL.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;
public ISqlite3DatabaseHandle Handle { get; private set; }
internal static readonly ISqlite3DatabaseHandle NullHandle = null;
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;
ISqlite3DatabaseHandle 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 != NullHandle) {
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 if not exists \"" + map.TableName + "\"(\n";
var decls = map.Columns.Select(p => Orm.SqlDecl(p));
var decl = string.Join(",\n", decls.ToArray());
query += decl;
query += ")";
var count = Execute(query);
if (count == 0) { //Possible bug: This always seems to return 0?
// Table already exists, migrate it
MigrateTable(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;
}
public 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;
#if DEBUG
Console.WriteLine("Finished in {0} ms ({1:0.0} s total)", _sw.ElapsedMilliseconds, _elapsedMilliseconds / 1000.0);
#endif
}
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 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.
/// The enumerator will call sqlite3_step on each call to MoveNext, so the database
/// connection must remain open for the lifetime of the enumerator.
/// </returns>
public IEnumerable<T> DeferredQuery<T>(string query, params object[] args) where T : new()
{
var cmd = CreateCommand(query, args);
return cmd.ExecuteDeferredQuery<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>
/// 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.
/// The enumerator will call sqlite3_step on each call to MoveNext, so the database
/// connection must remain open for the lifetime of the enumerator.
/// </returns>
public IEnumerable<object> DeferredQuery(TableMapping map, string query, params object[] args)
{
var cmd = CreateCommand(query, args);
return cmd.ExecuteDeferredQuery<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.Columns
where p != pk
select p;
var vals = from c in cols
select 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 cols
select "\"" + 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 != NullHandle) {
try {
foreach (var sqlInsertCommand in _mappings.Values) {
sqlInsertCommand.Dispose();
}
var r = SQLite3.Close(Handle);
if (r != SQLite3.Result.OK) {
string msg = SQLite3.GetErrmsg(Handle);
throw SQLiteException.New(r, msg);
}
} finally {
Handle = NullHandle;
_open = false;
}
}
}
}
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;
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, null));
}
}
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;
}
PreparedSqlLiteInsertCommand _insertCommand;
string _insertCommandExtra = null;
public PreparedSqlLiteInsertCommand GetInsertCommand(SQLiteConnection conn, string extra)
{
if (_insertCommand == null) {
_insertCommand = CreateInsertCommand(conn, extra);
_insertCommandExtra = extra;
} else if (_insertCommandExtra != extra) {
_insertCommand.Dispose();
_insertCommand = CreateInsertCommand(conn, extra);
_insertCommandExtra = extra;
}
return _insertCommand;
}
private PreparedSqlLiteInsertCommand CreateInsertCommand(SQLiteConnection conn, string extra)
{
var cols = InsertColumns;
var insertSql = string.Format("insert {3} into \"{0}\"({1}) values ({2})", TableName,
string.Join(",", (from c in cols
select "\"" + c.Name + "\"").ToArray()),
string.Join(",", (from c in cols
select "?").ToArray()), extra);
var insertCommand = new PreparedSqlLiteInsertCommand(conn);
insertCommand.CommandText = insertSql;
return insertCommand;
}
protected internal void Dispose()
{
if (_insertCommand != null) {
_insertCommand.Dispose();
_insertCommand = null;
}
}
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 instead
ColumnType = 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) {
#if DEBUG
Console.WriteLine("Executing: " + this);
#endif
}
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 IEnumerable<T> ExecuteDeferredQuery<T>() where T : new()
{
return ExecuteDeferredQuery<T>(_conn.GetMapping(typeof(T)));
}
public List<T> ExecuteQuery<T>() where T : new()
{
return ExecuteDeferredQuery<T>(_conn.GetMapping(typeof(T))).ToList();
}
public List<T> ExecuteQuery<T>(TableMapping map)
{
return ExecuteDeferredQuery<T>(map).ToList();
}
public IEnumerable<T> ExecuteDeferredQuery<T>(TableMapping map)
{
if (_conn.Trace) {
#if DEBUG
Console.WriteLine("Executing Query: " + this);
#endif
}
var stmt = Prepare();
try {
var cols = new TableMapping.Column[SQLite3.ColumnCount(stmt)];
for (int i = 0; i < cols.Length; i++) {
var name = 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);
}
yield return (T)obj;
}
} finally {
SQLite3.Finalize(stmt);
}
}
public T ExecuteScalar<T>()
{
if (_conn.Trace) {
#if DEBUG
Console.WriteLine("Executing Query: " + this);
#endif
}
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);
}
ISqlite3Statement Prepare()
{
var stmt = SQLite3.Prepare2(_conn.Handle, CommandText);
BindAll(stmt);
return stmt;
}
void Finalize(ISqlite3Statement stmt)
{
SQLite3.Finalize(stmt);
}
void BindAll(ISqlite3Statement 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(ISqlite3Statement 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(ISqlite3Statement 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 ISqlite3Statement Statement { get; set; }
internal static readonly ISqlite3Statement NullStatement = default(ISqlite3Statement);
internal PreparedSqlLiteInsertCommand(SQLiteConnection conn)
{
Connection = conn;
}
public int ExecuteNonQuery(object[] source)
{
if (Connection.Trace) {
#if DEBUG
Console.WriteLine("Executing: " + CommandText);
#endif
}
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 ISqlite3Statement 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 != NullStatement) {
try {
SQLite3.Finalize(Statement);
} finally {
Statement = NullStatement;
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;
q._deferred = _deferred;
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;
}
bool _deferred = false;
public TableQuery<T> Deferred()
{
var q = Clone();
q._deferred = true;
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);
else
text = "(" + 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) : 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) {
/*
* need to consider this...
#if SILVERLIGHT
val = Expression.Lambda(expr).Compile().DynamicInvoke();
#else
*/
var m = (FieldInfo)mem.Member;
var eee = Expression.Lambda (expr);
val = m.GetValue (obj);
/*
#endif
*/
} 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 ?)";
else
throw 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()
{
if (!_deferred)
return GenerateCommand("*").ExecuteQuery<T>().GetEnumerator();
return GenerateCommand("*").ExecuteDeferredQuery<T>().GetEnumerator();
}
System.Collections.IEnumerator System.Collections.IEnumerable.GetEnumerator()
{
return GetEnumerator();
}
}
public abstract class SQLite3
{
public static SQLite3 Instance { get; protected set; }
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
}
public enum ColType : int
{
Integer = 1,
Float = 2,
Text = 3,
Blob = 4,
Null = 5
}
public static Result Open(string filename, out ISqlite3DatabaseHandle db)
{
return Instance.OpenImpl(filename, out db);
}
public abstract Result OpenImpl(string filename, out ISqlite3DatabaseHandle db);
public static Result Close(ISqlite3DatabaseHandle db)
{
return Instance.CloseImpl(db);
}
public abstract Result CloseImpl(ISqlite3DatabaseHandle db);
public static Result BusyTimeout(ISqlite3DatabaseHandle db, int milliseconds)
{
return Instance.BusyTimeoutImpl(db, milliseconds);
}
public abstract Result BusyTimeoutImpl(ISqlite3DatabaseHandle db, int milliseconds);
public static int Changes(ISqlite3DatabaseHandle db)
{
return Instance.ChangesImpl(db);
}
public abstract int ChangesImpl(ISqlite3DatabaseHandle db);
public static ISqlite3Statement Prepare2(ISqlite3DatabaseHandle db, string query)
{
return Instance.Prepare2Impl(db, query);
}
public abstract ISqlite3Statement Prepare2Impl(ISqlite3DatabaseHandle db, string query);
public static Result Step(ISqlite3Statement stmt)
{
return Instance.StepImpl(stmt);
}
public abstract Result StepImpl(ISqlite3Statement stmt);
public static Result Reset(ISqlite3Statement stmt)
{
return Instance.ResetImpl(stmt);
}
public abstract Result ResetImpl(ISqlite3Statement stmt);
public static Result Finalize(ISqlite3Statement stmt)
{
return Instance.FinalizeImpl(stmt);
}
public abstract Result FinalizeImpl(ISqlite3Statement stmt);
public static long LastInsertRowid(ISqlite3DatabaseHandle db)
{
return Instance.LastInsertRowidImpl(db);
}
public abstract long LastInsertRowidImpl(ISqlite3DatabaseHandle db);
public static string GetErrmsg(ISqlite3DatabaseHandle db)
{
return Instance.GetErrmsgImpl(db);
}
public abstract string GetErrmsgImpl(ISqlite3DatabaseHandle db);
public static int BindParameterIndex(ISqlite3Statement stmt, string name)
{
return Instance.BindParameterIndexImpl(stmt, name);
}
public abstract int BindParameterIndexImpl(ISqlite3Statement stmt, string name);
public static int BindNull(ISqlite3Statement stmt, int index)
{
return Instance.BindNullImpl(stmt, index);
}
public abstract int BindNullImpl(ISqlite3Statement stmt, int index);
public static int BindInt(ISqlite3Statement stmt, int index, int val)
{
return Instance.BindIntImpl(stmt, index, val);
}
public abstract int BindIntImpl(ISqlite3Statement stmt, int index, int val);
public static int BindInt64(ISqlite3Statement stmt, int index, long val)
{
return Instance.BindInt64Impl(stmt, index, val);
}
public abstract int BindInt64Impl(ISqlite3Statement stmt, int index, long val);
public static int BindDouble(ISqlite3Statement stmt, int index, double val)
{
return Instance.BindDoubleImpl(stmt, index, val);
}
public abstract int BindDoubleImpl(ISqlite3Statement stmt, int index, double val);
public static int BindText(ISqlite3Statement stmt, int index, string val, int n, IntPtr free)
{
return Instance.BindTextImpl(stmt, index, val, n, free);
}
public abstract int BindTextImpl(ISqlite3Statement stmt, int index, string val, int n, IntPtr free);
public static int BindBlob(ISqlite3Statement stmt, int index, byte[] val, int n, IntPtr free)
{
return Instance.BindBlobImpl(stmt, index, val, n, free);
}
public abstract int BindBlobImpl(ISqlite3Statement stmt, int index, byte[] val, int n, IntPtr free);
public static int ColumnCount(ISqlite3Statement stmt)
{
return Instance.ColumnCountImpl(stmt);
}
public abstract int ColumnCountImpl(ISqlite3Statement stmt);
public static string ColumnName(ISqlite3Statement stmt, int index)
{
return Instance.ColumnNameImpl(stmt, index);
}
public abstract string ColumnNameImpl(ISqlite3Statement stmt, int index);
public static string ColumnName16(ISqlite3Statement stmt, int index)
{
return Instance.ColumnName16Impl(stmt, index);
}
public abstract string ColumnName16Impl(ISqlite3Statement stmt, int index);
public static ColType ColumnType(ISqlite3Statement stmt, int index)
{
return Instance.ColumnTypeImpl(stmt, index);
}
public abstract ColType ColumnTypeImpl(ISqlite3Statement stmt, int index);
public static int ColumnInt(ISqlite3Statement stmt, int index)
{
return Instance.ColumnIntImpl(stmt, index);
}
public abstract int ColumnIntImpl(ISqlite3Statement stmt, int index);
public static long ColumnInt64(ISqlite3Statement stmt, int index)
{
return Instance.ColumnInt64Impl(stmt, index);
}
public abstract long ColumnInt64Impl(ISqlite3Statement stmt, int index);
public static double ColumnDouble(ISqlite3Statement stmt, int index)
{
return Instance.ColumnDoubleImpl(stmt, index);
}
public abstract double ColumnDoubleImpl(ISqlite3Statement stmt, int index);
public static string ColumnText(ISqlite3Statement stmt, int index)
{
return Instance.ColumnTextImpl(stmt, index);
}
public abstract string ColumnTextImpl(ISqlite3Statement stmt, int index);
public static string ColumnText16(ISqlite3Statement stmt, int index)
{
return Instance.ColumnText16Impl(stmt, index);
}
public abstract string ColumnText16Impl(ISqlite3Statement stmt, int index);
public static byte[] ColumnBlob(ISqlite3Statement stmt, int index)
{
return Instance.ColumnBlobImpl(stmt, index);
}
public abstract byte[] ColumnBlobImpl(ISqlite3Statement stmt, int index);
public static int ColumnBytes(ISqlite3Statement stmt, int index)
{
return Instance.ColumnBytesImpl(stmt, index);
}
public abstract int ColumnBytesImpl(ISqlite3Statement stmt, int index);
public static string ColumnString(ISqlite3Statement stmt, int index)
{
return Instance.ColumnStringImpl(stmt, index);
}
public abstract string ColumnStringImpl(ISqlite3Statement stmt, int index);
public static byte[] ColumnByteArray(ISqlite3Statement stmt, int index)
{
return Instance.ColumnByteArrayImpl(stmt, index);
}
public abstract byte[] ColumnByteArrayImpl(ISqlite3Statement stmt, int index);
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment