97 lines
3.7 KiB
Forth
97 lines
3.7 KiB
Forth
namespace MyWebLog.Data.PostgreSql
|
|
|
|
open MyWebLog
|
|
open MyWebLog.Data
|
|
open Npgsql
|
|
open Npgsql.FSharp
|
|
|
|
/// PostgreSQL myWebLog uploaded file data implementation
|
|
type PostgreSqlUploadData (conn : NpgsqlConnection) =
|
|
|
|
/// The INSERT statement for an uploaded file
|
|
let upInsert =
|
|
"INSERT INTO upload (
|
|
id, web_log_id, path, updated_on, data
|
|
) VALUES (
|
|
@id, @webLogId, @path, @updatedOn, @data
|
|
)"
|
|
|
|
/// Parameters for adding an uploaded file
|
|
let upParams (upload : Upload) = [
|
|
webLogIdParam upload.WebLogId
|
|
"@id", Sql.string (UploadId.toString upload.Id)
|
|
"@path", Sql.string (Permalink.toString upload.Path)
|
|
"@updatedOn", Sql.timestamptz upload.UpdatedOn
|
|
"@data", Sql.bytea upload.Data
|
|
]
|
|
|
|
/// Save an uploaded file
|
|
let add upload = backgroundTask {
|
|
let! _ =
|
|
Sql.existingConnection conn
|
|
|> Sql.query upInsert
|
|
|> Sql.parameters (upParams upload)
|
|
|> Sql.executeNonQueryAsync
|
|
()
|
|
}
|
|
|
|
/// Delete an uploaded file by its ID
|
|
let delete uploadId webLogId = backgroundTask {
|
|
let theParams = [ "@id", Sql.string (UploadId.toString uploadId); webLogIdParam webLogId ]
|
|
let! path =
|
|
Sql.existingConnection conn
|
|
|> Sql.query "SELECT path FROM upload WHERE id = @id AND web_log_id = @webLogId"
|
|
|> Sql.parameters theParams
|
|
|> Sql.executeAsync (fun row -> row.string "path")
|
|
|> tryHead
|
|
if Option.isSome path then
|
|
let! _ =
|
|
Sql.existingConnection conn
|
|
|> Sql.query "DELETE FROM upload WHERE id = @id AND web_log_id = @webLogId"
|
|
|> Sql.parameters theParams
|
|
|> Sql.executeNonQueryAsync
|
|
return Ok path.Value
|
|
else return Error $"""Upload ID {UploadId.toString uploadId} not found"""
|
|
}
|
|
|
|
/// Find an uploaded file by its path for the given web log
|
|
let findByPath path webLogId =
|
|
Sql.existingConnection conn
|
|
|> Sql.query "SELECT * FROM upload WHERE web_log_id = @webLogId AND path = @path"
|
|
|> Sql.parameters [ webLogIdParam webLogId; "@path", Sql.string path ]
|
|
|> Sql.executeAsync (Map.toUpload true)
|
|
|> tryHead
|
|
|
|
/// Find all uploaded files for the given web log (excludes data)
|
|
let findByWebLog webLogId =
|
|
Sql.existingConnection conn
|
|
|> Sql.query "SELECT id, web_log_id, path, updated_on FROM upload WHERE web_log_id = @webLogId"
|
|
|> Sql.parameters [ webLogIdParam webLogId ]
|
|
|> Sql.executeAsync (Map.toUpload false)
|
|
|
|
/// Find all uploaded files for the given web log
|
|
let findByWebLogWithData webLogId =
|
|
Sql.existingConnection conn
|
|
|> Sql.query "SELECT * FROM upload WHERE web_log_id = @webLogId"
|
|
|> Sql.parameters [ webLogIdParam webLogId ]
|
|
|> Sql.executeAsync (Map.toUpload true)
|
|
|
|
/// Restore uploads from a backup
|
|
let restore uploads = backgroundTask {
|
|
for batch in uploads |> List.chunkBySize 5 do
|
|
let! _ =
|
|
Sql.existingConnection conn
|
|
|> Sql.executeTransactionAsync [
|
|
upInsert, batch |> List.map upParams
|
|
]
|
|
()
|
|
}
|
|
|
|
interface IUploadData with
|
|
member _.Add upload = add upload
|
|
member _.Delete uploadId webLogId = delete uploadId webLogId
|
|
member _.FindByPath path webLogId = findByPath path webLogId
|
|
member _.FindByWebLog webLogId = findByWebLog webLogId
|
|
member _.FindByWebLogWithData webLogId = findByWebLogWithData webLogId
|
|
member _.Restore uploads = restore uploads
|
|
|