使用 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)

开发完成后，代码目录结构截图如下：