usermngmt/database/db.go

package database

import (
	"fmt"

	"go.mongodb.org/mongo-driver/mongo"
)

// Table
var (
	tableUser = "users"
	tableRole = "roles"
)

var (
	db     *mongo.Database
	prefix string
)

// Set ...
func Set(instance *mongo.Database, tablePrefix string) {
	db = instance
	prefix = tablePrefix
}

// GetUserCol ...
func GetUserCol() *mongo.Collection {
	return db.Collection(fmt.Sprintf("%s-%s", prefix, tableUser))
}

// GetRoleCol ...
func GetRoleCol() *mongo.Collection {
	return db.Collection(fmt.Sprintf("%s-%s", prefix, tableRole))
}