deprecated-oc-catalog/services/mongo.go
2023-03-03 14:43:11 +01:00

158 lines
3.9 KiB
Go

package services
import (
"context"
"os"
"time"
"github.com/beego/beego/v2/core/logs"
beego "github.com/beego/beego/v2/server/web"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"go.mongodb.org/mongo-driver/x/bsonx"
)
type MongoCollectionNames struct {
DATA string
COMPUTING string
STORAGE string
DATACENTER string
WORKSPACE string
SCHEDULE string
}
var (
mngoClient *mongo.Client
mngoDB *mongo.Database
MngoCtx context.Context
MngoNamesCollection = MongoCollectionNames{
DATA: "datas",
COMPUTING: "computings",
STORAGE: "storages",
DATACENTER: "datacenters",
WORKSPACE: "WORKSPACE",
SCHEDULE: "SCHEDULE",
}
MngoCollData *mongo.Collection
MngoCollComputing *mongo.Collection
MngoCollStorage *mongo.Collection
MngoCollDatacenter *mongo.Collection
MngoCollWorkspace *mongo.Collection
MngoCollSchedule *mongo.Collection
)
// func GetMongoDBclient() *mongo.Client {
// return mongoDBclient
// }
func MongoDisconnect() {
if err := mngoClient.Disconnect(MngoCtx); err != nil {
panic(err)
}
}
func Mongoinit(DBname string) {
var baseConfig string
if len(os.Getenv("DOCKER_ENVIRONMENT")) == 0 {
baseConfig = "mongodb"
} else {
baseConfig = "mongodb_docker"
}
mongoURI, err := beego.AppConfig.String(baseConfig + "::url")
if err != nil {
logs.Critical("MongoDB URI error: %v", err)
panic(err)
}
logs.Info("Connecting to %v", mongoURI)
clientOptions := options.Client().ApplyURI(mongoURI)
// mngoClient, err = mongo.NewClient(options.Client().ApplyURI(mongoURI))
// if err = mngoClient.Connect(MngoCtx); err != nil {
// logs.Critical("Mongodb NewClient %v: %v", mongoURI, err)
// panic(err)
// }
MngoCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
// Ping the primary
if mngoClient, err = mongo.Connect(MngoCtx, clientOptions); err != nil {
logs.Critical("Mongodb Connect %v: %v", mongoURI, err)
panic(err)
}
if err = mngoClient.Ping(MngoCtx, nil); err != nil {
logs.Critical("Mongodb Ping %v: %v", mongoURI, err)
panic(err)
}
mngoDB = mngoClient.Database(DBname)
MngoCollData = mngoDB.Collection(MngoNamesCollection.DATA)
MngoCollComputing = mngoDB.Collection(MngoNamesCollection.COMPUTING)
MngoCollStorage = mngoDB.Collection(MngoNamesCollection.STORAGE)
MngoCollDatacenter = mngoDB.Collection(MngoNamesCollection.DATACENTER)
MngoCollWorkspace = mngoDB.Collection(MngoNamesCollection.WORKSPACE)
MngoCollSchedule = mngoDB.Collection(MngoNamesCollection.SCHEDULE)
if _, err = MngoCollData.Indexes().CreateMany(MngoCtx, []mongo.IndexModel{
{
Keys: bsonx.Doc{
{Key: "description", Value: bsonx.String("text")},
{Key: "example", Value: bsonx.String("text")},
},
},
}); err != nil && err.(mongo.CommandError).Code != 85 {
logs.Critical(err)
panic(err)
}
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 {
logs.Critical(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 {
logs.Critical(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 {
logs.Critical(err)
panic(err)
}
logs.Info("Database is READY")
}