oc-lib/mongo.go

package oclib

import (
	"context"
	"encoding/json"
	"errors"
	"os"
	"time"

	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"go.mongodb.org/mongo-driver/x/bsonx"
)

var (
	mngoClient *mongo.Client
	mngoDB     *mongo.Database
	MngoCtx    context.Context

	MngoCollData       *mongo.Collection
	MngoCollComputing  *mongo.Collection
	MngoCollStorage    *mongo.Collection
	MngoCollDatacenter *mongo.Collection
	MngoCollWorkspace  *mongo.Collection
	MngoCollSchedule   *mongo.Collection
)

// Trying to get vscode to display this
func MongoInit() {

	// var baseConfig string
	var err error
	var cancel context.CancelFunc
	var conf map[string]string
	var MongoURL string
	var DBname string
	
	db_conf, err := os.ReadFile("oclib_conf.json")
	if err != nil {
		logger.Fatal().Msg("Could not find configuration file")
	}

	json.Unmarshal(db_conf,&conf)
	
	
	if len(os.Getenv("DOCKER_ENVIRONMENT")) == 0 {
		MongoURL = conf["DB_URL_LOCAL"]
	} else {
		MongoURL = conf["DB_URL_DOCKER"]
	}

	DBname = conf["DCNAME"] + "-" + conf["DBPOINT"]

	// MongoURL, err := beego.AppConfig.String(baseConfig + "::url")
	// if err != nil {
	// 	logger.Critical().Msg("MongoDB URI error: %v", err)
	// 	panic(err)
	// }

	logger.Info().Msg("Connecting to" + MongoURL)
	MngoCtx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()

	createClient(MongoURL)

	logger.Info().Msg("Connecting mongo client to db " + DBname)
	prepareDB(conf["DCNAME"],conf["DBPOINT"])

	logger.Info().Msg("Database is READY")

}

// TODO : Soit retourner MngoCtx, soit comprendre comment passer sa référence et la mettre à jour
func createClient(MongoURL string){

	var err error

	clientOptions := options.Client().ApplyURI(MongoURL)
	
	mngoClient, _ = mongo.NewClient(options.Client().ApplyURI(MongoURL))

	if err = mngoClient.Connect(MngoCtx); err != nil {
		logger.Fatal().Msg("Mongodb NewClient " + MongoURL + ":"  + "err" )
		panic(err)
	}


	// Ping the primary
	if mngoClient, err = mongo.Connect(MngoCtx, clientOptions); err != nil {
		logger.Fatal().Msg("Mongodb connect " + MongoURL + ":"  + "err" )
		panic(err)
	}

	if err = mngoClient.Ping(MngoCtx, nil); err != nil {
		logger.Fatal().Msg("Mongodb ping " + MongoURL + ":"  + "err" )
		panic(err)
	}

}

func prepareDB(dc_name string, db_point string) {

	// var err error
	DBname := dc_name + "-"  + db_point

	mngoDB = mngoClient.Database(DBname)

	MngoCollData = mngoDB.Collection("Data")
	MngoCollComputing = mngoDB.Collection("Computing")
	MngoCollStorage = mngoDB.Collection("Storage")
	MngoCollDatacenter = mngoDB.Collection("Datacenter")
	MngoCollWorkspace = mngoDB.Collection("Workspace")
	MngoCollSchedule = mngoDB.Collection("Schedule")

	list_collection := [...]string{"Data","Computing","Storage","Datacenter","Workspace","Schedule"}

	for _, collection_name := range(list_collection){
		new_collection := mngoDB.Collection(collection_name)
		createCollection(new_collection)
	}



	

	// if _, err = MngoCollComputing.Indexes().CreateMany(MngoCtx, []mongo.IndexModel{
	// 	{
	// 		Keys: bsonx.Doc{
	// 			{Key: "description", Value: bsonx.String("text")},
	// 			{Key: "owner", Value: bsonx.String("text")},
	// 			{Key: "license", Value: bsonx.String("text")},
	// 		},
	// 	},
	// }); err != nil && err.(mongo.CommandError).Code != 85 {
	// 	logger.Critical().Msg(err)
	// 	panic(err)
	// }

	// if _, err = MngoCollStorage.Indexes().CreateMany(MngoCtx, []mongo.IndexModel{
	// 	{
	// 		Keys: bsonx.Doc{
	// 			{Key: "name", Value: bsonx.String("text")},
	// 			{Key: "description", Value: bsonx.String("text")},
	// 		},
	// 	},
	// }); err != nil && err.(mongo.CommandError).Code != 85 {
	// 	logger.Critical().Msg(err)
	// 	panic(err)
	// }

	// if _, err = MngoCollDatacenter.Indexes().CreateMany(MngoCtx, []mongo.IndexModel{
	// 	{
	// 		Keys: bsonx.Doc{
	// 			{Key: "name", Value: bsonx.String("text")},
	// 			{Key: "description", Value: bsonx.String("text")},
	// 			{Key: "owner", Value: bsonx.String("text")},
	// 		},
	// 	},
	// }); err != nil && err.(mongo.CommandError).Code != 85 {
	// 	logger.Critical().Msg(err)
	// 	panic(err)
	// }
}

func createCollection(new_collection *mongo.Collection)  {

	var err error 

	if _, err = new_collection.Indexes().CreateMany(MngoCtx, []mongo.IndexModel{
		{
			Keys: bsonx.Doc{
				{Key: "description", Value: bsonx.String("text")},
				{Key: "example", Value: bsonx.String("text")},
			},
		},
	}); err != nil {
		var cmdErr mongo.CommandError
		if errors.As(err, &cmdErr) && cmdErr.Code != 85 {
			logger.Fatal().Msg("It failed but I saw it: " + err.Error())
			panic(err)
		} else if !errors.As(err, &cmdErr) {
			logger.Fatal().Msg("Unexpected error: " + err.Error())
			panic(err)
		}
	}

}