首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >30 秒完美实现 Request 中"携带"用户信息,提升 10 倍性能方案

30 秒完美实现 Request 中"携带"用户信息,提升 10 倍性能方案

作者头像
python编程从入门到实践
修改2021-05-17 16:28:15
6060
修改2021-05-17 16:28:15
举报

/ python 30 秒完美实现 Request 中"携带"用户信息,提升 10 倍性能方案 /

之前在 大型 fastapi 项目实战 靠 python 中间件解决方案涨薪了 一文中 详细的介绍了通过中间件的方式去处理请求中的一些 case,没有看过的同学建议看看。

1

背景介绍

今天我们再看一下另外一个场景:当请求过来的时候,要对请求统一进行权限验证或者进行登陆保护。如果验证通过,那么就把验证过的用户信息附加到 Request 中,这样在具体的路径操作里,就可以直接使用用户信息,而不用再次查询当前用户。同时还有一个好处是减少了重复代码量,将以前的在路径操作部分的权限验证统一提取出来。今天我们通过把数据附加到 request.state 中,来解决这个问题。

2

案例实现

# -*- coding: utf-8 -*-
import time

@app.middleware("http")
async def process_authorization(request: Request, call_next):
    """
        在这个函数里统一对访问做权限 token 校验。
        1.如果是用户注册、登陆,那么不做 token 校验,由路径操作函数具体验证
        2.如果是其他操作,则需要从 header 或者 cookie 中取出 token 信息,解析出内容然后对用户身份进行验证,
            2.1.如果用户不存在则直接返回
            2.2.如果用户存在则将用户信息附加到 request 中,这样在后续的路径操作函数中可以直接使用。
    """

    start_time = time.time()

    if request.url.path == '/login' or request.url.path == '/register':
        # 是登陆或者注册的路由则直接通过
        logger.info('not need jwt verify, please do next task')
    else:
        logger.info('need jwt verify.')

        result = Auth.identifyAll(request)

        if result['status'] and result['data']:
            user = result['data']['user']
            logger.info('jwt verify success. user: %s ' % user.username)

            # state 中记录用户基本信息
            request.state.user = user
        else:
            return JSONResponse(content=result)

    # 进行接下来的操作处理
    response = await call_next(request)

    process_time = time.time() - start_time
    response.headers["X-Process-Time"] = str(process_time)
    return response

几点说明: 1.本案例中的方式其实质还是通过中间件的方式先拦截掉请求再附加信息进行处理,需要看清本质 2.我们会想一下是否在自己做过的系统中先是把请求放进来 然后在获取具体与人属性相关的信息的时候,是否都是先进行查询人属性的信息,然后基于此再做接下来的处理,那能否借鉴本案例中的方法进行处理呢?这个问题留给大家进行思考。至于标题中说到的提升 10 倍性能,这个对比基准不同测试得到的效果也有差异,请理性看待 3.若对中间件感兴趣的同学建议结合之前的一篇参照着来学习效果更佳,面试过程中发现很多同学一问都用过中间件,但自己没怎么自定义过中间件,这是不是值得思考呢?

我坚信:思考问题的方法远大于具体解决问题的方案,让我们继续一路前行,下期再见!

原创不易,只愿能帮助那些需要这些内容的同行或刚入行的小伙伴,你的每次 点赞、分享 都是我继续创作下去的动力,我希望能在推广 python 技术的道路上尽我一份力量,欢迎在评论区向我提问,我都会一一解答,记得一键三连支持一下哦!

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2021-05-09,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 python编程从入门到实践 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
即时通信 IM
即时通信 IM(Instant Messaging)基于腾讯二十余年的 IM 技术积累,支持Android、iOS、Mac、Windows、Web、H5、小程序平台且跨终端互通,低代码 UI 组件助您30分钟集成单聊、群聊、关系链、消息漫游、群组管理、资料管理、直播弹幕和内容审核等能力。适用于直播互动、电商带货、客服咨询、社交沟通、在线课程、企业办公、互动游戏、医疗健康等场景。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档