core/oc-lib
6
0
Fork 0
Code Issues 7 Pull Requests Packages Projects Releases Wiki Activity

module name

Browse Source
This commit is contained in:
mr
2024-07-18 13:35:14 +02:00
parent 7d5bdadbef
commit 1999213f8b
19 changed files with 103 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
dbs/mongo/mongo.go
View File

@@ -4,11 +4,14 @@ import (
"context"
"encoding/json"
"errors"
lib "oc-lib"
"oc-lib/dbs"
"os"
"time"
lib "cloud.o-forge.io/core/oc-lib"
"cloud.o-forge.io/core/oc-lib/dbs"
"cloud.o-forge.io/core/oc-lib/logs"
"github.com/rs/zerolog"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo"
@@ -26,39 +29,45 @@ var (
ResourceMap map[string]interface{}
)
func Init(collections []string) {
var MONGOService = MongoDB{}
type MongoDB struct {
Logger zerolog.Logger
}
func (m *MongoDB) Init(collections []string) {
// var baseConfig string
var err error
var conf map[string]string
var DBname string
m.Logger = logs.CreateLogger("oc-lib", "")
ResourceMap = make(map[string]interface{})
lib.Logger = lib.CreateLogger("oclib", "")
logger := logs.CreateLogger("oclib", "")
db_conf, err := os.ReadFile("tests/oclib_conf.json")
if err != nil {
lib.Logger.Fatal().Msg("Could not find configuration file")
logger.Fatal().Msg("Could not find configuration file")
}
json.Unmarshal(db_conf, &conf)
DBname = lib.GetConfig().DCNAME + "-" + lib.GetConfig().DBPOINT
lib.Logger.Info().Msg("Connecting to" + lib.GetConfig().MongoURL)
logger.Info().Msg("Connecting to" + lib.GetConfig().MongoURL)
MngoCtx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
createClient(lib.GetConfig().MongoURL)
m.createClient(lib.GetConfig().MongoURL)
lib.Logger.Info().Msg("Connecting mongo client to db " + DBname)
prepareDB(collections, lib.GetConfig().DCNAME, lib.GetConfig().DBPOINT)
logger.Info().Msg("Connecting mongo client to db " + DBname)
m.prepareDB(collections, lib.GetConfig().DCNAME, lib.GetConfig().DBPOINT)
lib.Logger.Info().Msg("Database is READY")
logger.Info().Msg("Database is READY")
}
func createClient(MongoURL string) {
func (m *MongoDB) createClient(MongoURL string) {
var err error
@@ -71,30 +80,30 @@ func createClient(MongoURL string) {
clientOptions := options.Client().ApplyURI(MongoURL).SetBSONOptions(bsonOpts)
mngoClient, err = mongo.Connect(MngoCtx, clientOptions)
if err != nil {
lib.Logger.Fatal().Msg("Mongodb NewClient " + MongoURL + ":" + "err")
m.Logger.Fatal().Msg("Mongodb NewClient " + MongoURL + ":" + "err")
panic(err)
}
// Ping the primary
if mngoClient, err = mongo.Connect(MngoCtx, clientOptions); err != nil {
lib.Logger.Fatal().Msg("Mongodb connect " + MongoURL + ":" + "err")
m.Logger.Fatal().Msg("Mongodb connect " + MongoURL + ":" + "err")
panic(err)
}
if err = mngoClient.Ping(MngoCtx, nil); err != nil {
lib.Logger.Fatal().Msg("Mongodb ping " + MongoURL + ":" + "err")
m.Logger.Fatal().Msg("Mongodb ping " + MongoURL + ":" + "err")
panic(err)
}
}
func prepareDB(list_collection []string, dc_name string, db_point string) {
func (m *MongoDB) prepareDB(list_collection []string, dc_name string, db_point string) {
var err error
DBname := dc_name + "-" + db_point
mngoDB = mngoClient.Database(DBname)
existingCollections, err = mngoDB.ListCollectionNames(MngoCtx, bson.D{})
if err != nil {
lib.Logger.Fatal().Msg("Error contacting MongoDB\n" + err.Error())
m.Logger.Fatal().Msg("Error contacting MongoDB\n" + err.Error())
}
collectionMap := make(map[string]bool)
for _, name := range existingCollections {
@@ -105,17 +114,16 @@ func prepareDB(list_collection []string, dc_name string, db_point string) {
for _, collection_name := range list_collection {
new_collection := mngoDB.Collection(collection_name)
if _, exists := collectionMap[collection_name]; !exists {
createCollection(collection_name, new_collection)
m.createCollection(collection_name, new_collection)
} else {
CollectionMap[collection_name] = new_collection
}
}
}
// Creates the collection with index specified in mongo/mongo_collections
// or use the basic collection creation function
func createCollection(collection_name string, new_collection *mongo.Collection) {
func (m *MongoDB) createCollection(collection_name string, new_collection *mongo.Collection) {
var err error
CollectionMap[collection_name] = new_collection
_, exists := IndexesMap[collection_name]
@@ -123,10 +131,10 @@ func createCollection(collection_name string, new_collection *mongo.Collection)
if _, err = new_collection.Indexes().CreateMany(MngoCtx, IndexesMap[collection_name]); err != nil {
var cmdErr mongo.CommandError
if errors.As(err, &cmdErr) && cmdErr.Code != 85 {
lib.Logger.Fatal().Msg("Error creating indexes for " + collection_name + " collection : \n" + err.Error())
m.Logger.Fatal().Msg("Error creating indexes for " + collection_name + " collection : \n" + err.Error())
panic(err)
} else if !errors.As(err, &cmdErr) {
lib.Logger.Fatal().Msg("Unexpected error: " + err.Error())
m.Logger.Fatal().Msg("Unexpected error: " + err.Error())
panic(err)
}
}
@@ -136,7 +144,7 @@ func createCollection(collection_name string, new_collection *mongo.Collection)
}
func DeleteOne(id string, collection_name string) (int64, error) {
func (m *MongoDB) DeleteOne(id string, collection_name string) (int64, error) {
filter := bson.M{"_id": GetObjIDFromString(id)}
targetDBCollection := CollectionMap[collection_name]
opts := options.Delete().SetHint(bson.D{{Key: "_id", Value: 1}})
@@ -145,13 +153,13 @@ func DeleteOne(id string, collection_name string) (int64, error) {
result, err := targetDBCollection.DeleteOne(MngoCtx, filter, opts)
if err != nil {
lib.Logger.Error().Msg("Couldn't insert resource: " + err.Error())
m.Logger.Error().Msg("Couldn't insert resource: " + err.Error())
return 0, err
}
return result.DeletedCount, nil
}
func UpdateOne(set map[string]interface{}, id string, collection_name string) (string, error) {
func (m *MongoDB) UpdateOne(set map[string]interface{}, id string, collection_name string) (string, error) {
filter := bson.M{"_id": GetObjIDFromString(id)}
targetDBCollection := CollectionMap[collection_name]
MngoCtx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
@@ -159,13 +167,13 @@ func UpdateOne(set map[string]interface{}, id string, collection_name string) (s
result, err := targetDBCollection.UpdateOne(MngoCtx, filter, dbs.InputToBson(set, true))
if err != nil {
lib.Logger.Error().Msg("Couldn't insert resource: " + err.Error())
m.Logger.Error().Msg("Couldn't insert resource: " + err.Error())
return "", err
}
return result.UpsertedID.(primitive.ObjectID).Hex(), nil
}
func StoreOne(obj interface{}, collection_name string) (string, error) {
func (m *MongoDB) StoreOne(obj interface{}, collection_name string) (string, error) {
targetDBCollection := CollectionMap[collection_name]
MngoCtx, cancel = context.WithTimeout(context.Background(), 10*time.Second)
@@ -173,14 +181,14 @@ func StoreOne(obj interface{}, collection_name string) (string, error) {
result, err := targetDBCollection.InsertOne(MngoCtx, obj)
if err != nil {
lib.Logger.Error().Msg("Couldn't insert resource: " + err.Error())
m.Logger.Error().Msg("Couldn't insert resource: " + err.Error())
return "", err
}
return result.InsertedID.(primitive.ObjectID).Hex(), nil
}
func LoadOne(id string, collection_name string) (res *mongo.SingleResult, err error) {
func (m *MongoDB) LoadOne(id string, collection_name string) (res *mongo.SingleResult, err error) {
filter := bson.M{"_id": GetObjIDFromString(id)}
targetDBCollection := CollectionMap[collection_name]
@@ -189,7 +197,7 @@ func LoadOne(id string, collection_name string) (res *mongo.SingleResult, err er
res = targetDBCollection.FindOne(MngoCtx, filter)
if res.Err() != nil {
lib.Logger.Error().Msg("Couldn't find resource " + id + ". Error : " + res.Err().Error())
m.Logger.Error().Msg("Couldn't find resource " + id + ". Error : " + res.Err().Error())
err = res.Err()
return nil, err
}