Commit 69db1f45 authored by xianyang's avatar xianyang

优化拦截

parent 7ddbfc38
...@@ -28,13 +28,13 @@ app.add_middleware( ...@@ -28,13 +28,13 @@ app.add_middleware(
@app.middleware("http") @app.middleware("http")
async def add_process_time_header(request: Request, call_next): async def add_process_time_header(request: Request, call_next):
hs = request.headers hs = request.headers
token=hs.get("authorization").replace('Bearer','').replace(' ','') token=hs.get("authorization")
start_time = time.time() start_time = time.time()
response = await call_next(request) response = await call_next(request)
process_time = time.time() - start_time process_time = time.time() - start_time
response.headers["X-Process-Time"] = str(process_time) response.headers["X-Process-Time"] = str(process_time)
if token: if token:
payload = jwt.decode(token, env.SECRET_KEY, algorithms=[env.ALGORITHM]) payload = jwt.decode(token.replace('Bearer','').replace(' ',''), env.SECRET_KEY, algorithms=[env.ALGORITHM])
timestamp=payload.get("exp") timestamp=payload.get("exp")
access_token_expires = timedelta(hours=time_format(timestamp)) access_token_expires = timedelta(hours=time_format(timestamp))
create_access_token({'username':payload.get("xup"),'password':payload.get("password")},expires_delta=access_token_expires) #更新token时间 create_access_token({'username':payload.get("xup"),'password':payload.get("password")},expires_delta=access_token_expires) #更新token时间
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment