f33f7b5c29
+ Refactored all imports + Fixed a bug for the logging regarding removing \n \t \r + Updated to current MGO release + Changed the name of ICCC
62 lines
2.4 KiB
Go
62 lines
2.4 KiB
Go
package CustomerDB
|
|
|
|
import (
|
|
"github.com/SommerEngineering/Ocean/ConfigurationDB"
|
|
"github.com/SommerEngineering/Ocean/Log"
|
|
LM "github.com/SommerEngineering/Ocean/Log/Meta"
|
|
"gopkg.in/mgo.v2"
|
|
)
|
|
|
|
func init() {
|
|
|
|
Log.LogShort(senderName, LM.CategorySYSTEM, LM.LevelINFO, LM.MessageNameDATABASE, `Init the customer database.`)
|
|
|
|
databaseHost := ConfigurationDB.Read(`CustomerDBHost`)
|
|
databaseDB = ConfigurationDB.Read(`CustomerDBDatabase`)
|
|
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`)
|
|
|
|
Log.LogShort(senderName, LM.CategorySYSTEM, LM.LevelINFO, LM.MessageNameDATABASE, `Customer database is now ready.`)
|
|
}
|