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

ConfigurationDB/ConnectDatabase.go
+51
View File
@@ -0,0 +1,51 @@
package ConfigurationDB
import (
"fmt"
"github.com/SommerEngineering/Ocean/Configuration/Meta"
"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(config Meta.Configuration) {
defer func() {
if err := recover(); err != nil {
fmt.Printf("[Error] Was not able to connect to the configuration database: %s. Please read https://github.com/SommerEngineering/Ocean.\n", err)
os.Exit(0)
}
}()
// Connect to MongoDB:
if newSession, errDial := mgo.Dial(config.ConfigDBHostname); 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 `+config.ConfigDBHostname, errDial.Error())
fmt.Printf("[Error] Was not able to connect to the configuration database: %s. Please read https://github.com/SommerEngineering/Ocean.\n", errDial.Error())
os.Exit(0)
} else {
session = newSession
}
// Use the correct database:
db = session.DB(config.ConfigDBDatabase)
// Login:
if errLogin := db.Login(config.ConfigDBConfigurationCollectionUsername, config.ConfigDBConfigurationCollectionPassword); errLogin != nil {
Log.LogFull(senderName, LM.CategorySYSTEM, LM.LevelSECURITY, LM.SeverityUnknown, LM.ImpactUnknown, LM.MessageNameDATABASE, `It was not possible to login the user `+config.ConfigDBConfigurationCollectionUsername, errLogin.Error())
fmt.Printf("[Error] Was not able to connect to the configuration 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:
session.SetSafe(&mgo.Safe{WMode: "majority"})
// Set the consistency mode to read from any secondary server and write to the primary.
session.SetMode(mgo.Eventual, true)
// Get the collection:
collection = db.C(config.ConfigDBConfigurationCollection)
// Take care about the index:
collection.EnsureIndexKey(`Name`)
}
ConfigurationDB/Init.go
+2 -29
View File
@@ -4,41 +4,14 @@ import (
"github.com/SommerEngineering/Ocean/Configuration"
"github.com/SommerEngineering/Ocean/Log"
LM "github.com/SommerEngineering/Ocean/Log/Meta"
"gopkg.in/mgo.v2"
)
// The init function for this package.
func init() {
config := Configuration.Read()
// Connect to MongoDB:
if newSession, errDial := mgo.Dial(config.ConfigDBHostname); 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 `+config.ConfigDBHostname, errDial.Error())
return
} else {
session = newSession
}
// Use the correct database:
db = session.DB(config.ConfigDBDatabase)
// Login:
if errLogin := db.Login(config.ConfigDBConfigurationCollectionUsername, config.ConfigDBConfigurationCollectionPassword); errLogin != nil {
Log.LogFull(senderName, LM.CategorySYSTEM, LM.LevelSECURITY, LM.SeverityUnknown, LM.ImpactUnknown, LM.MessageNameDATABASE, `It was not possible to login the user `+config.ConfigDBConfigurationCollectionUsername, errLogin.Error())
return
}
// In case of write operations, wait for the majority of servers to be done:
session.SetSafe(&mgo.Safe{WMode: "majority"})
// Set the consistency mode to read from any secondary server and write to the primary.
session.SetMode(mgo.Eventual, true)
// Get the collection:
collection = db.C(config.ConfigDBConfigurationCollection)
// Take care about the index:
collection.EnsureIndexKey(`Name`)
// Connect to the database:
connectDatabase(config)
// Check the system configuration:
checkConfiguration()