Bug: JWTAuth How to get db_session

[renjianguo666]

Description

https://docs.litestar.dev/2/usage/security/jwt.html

retrieve_user_handler How to get db_session ???

URL to code causing the issue

No response

MCVE

Steps to reproduce

1. Go to '...'
2. Click on '....'
3. Scroll down to '....'
4. See error

Screenshots

No response

Logs

Litestar Version

2.18.0

Platform

• Linux
• Mac
• Windows
• Other (Please specify in the description above)

[euri10]

maybe someone more familiar with the sqlalchemy plugin can confirm but generally speaking the session should be accessible in the ASGIConnection object in the app state, now i dont know under which key so something like connection.app.state["idk_what_key"] should do the job

[renjianguo666]

My initial thought was also to get it from connection.app.state. I saw 'db_session' in the dependency injection, so I naturally tried to use connection.app.state['db_session']. However, it threw an error.

https://docs.advanced-alchemy.litestar.dev/latest/_modules/advanced_alchemy/extensions/litestar/plugins/init/config/asyncio.html#SQLAlchemyAsyncConfig

[renjianguo666]

I found it.

[euri10]

next time can you paste code instead of screenshot please @renjianguo666 it's really hard to read ;)

[renjianguo666]

sorry, I have learned paste code. using ```python

[provinzkraut]

You can actually do this without having to go through the state:

async def retrieve_user_handler(
    handler: BaseRouteHandler, 
    connection: ASGIConnection
) -> User | None:
    sqlalchemy_plugin = connection.app.plugins.get(SQLAlchemyPlugin)
    config = sqlalchemy_plugin.config
    async with config.create_session_maker() as session:
        pass

[renjianguo666]

very good,
but sqlalchemy_plugin.config is a list type.
Need to go through sqlalchemy_plugin.config[0] obtain.

async def retrieve_user_handler(
    token: Token,
    connection: ASGIConnection,
) -> User | None:
    sqlalchemy_plugin = connection.app.plugins.get("SQLAlchemyPlugin")
    session_maker = sqlalchemy_plugin.config[0].create_session_maker()
    async with session_maker() as session:
        print(session)
        return await UserService(session).get_one_or_none(id=token.sub)