-
Notifications
You must be signed in to change notification settings - Fork 0
/
one.go
56 lines (44 loc) · 1.6 KB
/
one.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
/*
* Tencent is pleased to support the open source community by making 蓝鲸 available.,
* Copyright (C) 2017-2018 THL A29 Limited, a Tencent company. All rights reserved.
* Licensed under the MIT License (the ",License",); you may not use this file except
* in compliance with the License. You may obtain a copy of the License at
* http://opensource.org/licenses/MIT
* Unless required by applicable law or agreed to in writing, software distributed under
* the License is distributed on an ",AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
* either express or implied. See the License for the specific language governing permissions and
* limitations under the License.
*/
package findopt
import (
"github.com/mongodb/mongo-go-driver/bson/primitive"
"github.com/mongodb/mongo-go-driver/mongo/options"
)
// ConvertToMongoOptions convert find many opt into mongo options array
func (m *One) ConvertToMongoOptions() *options.FindOneOptions {
option := &options.FindOneOptions{}
option.Skip = &m.Skip
sortD := primitive.D{}
for _, sortItem := range m.Sort {
if sortItem.Descending {
sortD = append(sortD, primitive.E{Key: sortItem.Name, Value: -1})
continue
}
sortD = append(sortD, primitive.E{Key: sortItem.Name, Value: 1})
}
if 0 != len(sortD) {
option.Sort = sortD
}
fieldD := primitive.D{}
for _, fieldItem := range m.Fields {
if fieldItem.Hide {
fieldD = append(fieldD, primitive.E{Key: fieldItem.Name, Value: 0})
continue
}
fieldD = append(fieldD, primitive.E{Key: fieldItem.Name, Value: 1})
}
if 0 != len(fieldD) {
option.Projection = fieldD
}
return option
}