usermngmt/model/query.go

package model

import (
	"github.com/Selly-Modules/mongodb"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo/options"
)

// CommonQuery ...
type CommonQuery struct {
	Page    int64
	Limit   int64
	Keyword string
	RoleID  string
	Status  string
	Deleted string
	Sort    interface{}
	Other   map[string]interface{}
}

// AssignDeleted ...
func (q *CommonQuery) AssignDeleted(cond bson.M) {
	if q.Deleted == "true" {
		cond["deleted"] = true
	}
	if q.Deleted == "false" {
		cond["deleted"] = false
	}
}

// AssignKeyword ...
func (q *CommonQuery) AssignKeyword(cond bson.M) {
	if q.Keyword != "" {
		cond["searchString"] = mongodb.GenerateQuerySearchString(q.Keyword)
	}
}

// AssignRoleID ...
func (q *CommonQuery) AssignRoleID(cond bson.M) {
	if q.RoleID != "" {
		if id, isValid := mongodb.NewIDFromString(q.RoleID); isValid {
			cond["roleId"] = id
		}
	}
}

// AssignStatus ...
func (q *CommonQuery) AssignStatus(cond bson.M) {
	if q.Status != "" {
		cond["status"] = q.Status
	}
}

// GetFindOptionsUsingPage ...
func (q *CommonQuery) GetFindOptionsUsingPage() *options.FindOptions {
	opts := options.Find()
	if q.Limit > 0 {
		opts.SetLimit(q.Limit).SetSkip(q.Limit * q.Page)
	}
	if q.Sort != nil {
		opts.SetSort(q.Sort)
	}
	return opts
}

// SetDefaultLimit ...
func (q *CommonQuery) SetDefaultLimit() {
	if q.Limit <= 0 || q.Limit > 20 {
		q.Limit = 20
	}
}

// AssignOther ...
func (q *CommonQuery) AssignOther(cond bson.M) {
	// Query fields in other object
	if len(q.Other) > 0 {
		for key, value := range q.Other {
			cond["other."+key] = value
		}
	}
}