flask celery hdfs 异步上传

在Flask和Celery环境下实现异步上传文件到HDFS(Hadoop Distributed File System)的功能,可以大大提高Web应用的性能和用户体验。以下是一个详细的步骤和代码示例,帮助你实现这一功能。

步骤 1: 环境准备

  1. 安装必要的库

    复制代码
    pip install Flask Celery hdfs pip install redis # 如果使用Redis作为消息代
  2. 设置Redis‌(Celery需要消息代理,这里以Redis为例)

    复制代码
    redis-server

步骤 2: 创建Flask应用和Celery任务

  1. 创建Flask应用

    复制代码
    from flask import Flask, request, jsonify app = Flask(__name__
  2. 配置Celery

    复制代码
    from celery import Celery 
    
    def make_celery(app): 
        celery = Celery( 
            app.import_name, 
            broker=app.config['CELERY_BROKER_URL'], 
            backend=app.config['CELERY_RESULT_BACKEND'] 
        ) 
        celery.conf.update(app.config) 
        TaskBase = celery.Task 
        class ContextTask(TaskBase): 
            abstract = True 
            def __call__(self, *args, **kwargs): 
                with app.app_context(): 
                    return TaskBase.__call__(self, *args, **kwargs) 
        celery.Task = ContextTask 
        return celery 
    
    app.config['CELERY_BROKER_URL'] = 'redis://localhost:6379/0' 
    app.config['CELERY_RESULT_BACKEND'] = 'redis://localhost:6379/0' 
    celery = make_celery(app)
  3. 定义上传任务

    复制代码
    from hdfs import InsecureClient 
    
    hdfs = InsecureClient('http://localhost:50070', user='your_username') 
    
    @celery.task() 
    def upload_to_hdfs(file_path, hdfs_path): 
        with open(file_path, 'rb') as f: 
            hdfs.write(hdfs_path, f, overwrite=True) 
        return f'File uploaded to {hdfs_path}'

步骤 3: 创建上传接口

  1. 定义Flask路由

    复制代码
    @app.route('/upload', methods=['POST']) 
    def upload(): 
        file = request.files['file'] if file: 
            file_path = 'temp/' + file.filename # 存储在本地临时位置 
            file.save(file_path) 
            hdfs_path = '/user/your_username/' + file.filename # HDFS路径 
            upload_to_hdfs.delay(file_path, hdfs_path) # 异步上传到HDFS 
            return jsonify({'status': 'success', 'message': 'Upload in progress'}) 
        return jsonify({'status': 'error', 'message': 'No file provided'}), 400

步骤 4: 运行应用和Celery worker

  1. 运行Flask应用

    复制代码
    FLASK_APP=your_application.py flask run --port=5000

    其中your_application.py是你的Flask应用文件。

  2. 启动Celery worker

    复制代码
    celery -A your_application.celery worker --loglevel=inf

    同样,your_application.py是你的Flask应用文件。

步骤 5: 测试上传功能

你可以使用Postman或者curl来测试上传接口:

复制代码
curl -X POST -F "file=@path_to_your_file" http://localhost:5000/uploa

确保替换path_to_your_file为你的文件路径。

以上步骤和代码示例将帮助你在Flask和Celery环境下实现异步上传文件到HDFS的功能。

相关推荐
Scott9999HH2 小时前
【IIoT流量实战】蒸汽管道阀门全关却仍有流量?用 Python 实现涡街信号 FFT 频谱分析与温压全补偿积算网关,深度拆解靠谱的涡街流量计厂家硬核技术标准
开发语言·python
AI云海3 小时前
python 列表、元组、集合和字典
开发语言·python
二十雨辰3 小时前
[爬虫]-Urllib
爬虫·python
玉鸯5 小时前
Agent Hook:在概率推理之上,为 Agent 叠加确定性控制
python·langchain·agent
weixin_446260856 小时前
HACO:面向动态部署环境的对冲式智能计算可靠多智能体调度框架
后端·python·flask
我的xiaodoujiao6 小时前
API 接口自动化测试详细图文教程学习系列32--Allure测试报告2
python·学习·测试工具·pytest
qetfw6 小时前
MXU:Tauri 2 + React 的 MaaFramework 跨平台 GUI 源码
前端·python·react.js·前端框架·开源项目·效率工具
用户8356290780517 小时前
Python 实现 Excel 页面布局与打印设置自动化
后端·python
一次旅行7 小时前
Python+大模型端到端自动化日报系统
开发语言·python·自动化