aiodataloader_basic_batch_key_fetching_with_asyncio.py

1import asyncio
2from aiodataloader import DataLoader
3
4class UserLoader(DataLoader):
5    async def batch_load_fn(self, keys):
6        # Here you would fetch from a database
7        # For this example, we return a list of strings
8        return [f"User {key}" for key in keys]
9
10async def run():
11    loader = UserLoader()
12    
13    # Load keys (these will be batched together)
14    promise1 = loader.load(1)
15    promise2 = loader.load(2)
16    
17    # Resolve the promises
18    user1, user2 = await asyncio.gather(promise1, promise2)
19    
20    print(user1)
21    print(user2)
22
23if __name__ == '__main__':
24    asyncio.run(run())