/
query.go
67 lines (52 loc) · 1.27 KB
/
query.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package impl
import (
"github.com/infraboard/keyauth/apps/micro"
"github.com/infraboard/mcube/exception"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo/options"
)
func newPaggingQuery(req *micro.QueryMicroRequest) *queryRequest {
return &queryRequest{req}
}
type queryRequest struct {
*micro.QueryMicroRequest
}
func (r *queryRequest) FindOptions() *options.FindOptions {
pageSize := int64(r.Page.PageSize)
skip := int64(r.Page.PageSize) * int64(r.Page.PageNumber-1)
opt := &options.FindOptions{
Sort: bson.D{bson.E{Key: "create_at", Value: -1}},
Limit: &pageSize,
Skip: &skip,
}
return opt
}
func (r *queryRequest) FindFilter() bson.M {
filter := bson.M{}
if r.Type != micro.Type_NULL {
filter["type"] = r.Type
}
return filter
}
func newDescribeQuery(req *micro.DescribeMicroRequest) (*describeRequest, error) {
if err := req.Validate(); err != nil {
return nil, exception.NewBadRequest(err.Error())
}
return &describeRequest{req}, nil
}
type describeRequest struct {
*micro.DescribeMicroRequest
}
func (r *describeRequest) FindFilter() bson.M {
filter := bson.M{}
if r.Name != "" {
filter["name"] = r.Name
}
if r.Id != "" {
filter["_id"] = r.Id
}
if r.ClientId != "" {
filter["client_id"] = r.ClientId
}
return filter
}