web-dev-qa-db-ja.com

バックグラウンドスレッドをフラスコに追加するにはどうすればよいですか?

フラスコを試すために小さなゲームサーバーを書くのに忙しいです。ゲームはRESTを介してAPIをユーザーに公開します。ユーザーがアクションを実行してデータを照会するのは簡単ですが、app.run()ループの外側で「ゲームワールド」にサービスを提供して、ゲームエンティティなどを更新したいと思います。Flaskは非常にきれいです実装されています。これを行うためのFlask方法があるかどうかを確認したいと思います。

70
Marinus

追加のスレッドは、WSGIサーバーによって呼び出されるのと同じアプリから開始する必要があります。

以下の例では、5秒ごとに実行するバックグラウンドスレッドを作成し、Flaskルーティングされた関数でも使用できるデータ構造を操作します。

import threading
import atexit
from flask import Flask

POOL_TIME = 5 #Seconds

# variables that are accessible from anywhere
commonDataStruct = {}
# lock to control access to variable
dataLock = threading.Lock()
# thread handler
yourThread = threading.Thread()

def create_app():
    app = Flask(__name__)

    def interrupt():
        global yourThread
        yourThread.cancel()

    def doStuff():
        global commonDataStruct
        global yourThread
        with dataLock:
        # Do your stuff with commonDataStruct Here

        # Set the next thread to happen
        yourThread = threading.Timer(POOL_TIME, doStuff, ())
        yourThread.start()   

    def doStuffStart():
        # Do initialisation stuff here
        global yourThread
        # Create your thread
        yourThread = threading.Timer(POOL_TIME, doStuff, ())
        yourThread.start()

    # Initiate
    doStuffStart()
    # When you kill Flask (SIGTERM), clear the trigger for the next thread
    atexit.register(interrupt)
    return app

app = create_app()          

Gunicornから次のように呼び出します:

gunicorn -b 0.0.0.0:5000 --log-config log.conf --pid=app.pid myfile:app
67
caio

純粋なスレッドまたはCeleryキューを使用することに加えて(flask-celeryは不要になったことに注意してください)、flask-apschedulerを確認することもできます。

https://github.com/viniciuschiele/flask-apscheduler

https://github.com/viniciuschiele/flask-apscheduler/blob/master/examples/jobs.py からコピーした簡単な例:

from flask import Flask
from flask_apscheduler import APScheduler


class Config(object):
    JOBS = [
        {
            'id': 'job1',
            'func': 'jobs:job1',
            'args': (1, 2),
            'trigger': 'interval',
            'seconds': 10
        }
    ]

    SCHEDULER_API_ENABLED = True


def job1(a, b):
    print(str(a) + ' ' + str(b))

if __== '__main__':
    app = Flask(__name__)
    app.config.from_object(Config())

    scheduler = APScheduler()
    # it is also possible to enable the API directly
    # scheduler.api_enabled = True
    scheduler.init_app(app)
    scheduler.start()

    app.run()
3

ハック的な方法 があるように見えますが、技術的にサポートされているとは思いません。

また、 この回答 を見つけました。

3
girasquid

RQ をご覧ください。

また、MQuel GreenbergのRQとFlaskを使用した バックグラウンドジョブ に関するすばらしいチュートリアルもご覧ください。

0
Levon