95 lines
2.5 KiB
Go
95 lines
2.5 KiB
Go
package txhash
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
"strings"
|
|
|
|
gossipv1 "github.com/certusone/wormhole/node/pkg/proto/gossip/v1"
|
|
"github.com/hashicorp/go-multierror"
|
|
"github.com/wormhole-foundation/wormhole-explorer/common/domain"
|
|
"github.com/wormhole-foundation/wormhole-explorer/fly/internal/metrics"
|
|
"github.com/wormhole-foundation/wormhole/sdk/vaa"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type composite struct {
|
|
hashStores []TxHashStore
|
|
metrics metrics.Metrics
|
|
logger *zap.Logger
|
|
}
|
|
|
|
func NewComposite(
|
|
hashStores []TxHashStore,
|
|
metrics metrics.Metrics,
|
|
logger *zap.Logger) *composite {
|
|
return &composite{
|
|
hashStores: hashStores,
|
|
metrics: metrics,
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
func (t *composite) Set(ctx context.Context, vaaID string, txHash TxHash) error {
|
|
var result multierror.Error
|
|
for _, store := range t.hashStores {
|
|
if err := store.Set(ctx, vaaID, txHash); err != nil {
|
|
t.logger.Error("Error setting tx hash",
|
|
zap.String("vaaId", vaaID),
|
|
zap.String("store", store.GetName()),
|
|
zap.Error(err))
|
|
result.Errors = append(result.Errors, err)
|
|
}
|
|
}
|
|
return result.ErrorOrNil()
|
|
}
|
|
|
|
func (t *composite) SetObservation(ctx context.Context, o *gossipv1.SignedObservation) error {
|
|
vaaID := strings.Split(o.MessageId, "/")
|
|
chainIDStr, emitter, sequenceStr := vaaID[0], vaaID[1], vaaID[2]
|
|
|
|
chainID, err := strconv.ParseUint(chainIDStr, 10, 16)
|
|
if err != nil {
|
|
t.logger.Error("Error parsing chainId", zap.Error(err))
|
|
return err
|
|
}
|
|
|
|
txHash, err := domain.EncodeTrxHashByChainID(vaa.ChainID(chainID), o.GetTxHash())
|
|
if err != nil {
|
|
t.logger.Warn("Error encoding tx hash",
|
|
zap.Uint64("chainId", chainID),
|
|
zap.ByteString("txHash", o.GetTxHash()),
|
|
zap.Error(err))
|
|
}
|
|
|
|
vaaTxHash := TxHash{
|
|
ChainID: vaa.ChainID(chainID),
|
|
Emitter: emitter,
|
|
Sequence: sequenceStr,
|
|
TxHash: txHash,
|
|
}
|
|
return t.Set(ctx, o.MessageId, vaaTxHash)
|
|
}
|
|
|
|
func (t *composite) Get(ctx context.Context, vaaID string) (*TxHash, error) {
|
|
log := t.logger.With(zap.String("vaaId", vaaID))
|
|
for _, store := range t.hashStores {
|
|
txHash, err := store.Get(ctx, vaaID)
|
|
if err == nil {
|
|
t.metrics.IncFoundTxHash(store.GetName())
|
|
return txHash, nil
|
|
}
|
|
if err != ErrTxHashNotFound {
|
|
log.Error("Error getting tx hash", zap.String("store", store.GetName()), zap.Error(err))
|
|
} else {
|
|
log.Info("Cannot get txHash", zap.String("store", store.GetName()))
|
|
}
|
|
t.metrics.IncNotFoundTxHash(store.GetName())
|
|
}
|
|
return nil, ErrTxHashNotFound
|
|
}
|
|
|
|
func (t *composite) GetName() string {
|
|
return "multi"
|
|
}
|