内容介绍
本专题将带你使用 axum 构建一个短链接服务。短链接,又称短网址,是指将一个可能比较长的链接变成固定长度的短链接,便于在社交媒体、短信等渠道发布。另外,它也可以隐藏原链接里的一些敏感信息,比如带邀请码的推广链接。短链接的核心算法
本章将对短链接的原理及实现算法进行讲解。准备工作
搞定了短链接的核心算法,我们可以开始进行业务开发。本章将进行一系列的准备工作,包括:配置、日志、自定义错误、handler、数据库操作原型、模板等。创建短链接
本章将实现创建短链接功能。短链接跳转到原始链接
我们继续实现功能,本章要实现的是通过短链接跳转到原始链接的功能。注意,跳转之前,我们需要将它的访问量加一。排行榜
本章继续完善短链接服务:显示排行。总结与代码清理
目前为止,我们的短链接服务基本已经完成了。但有一些地方不完善,同时有些警告没有处理掉。
短链接跳转到原始链接
handler
/// 跳转到目标URL
pub async fn goto_url(
Extension(state): Extension<AppState>,
Path(id): Path<String>,
) -> HandlerRedirectResult {
let handler_name = "goto_url";
let client = get_client(&state, handler_name).await?;
let result = db::goto_url(&client, id)
.await
.map_err(log_error(handler_name.to_string()))?;
Ok(redirect(result.url.as_str()))
}
数据库
pub async fn goto_url(client: &Client, id: String) -> Result<UrlTarget> {
let result = super::query_one(
client,
"UPDATE url SET visit=visit+1 WHERE id=$1 RETURNING url",
&[&id],
)
.await?;
Ok(result)
}
为什么不用事务
利益于 PostgreSQL 数据库的强大,我们可以使用单条 SQL 语句就实现既增加访问量,又返回原始链接。如果是其它数据库,则需要使用事务来分步操作:
// 1. 执行 UPDATE url SET visit=visit+1 WHERE id=?
// 2. 判断该 UPDATE 的结果,如果出错回滚事务
// 3. 执行 SELECT url FROM url WHERE id=?
// 4. 提交事务
// 5. 返回 SELECT 的结果