Go/Ocean
1
0 Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improved the startup for a new environment

Browse Source
This commit is contained in:
Thorsten Sommer committed 2015-06-21 20:20:27 +02:00
1 parent 16ac349049
commit e1465c2fd0
7 files changed
+148 -85
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

CustomerDB/ConnectDatabase.go
+64
View File
@@ -0,0 +1,64 @@
package CustomerDB
import (
"fmt"
"github.com/SommerEngineering/Ocean/Log"
LM "github.com/SommerEngineering/Ocean/Log/Meta"
"gopkg.in/mgo.v2"
"os"
)
// Try to connect to the database.
func connectDatabase(host, username, password, database string) {
defer func() {
if err := recover(); err != nil {
fmt.Printf("[Error] Was not able to connect to the customer database: %s. Please read https://github.com/SommerEngineering/Ocean.\n", err)
os.Exit(0)
}
}()
// Connect to MongoDB:
if newSession, errDial := mgo.Dial(host); errDial != nil {
Log.LogFull(senderName, LM.CategorySYSTEM, LM.LevelERROR, LM.SeverityUnknown, LM.ImpactUnknown, LM.MessageNameDATABASE, `It was not possible to connect to the MongoDB host `+host, errDial.Error())
fmt.Printf("[Error] Was not able to connect to the customer database: %s. Please read https://github.com/SommerEngineering/Ocean.\n", errDial.Error())
os.Exit(0)
} else {
mainSession = newSession
}
// Use the correct database:
db := mainSession.DB(database)
if db == nil {
Log.LogFull(senderName, LM.CategorySYSTEM, LM.LevelERROR, LM.SeverityCritical, LM.ImpactCritical, LM.MessageNameDATABASE, `Was not able to get the customer database.`)
fmt.Printf("[Error] Was not able to connect to the customer database. Please read https://github.com/SommerEngineering/Ocean.\n")
os.Exit(0)
}
// Login:
if errLogin := db.Login(username, password); errLogin != nil {
Log.LogFull(senderName, LM.CategorySYSTEM, LM.LevelSECURITY, LM.SeverityUnknown, LM.ImpactUnknown, LM.MessageNameDATABASE, `It was not possible to login the user `+databaseUsername, errLogin.Error())
fmt.Printf("[Error] Was not able to connect to the customer database: %s. Please read https://github.com/SommerEngineering/Ocean.\n", errLogin.Error())
os.Exit(0)
}
// In case of write operations, wait for the majority of servers to be done:
mainSession.SetSafe(&mgo.Safe{WMode: "majority"})
// Set the consistency mode to read from any secondary server and write to the primary.
// Copied sessions can overwrite this setting of necessary.
mainSession.SetMode(mgo.Eventual, true)
// Get the GridFS:
gridFS := db.GridFS(`fs`)
if gridFS == nil {
Log.LogFull(senderName, LM.CategorySYSTEM, LM.LevelERROR, LM.SeverityCritical, LM.ImpactCritical, LM.MessageNameDATABASE, `Was not able to get the GridFS from the database.`)
fmt.Printf("[Error] Was not able to connect to the customer database. Please read https://github.com/SommerEngineering/Ocean.\n")
os.Exit(0)
}
// Ensure the indexes for the GridFS:
filesCollection := gridFS.Files
filesCollection.EnsureIndexKey(`uploadDate`)
filesCollection.EnsureIndexKey(`filename`)
filesCollection.EnsureIndexKey(`filename`, `uploadDate`)
}
CustomerDB/Init.go
+2 -42
View File
@@ -4,7 +4,6 @@ import (
"github.com/SommerEngineering/Ocean/ConfigurationDB"
"github.com/SommerEngineering/Ocean/Log"
LM "github.com/SommerEngineering/Ocean/Log/Meta"
"gopkg.in/mgo.v2"
)
// The init function for this package.
@@ -17,46 +16,7 @@ func init() {
databaseUsername = ConfigurationDB.Read(`CustomerDBUsername`)
databasePassword = ConfigurationDB.Read(`CustomerDBPassword`)
// Connect to MongoDB:
if newSession, errDial := mgo.Dial(databaseHost); errDial != nil {
Log.LogFull(senderName, LM.CategorySYSTEM, LM.LevelERROR, LM.SeverityUnknown, LM.ImpactUnknown, LM.MessageNameDATABASE, `It was not possible to connect to the MongoDB host `+databaseHost, errDial.Error())
return
} else {
mainSession = newSession
}
// Use the correct database:
db := mainSession.DB(databaseDB)
if db == nil {
Log.LogFull(senderName, LM.CategorySYSTEM, LM.LevelERROR, LM.SeverityCritical, LM.ImpactCritical, LM.MessageNameDATABASE, `Was not able to get the customer database.`)
return
}
// Login:
if errLogin := db.Login(databaseUsername, databasePassword); errLogin != nil {
Log.LogFull(senderName, LM.CategorySYSTEM, LM.LevelSECURITY, LM.SeverityUnknown, LM.ImpactUnknown, LM.MessageNameDATABASE, `It was not possible to login the user `+databaseUsername, errLogin.Error())
return
}
// In case of write operations, wait for the majority of servers to be done:
mainSession.SetSafe(&mgo.Safe{WMode: "majority"})
// Set the consistency mode to read from any secondary server and write to the primary.
// Copied sessions can overwrite this setting of necessary.
mainSession.SetMode(mgo.Eventual, true)
// Get the GridFS:
gridFS := db.GridFS(`fs`)
if gridFS == nil {
Log.LogFull(senderName, LM.CategorySYSTEM, LM.LevelERROR, LM.SeverityCritical, LM.ImpactCritical, LM.MessageNameDATABASE, `Was not able to get the GridFS from the database.`)
return
}
// Ensure the indexes for the GridFS:
filesCollection := gridFS.Files
filesCollection.EnsureIndexKey(`uploadDate`)
filesCollection.EnsureIndexKey(`filename`)
filesCollection.EnsureIndexKey(`filename`, `uploadDate`)
// Try to connect to the database:
connectDatabase(databaseHost, databaseUsername, databasePassword, databaseDB)
Log.LogShort(senderName, LM.CategorySYSTEM, LM.LevelINFO, LM.MessageNameDATABASE, `Customer database is now ready.`)
}