2023-05-24 12:52:22 -07:00
|
|
|
package vaa
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
)
|
|
|
|
|
|
|
|
type Repository struct {
|
|
|
|
db *mongo.Database
|
|
|
|
logger *zap.Logger
|
|
|
|
vaas *mongo.Collection
|
|
|
|
}
|
|
|
|
|
|
|
|
type VaaDoc struct {
|
|
|
|
ID string `bson:"_id" json:"id"`
|
|
|
|
Vaa []byte `bson:"vaas" json:"vaa"`
|
|
|
|
}
|
|
|
|
|
|
|
|
// NewRepository create a new Repository.
|
|
|
|
func NewRepository(db *mongo.Database, logger *zap.Logger) *Repository {
|
|
|
|
return &Repository{db: db,
|
|
|
|
logger: logger.With(zap.String("module", "VaaRepository")),
|
|
|
|
vaas: db.Collection("vaas"),
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *Repository) 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
|
|
|
|
}
|
|
|
|
|
2023-07-12 08:51:17 -07:00
|
|
|
func (r *Repository) FindPageByTimeRange(ctx context.Context, startTime time.Time, endTime time.Time, page, pageSize int64, sortAsc bool) ([]*VaaDoc, error) {
|
2023-05-24 12:52:22 -07:00
|
|
|
filter := bson.M{
|
|
|
|
"timestamp": bson.M{
|
|
|
|
"$gte": startTime,
|
|
|
|
"$lt": endTime,
|
|
|
|
},
|
|
|
|
}
|
2023-07-12 08:51:17 -07:00
|
|
|
sort := -1
|
|
|
|
if sortAsc {
|
|
|
|
sort = 1
|
|
|
|
}
|
|
|
|
|
2023-05-24 12:52:22 -07:00
|
|
|
skip := page * pageSize
|
2023-07-12 08:51:17 -07:00
|
|
|
opts := &options.FindOptions{Skip: &skip, Limit: &pageSize, Sort: bson.M{"timestamp": sort}}
|
2023-05-24 12:52:22 -07:00
|
|
|
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
|
|
|
|
}
|