linkletter 0.1.0

URL replacement and analytics service
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

use url::Url;
use uuid::Uuid;

/// Persist URL with its given document appearance.
/// Returns assigned ID.
pub(crate) async fn persist<'e, E>(
    url: &Url,
    doc_id: &Uuid,
    index_in_doc: u16,
    executor: E,
) -> sqlx::Result<Uuid>
where
    E: sqlx::Executor<'e, Database = sqlx::Sqlite>,
{
    let url_id = Uuid::new_v4();

    sqlx::query!(
        "INSERT INTO url(url_id, doc_id, index_in_doc, url) VALUES($1, $2, $3, $4)",
        url_id.as_bytes().as_slice(),
        doc_id.as_bytes().as_slice(),
        index_in_doc,
        url.as_str()
    )
    .execute(executor)
    .await?;

    Ok(url_id)
}

pub(crate) async fn fetch<'e, E>(url_id: &Uuid, executor: E) -> sqlx::Result<Url>
where
    E: sqlx::Executor<'e, Database = sqlx::Sqlite>,
{
    let url = sqlx::query_scalar!(
        "SELECT url FROM url WHERE url_id = $1",
        url_id.as_bytes().as_slice()
    )
    .fetch_one(executor)
    .await?;

    Url::parse(&url).map_err(|err| sqlx::Error::ColumnDecode {
        index: "url".into(),
        source: Box::new(err),
    })
}

#[derive(sqlx::FromRow)]
pub(crate) struct UrlInfo {
    #[sqlx(try_from = "Vec<u8>")]
    pub(crate) url_id: Uuid,
    #[sqlx(try_from = "&'a str")]
    pub(crate) url: Url,
}

pub(crate) async fn fetch_all_in_doc<'e, E>(doc_id: &Uuid, executor: E) -> sqlx::Result<Vec<UrlInfo>>
where
    E: sqlx::Executor<'e, Database = sqlx::Sqlite>,
{
    sqlx::query_as(
        "SELECT url_id, url FROM url WHERE doc_id = $1",
    )
    .bind(doc_id.as_bytes().as_slice())
    .fetch_all(executor)
    .await
}