Merge pull request #1601 from stakwork/feat/mutex_lock_withdrawal

Added mutex lock to budget withdrawal to prevent race conditions
stakwork · Mar 8, 2024 · aeb39f9 · aeb39f9
2 parents 940bea3 + 5c9a8d7
commit aeb39f9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/handlers/bounty.go b/handlers/bounty.go
@@ -543,6 +543,9 @@ func (h *bountyHandler) MakeBountyPayment(w http.ResponseWriter, r *http.Request
 }
 
 func (h *bountyHandler) BountyBudgetWithdraw(w http.ResponseWriter, r *http.Request) {
+	var m sync.Mutex
+	m.Lock()
+
 	ctx := r.Context()
 	pubKeyFromAuth, _ := ctx.Value(auth.ContextKey).(string)
 
@@ -599,6 +602,8 @@ func (h *bountyHandler) BountyBudgetWithdraw(w http.ResponseWriter, r *http.Requ
 		errMsg := formatPayError("Could not pay lightning invoice")
 		json.NewEncoder(w).Encode(errMsg)
 	}
+
+	m.Unlock()
 }
 
 func formatPayError(errorMsg string) db.InvoicePayError {