-
Notifications
You must be signed in to change notification settings - Fork 4
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
5 changed files
with
188 additions
and
24 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,40 +1,129 @@ | ||
| module VahterBanBot.DB | ||
|
|
||
| open System.Threading.Tasks | ||
| open Npgsql | ||
| open VahterBanBot.Types | ||
| open Dapper | ||
| open VahterBanBot.Utils | ||
|
|
||
| let private connString = getEnv "DATABASE_URL" | ||
|
|
||
| let upsertUser (user: User) = | ||
| let upsertUser (user: DbUser): Task<DbUser> = | ||
| task { | ||
| use conn = new NpgsqlConnection(connString) | ||
|
|
||
| //language=postgresql | ||
| let sql = | ||
| """ | ||
| INSERT INTO "user" (id, username) | ||
| VALUES (@id, @username) | ||
| INSERT INTO "user" (id, username, ban_reason, banned_at, banned_by, created_at, updated_at) | ||
| VALUES (@id, @username, @banReason, @bannedAt, @bannedBy, @createdAt, @updatedAt) | ||
| ON CONFLICT (id) DO UPDATE | ||
| SET username = | ||
| CASE | ||
| WHEN EXCLUDED.username != "user".username THEN EXCLUDED.username | ||
| WHEN EXCLUDED.username != "user".username THEN COALESCE(EXCLUDED.username, "user".username) | ||
| ELSE "user".username | ||
| END, | ||
| updated_at = | ||
| CASE | ||
| WHEN EXCLUDED.username != "user".username THEN timezone('utc'::TEXT, NOW()) | ||
| WHEN EXCLUDED.updated_at != "user".updated_at THEN EXCLUDED.username | ||
| ELSE "user".updated_at | ||
| END, | ||
| ban_reason = | ||
| CASE | ||
| WHEN EXCLUDED.ban_reason != "user".ban_reason THEN EXCLUDED.ban_reason | ||
| ELSE "user".ban_reason | ||
| END, | ||
| banned_at = | ||
| CASE | ||
| WHEN EXCLUDED.banned_at != "user".banned_at THEN EXCLUDED.ban_reason | ||
| ELSE "user".banned_at | ||
| END, | ||
| banned_by = | ||
| CASE | ||
| WHEN EXCLUDED.banned_by != "user".banned_by THEN EXCLUDED.ban_reason | ||
| ELSE "user".banned_by | ||
| END, | ||
| created_at = | ||
| CASE | ||
| WHEN EXCLUDED.created_at != "user".created_at THEN EXCLUDED.ban_reason | ||
| ELSE "user".created_at | ||
| END, | ||
| updated_at = | ||
| CASE | ||
| WHEN EXCLUDED.updated_at != "user".updated_at THEN EXCLUDED.ban_reason | ||
| ELSE "user".updated_at | ||
| END | ||
| RETURNING *; | ||
| """ | ||
|
|
||
| let! insertedUser = | ||
| conn.QueryAsync<User>( | ||
| conn.QueryAsync<DbUser>( | ||
| sql, | ||
| {| id = user.Id; username = user.Username |} | ||
| {| id = user.Id | ||
| username = user.Username | ||
| banReason = user.Ban_Reason | ||
| bannedAt = user.Banned_At | ||
| bannedBy = user.Banned_By | ||
| createdAt = user.Created_At | ||
| updatedAt = user.Updated_At |} | ||
| ) | ||
|
|
||
| return insertedUser |> Seq.head | ||
| } | ||
|
|
||
| let insertMessage (message: DbMessage): Task<DbMessage> = | ||
| task { | ||
| use conn = new NpgsqlConnection(connString) | ||
|
|
||
| //language=postgresql | ||
| let sql = | ||
| """ | ||
| INSERT INTO message (chat_id, message_id, user_id, created_at) | ||
| VALUES (@chatId, @messageId, @userId, @createdAt) | ||
| ON CONFLICT (chat_id, message_id) DO NOTHING RETURNING *; | ||
| """ | ||
|
|
||
| let! insertedMessage = | ||
| conn.QueryAsync<DbMessage>( | ||
| sql, | ||
| {| chatId = message.Chat_Id | ||
| messageId = message.Message_Id | ||
| userId = message.User_Id | ||
| createdAt = message.Created_At |} | ||
| ) | ||
|
|
||
| return | ||
| insertedMessage | ||
| |> Seq.tryHead | ||
| |> Option.defaultValue message | ||
| } | ||
|
|
||
| let getUserMessages (userId: int64): Task<DbMessage array> = | ||
| task { | ||
| use conn = new NpgsqlConnection(connString) | ||
|
|
||
| //language=postgresql | ||
| let sql = "SELECT * FROM message WHERE user_id = @userId" | ||
|
|
||
| let! messages = | ||
| conn.QueryAsync<DbMessage>( | ||
| sql, | ||
| {| userId = userId |} | ||
| ) | ||
| return Array.ofSeq messages | ||
| } | ||
|
|
||
| let deleteUserMessages (userId: int64): Task<int> = | ||
| task { | ||
| use conn = new NpgsqlConnection(connString) | ||
|
|
||
| //language=postgresql | ||
| let sql = "DELETE FROM message WHERE user_id = @userId" | ||
|
|
||
| let! messagesDeleted = | ||
| conn.ExecuteAsync( | ||
| sql, | ||
| {| userId = userId |} | ||
| ) | ||
| return messagesDeleted | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,20 @@ | ||
| DROP TABLE message; | ||
|
|
||
| CREATE TABLE "message" | ||
| ( | ||
| id BIGSERIAL NOT NULL PRIMARY KEY, | ||
| chat_id BIGINT NOT NULL, | ||
| message_id INT NOT NULL, | ||
| user_id BIGINT NOT NULL | ||
| CONSTRAINT message_user_id_fkey | ||
| REFERENCES "user" (id) | ||
| ON DELETE CASCADE, | ||
|
|
||
| created_at TIMESTAMPTZ NOT NULL DEFAULT timezone('utc'::TEXT, NOW()) | ||
| ); | ||
|
|
||
| CREATE UNIQUE INDEX message_chat_id_message_id_uindex | ||
| ON "message" (chat_id, message_id); | ||
|
|
||
| CREATE INDEX message_user_id_index | ||
| ON "message" (user_id); |