github.com/amazechain/amc@v0.1.3/internal/api/addrlock.go (about)

     1  // Copyright 2022 The AmazeChain Authors
     2  // This file is part of the AmazeChain library.
     3  //
     4  // The AmazeChain library is free software: you can redistribute it and/or modify
     5  // it under the terms of the GNU Lesser General Public License as published by
     6  // the Free Software Foundation, either version 3 of the License, or
     7  // (at your option) any later version.
     8  //
     9  // The AmazeChain library is distributed in the hope that it will be useful,
    10  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    12  // GNU Lesser General Public License for more details.
    13  //
    14  // You should have received a copy of the GNU Lesser General Public License
    15  // along with the AmazeChain library. If not, see <http://www.gnu.org/licenses/>.
    16  
    17  package api
    18  
    19  import (
    20  	"github.com/amazechain/amc/common/types"
    21  	"sync"
    22  )
    23  
    24  type AddrLocker struct {
    25  	mu    sync.Mutex
    26  	locks map[types.Address]*sync.Mutex
    27  }
    28  
    29  // lock returns the lock of the given address.
    30  func (l *AddrLocker) lock(address types.Address) *sync.Mutex {
    31  	l.mu.Lock()
    32  	defer l.mu.Unlock()
    33  	if l.locks == nil {
    34  		l.locks = make(map[types.Address]*sync.Mutex)
    35  	}
    36  	if _, ok := l.locks[address]; !ok {
    37  		l.locks[address] = new(sync.Mutex)
    38  	}
    39  	return l.locks[address]
    40  }
    41  
    42  // LockAddr locks an account's mutex. This is used to prevent another tx getting the
    43  // same nonce until the lock is released. The mutex prevents the (an identical nonce) from
    44  // being read again during the time that the first transaction is being signed.
    45  func (l *AddrLocker) LockAddr(address types.Address) {
    46  	l.lock(address).Lock()
    47  }
    48  
    49  // UnlockAddr unlocks the mutex of the given account.
    50  func (l *AddrLocker) UnlockAddr(address types.Address) {
    51  	l.lock(address).Unlock()
    52  }