88 lines
2.5 KiB
Go
88 lines
2.5 KiB
Go
package repository
|
||
|
||
import (
|
||
"context"
|
||
|
||
"apps/backend/internal/module/radar/domain"
|
||
|
||
"github.com/zeromicro/go-zero/core/stores/mon"
|
||
"go.mongodb.org/mongo-driver/bson"
|
||
"go.mongodb.org/mongo-driver/mongo/options"
|
||
)
|
||
|
||
func (s *MonStore) SaveWatch(ctx context.Context, w *domain.RadarWatch) error {
|
||
_, err := s.watches.ReplaceOne(ctx, bson.M{"_id": w.ID}, w, options.Replace().SetUpsert(true))
|
||
return err
|
||
}
|
||
|
||
func (s *MonStore) GetWatch(ctx context.Context, id string) (*domain.RadarWatch, error) {
|
||
var w domain.RadarWatch
|
||
err := s.watches.FindOne(ctx, &w, bson.M{"_id": id})
|
||
if err == mon.ErrNotFound {
|
||
return nil, domain.ErrNotFound
|
||
}
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
return &w, nil
|
||
}
|
||
|
||
func (s *MonStore) ListWatches(ctx context.Context, ownerUID int64, f domain.WatchListFilter) ([]*domain.RadarWatch, int64, error) {
|
||
q := bson.M{"owner_uid": ownerUID}
|
||
if f.Status != "" {
|
||
q["status"] = f.Status
|
||
}
|
||
total, err := s.watches.CountDocuments(ctx, q)
|
||
if err != nil {
|
||
return nil, 0, err
|
||
}
|
||
page, ps := f.Page, f.PageSize
|
||
if page < 1 {
|
||
page = 1
|
||
}
|
||
if ps < 1 {
|
||
ps = 20
|
||
}
|
||
var list []*domain.RadarWatch
|
||
err = s.watches.Find(ctx, &list, q, options.Find().
|
||
SetSort(bson.D{{Key: "created_at", Value: -1}}).
|
||
SetSkip(int64((page-1)*ps)).
|
||
SetLimit(int64(ps)))
|
||
return list, total, err
|
||
}
|
||
|
||
func (s *MonStore) ListActiveWatches(ctx context.Context, ownerUID int64) ([]*domain.RadarWatch, error) {
|
||
var list []*domain.RadarWatch
|
||
err := s.watches.Find(ctx, &list,
|
||
bson.M{"owner_uid": ownerUID, "status": domain.WatchActive},
|
||
options.Find().SetSort(bson.D{{Key: "created_at", Value: 1}}))
|
||
return list, err
|
||
}
|
||
|
||
func (s *MonStore) ListAllActiveWatches(ctx context.Context) ([]*domain.RadarWatch, error) {
|
||
var list []*domain.RadarWatch
|
||
err := s.watches.Find(ctx, &list,
|
||
bson.M{"status": domain.WatchActive},
|
||
options.Find().SetSort(bson.D{
|
||
{Key: "owner_uid", Value: 1},
|
||
{Key: "created_at", Value: 1},
|
||
}))
|
||
return list, err
|
||
}
|
||
|
||
func (s *MonStore) CountActiveWatches(ctx context.Context, ownerUID int64) (int64, error) {
|
||
return s.watches.CountDocuments(ctx, bson.M{"owner_uid": ownerUID, "status": domain.WatchActive})
|
||
}
|
||
|
||
// TouchWatchSweptAt 只動 last_swept_at，避免與使用者同時編輯關鍵字互相覆蓋。
|
||
func (s *MonStore) TouchWatchSweptAt(ctx context.Context, id string, at int64) error {
|
||
res, err := s.watches.UpdateOne(ctx, bson.M{"_id": id}, bson.M{"$set": bson.M{"last_swept_at": at}})
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if res.MatchedCount == 0 {
|
||
return domain.ErrNotFound
|
||
}
|
||
return nil
|
||
}
|