add 60 second buffer for rds token refresh

This commit is contained in:
Brendan Shanahan 2026-04-10 12:56:42 -04:00
parent 65d718344f
commit 5b93655f42

View file

@ -145,7 +145,7 @@ class RDSIAMConfig:
def check_token(self, dialect, connection_record, connection_args, connection_kwargs) -> None:
now = datetime.now(tz=timezone.utc)
if not self.token or now >= self.token.expiration:
if not self.token or now >= self.token.expiration - timedelta(seconds=60):
self.token = self.get_token()
connection_kwargs["password"] = self.token.token.get_secret_value()