audit/audit.go

package audit

import (
	"errors"
	"fmt"

	"github.com/Selly-Modules/mongodb"
	"go.mongodb.org/mongo-driver/mongo"
)

// MongoDBConfig ...
type MongoDBConfig struct {
	Host, User, Password, DBName, Mechanism, Source string
}

// Config ...
type Config struct {
	// Targets: staff, article, ...
	Targets []string
	// MongoDB config, for save documents
	MongoDB MongoDBConfig
}

// Service ...
type Service struct {
	Config
	DB *mongo.Database
}

var s *Service

// NewInstance ...
func NewInstance(config Config) error {
	if len(config.Targets) == 0 || config.MongoDB.Host == "" {
		return errors.New("please provide all necessary information: targets, mongodb")
	}

	// Connect MongoDB
	err := mongodb.Connect(
		config.MongoDB.Host,
		config.MongoDB.User,
		config.MongoDB.Password,
		config.MongoDB.DBName,
		config.MongoDB.Mechanism,
		config.MongoDB.Source,
	)
	if err != nil {
		fmt.Println("Cannot init module AUDIT", err)
		return err
	}

	s = &Service{
		Config: config,
		DB:     mongodb.GetInstance(),
	}

	// index mongo
	s.indexDB()

	return nil
}

// GetInstance ...
func GetInstance() *Service {
	return s
}

// getColName ...
func getColName(target string) string {
	return target
}