Serverless 中怎么创建一个短网址服务

发布时间：2021-07-28 15:30:15 阅读：270 作者：Leah 栏目：云计算

亿速云云数据库，读写分离，安全稳定，弹性扩容，低至0.3元/天！！点击查看>>

本篇文章为大家展示了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)