The strawberry-ratelimit from stypr

strawberry-ratelimit's Introduction

Strawberry GraphQL Ratelimit Extension

Installation

pip3 install strawberry-ratelimit

Usage

from strawberry_ratelimit.ratelimit import ExtensionRatelimit

...


DEPTH_MAX = 50
CALL_MAX = 500

my_schema = strawberry.Schema(
    query=Query,
    mutation=Mutation,
    extensions=[
        # 60req/min
        ExtensionRatelimit(
            type_name=['getUser', 'updateUser', 'user_friends'], # queries, mutations, internal funcs.
            rate_max=60,
            rate_seconds=60,
            depth_max=DEPTH_MAX, # Maximum depth of the query
            call_max=CALL_MAX # Maximum call count
        ),
        # 1000req/hr
        ExtensionRatelimit(
            type_name=['getFriend', 'updateFriend', 'user_profile'], # queries, mutations, internal funcs.
            rate_max=1000,
            rate_seconds=3600,
            depth_max=DEPTH_MAX, # Maximum depth of the query
            call_max=CALL_MAX # Maximum call count
        ),        
    ]
)

Limitations

Consider this project as a reference code. Do not use it for production.
- Currently, the code uses a single global dict variable to store users' query/mutation accesses.
- Although there is a code to flush the list of user's IP after a certain amount of logs, the code itself does not collect garbage and it may consume a lot of memory depending on the number of users.
- It is recommended to re-write this code with DB support for optimization.

Recommend Projects