短链生成接口,

Wd提问: 短链生成接口

Ai批改: 短链生成接口通常用于将长URL转换为短小易记的URL，方便分享和记录。以下是一个使用Python Flask框架实现的简单短链生成接口的示例：

依赖

Flask：用于快速构建Web服务
sqlite3：用于存储短链与原链的映射关系

代码示例

python
from flask import Flask, request, jsonify
import sqlite3
import uuid

app = Flask(__name__)

创建一个SQLite数据库来存储短链和原链的映射
conn = sqlite3.connect('short_url.db')
cursor = conn.cursor()

创建表，如果不存在的话
cursor.execute('''
CREATE TABLE IF NOT EXISTS urls
(id TEXT PRIMARY KEY, original_url TEXT)
''')
conn.commit()
conn.close()

@app.route('/shorten', methods=['POST'])
def shorten_url():
data = request.json
if 'url' not in data:
return jsonify({'error': 'Missing URL parameter'}), 400

original_url = data['url']

生成一个唯一的短链码
short_code = str(uuid.uuid4())[:6] 取UUID的前6位作为短链码

连接数据库，插入新记录
conn = sqlite3.connect('short_url.db')
cursor = conn.cursor()
cursor.execute('INSERT INTO urls (id, original_url) VALUES (?, ?)', (short_code, original_url))
conn.commit()
conn.close()

return jsonify({'short_url': f'http://localhost:5000/{short_code}'}), 201

@app.route('/<short_code', methods=['GET'])
def redirect_to_original(short_code):
conn = sqlite3.connect('short_url.db')
cursor = conn.cursor()
cursor.execute('SELECT original_url FROM urls WHERE id = ?', (short_code,))
row = cursor.fetchone()
conn.close()

if row:
return jsonify({'url': row[0]}), 301 这里返回301状态码，浏览器会自动跳转
else:
return jsonify({'error': 'Short URL not found'}), 404

if __name__ == '__main__':
app.run(debug=True)

接口说明

1、短链生成接口：

URL：`http://localhost:5000/shorten`
方法：`POST`
请求参数：
`url`: 原始链接
返回：
短链地址

2、短链跳转接口：

URL：`http://localhost:5000/<short_code`
方法：`GET`
请求参数：
`short_code`: 短链码
返回：
原始链接

使用方法

1、运行上述Python脚本，启动Flask服务。
2、使用POST请求`http://localhost:5000/shorten`，并在请求体中传递原始URL：
json
{
url: https://www.example.com/very-long-url
}

3、服务端返回一个短URL：`http://localhost:5000/xxxxx`
4、访问短URL，浏览器会自动跳转到原始URL。

注意事项

这里使用SQLite作为简单的存储方案，如果用于生产环境，建议使用更可靠的数据库。
短链码生成策略可以调整为更短或者按照特定规则生成。
请确保服务器的安全性，短链服务可能会被用来进行钓鱼攻击或者垃圾信息传播。