Merge pull request #8 from hellofresh/feature/checker-mongo

Feature/checker mongo

README.md

@@ -6,6 +6,7 @@
   * PostgreSQL
   * Redis
   * HTTP
+  * MongoDB
 
 ## Usage
 

checks/http/check.go

@@ -24,15 +24,15 @@ type Config struct {
 // - getting response status from defined URL
 // - verifying that status code is less than 500
 func New(config Config) func() error {
-	return func() error {
-		if config.LogFunc == nil {
-			config.LogFunc = func(err error, details string, extra ...interface{}) {}
-		}
+	if config.LogFunc == nil {
+		config.LogFunc = func(err error, details string, extra ...interface{}) {}
+	}
 
-		if config.RequestTimeout == 0 {
-			config.RequestTimeout = time.Second * 5
-		}
+	if config.RequestTimeout == 0 {
+		config.RequestTimeout = time.Second * 5
+	}
 
+	return func() error {
 		req, err := http.NewRequest(http.MethodGet, config.URL, nil)
 		if err != nil {
 			config.LogFunc(err, "Creating the request for the health check failed")

checks/mongo/check.go

@@ -0,0 +1,38 @@
+package mongo
+
+import "gopkg.in/mgo.v2"
+
+// Config is the MongoDB checker configuration settings container.
+type Config struct {
+	// DSN is the MongoDB instance connection DSN. Required.
+	DSN string
+	// LogFunc is the callback function for errors logging during check.
+	// If not set logging is skipped.
+	LogFunc func(err error, details string, extra ...interface{})
+}
+
+// New creates new MongoDB health check that verifies the following:
+// - connection establishing
+// - doing the ping command
+func New(config Config) func() error {
+	if config.LogFunc == nil {
+		config.LogFunc = func(err error, details string, extra ...interface{}) {}
+	}
+
+	return func() error {
+		session, err := mgo.Dial(config.DSN)
+		if err != nil {
+			config.LogFunc(err, "MongoDB health check failed during connect")
+			return err
+		}
+		defer session.Close()
+
+		err = session.Ping()
+		if err != nil {
+			config.LogFunc(err, "MongoDB health check failed during ping")
+			return err
+		}
+
+		return nil
+	}
+}

checks/postgres/check.go

@@ -32,11 +32,11 @@ type Config struct {
 // - selecting inserted row
 // - deleting inserted row
 func New(config Config) func() error {
-	return func() error {
-		if config.LogFunc == nil {
-			config.LogFunc = func(err error, details string, extra ...interface{}) {}
-		}
+	if config.LogFunc == nil {
+		config.LogFunc = func(err error, details string, extra ...interface{}) {}
+	}
 
+	return func() error {
 		db, err := sql.Open("postgres", config.DSN)
 		if err != nil {
 			config.LogFunc(err, "PostgreSQL health check failed during connect")

checks/rabbitmq/check.go

@@ -41,32 +41,31 @@ type Config struct {
 // - publishing a message to the exchange with the defined routing key
 // - consuming published message
 func New(config Config) func() error {
-	return func() error {
-		if config.LogFunc == nil {
-			config.LogFunc = func(err error, details string, extra ...interface{}) {}
-		}
+	if config.LogFunc == nil {
+		config.LogFunc = func(err error, details string, extra ...interface{}) {}
+	}
 
-		if config.Exchange == "" {
-			config.Exchange = defaultExchange
-		}
+	if config.Exchange == "" {
+		config.Exchange = defaultExchange
+	}
 
-		if config.RoutingKey == "" {
-			host, err := os.Hostname()
-			if nil != err {
-				config.LogFunc(err, "RabbitMQ health check failed on settings default value for unset routing key")
-				return err
-			}
-			config.RoutingKey = host
+	if config.RoutingKey == "" {
+		host, err := os.Hostname()
+		if nil != err {
+			config.RoutingKey = "-unknown-"
 		}
+		config.RoutingKey = host
+	}
 
-		if config.Queue == "" {
-			config.Queue = fmt.Sprintf("%s.%s", config.Exchange, config.RoutingKey)
-		}
+	if config.Queue == "" {
+		config.Queue = fmt.Sprintf("%s.%s", config.Exchange, config.RoutingKey)
+	}
 
-		if config.ConsumeTimeout == 0 {
-			config.ConsumeTimeout = time.Second * 3
-		}
+	if config.ConsumeTimeout == 0 {
+		config.ConsumeTimeout = time.Second * 3
+	}
 
+	return func() error {
 		conn, err := amqp.Dial(config.DSN)
 		if err != nil {
 			config.LogFunc(err, "RabbitMQ health check failed on dial phase")

checks/redis/check.go

@@ -20,11 +20,11 @@ type Config struct {
 // - connection establishing
 // - doing the PING command and verifying the response
 func New(config Config) func() error {
-	return func() error {
-		if config.LogFunc == nil {
-			config.LogFunc = func(err error, details string, extra ...interface{}) {}
-		}
+	if config.LogFunc == nil {
+		config.LogFunc = func(err error, details string, extra ...interface{}) {}
+	}
 
+	return func() error {
 		pool := &redis.Pool{
 			MaxIdle:     1,
 			IdleTimeout: 10 * time.Second,