Add lock in GetPaymentLocal

keybase · Sep 9, 2019 · b5f8897 · b5f8897
1 parent 2efe0a2
commit b5f8897
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/go/stellar/loader.go b/go/stellar/loader.go
@@ -90,6 +90,12 @@ func DefaultLoader(g *libkb.GlobalContext) *Loader {
 }
 
 func (p *Loader) GetPaymentLocal(ctx context.Context, paymentID stellar1.PaymentID) (*stellar1.PaymentLocal, bool) {
+	p.Lock()
+	defer p.Unlock()
+	return p.getPaymentLocalLocked(ctx, paymentID)
+}
+
+func (p *Loader) getPaymentLocalLocked(ctx context.Context, paymentID stellar1.PaymentID) (*stellar1.PaymentLocal, bool) {
 	pmt, ok := p.payments[paymentID]
 	return pmt, ok
 }
@@ -125,7 +131,7 @@ func (p *Loader) LoadPayment(ctx context.Context, convID chat1.ConversationID, m
 		m.Warning("existing payment message info does not match load info: (%v, %v) != (%v, %v)", msg.convID, msg.msgID, convID, msgID)
 	}
 
-	payment, ok := p.GetPaymentLocal(ctx, paymentID)
+	payment, ok := p.getPaymentLocalLocked(ctx, paymentID)
 	if ok {
 		info := p.uiPaymentInfo(m, payment, msg)
 		p.G().NotifyRouter.HandleChatPaymentInfo(m.Ctx(), p.G().ActiveDevice.UID(), convID, msgID, *info)