Robosats/robosats
1
0
Fork 0
mirror of https://github.com/RoboSats/robosats.git synced 2025-07-20 09:43:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Perf reduce DB writes on last_login updates

Browse Source
This commit is contained in:
Reckless_Satoshi
2023-09-11 08:52:50 -07:00
parent c7a14fbc56
commit b3598f5e9c
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
robosats/middleware.py
View File

@ -1,10 +1,12 @@
import hashlib import hashlib
from datetime import timedelta
from pathlib import Path from pathlib import Path
from channels.db import database_sync_to_async from channels.db import database_sync_to_async
from channels.middleware import BaseMiddleware from channels.middleware import BaseMiddleware
from django.conf import settings from django.conf import settings
from django.contrib.auth.models import AnonymousUser, User, update_last_login from django.contrib.auth.models import AnonymousUser, User, update_last_login
from django.utils import timezone
from django.utils.deprecation import MiddlewareMixin from django.utils.deprecation import MiddlewareMixin
from rest_framework.authtoken.models import Token from rest_framework.authtoken.models import Token
from rest_framework.exceptions import AuthenticationFailed from rest_framework.exceptions import AuthenticationFailed
@ -84,7 +86,9 @@ class RobotTokenSHA256AuthenticationMiddleWare:
# Check if it is an existing robot. # Check if it is an existing robot.
try: try:
token = Token.objects.get(key=token_sha256_b91) token = Token.objects.get(key=token_sha256_b91)
update_last_login(None, token.user) # Update last login every 2 minutes (avoid too many DB writes)
if token.user.last_login < timezone.now() - timedelta(minutes=2):
update_last_login(None, token.user)
except Token.DoesNotExist: except Token.DoesNotExist:
# If we get here the user does not have a robot on this coordinator # If we get here the user does not have a robot on this coordinator