diff --git a/configs/config.go b/configs/config.go index 204e71c..445e617 100644 --- a/configs/config.go +++ b/configs/config.go @@ -23,6 +23,7 @@ type MuffinConfig struct { Bot botConfig Train trainConfig DatabaseURL string + DBName string } func loadConfig() *MuffinConfig { @@ -57,6 +58,7 @@ func setConfig(config *MuffinConfig) { config.Train.UserID = getValue("TRAIN_USER_ID") config.DatabaseURL = getRequiredValue("DATABASE_URL") + config.DBName = getRequiredValue("DATABASE_NAME") } var Config *MuffinConfig = loadConfig() diff --git a/databases/Learn.go b/databases/Learn.go index 8aeddde..f0d25aa 100644 --- a/databases/Learn.go +++ b/databases/Learn.go @@ -3,6 +3,7 @@ package databases import ( "time" + "git.wh64.net/muffin/goMuffin/configs" "go.mongodb.org/mongo-driver/v2/bson" "go.mongodb.org/mongo-driver/v2/mongo" ) @@ -22,4 +23,4 @@ type Learn struct { CreatedAt time.Time `bson:"created_at"` } -var Learns *mongo.Collection = Client.Database("muffin_ai_test").Collection("learn") +var Learns *mongo.Collection = Client.Database(configs.Config.DBName).Collection("learn") diff --git a/databases/Text.go b/databases/Text.go index 295d688..06dc147 100644 --- a/databases/Text.go +++ b/databases/Text.go @@ -3,6 +3,7 @@ package databases import ( "time" + "git.wh64.net/muffin/goMuffin/configs" "go.mongodb.org/mongo-driver/v2/bson" "go.mongodb.org/mongo-driver/v2/mongo" ) @@ -20,4 +21,4 @@ type Text struct { CreatedAt time.Time `bson:"created_at"` } -var Texts *mongo.Collection = Client.Database("muffin_ai_test").Collection("text") +var Texts *mongo.Collection = Client.Database(configs.Config.DBName).Collection("text")