diff --git a/database/connect_mongo.go b/database/connect_mongo.go index 8b16f73..350dfda 100644 --- a/database/connect_mongo.go +++ b/database/connect_mongo.go @@ -5,6 +5,7 @@ import ( "os" "time" + "github.com/PretendoNetwork/friends-secure/globals" "go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo/options" ) @@ -21,4 +22,6 @@ func connectMongo() { mongoDatabase = mongoClient.Database("pretendo") MongoCollection = mongoDatabase.Collection("pnids") + + globals.Logger.Success("Connected to Mongo!") } diff --git a/database/connect_postgres.go b/database/connect_postgres.go index 20c9619..b2b2134 100644 --- a/database/connect_postgres.go +++ b/database/connect_postgres.go @@ -19,6 +19,8 @@ func connectPostgres() { globals.Logger.Critical(err.Error()) } + globals.Logger.Success("Connected to Postgres!") + initPostgresWiiU() // TODO: 3DS database } diff --git a/database/init_postgres_wiiu.go b/database/init_postgres_wiiu.go index 157e211..c53526b 100644 --- a/database/init_postgres_wiiu.go +++ b/database/init_postgres_wiiu.go @@ -11,6 +11,8 @@ func initPostgresWiiU() { return } + globals.Logger.Success("[Wii U] Postgres schema created") + _, err = Postgres.Exec(`CREATE TABLE IF NOT EXISTS wiiu.user_data ( pid integer PRIMARY KEY, show_online boolean, @@ -62,4 +64,6 @@ func initPostgresWiiU() { globals.Logger.Critical(err.Error()) return } + + globals.Logger.Success("[Wii U] Postgres tables created") }