使用 Serverless 创建一个简单的短网址服务
本文首发于 ServerLife 公众号。
短网址服务，是一个能够将冗长的网址转换缩短为更简短、方便的短网址的工具。
经过短网址服务缩短的网址很短很容易使用和传播。
例如我们经常会在营销短信中会看到短网址。
本文将介绍使用 Serverless 创建一个简单的短网址服务。
使用的技术
使用的产品与服务：
-
Serverless Framework：一个免费开源的 Serverless 框架。
-
Tencent SCF：腾讯云云函数服务。
-
Lambda Store：全球第一个 Serverless Redis 服务。
语言及框架：
-
Python 3.6
-
Flask：一个微型的 Python 开发的 Web 框架。
项目初始化
通过 npm 全局安装 Serverless 命令行工具：
npm install -g serverless
使用模板初始化项目：
serverless init flask-starter --name url-shortener
业务设计与实现
这个简单的短网址服务，主要有以下几个接口，目前没有前端页面：
1、将长网址转换为短网址
2、访问短网址时将其重定向到原始的长网址
3、将短网址还原为原始的长网址
数据目前存储到 redis 中，这里用到了 Lambda Store 服务。
在生成短网址时，会生成一个6位的随机标识符（如果标识符已存在，会重新生成，最多尝试20次，如果还是失败，则返回错误信息），然后以 key 为短网址标识符，以 value 为原始的长网址，将其存储到 redis 中。
访问短网址时，首先以标识符为 key 从 redis 获取相应的 原始的长网址，如果获取成功，执行重定向操作，否则返回404。
还原短网址时，也是以标识符为 key 从 redis 获取相应的 原始的长网址，如果获取成功，则返回相应的原始长网址，否则返回错误信息。
已将代码上传到 GitHub： https://github.com/donhui/url-shortener， 核心代码如下：
import random
import string
from flask import Flask, jsonify, request, abort, redirect, render_template
import redis
from settings import redis_settings
app = Flask(__name__)
def generate_identifier(n=6):
identifier = ""
for i in range(n):
identifier += random.choice(string.ascii_letters)
return identifier
def get_redis_instance():
return redis.Redis(
host=redis_settings.get('host'),
port=redis_settings.get('port'),
password=redis_settings.get('password'))
redis_instance = get_redis_instance()
@app.route("/")
def index():
return render_template("index.html")
@app.route("/generate/<path:address>/")
def generate(address):
identifier = ''
i = 0
while True:
if i == 20:
break
identifier = generate_identifier()
if redis_instance.exists(identifier):
i = i + 1
continue
else:
break
if identifier == '':
return jsonify({"status": "fail", "error_msg": "generate shortened_url fail, please try again."})
if not (address.startswith("http://") or address.startswith("https://")):
address = "http://" + address
redis_instance.set(identifier, address)
shortened_url = request.host_url + identifier
return jsonify({"identifier": identifier, "shortened_url": shortened_url})
@app.route("/<string:identifier>/")
def fetch_original(identifier):
try:
origin_url = redis_instance.get(identifier)
except:
abort(404)
return redirect(origin_url)
@app.route("/restore/<string:identifier>/")
def restore(identifier):
try:
original_url = redis_instance.get(identifier)
except:
pass
if original_url is None:
return jsonify({"status": "fail", "error_msg": "get original_url fail, please check the identifier."})
return jsonify({"identifier": identifier, "original_url": str(original_url)})
if __name__ == "__main__":
app.run(debug=True)
开发完成后，代码目录结构截图如下：
部署到腾讯云 SCF
使用 serverless deploy 命令即可一键快速将服务部署到腾讯云 SCF。
访问短网址相关服务
短网址顾名思义网址比较短，一般都会有一个短的域名。
理论上 SCF 支持自定义域名，当然最好是个短域名。
下面的 Demo 出于演示的目的，权且使用腾讯云自带的 API 网关地址。
首先进入首页：
生成一个短网址：
生成这个短网址后，使用浏览器访问它，它会跳转到原始的网址。
还原短网址：
总结
本文使用 Serverless 创建了一个简单的短网址服务。用到了 Serverless Framework 及其组件 tencent-flask，还用到了 Lambda Store 服务，创建了一个完完全全的 Serverless 服务。
当然从功能的角度来讲，还有很多可以优化的地方，比如：友好的前端页面、更友好的提示、更好的标识符生成算法等等。
