fix ip header

conf/tests.py

@@ -1,6 +1,7 @@
 import hashlib
 from unittest import mock
 
+from django.conf import settings
 from django.utils import timezone
 
 from options.options import SysOptions
@@ -87,7 +88,7 @@ class JudgeServerHeartbeatTest(APITestCase):
         self.token = "test"
         self.hashed_token = hashlib.sha256(self.token.encode("utf-8")).hexdigest()
         SysOptions.judge_server_token = self.token
-        self.headers = {"HTTP_X_JUDGE_SERVER_TOKEN": self.hashed_token, "HTTP_X_REAL_IP": "1.2.3.4"}
+        self.headers = {"HTTP_X_JUDGE_SERVER_TOKEN": self.hashed_token, settings.IP_HEADER: "1.2.3.4"}
 
     def test_new_heartbeat(self):
         resp = self.client.post(self.url, data=self.data, **self.headers)

conf/views.py

@@ -136,7 +136,7 @@ class JudgeServerHeartbeatAPI(CSRFExemptAPIView):
             server.memory_usage = data["memory"]
             server.cpu_usage = data["cpu"]
             server.service_url = data["service_url"]
-            server.ip = request.META["HTTP_X_REAL_IP"]
+            server.ip = request.ip
             server.last_heartbeat = timezone.now()
             server.save()
         except JudgeServer.DoesNotExist: