feat: Added DSN
This commit is contained in:
parent
b444011d09
commit
212cf19d05
31
dsn.go
Normal file
31
dsn.go
Normal file
@ -0,0 +1,31 @@
|
||||
package gorm
|
||||
|
||||
import (
|
||||
"net/url"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type DSN struct {
|
||||
Host string
|
||||
Port int
|
||||
User string
|
||||
Pass string
|
||||
Db string
|
||||
Options map[string]string
|
||||
}
|
||||
|
||||
func (d DSN) String() string {
|
||||
dsn := d.User + ":" + d.Pass + "@tcp(" + d.Host + ":" + strconv.Itoa(d.Port) + ")/" + d.Db
|
||||
|
||||
if d.Options != nil {
|
||||
value := url.Values{}
|
||||
|
||||
for k, v := range d.Options {
|
||||
value.Add(k, v)
|
||||
}
|
||||
|
||||
dsn += "?" + value.Encode()
|
||||
}
|
||||
|
||||
return dsn
|
||||
}
|
30
dsn_test.go
Normal file
30
dsn_test.go
Normal file
@ -0,0 +1,30 @@
|
||||
package gorm
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestDsn(t *testing.T) {
|
||||
dsn := DSN{
|
||||
Host: "127.0.0.1",
|
||||
Port: 3306,
|
||||
User: "root",
|
||||
Pass: "password",
|
||||
Db: "gorm",
|
||||
}
|
||||
|
||||
t.Run("dsn string", func(t *testing.T) {
|
||||
if dsn.String() != "root:password@tcp(127.0.0.1:3306)/gorm" {
|
||||
t.Error("dsn string error")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("dsn string with options", func(t *testing.T) {
|
||||
dsn.Options = map[string]string{
|
||||
"charset": "utf8mb4",
|
||||
"parseTime": "True",
|
||||
}
|
||||
|
||||
if dsn.String() != "root:password@tcp(127.0.0.1:3306)/gorm?charset=utf8mb4&parseTime=True" {
|
||||
t.Error("dsn string with options error")
|
||||
}
|
||||
})
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user