Archives/loop
2
0
Fork 0
mirror of https://github.com/lightninglabs/loop synced 2024-11-09 19:10:47 +00:00
Code Issues Releases Wiki Activity

Merge pull request #819 from bhandras/store-mock-mx

Browse Source 
loopdb: properly lock store mock for concurrent access
This commit is contained in:
András Bánki-Horváth 2024-09-02 14:23:41 +02:00 committed by GitHub
commit 634fbbcba2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 33 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
loopdb/store_mock.go
View File

@ -4,6 +4,7 @@ import (
"context"
"errors"
"fmt"
"sync"
"testing"
"time"
@ -14,6 +15,8 @@ import (
// StoreMock implements a mock client swap store.
type StoreMock struct {
sync.RWMutex
LoopOutSwaps map[lntypes.Hash]*LoopOutContract
LoopOutUpdates map[lntypes.Hash][]SwapStateData
loopOutStoreChan chan LoopOutContract
@ -50,6 +53,9 @@ func NewStoreMock(t *testing.T) *StoreMock {
//
// NOTE: Part of the SwapStore interface.
func (s *StoreMock) FetchLoopOutSwaps(ctx context.Context) ([]*LoopOut, error) {
s.RLock()
defer s.RUnlock()
result := []*LoopOut{}
for hash, contract := range s.LoopOutSwaps {
@ -80,6 +86,9 @@ func (s *StoreMock) FetchLoopOutSwaps(ctx context.Context) ([]*LoopOut, error) {
func (s *StoreMock) FetchLoopOutSwap(ctx context.Context,
hash lntypes.Hash) (*LoopOut, error) {
s.RLock()
defer s.RUnlock()
contract, ok := s.LoopOutSwaps[hash]
if !ok {
return nil, errors.New("swap not found")
@ -110,6 +119,9 @@ func (s *StoreMock) FetchLoopOutSwap(ctx context.Context,
func (s *StoreMock) CreateLoopOut(ctx context.Context, hash lntypes.Hash,
swap *LoopOutContract) error {
s.Lock()
defer s.Unlock()
_, ok := s.LoopOutSwaps[hash]
if ok {
return errors.New("swap already exists")
@ -126,6 +138,9 @@ func (s *StoreMock) CreateLoopOut(ctx context.Context, hash lntypes.Hash,
func (s *StoreMock) FetchLoopInSwaps(ctx context.Context) ([]*LoopIn,
error) {
s.RLock()
defer s.RUnlock()
result := []*LoopIn{}
for hash, contract := range s.LoopInSwaps {
@ -156,6 +171,9 @@ func (s *StoreMock) FetchLoopInSwaps(ctx context.Context) ([]*LoopIn,
func (s *StoreMock) CreateLoopIn(ctx context.Context, hash lntypes.Hash,
swap *LoopInContract) error {
s.Lock()
defer s.Unlock()
_, ok := s.LoopInSwaps[hash]
if ok {
return errors.New("swap already exists")
@ -176,6 +194,9 @@ func (s *StoreMock) CreateLoopIn(ctx context.Context, hash lntypes.Hash,
func (s *StoreMock) UpdateLoopOut(ctx context.Context, hash lntypes.Hash,
time time.Time, state SwapStateData) error {
s.Lock()
defer s.Unlock()
updates, ok := s.LoopOutUpdates[hash]
if !ok {
return errors.New("swap does not exists")
@ -196,6 +217,9 @@ func (s *StoreMock) UpdateLoopOut(ctx context.Context, hash lntypes.Hash,
func (s *StoreMock) UpdateLoopIn(ctx context.Context, hash lntypes.Hash,
time time.Time, state SwapStateData) error {
s.Lock()
defer s.Unlock()
updates, ok := s.LoopInUpdates[hash]
if !ok {
return errors.New("swap does not exists")
@ -347,6 +371,9 @@ func (b *StoreMock) BatchInsertUpdate(ctx context.Context,
func (s *StoreMock) BatchUpdateLoopOutSwapCosts(ctx context.Context,
costs map[lntypes.Hash]SwapCost) error {
s.Lock()
defer s.Unlock()
for hash, cost := range costs {
if _, ok := s.LoopOutUpdates[hash]; !ok {
return fmt.Errorf("swap has no updates: %v", hash)
@ -367,6 +394,9 @@ func (s *StoreMock) BatchUpdateLoopOutSwapCosts(ctx context.Context,
func (s *StoreMock) HasMigration(ctx context.Context, migrationID string) (
bool, error) {
s.RLock()
defer s.RUnlock()
_, ok := s.migrations[migrationID]
return ok, nil
@ -376,6 +406,9 @@ func (s *StoreMock) HasMigration(ctx context.Context, migrationID string) (
func (s *StoreMock) SetMigration(ctx context.Context,
migrationID string) error {
s.Lock()
defer s.Unlock()
if _, ok := s.migrations[migrationID]; ok {
return errors.New("migration already done")
}