fix: cancel timer when server shuts down
This commit is contained in:
17
server.py
17
server.py
@@ -1,16 +1,27 @@
|
|||||||
import threading
|
import threading
|
||||||
|
|
||||||
# from generate import generate
|
# from generate import generate
|
||||||
|
from contextlib import asynccontextmanager
|
||||||
from fastapi import FastAPI
|
from fastapi import FastAPI
|
||||||
from fastapi.responses import FileResponse
|
from fastapi.responses import FileResponse
|
||||||
from fastapi.staticfiles import StaticFiles
|
from fastapi.staticfiles import StaticFiles
|
||||||
|
|
||||||
app = FastAPI()
|
# the index of the current audio track from 0 to 9
|
||||||
current_index = -1
|
current_index = -1
|
||||||
|
# the timer that periodically advances the current audio track
|
||||||
|
t = None
|
||||||
|
|
||||||
|
|
||||||
|
@asynccontextmanager
|
||||||
|
async def lifespan(app: FastAPI):
|
||||||
|
advance()
|
||||||
|
yield
|
||||||
|
if t:
|
||||||
|
t.cancel()
|
||||||
|
|
||||||
|
|
||||||
def advance():
|
def advance():
|
||||||
global current_index
|
global current_index, t
|
||||||
|
|
||||||
# if current_index == 0:
|
# if current_index == 0:
|
||||||
# generate(offset=5)
|
# generate(offset=5)
|
||||||
@@ -28,7 +39,7 @@ def advance():
|
|||||||
t.start()
|
t.start()
|
||||||
|
|
||||||
|
|
||||||
advance()
|
app = FastAPI(lifespan=lifespan)
|
||||||
|
|
||||||
|
|
||||||
@app.get("/current.mp3")
|
@app.get("/current.mp3")
|
||||||
|
Reference in New Issue
Block a user