Skip to content

Conversation

ptiurin
Copy link
Contributor

@ptiurin ptiurin commented Mar 1, 2024

FIR-30447

More intelligent auth lock that doesn't lock on each request, but only when auth flow is executed.
Added tests that verify that we're truly async.

@ptiurin ptiurin marked this pull request as ready for review March 5, 2024 10:26
@ptiurin ptiurin requested a review from a team as a code owner March 5, 2024 10:26
@ptiurin ptiurin marked this pull request as draft March 5, 2024 10:27
@ptiurin
Copy link
Contributor Author

ptiurin commented Mar 5, 2024

I want to add tests before marking it ready for review again.

@ptiurin ptiurin marked this pull request as ready for review March 8, 2024 18:42
finally:
# token gets updated only after flow.send is called
# so unlock only after that
if self._lock.locked() and self._lock._owner_task == get_current_task():
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Are there situations where we end up here, but the task that has acquired the lock if different?
In case no, can we still use context manager?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yeah, I remember reproducing it when the lock was still held but another task has reached this point. I don't remember specifics anymore (it's been a while) but I think there is a point when the token has been updated but the lock is not yet released so another thread progresses up until this point by skipping lock on L152.

Using a context manager is not really possible as we want to lock it only when doing the first auth. However, async_auth_flow is called on every request so we want to avoid always locking like we did before, since it leads to synchronous execution.

Copy link

sonarqubecloud bot commented Apr 3, 2024

@ptiurin ptiurin merged commit 10435d4 into main Apr 3, 2024
@ptiurin ptiurin deleted the fix-async-execution branch April 3, 2024 13:26
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants