Merge pull request #78 from Scalingo/feat/SRE-570/implement_MySQL_probe

feat(mysqlprobe): implement MySQL probe

Author: stopostit

CHANGELOG.md

@@ -2,6 +2,8 @@
 
 ## To Be Released
 
+* feat(mysqlprobe): implement MySQL probe
+
 ## v4.4.5
 
 * feat(prober): change the exposed CheckOneProbe to make it usable outside of this package

go.mod

@@ -5,10 +5,12 @@ go 1.17
 require (
 	github.com/Scalingo/go-utils/logger v1.1.0
 	github.com/fsouza/go-dockerclient v1.7.11
+	github.com/go-sql-driver/mysql v1.6.0
 	github.com/gorilla/mux v1.8.0
 	github.com/goware/httpmock v0.0.0-20150807175315-4fdb484c3e70
 	github.com/lib/pq v1.10.6
 	github.com/ncw/swift v1.0.53
+	github.com/pkg/errors v0.9.1
 	github.com/sirupsen/logrus v1.8.1
 	github.com/stretchr/testify v1.7.1
 	go.etcd.io/etcd/client/v2 v2.305.4
@@ -41,7 +43,6 @@ require (
 	github.com/opencontainers/go-digest v1.0.0 // indirect
 	github.com/opencontainers/image-spec v1.0.2 // indirect
 	github.com/opencontainers/runc v1.1.0 // indirect
-	github.com/pkg/errors v0.9.1 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
 	go.etcd.io/etcd/api/v3 v3.5.4 // indirect
 	go.etcd.io/etcd/client/pkg/v3 v3.5.4 // indirect

go.sum

@@ -386,6 +386,8 @@ github.com/go-openapi/swag v0.0.0-20160704191624-1d0bd113de87/go.mod h1:DXUve3Dp
 github.com/go-openapi/swag v0.19.2/go.mod h1:POnQmlKehdgb5mhVOsnJFsivZCEZ/vjK9gh66Z9tfKk=
 github.com/go-openapi/swag v0.19.5/go.mod h1:POnQmlKehdgb5mhVOsnJFsivZCEZ/vjK9gh66Z9tfKk=
 github.com/go-openapi/swag v0.19.14/go.mod h1:QYRuS/SOXUCsnplDa677K7+DxSOj6IPNl/eQntq43wQ=
+github.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE=
+github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg=
 github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY=
 github.com/go-task/slim-sprig v0.0.0-20210107165309-348f09dbbbc0/go.mod h1:fyg7847qk6SyHyPtNmDHnmrv/HOrqktSC+C9fM+CJOE=
 github.com/godbus/dbus v0.0.0-20151105175453-c7fdd8b5cd55/go.mod h1:/YcGZj5zSblfDWMMoOzV4fas9FZnQYTkDnsGvmh2Grw=

mysqlprobe/mysqlprobe.go

@@ -0,0 +1,44 @@
+package mysqlprobe
+
+import (
+	"database/sql"
+
+	// Mandatory for sql.Open to work
+	_ "github.com/go-sql-driver/mysql"
+
+	errgo "gopkg.in/errgo.v1"
+)
+
+type MySQLProbe struct {
+	name             string
+	connectionString string
+}
+
+// NewMySQLProbe instantiate a new MySQL probe:
+// - name: probe name
+// - connectionString: connection string with the form "mysql://username:[email protected]"
+func NewMySQLProbe(name, connectionString string) MySQLProbe {
+	return MySQLProbe{
+		name:             name,
+		connectionString: connectionString,
+	}
+}
+
+func (p MySQLProbe) Name() string {
+	return p.name
+}
+
+func (p MySQLProbe) Check() error {
+	client, err := sql.Open("mysql", p.connectionString)
+	if err != nil {
+		return errgo.Notef(err, "fail to open a new connection to MySQL")
+	}
+	defer client.Close()
+
+	err = client.Ping()
+	if err != nil {
+		return errgo.Notef(err, "unable to contact MySQL host")
+	}
+
+	return nil
+}