创建短链接

本章将实现创建短链接功能。

模板文件

可以看到，我们只是简单的做了一个表单。

数据库操作

pub async fn create(client: &Client, cu: form::CreateUrl, id: String) -> Result<UrlID> {
    // 是否存在
    let result = super::query_one(client, "SELECT id FROM url WHERE id=$1", &[&id]).await;
    match result {
        // 如果已存在，直接返回
        Ok(result) => return Ok(result),
        // 如果不是“未找到”的错误，直接返回
        Err(err) if !err.is_not_found() => return Err(err),
        // 如果不存在，什么也不做，继续下面的代码
        _ => {}
    };
    let result = super::query_one(
        client,
        "INSERT INTO url(id, url, email) VALUES ($1,$2,$3) RETURNING id",
        &[&id, &cu.url, &cu.email],
    )
    .await?;
    Ok(result)
}

首先判断提交的原始链接是否存在，如果存在，直接返回对应的短链接。否则，将提交的数据写入数据库。注意，虽然这个函数有两个数据库操作，但并不需要使用事务来保证原子性，因为这两个操作是无关联的。