Merge pull request #6 from Selly-Modules/add-filter-author-and-time

add filter author and time
This commit is contained in:
Sinh Luu 2022-07-06 09:08:16 +07:00 committed by GitHub
commit e7c7c46b75
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 39 additions and 7 deletions

View File

@ -3,6 +3,7 @@ package audit
import (
"context"
"sync"
"time"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo/options"
@ -12,9 +13,16 @@ import (
type AllQuery struct {
Target string
TargetID string
Page int64
Limit int64
Sort interface{}
// Additional filter
Author string
CreateTimeFrom time.Time
CreateTimeTo time.Time
// Pagination
Page int64
Limit int64
Sort interface{}
}
// All ...
@ -25,10 +33,7 @@ func (s Service) All(query AllQuery) (result []Audit, total int64) {
skip = query.Page * query.Limit
wg sync.WaitGroup
)
cond := bson.D{
{"target", query.Target},
{"targetId", query.TargetID},
}
cond := s.getQueryCondition(query)
wg.Add(1)
go func() {
defer wg.Done()
@ -54,3 +59,30 @@ func (s Service) All(query AllQuery) (result []Audit, total int64) {
wg.Wait()
return result, total
}
func (s Service) getQueryCondition(query AllQuery) bson.D {
cond := bson.D{
{"target", query.Target},
{"targetId", query.TargetID},
}
if query.Author != "" {
cond = append(cond, bson.E{
Key: "author.id",
Value: query.Author,
})
}
if !query.CreateTimeFrom.IsZero() || !query.CreateTimeTo.IsZero() {
v := bson.M{}
if !query.CreateTimeFrom.IsZero() {
v["$gte"] = query.CreateTimeFrom
}
if !query.CreateTimeTo.IsZero() {
v["$lt"] = query.CreateTimeTo
}
cond = append(cond, bson.E{
Key: "createdAt",
Value: v,
})
}
return cond
}