diff --git a/src/Sqlite/Library.fs b/src/Sqlite/Library.fs index dd1063d..6cf6dc9 100644 --- a/src/Sqlite/Library.fs +++ b/src/Sqlite/Library.fs @@ -1,4 +1,5 @@ -module BitBadger.Documents.Sqlite +/// Document store implementation for SQLite +module BitBadger.Documents.Sqlite open BitBadger.Documents open Microsoft.Data.Sqlite @@ -27,6 +28,7 @@ module Configuration = | None -> invalidOp "Please provide a connection string before attempting data access" +/// Query definitions [] module Query = @@ -39,45 +41,65 @@ module Query = Query.Definition.ensureTableFor name "TEXT" -/// Create an ID parameter (key will be treated as a string) -[] -let idParam (key: 'TKey) = - SqliteParameter("@id", string key) +/// Parameter handling helpers +[] +module Parameters = + + /// Create an ID parameter (name "@id", key will be treated as a string) + [] + let idParam (key: 'TKey) = + SqliteParameter("@id", string key) -/// Create a parameter with a JSON value -[] -let jsonParam name (it: 'TJson) = - SqliteParameter(name, Configuration.serializer().Serialize it) + /// Create a parameter with a JSON value + [] + let jsonParam name (it: 'TJson) = + SqliteParameter(name, Configuration.serializer().Serialize it) -/// Create a JSON field parameter -[] -let fieldParam (value: obj) = - SqliteParameter("@field", value) + /// Create a JSON field parameter (name "@field") + [] + let fieldParam (value: obj) = + SqliteParameter("@field", value) -/// Create a domain item from a document, specifying the field in which the document is found -[] -let fromDocument<'TDoc> field (rdr: SqliteDataReader) : 'TDoc = - Configuration.serializer().Deserialize<'TDoc>(rdr.GetString(rdr.GetOrdinal(field))) -/// Create a domain item from a document -[] -let fromData<'TDoc> rdr = - fromDocument<'TDoc> "data" rdr +/// Helper functions for handling results +[] +module Results = + + /// Create a domain item from a document, specifying the field in which the document is found + [] + let fromDocument<'TDoc> field (rdr: SqliteDataReader) : 'TDoc = + Configuration.serializer().Deserialize<'TDoc>(rdr.GetString(rdr.GetOrdinal(field))) -/// Create a list of items for the results of the given command, using the specified mapping function -[] -let toCustomList<'TDoc> (cmd: SqliteCommand) (mapFunc: SqliteDataReader -> 'TDoc) = backgroundTask { - use! rdr = cmd.ExecuteReaderAsync() - let mutable it = Seq.empty<'TDoc> - while! rdr.ReadAsync() do - it <- Seq.append it (Seq.singleton (mapFunc rdr)) - return List.ofSeq it -} + /// Create a domain item from a document + [] + let fromData<'TDoc> rdr = + fromDocument<'TDoc> "data" rdr + + /// Create a list of items for the results of the given command, using the specified mapping function + [] + let toCustomList<'TDoc> (cmd: SqliteCommand) (mapFunc: SqliteDataReader -> 'TDoc) = backgroundTask { + use! rdr = cmd.ExecuteReaderAsync() + let mutable it = Seq.empty<'TDoc> + while! rdr.ReadAsync() do + it <- Seq.append it (Seq.singleton (mapFunc rdr)) + return List.ofSeq it + } + + /// Create a list of items for the results of the given command, using the specified mapping function + let ToCustomList<'TDoc>(cmd, mapFunc: System.Func) = backgroundTask { + let! results = toCustomList<'TDoc> cmd mapFunc.Invoke + return ResizeArray<'TDoc> results + } + + /// Create a list of items for the results of the given command + [] + let toDocumentList<'TDoc> (cmd: SqliteCommand) = + toCustomList<'TDoc> cmd fromData + + /// Create a list of items for the results of the given command + let ToDocumentList<'TDoc> cmd = + ToCustomList<'TDoc>(cmd, fromData<'TDoc>) -/// Create a list of items for the results of the given command -[] -let toDocumentList<'TDoc> (cmd: SqliteCommand) = - toCustomList<'TDoc> cmd fromData /// Execute a non-query command let internal write (cmd: SqliteCommand) = backgroundTask { @@ -86,45 +108,6 @@ let internal write (cmd: SqliteCommand) = backgroundTask { } -/// Command creation helper functions -[] -module private Helpers = - - let addParam (cmd: SqliteCommand) it = - cmd.Parameters.Add it |> ignore - - /// Add an ID parameter to a command - let addIdParam (cmd: SqliteCommand) (key: 'TKey) = - addParam cmd (idParam key) - - /// Add a JSON document parameter to a command - let addJsonParam (cmd: SqliteCommand) name (it: 'TJson) = - addParam cmd (jsonParam name it) - - /// Add ID (@id) and document (@data) parameters to a command - let addIdAndDocParams cmd (docId: 'TKey) (doc: 'TDoc) = - addIdParam cmd docId - addJsonParam cmd "@data" doc - - /// Add a parameter to a SQLite command, ignoring the return value (can still be accessed on cmd via indexing) - let addFieldParam (cmd: SqliteCommand) (value: obj) = - addParam cmd (SqliteParameter("@field", value)) - - /// Execute a non-query statement to manipulate a document - let executeNonQuery query (document: 'T) (conn: SqliteConnection) = - use cmd = conn.CreateCommand() - cmd.CommandText <- query - addJsonParam cmd "@data" document - write cmd - - /// Execute a non-query statement to manipulate a document with an ID specified - let executeNonQueryWithId query (docId: 'TKey) (document: 'TDoc) (conn: SqliteConnection) = - use cmd = conn.CreateCommand() - cmd.CommandText <- query - addIdAndDocParams cmd docId document - write cmd - - /// Versions of queries that accept a SqliteConnection as the last parameter module WithConn = @@ -331,21 +314,68 @@ module WithConn = let byField tableName fieldName op (value: obj) conn = Custom.nonQuery (Query.Delete.byField tableName fieldName op) [ fieldParam value ] conn + +/// Commands to execute custom SQL queries +[] +module Custom = + + /// Execute a query that returns a list of results + [] + let list<'TDoc> query parameters (mapFunc: SqliteDataReader -> 'TDoc) = + use conn = Configuration.dbConn () + WithConn.Custom.list<'TDoc> query parameters mapFunc conn + + /// Execute a query that returns a list of results + let List<'TDoc>(query, parameters, mapFunc: System.Func) = + use conn = Configuration.dbConn () + WithConn.Custom.List<'TDoc>(query, parameters, mapFunc, conn) + + /// Execute a query that returns one or no results (returns None if not found) + [] + let single<'TDoc> query parameters (mapFunc: SqliteDataReader -> 'TDoc) = + use conn = Configuration.dbConn () + WithConn.Custom.single<'TDoc> query parameters mapFunc conn + + /// Execute a query that returns one or no results (returns null if not found) + let Single<'TDoc when 'TDoc: null>(query, parameters, mapFunc: System.Func) = + use conn = Configuration.dbConn () + WithConn.Custom.Single<'TDoc>(query, parameters, mapFunc, conn) + + /// Execute a query that does not return a value + [] + let nonQuery query parameters = + use conn = Configuration.dbConn () + WithConn.Custom.nonQuery query parameters conn + + /// Execute a query that returns a scalar value + [] + let scalar<'T when 'T: struct> query parameters (mapFunc: SqliteDataReader -> 'T) = + use conn = Configuration.dbConn () + WithConn.Custom.scalar<'T> query parameters mapFunc conn + + /// Execute a query that returns a scalar value + let Scalar<'T when 'T: struct>(query, parameters, mapFunc: System.Func) = + use conn = Configuration.dbConn () + WithConn.Custom.Scalar<'T>(query, parameters, mapFunc, conn) + /// Functions to create tables and indexes [] module Definition = /// Create a document table + [] let ensureTable name = use conn = Configuration.dbConn () WithConn.Definition.ensureTable name conn /// Insert a new document +[] let insert<'TDoc> tableName (document: 'TDoc) = use conn = Configuration.dbConn () WithConn.insert tableName document conn /// Save a document, inserting it if it does not exist and updating it if it does (AKA "upsert") +[] let save<'TDoc> tableName (document: 'TDoc) = use conn = Configuration.dbConn () WithConn.save tableName document conn @@ -355,11 +385,13 @@ let save<'TDoc> tableName (document: 'TDoc) = module Count = /// Count all documents in a table + [] let all tableName = use conn = Configuration.dbConn () WithConn.Count.all tableName conn /// Count matching documents using a comparison on a JSON field + [] let byField tableName fieldName op (value: obj) = use conn = Configuration.dbConn () WithConn.Count.byField tableName fieldName op value conn @@ -369,11 +401,13 @@ module Count = module Exists = /// Determine if a document exists for the given ID + [] let byId tableName (docId: 'TKey) = use conn = Configuration.dbConn () WithConn.Exists.byId tableName docId conn /// Determine if a document exists using a comparison on a JSON field + [] let byField tableName fieldName op (value: obj) = use conn = Configuration.dbConn () WithConn.Exists.byField tableName fieldName op value conn @@ -383,45 +417,78 @@ module Exists = module Find = /// Retrieve all documents in the given table + [] let all<'TDoc> tableName = use conn = Configuration.dbConn () WithConn.Find.all<'TDoc> tableName conn - /// Retrieve a document by its ID + /// Retrieve all documents in the given table + let All<'TDoc> tableName = + use conn = Configuration.dbConn () + WithConn.Find.All<'TDoc>(tableName, conn) + + /// Retrieve a document by its ID (returns None if not found) + [] let byId<'TKey, 'TDoc> tableName docId = use conn = Configuration.dbConn () WithConn.Find.byId<'TKey, 'TDoc> tableName docId conn + /// Retrieve a document by its ID (returns null if not found) + let ById<'TKey, 'TDoc when 'TDoc: null>(tableName, docId) = + use conn = Configuration.dbConn () + WithConn.Find.ById<'TKey, 'TDoc>(tableName, docId, conn) + /// Retrieve documents via a comparison on a JSON field + [] let byField<'TDoc> tableName fieldName op value = use conn = Configuration.dbConn () WithConn.Find.byField<'TDoc> tableName fieldName op value conn + /// Retrieve documents via a comparison on a JSON field + let ByField<'TDoc>(tableName, fieldName, op, value) = + use conn = Configuration.dbConn () + WithConn.Find.ByField<'TDoc>(tableName, fieldName, op, value, conn) + /// Retrieve documents via a comparison on a JSON field, returning only the first result + [] let firstByField<'TDoc> tableName fieldName op value = use conn = Configuration.dbConn () WithConn.Find.firstByField<'TDoc> tableName fieldName op value conn + /// Retrieve documents via a comparison on a JSON field, returning only the first result + let FirstByField<'TDoc when 'TDoc: null>(tableName, fieldName, op, value) = + use conn = Configuration.dbConn () + WithConn.Find.FirstByField<'TDoc>(tableName, fieldName, op, value, conn) + /// Commands to update documents [] module Update = /// Update an entire document + [] let full tableName (docId: 'TKey) (document: 'TDoc) = use conn = Configuration.dbConn () WithConn.Update.full tableName docId document conn /// Update an entire document + [] let fullFunc tableName (idFunc: 'TDoc -> 'TKey) (document: 'TDoc) = use conn = Configuration.dbConn () WithConn.Update.fullFunc tableName idFunc document conn + /// Update an entire document + let FullFunc(tableName, idFunc: System.Func<'TDoc, 'TKey>, document: 'TDoc) = + use conn = Configuration.dbConn () + WithConn.Update.FullFunc(tableName, idFunc, document, conn) + /// Update a partial document + [] let partialById tableName (docId: 'TKey) (partial: 'TPatch) = use conn = Configuration.dbConn () WithConn.Update.partialById tableName docId partial conn /// Update partial documents using a comparison on a JSON field in the WHERE clause + [] let partialByField tableName fieldName op (value: obj) (partial: 'TPatch) = use conn = Configuration.dbConn () WithConn.Update.partialByField tableName fieldName op value partial conn @@ -431,44 +498,40 @@ module Update = module Delete = /// Delete a document by its ID + [] let byId tableName (docId: 'TKey) = use conn = Configuration.dbConn () WithConn.Delete.byId tableName docId conn /// Delete documents by matching a comparison on a JSON field + [] let byField tableName fieldName op (value: obj) = use conn = Configuration.dbConn () WithConn.Delete.byField tableName fieldName op value conn -/// Commands to execute custom SQL queries -[] -module Custom = - - /// Execute a query that returns a list of results - let list<'TDoc> query parameters (mapFunc: SqliteDataReader -> 'TDoc) = - use conn = Configuration.dbConn () - WithConn.Custom.list<'TDoc> query parameters mapFunc conn - - /// Execute a query that returns one or no results - let single<'TDoc> query parameters (mapFunc: SqliteDataReader -> 'TDoc) = - use conn = Configuration.dbConn () - WithConn.Custom.single<'TDoc> query parameters mapFunc conn - - /// Execute a query that does not return a value - let nonQuery query parameters = - use conn = Configuration.dbConn () - WithConn.Custom.nonQuery query parameters conn - - /// Execute a query that returns a scalar value - let scalar<'T when 'T : struct> query parameters (mapFunc: SqliteDataReader -> 'T) = - use conn = Configuration.dbConn () - WithConn.Custom.scalar<'T> query parameters mapFunc conn +/// F# extensions for the SqliteConnection type [] module Extensions = type SqliteConnection with + /// Execute a query that returns a list of results + member conn.customList<'TDoc> query parameters mapFunc = + WithConn.Custom.list<'TDoc> query parameters mapFunc conn + + /// Execute a query that returns one or no results + member conn.customSingle<'TDoc> query parameters mapFunc = + WithConn.Custom.single<'TDoc> query parameters mapFunc conn + + /// Execute a query that does not return a value + member conn.customNonQuery query parameters = + WithConn.Custom.nonQuery query parameters conn + + /// Execute a query that returns a scalar value + member conn.customScalar<'T when 'T: struct> query parameters mapFunc = + WithConn.Custom.scalar<'T> query parameters mapFunc conn + /// Create a document table member conn.ensureTable name = WithConn.Definition.ensureTable name conn @@ -541,18 +604,121 @@ module Extensions = member conn.deleteByField tableName fieldName op (value: obj) = WithConn.Delete.byField tableName fieldName op value conn - /// Execute a query that returns a list of results - member conn.customList<'TDoc> query parameters mapFunc = - WithConn.Custom.list<'TDoc> query parameters mapFunc conn - /// Execute a query that returns one or no results - member conn.customSingle<'TDoc> query parameters mapFunc = - WithConn.Custom.single<'TDoc> query parameters mapFunc conn - - /// Execute a query that does not return a value - member conn.customNonQuery query parameters = - WithConn.Custom.nonQuery query parameters conn +open System.Runtime.CompilerServices - /// Execute a query that returns a scalar value - member conn.customScalar<'T when 'T: struct> query parameters mapFunc = - WithConn.Custom.scalar<'T> query parameters mapFunc conn +/// C# extensions on the SqliteConnection type +[] +type SqliteConnectionCSharpExtensions = + + /// Execute a query that returns a list of results + [] + static member inline CustomList<'TDoc>(conn, query, parameters, mapFunc: System.Func) = + WithConn.Custom.List<'TDoc>(query, parameters, mapFunc, conn) + + /// Execute a query that returns one or no results + [] + static member inline CustomSingle<'TDoc when 'TDoc: null>( + conn, query, parameters, mapFunc: System.Func) = + WithConn.Custom.Single<'TDoc>(query, parameters, mapFunc, conn) + + /// Execute a query that does not return a value + [] + static member inline CustomNonQuery(conn, query, parameters) = + WithConn.Custom.nonQuery query parameters conn + + /// Execute a query that returns a scalar value + [] + static member inline CustomScalar<'T when 'T: struct>( + conn, query, parameters, mapFunc: System.Func) = + WithConn.Custom.Scalar<'T>(query, parameters, mapFunc, conn) + + /// Create a document table + [] + static member inline EnsureTable(conn, name) = + WithConn.Definition.ensureTable name conn + + /// Create an index on one or more fields in a document table + [] + static member inline EnsureIndex(conn, tableName, indexName, fields) = + WithConn.Definition.ensureIndex tableName indexName fields conn + + /// Insert a new document + [] + static member inline Insert<'TDoc>(conn, tableName, document: 'TDoc) = + WithConn.insert<'TDoc> tableName document conn + + /// Save a document, inserting it if it does not exist and updating it if it does (AKA "upsert") + [] + static member inline Save<'TDoc>(conn, tableName, document: 'TDoc) = + WithConn.save<'TDoc> tableName document conn + + /// Count all documents in a table + [] + static member inline CountAll(conn, tableName) = + WithConn.Count.all tableName conn + + /// Count matching documents using a comparison on a JSON field + [] + static member inline CountByField(conn, tableName, fieldName, op, value: obj) = + WithConn.Count.byField tableName fieldName op value conn + + /// Determine if a document exists for the given ID + [] + static member inline ExistsById<'TKey>(conn, tableName, docId: 'TKey) = + WithConn.Exists.byId tableName docId conn + + /// Determine if a document exists using a comparison on a JSON field + [] + static member inline ExistsByField(conn, tableName, fieldName, op, value: obj) = + WithConn.Exists.byField tableName fieldName op value conn + + /// Retrieve all documents in the given table + [] + static member inline FindAll<'TDoc>(conn, tableName) = + WithConn.Find.All<'TDoc>(tableName, conn) + + /// Retrieve a document by its ID + [] + static member inline FindById<'TKey, 'TDoc when 'TDoc: null>(conn, tableName, docId: 'TKey) = + WithConn.Find.ById<'TKey, 'TDoc>(tableName, docId, conn) + + /// Retrieve documents via a comparison on a JSON field + [] + static member inline FindByField<'TDoc>(conn, tableName, fieldName, op, value) = + WithConn.Find.ByField<'TDoc>(tableName, fieldName, op, value, conn) + + /// Retrieve documents via a comparison on a JSON field, returning only the first result + [] + static member inline FindFirstByField<'TDoc when 'TDoc: null>(conn, tableName, fieldName, op, value: obj) = + WithConn.Find.FirstByField<'TDoc>(tableName, fieldName, op, value, conn) + + /// Update an entire document + [] + static member inline UpdateFull<'TKey, 'TDoc>(conn, tableName, docId: 'TKey, document: 'TDoc) = + WithConn.Update.full tableName docId document conn + + /// Update an entire document + [] + static member inline UpdateFullFunc<'TKey, 'TDoc>(conn, tableName, idFunc: System.Func<'TDoc, 'TKey>, doc: 'TDoc) = + WithConn.Update.FullFunc(tableName, idFunc, doc, conn) + + /// Update a partial document + [] + static member inline UpdatePartialById<'TKey, 'TPatch>(conn, tableName, docId: 'TKey, partial: 'TPatch) = + WithConn.Update.partialById tableName docId partial conn + + /// Update partial documents using a comparison on a JSON field + [] + static member inline UpdatePartialByField<'TPatch>(conn, tableName, fieldName, op, value: obj, partial: 'TPatch) = + WithConn.Update.partialByField tableName fieldName op value partial conn + + /// Delete a document by its ID + [] + static member inline DeleteById<'TKey>(conn, tableName, docId: 'TKey) = + WithConn.Delete.byId tableName docId conn + + /// Delete documents by matching a comparison on a JSON field + [] + static member inline DeleteByField(conn, tableName, fieldName, op, value: obj) = + WithConn.Delete.byField tableName fieldName op value conn