search_builder.go

package mongo

import (
	"context"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"reflect"
)

type SearchBuilder struct {
	Collection *mongo.Collection
	BuildQuery func(m interface{}) (bson.D, bson.M)
	GetSort    func(m interface{}) string
	BuildSort  func(s string, modelType reflect.Type) bson.D
}

func NewSearchQueryWithSort(db *mongo.Database, collectionName string, buildQuery func(interface{}) (bson.D, bson.M), getSort func(interface{}) string, buildSort func(string, reflect.Type) bson.D) *SearchBuilder {
	return NewSearchBuilderWithSort(db, collectionName, buildQuery, getSort, buildSort)
}
func NewSearchQuery(db *mongo.Database, collectionName string, buildQuery func(interface{}) (bson.D, bson.M), getSort func(interface{}) string) *SearchBuilder {
	return NewSearchBuilderWithSort(db, collectionName, buildQuery, getSort, BuildSort)
}
func NewSearchBuilderWithSort(db *mongo.Database, collectionName string, buildQuery func(interface{}) (bson.D, bson.M), getSort func(interface{}) string, buildSort func(string, reflect.Type) bson.D) *SearchBuilder {
	collection := db.Collection(collectionName)
	builder := &SearchBuilder{Collection: collection, BuildQuery: buildQuery, GetSort: getSort, BuildSort: buildSort}
	return builder
}
func NewSearchBuilder(db *mongo.Database, collectionName string, buildQuery func(interface{}) (bson.D, bson.M), getSort func(interface{}) string) *SearchBuilder {
	return NewSearchBuilderWithSort(db, collectionName, buildQuery, getSort, BuildSort)
}
func (b *SearchBuilder) Search(ctx context.Context, m interface{}, results interface{}, limit int64, skip int64) (int64, error) {
	query, fields := b.BuildQuery(m)

	var sort = bson.D{}
	s := b.GetSort(m)
	modelType := reflect.TypeOf(results).Elem().Elem()
	sort = b.BuildSort(s, modelType)
	if skip < 0 {
		skip = 0
	}
	return BuildSearchResult(ctx, b.Collection, results, query, fields, sort, limit, skip)
}