Merge pull request #91 from deliveryhero/mysql-add-reject-read-only

Mysql add reject read only

README.md

@@ -14,4 +14,4 @@ See more on ```example.go```
 
 # Libraries used
 
-See ```Gopkg.toml```
+See ```go.mod```

fddb/config.go

@@ -32,9 +32,10 @@ type DBConfig struct {
 }
 
 type MysqlOptions struct {
-	Timeout      time.Duration
-	ReadTimeout  time.Duration
-	WriteTimeout time.Duration
+	Timeout        time.Duration
+	ReadTimeout    time.Duration
+	WriteTimeout   time.Duration
+	RejectReadOnly bool
 }
 
 var availableDrivers = map[string]DBConfig{
@@ -148,6 +149,10 @@ func (c DBConfig) mySqlConnString(host, usrPwd string) string {
 		dsnParams = append(dsnParams, "writeTimeout="+c.MysqlOptions.WriteTimeout.String())
 	}
 
+	if c.MysqlOptions.RejectReadOnly {
+		dsnParams = append(dsnParams, "rejectReadOnly=true")
+	}
+
 	if len(dsnParams) > 0 {
 		return fmt.Sprintf("%s@tcp(%s)/%s?%s", usrPwd, host, c.DB, strings.Join(dsnParams, "&"))
 	}

fddb/config_test.go

@@ -1,8 +1,9 @@
 package fddb
 
 import (
-	"github.com/stretchr/testify/assert"
 	"testing"
+
+	"github.com/stretchr/testify/assert"
 )
 
 func TestDBConfig(t *testing.T) {
@@ -153,15 +154,16 @@ func TestDBConfigConnFullString_MySQL(t *testing.T) {
 		User:   "root",
 		DB:     "test",
 		MysqlOptions: MysqlOptions{
-			Timeout:      10000000,
-			ReadTimeout:  20000000,
-			WriteTimeout: 30000000,
+			Timeout:        10000000,
+			ReadTimeout:    20000000,
+			WriteTimeout:   30000000,
+			RejectReadOnly: true,
 		},
 	}
-	assert.Equal(t, c.ConnString(), "root@tcp(127.0.0.1:3306)/test?timeout=10ms&readTimeout=20ms&writeTimeout=30ms")
+	assert.Equal(t, c.ConnString(), "root@tcp(127.0.0.1:3306)/test?timeout=10ms&readTimeout=20ms&writeTimeout=30ms&rejectReadOnly=true")
 
 	c.Password = "r007"
-	assert.Equal(t, c.ConnString(), "root:r007@tcp(127.0.0.1:3306)/test?timeout=10ms&readTimeout=20ms&writeTimeout=30ms")
+	assert.Equal(t, c.ConnString(), "root:r007@tcp(127.0.0.1:3306)/test?timeout=10ms&readTimeout=20ms&writeTimeout=30ms&rejectReadOnly=true")
 }
 
 func TestDBConfigConnString_Postgres(t *testing.T) {