sign_msg_cache.go

package walletutils

import (
	"sync"

	"github.com/ethereum/go-ethereum/common"
	lotustypes "github.com/filecoin-project/lotus/chain/types"
)

// maps ethereum tx hashes to filecoin signed messages
type SignedMessageCache struct {
	signedMessages map[common.Hash]*lotustypes.SignedMessage
	txHashLookup   map[common.Hash]common.Hash
	mutex          sync.Mutex
}

func (smc *SignedMessageCache) Add(txHash common.Hash, signedMsg *lotustypes.SignedMessage) {
	smc.mutex.Lock()
	defer smc.mutex.Unlock()

	smc.signedMessages[txHash] = signedMsg
}

func (smc *SignedMessageCache) Get(txHash common.Hash) *lotustypes.SignedMessage {
	smc.mutex.Lock()
	defer smc.mutex.Unlock()

	return smc.signedMessages[txHash]
}

func (smc *SignedMessageCache) Delete(txHash common.Hash) {
	smc.mutex.Lock()
	defer smc.mutex.Unlock()

	delete(smc.signedMessages, txHash)
}

func (smc *SignedMessageCache) MapHash(innerHash, outerHash common.Hash) {
	smc.mutex.Lock()
	defer smc.mutex.Unlock()

	smc.txHashLookup[innerHash] = outerHash
}

func (smc *SignedMessageCache) GetOuterHash(innerHash common.Hash) common.Hash {
	smc.mutex.Lock()
	defer smc.mutex.Unlock()

	return smc.txHashLookup[innerHash]
}

func NewSignedMsgCache() *SignedMessageCache {
	return &SignedMessageCache{
		signedMessages: make(map[common.Hash]*lotustypes.SignedMessage),
		// maps the outer filecoin tx hash to the inner filecoin tx hash
		txHashLookup: make(map[common.Hash]common.Hash),
	}
}