initial commit

2023-03-03 14:43:11 +01:00
parent 7229007847
commit 88c21d1828
142 changed files with 13975 additions and 22 deletions
--- a/services/mongo.go
+++ b/services/mongo.go
@@ -0,0 +1,157 @@
+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")
+
+}