Trying to fix globocom#151

brunojuliao · Oct 27, 2021 · 2d011c5 · 2d011c5
1 parent 9ab53f8
commit 2d011c5
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/networkapi/healthcheckexpect/models.py b/networkapi/healthcheckexpect/models.py
@@ -14,6 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import logging
+import threading
 
 from django.core.exceptions import ObjectDoesNotExist
 from django.db import models
@@ -181,6 +182,7 @@ def insert_expect_string(self, authenticated_user, expect_string, ambiente=None)
             raise HealthcheckExpectError(
                 e, u'Falha ao inserir healthcheck_expect.')
 
+lock = threading.Lock()
 
 class Healthcheck(BaseModel):
     id = models.AutoField(primary_key=True, db_column='id_healthcheck')
@@ -197,10 +199,11 @@ class Meta(BaseModel.Meta):
         managed = True
 
     def get_create_healthcheck(self, healthcheck):
-        try:
-            hc = Healthcheck.objects.get(**healthcheck)
-        except ObjectDoesNotExist:
-            hc = Healthcheck(**healthcheck)
-            hc.save()
+        with lock:
+            try:
+                hc = Healthcheck.objects.get(**healthcheck)
+            except ObjectDoesNotExist:
+                hc = Healthcheck(**healthcheck)
+                hc.save()
 
         return hc