This commit is contained in:
mr
2024-07-18 12:05:32 +02:00
parent 85fc43368b
commit 1dfff4635e
2 changed files with 12 additions and 6 deletions

View File

@@ -26,7 +26,7 @@ var (
ResourceMap map[string]interface{}
)
func init() {
func Init(collections []string) {
// var baseConfig string
var err error
var conf map[string]string
@@ -52,7 +52,7 @@ func init() {
createClient(lib.GetConfig().MongoURL)
lib.Logger.Info().Msg("Connecting mongo client to db " + DBname)
prepareDB(lib.GetConfig().DCNAME, lib.GetConfig().DBPOINT)
prepareDB(collections, lib.GetConfig().DCNAME, lib.GetConfig().DBPOINT)
lib.Logger.Info().Msg("Database is READY")
@@ -88,13 +88,10 @@ func createClient(MongoURL string) {
}
func prepareDB(dc_name string, db_point string) {
func prepareDB(list_collection []string, dc_name string, db_point string) {
var err error
DBname := dc_name + "-" + db_point
mngoDB = mngoClient.Database(DBname)
list_collection := [...]string{"data", "processing", "storage", "datacenter", "workspace", "schedule", "workflow"}
existingCollections, err = mngoDB.ListCollectionNames(MngoCtx, bson.D{})
if err != nil {
lib.Logger.Fatal().Msg("Error contacting MongoDB\n" + err.Error())