47 lines
978 B
Go
47 lines
978 B
Go
|
package impl
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"122.51.31.227/go-course/go18/devcloud/audit/apps/event"
|
||
|
|
"github.com/infraboard/mcube/v2/types"
|
||
|
|
"go.mongodb.org/mongo-driver/bson"
|
||
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
||
|
|
)
|
||
|
|
|
||
|
|
// 存储
|
||
|
|
func (s *EventServiceImpl) SaveEvent(ctx context.Context, in *types.Set[*event.Event]) error {
|
||
|
|
s.log.Debug().Msgf("events: %s", in)
|
||
|
|
|
||
|
|
_, err := s.col.InsertMany(ctx, in.ToAny())
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *EventServiceImpl) QueryEvent(ctx context.Context, in *event.QueryEventRequest) (*types.Set[*event.Event], error) {
|
||
|
|
set := types.NewSet[*event.Event]()
|
||
|
|
|
||
|
|
// 查询条件
|
||
|
|
filter := bson.M{}
|
||
|
|
|
||
|
|
opt := options.Find()
|
||
|
|
opt.SetLimit(int64(in.PageSize))
|
||
|
|
opt.SetSkip(in.ComputeOffset())
|
||
|
|
|
||
|
|
cursor, err := s.col.Find(ctx, filter, opt)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
for cursor.Next(ctx) {
|
||
|
|
e := event.NewEvent()
|
||
|
|
if err := cursor.Decode(e); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
set.Add(e)
|
||
|
|
}
|
||
|
|
return set, nil
|
||
|
|
}
|