《短网址系统设计与实现》 --- 详细介绍和代码实现（二）

悠悠 2021-10-15 04:41 242阅读 0赞

### 1.短网址服务的编码实现 ###

首先需要你**安装MySQL数据库**，只安装服务器端就行，安装的时候可以选择。

**  安装Redis 非关系型数据库**

**      接下来是一些需要的Python插件安装：**

# 安装flask
    
    (python3) C:\Users\allar\system_design>pip  install  -i  https://pypi.doubanio.com/simple/  --trusted-host pypi.doubanio.com  flask
    
    # 安装flask_mysqldb
    (python3) C:\Users\allar\system_design>pip install flask_mysqldb
    
    # 安装flask_redis
    (python3) C:\Users\allar\system_design>pip install flask-redis
    
    # 安装requests
    
    (python3) C:\Users\allar\system_design>pip install requests

**（1） 在数据库里面创建一个 short\_url 表**

CREATE TABLE short_url (
        id bigint unsigned NOT NULL AUTO_INCREMENT,
        token varchar(10),
        url varchar(2048),
        created_at timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
        PRIMARY KEY (`id`),
        KEY `idx_token` (`token`)
    );

**（2）使用 Flask 框架演示本系统实现**

代码里实现了短网址生成算法

数据库使用 Mysql

计数器使用 Redis

**完整代码地址：**

https://github.com/AllardZhao/TinyUrl-Service/archive/master.zip

**代码实现：**

import os
    
    from flask import Flask, jsonify, render_template, request
    from flask_mysqldb import MySQL
    # from flask.ext.redis import FlaskRedis
    from flask_redis import FlaskRedis
    
    
    app = Flask(__name__)
    app.config['MYSQL_USER'] = 'root'
    app.config['MYSQL_PASSWORD'] = '123456'
    app.config['MYSQL_DB'] = 'test'
    app.config['MYSQL_CUESORCLASS'] = 'DictCursor'
    
    mysql = MySQL(app)
    redis_store = FlaskRedis(app)
    
    CHARS="abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
    def encode(num):
        '''10进制转化成62进制串'''
        if num == 0:
            return CHARS[0]
        res = []
        while num:
            num, rem = divmod(num, len(CHARS))
            res.append(CHARS[rem])
        return ''.join(reversed(res))
        
    @app.route('/shorten', methods=['POST'])
    def shorten_url():
        # 获取页面长的url
        long_url = request.json['url']
        # 从Redis获取自增的index
        index = int(redis_store.incr('SHORT_CNT'))
        # 生成一个token，把一个index10进制数转化成62进制串
        token = encode(index)
        sql = "INSERT INTO short_url(token, url) VALUES(%s, %s)"
        # 获取mysql连接
        cur = mysql.connection.cursor()
        cur.execute(sql, (token, long_url))
        mysql.connection.commit()
        # 构造短网址
        short_url = 'https://short.com/' + token
        return jsonify(url=short_url) # 返回短网址的json数据
        
    @app.route('/')
    def index():
        return render_template('index.html')
        
    if __name__ == '__main__':
        app.run(debug=1)