123 lines
2.9 KiB
Go
123 lines
2.9 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
sdk "github.com/wormhole-foundation/wormhole/sdk/vaa"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// VaaRepository is a repository for VAA.
|
|
type VaaRepository struct {
|
|
db *mongo.Database
|
|
logger *zap.Logger
|
|
vaas *mongo.Collection
|
|
}
|
|
|
|
// VaaDoc is a document for VAA.
|
|
type VaaDoc struct {
|
|
ID string `bson:"_id" json:"id"`
|
|
Vaa []byte `bson:"vaas" json:"vaa"`
|
|
}
|
|
|
|
// VaaQuery is a query for VAA.
|
|
type VaaQuery struct {
|
|
StartTime *time.Time
|
|
EndTime *time.Time
|
|
EmitterChainID *sdk.ChainID
|
|
EmitterAddress *string
|
|
Sequence *string
|
|
}
|
|
|
|
// Pagination is a pagination for VAA.
|
|
type Pagination struct {
|
|
Page int64
|
|
PageSize int64
|
|
SortAsc bool
|
|
}
|
|
|
|
// NewVaaRepository create a new Vaa repository.
|
|
func NewVaaRepository(db *mongo.Database, logger *zap.Logger) *VaaRepository {
|
|
return &VaaRepository{db: db,
|
|
logger: logger.With(zap.String("module", "VaaRepository")),
|
|
vaas: db.Collection(Vaas),
|
|
}
|
|
}
|
|
|
|
// FindById finds VAA by id.
|
|
func (r *VaaRepository) FindById(ctx context.Context, id string) (*VaaDoc, error) {
|
|
var vaaDoc VaaDoc
|
|
err := r.vaas.FindOne(ctx, bson.M{"_id": id}).Decode(&vaaDoc)
|
|
return &vaaDoc, err
|
|
}
|
|
|
|
// FindPageByTimeRange finds VAA by time range.
|
|
func (r *VaaRepository) FindPageByTimeRange(ctx context.Context, startTime time.Time, endTime time.Time, page, pageSize int64, sortAsc bool) ([]*VaaDoc, error) {
|
|
filter := bson.M{
|
|
"timestamp": bson.M{
|
|
"$gte": startTime,
|
|
"$lt": endTime,
|
|
},
|
|
}
|
|
sort := -1
|
|
if sortAsc {
|
|
sort = 1
|
|
}
|
|
|
|
skip := page * pageSize
|
|
opts := &options.FindOptions{Skip: &skip, Limit: &pageSize, Sort: bson.M{"timestamp": sort}}
|
|
cur, err := r.vaas.Find(ctx, filter, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var vaas []*VaaDoc
|
|
err = cur.All(ctx, &vaas)
|
|
return vaas, err
|
|
}
|
|
|
|
// FindPage finds VAA by query and pagination.
|
|
func (r *VaaRepository) FindPage(ctx context.Context, query VaaQuery, pagination Pagination) ([]*VaaDoc, error) {
|
|
|
|
filter := bson.M{}
|
|
|
|
if query.StartTime != nil || query.EndTime != nil {
|
|
rangeTimestamp := bson.M{}
|
|
if query.StartTime != nil {
|
|
rangeTimestamp["$gte"] = query.StartTime
|
|
}
|
|
if query.EndTime != nil {
|
|
rangeTimestamp["$lt"] = query.EndTime
|
|
}
|
|
filter["timestamp"] = rangeTimestamp
|
|
}
|
|
|
|
if query.EmitterChainID != nil {
|
|
filter["emitterChain"] = query.EmitterChainID
|
|
}
|
|
if query.EmitterAddress != nil {
|
|
filter["emitterAddr"] = query.EmitterAddress
|
|
}
|
|
if query.Sequence != nil {
|
|
filter["sequence"] = query.Sequence
|
|
}
|
|
|
|
sort := -1
|
|
if pagination.SortAsc {
|
|
sort = 1
|
|
}
|
|
|
|
skip := pagination.Page * pagination.PageSize
|
|
opts := &options.FindOptions{Skip: &skip, Limit: &pagination.PageSize, Sort: bson.M{"timestamp": sort}}
|
|
cur, err := r.vaas.Find(ctx, filter, opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var vaas []*VaaDoc
|
|
err = cur.All(ctx, &vaas)
|
|
return vaas, err
|
|
}
|