99 lines
2.5 KiB
Rust
99 lines
2.5 KiB
Rust
use sqlx::Row;
|
|
|
|
pub async fn insert(
|
|
pool: &sqlx::PgPool,
|
|
file_key: &str,
|
|
bucket: &str,
|
|
region: &str,
|
|
song_queue_id: &uuid::Uuid,
|
|
) -> Result<uuid::Uuid, sqlx::Error> {
|
|
match sqlx::query(
|
|
r#"
|
|
INSERT INTO "songQueueData" (file_key, bucket, region, song_queue_id) VALUES($1, $2, $3, $4) RETURNING id;
|
|
"#,
|
|
)
|
|
.bind(file_key)
|
|
.bind(bucket)
|
|
.bind(region)
|
|
.bind(song_queue_id)
|
|
.fetch_one(pool).await {
|
|
Ok(row) => {
|
|
let id: uuid::Uuid = row.try_get("id")?;
|
|
Ok(id)
|
|
}
|
|
Err(_err) => Err(sqlx::Error::RowNotFound),
|
|
}
|
|
}
|
|
|
|
pub async fn update_file_key(
|
|
pool: &sqlx::PgPool,
|
|
id: &uuid::Uuid,
|
|
file_key: &str,
|
|
) -> Result<(), sqlx::Error> {
|
|
match sqlx::query(
|
|
r#"
|
|
UPDATE "songQueueData" SET file_key = $1 WHERE id = $2;
|
|
"#,
|
|
)
|
|
.bind(file_key)
|
|
.bind(id)
|
|
.execute(pool)
|
|
.await
|
|
{
|
|
Ok(_row) => Ok(()),
|
|
Err(_) => Err(sqlx::Error::RowNotFound),
|
|
}
|
|
}
|
|
|
|
pub async fn get(
|
|
pool: &sqlx::PgPool,
|
|
id: &uuid::Uuid,
|
|
) -> Result<(uuid::Uuid, String, String, String, uuid::Uuid), sqlx::Error> {
|
|
match sqlx::query(
|
|
r#"
|
|
SELECT id, file_key, bucket, region, song_queue_id FROM "songQueueData"
|
|
WHERE id = $1
|
|
"#,
|
|
)
|
|
.bind(id)
|
|
.fetch_one(pool)
|
|
.await
|
|
{
|
|
Ok(row) => {
|
|
let file_key: String = row.try_get("file_key")?;
|
|
let bucket: String = row.try_get("bucket")?;
|
|
let region: String = row.try_get("region")?;
|
|
let song_queue_id: uuid::Uuid = row.try_get("song_queue_id")?;
|
|
|
|
Ok((*id, file_key, bucket, region, song_queue_id))
|
|
}
|
|
Err(err) => Err(err),
|
|
}
|
|
}
|
|
|
|
pub async fn get_with_song_queue_id(
|
|
pool: &sqlx::PgPool,
|
|
song_queue_id: &uuid::Uuid,
|
|
) -> Result<(uuid::Uuid, String, String, String, uuid::Uuid), sqlx::Error> {
|
|
match sqlx::query(
|
|
r#"
|
|
SELECT id, file_key, bucket, region, song_queue_id FROM "songQueueData"
|
|
WHERE song_queue_id = $1
|
|
"#,
|
|
)
|
|
.bind(song_queue_id)
|
|
.fetch_one(pool)
|
|
.await
|
|
{
|
|
Ok(row) => {
|
|
let id: uuid::Uuid = row.try_get("id")?;
|
|
let file_key: String = row.try_get("file_key")?;
|
|
let bucket: String = row.try_get("bucket")?;
|
|
let region: String = row.try_get("region")?;
|
|
|
|
Ok((id, file_key, bucket, region, *song_queue_id))
|
|
}
|
|
Err(err) => Err(err),
|
|
}
|
|
}
|