add find collation option (#254)

Author: wanghao-bianjie

interface.go

@@ -13,6 +13,8 @@
 
 package qmgo
 
+import "go.mongodb.org/mongo-driver/mongo/options"
+
 // CollectionI
 // 集合操作接口
 //type CollectionI interface {
@@ -48,6 +50,7 @@ type CursorI interface {
 
 // QueryI Query interface
 type QueryI interface {
+	Collation(collation *options.Collation) QueryI
 	Sort(fields ...string) QueryI
 	Select(selector interface{}) QueryI
 	Skip(n int64) QueryI

query.go

@@ -37,13 +37,20 @@ type Query struct {
 	skip            *int64
 	batchSize       *int64
 	noCursorTimeout *bool
+	collation       *options.Collation
 
 	ctx        context.Context
 	collection *mongo.Collection
 	opts       []qOpts.FindOptions
 	registry   *bsoncodec.Registry
 }
 
+func (q *Query) Collation(collation *options.Collation) QueryI {
+	newQ := q
+	newQ.collation = collation
+	return newQ
+}
+
 func (q *Query) NoCursorTimeout(n bool) QueryI {
 	newQ := q
 	newQ.noCursorTimeout = &n
@@ -128,6 +135,9 @@ func (q *Query) One(result interface{}) error {
 	}
 	opt := options.FindOne()
 
+	if q.collation != nil {
+		opt.SetCollation(q.collation)
+	}
 	if q.sort != nil {
 		opt.SetSort(q.sort)
 	}
@@ -163,6 +173,9 @@ func (q *Query) All(result interface{}) error {
 		}
 	}
 	opt := options.Find()
+	if q.collation != nil {
+		opt.SetCollation(q.collation)
+	}
 	if q.sort != nil {
 		opt.SetSort(q.sort)
 	}