fix: use mysql to parse dsn (#2838)

Signed-off-by: Zhang Wei <kweizh@gmail.com>
pull/2843/head
Wei Zhang 1 year ago committed by GitHub
parent f8b9a83d4a
commit ddc89029b7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -4,7 +4,6 @@ import (
"context" "context"
"database/sql" "database/sql"
"fmt" "fmt"
"net/url"
"github.com/go-sql-driver/mysql" "github.com/go-sql-driver/mysql"
"github.com/pkg/errors" "github.com/pkg/errors"
@ -24,9 +23,7 @@ func NewDB(profile *profile.Profile) (store.Driver, error) {
// Open MySQL connection with parameter. // Open MySQL connection with parameter.
// multiStatements=true is required for migration. // multiStatements=true is required for migration.
// See more in: https://github.com/go-sql-driver/mysql#multistatements // See more in: https://github.com/go-sql-driver/mysql#multistatements
dsn, err := mergeDSNWithParams(profile.DSN, map[string]string{ dsn, err := mergeDSN(profile.DSN)
"multiStatements": "true",
})
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -113,16 +110,12 @@ func (d *DB) Close() error {
return d.db.Close() return d.db.Close()
} }
func mergeDSNWithParams(baseDSN string, params map[string]string) (string, error) { func mergeDSN(baseDSN string) (string, error) {
parsedDSN, err := url.Parse(baseDSN) config, err := mysql.ParseDSN(baseDSN)
if err != nil { if err != nil {
return "", errors.Wrapf(err, "failed to parse DSN: %s", baseDSN) return "", errors.Wrapf(err, "failed to parse DSN: %s", baseDSN)
} }
existingParams := parsedDSN.Query() config.MultiStatements = true
for key, value := range params { return config.FormatDSN(), nil
existingParams.Add(key, value)
}
parsedDSN.RawQuery = existingParams.Encode()
return parsedDSN.String(), nil
} }

Loading…
Cancel
Save