env fix

config/database.go

@@ -3,6 +3,7 @@ package config
 import (
 	"context"
 	"log"
+	"os"
 	"time"
 
 	"go.mongodb.org/mongo-driver/mongo"
@@ -14,17 +15,22 @@ var (
 )
 
 func ConnectDB() {
-	client, err := mongo.NewClient(options.Client().ApplyURI("mongodb://localhost:27017"))
+	mongoURL := os.Getenv("MONGO_URL")
+	if mongoURL == "" {
+		log.Fatal("MONGO_URL is not set in environment variables")
+	}
+
+	client, err := mongo.NewClient(options.Client().ApplyURI(mongoURL))
 	if err != nil {
-		log.Fatal(err)
+		log.Fatal("Error creating MongoDB client: ", err)
 	}
 
 	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
 	defer cancel()
 
 	err = client.Connect(ctx)
 	if err != nil {
-		log.Fatal(err)
+		log.Fatal("Error connecting to MongoDB: ", err)
 	}
 
 	MongoDB = client.Database("testdb")

invitation_seed.go

@@ -0,0 +1,42 @@
+package main
+
+import (
+	"context"
+	"log"
+	"os"
+	"time"
+
+	"github.com/rajivgeraev/invitations/config"
+	"github.com/rajivgeraev/invitations/models"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+)
+
+func main() {
+	// Подключение к базе данных
+	mongoURL := os.Getenv("MONGO_URL")
+	if mongoURL == "" {
+		log.Fatal("MONGO_URL is not set in environment variables")
+	}
+
+	config.ConnectDB()
+
+	collection := config.MongoDB.Collection("invitations")
+
+	invitations := []models.Invitation{
+		{ID: primitive.NewObjectID(), Code: "twitter-reg1", MaxUses: 100, UsedCount: 0, Emails: []string{}},
+		{ID: primitive.NewObjectID(), Code: "telegram-test", MaxUses: 100, UsedCount: 0, Emails: []string{}},
+		{ID: primitive.NewObjectID(), Code: "instagram-hello", MaxUses: 100, UsedCount: 0, Emails: []string{}},
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+
+	for _, invitation := range invitations {
+		_, err := collection.InsertOne(ctx, invitation)
+		if err != nil {
+			log.Fatal("Error inserting invitation: ", err)
+		}
+	}
+
+	log.Println("Data seeded successfully!")
+}

main.go

@@ -3,21 +3,25 @@ package main
 import (
 	"log"
 	"net/http"
+	"os"
 
 	"github.com/gorilla/mux"
-	"github.com/joho/godotenv"
-
+	"github.com/rajivgeraev/invitations/config"
 	"github.com/rajivgeraev/invitations/routes"
 )
 
 func main() {
-	if err := godotenv.Load(); err != nil {
-		log.Fatal("Error loading .env file")
+	// Подключение к базе данных
+	mongoURL := os.Getenv("MONGO_URL")
+	if mongoURL == "" {
+		log.Fatal("MONGO_URL is not set in environment variables")
 	}
 
-	router := mux.NewRouter()
+	config.ConnectDB()
 
+	router := mux.NewRouter()
 	routes.InitializeRoutes(router)
 
+	log.Println("Server started on port 8080")
 	log.Fatal(http.ListenAndServe(":8080", router))
 }