forked from fajran/go-monetdb
-
Notifications
You must be signed in to change notification settings - Fork 4
/
driver.go
83 lines (70 loc) · 2.45 KB
/
driver.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
package monetdb
import (
"database/sql"
"database/sql/driver"
"fmt"
"regexp"
"strconv"
)
func init() {
sql.Register("monetdb", &Driver{})
}
type Driver struct {
}
type config struct {
Username string
Password string
Hostname string
Database string
Port int
}
func (*Driver) Open(name string) (driver.Conn, error) {
c, err := parseDSN(name)
if err != nil {
return nil, err
}
return newConn(c)
}
func parseDSN(name string) (config, error) {
re := regexp.MustCompile(`^((?P<username>[^:]+?)(:(?P<password>[^@]+?))?@)?(?P<hostname>[a-zA-Z0-9.]+?)(:(?P<port>\d+?))?/(?P<database>.+?)$`)
ipv6_re := regexp.MustCompile(`^((?P<username>[^:]+?)(:(?P<password>[^@]+?))?@)?\[(?P<hostname>(([0-9a-fA-F]{1,4}:){7,7}[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,7}:|([0-9a-fA-F]{1,4}:){1,6}:[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,5}(:[0-9a-fA-F]{1,4}){1,2}|([0-9a-fA-F]{1,4}:){1,4}(:[0-9a-fA-F]{1,4}){1,3}|([0-9a-fA-F]{1,4}:){1,3}(:[0-9a-fA-F]{1,4}){1,4}|([0-9a-fA-F]{1,4}:){1,2}(:[0-9a-fA-F]{1,4}){1,5}|[0-9a-fA-F]{1,4}:((:[0-9a-fA-F]{1,4}){1,6})|:((:[0-9a-fA-F]{1,4}){1,7}|:)|fe80:(:[0-9a-fA-F]{0,4}){0,4}%[0-9a-zA-Z]{1,}|::(ffff(:0{1,4}){0,1}:){0,1}((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])|([0-9a-fA-F]{1,4}:){1,4}:((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9]))+?)\](:(?P<port>\d+?))?\/(?P<database>.+?)$`)
m := make([]string, 0)
n := make([]string, 0)
if re.MatchString(name) {
m = re.FindAllStringSubmatch(name, -1)[0]
n = re.SubexpNames()
return getConfig(m, n, false), nil
} else if ipv6_re.MatchString(name) {
m = ipv6_re.FindAllStringSubmatch(name, -1)[0]
n = ipv6_re.SubexpNames()
return getConfig(m, n, true), nil
}
return config{}, fmt.Errorf("Invalid DSN")
}
func getConfig(m []string, n []string, ipv6 bool) config {
c := config{
Hostname: "localhost",
Port: 50000,
}
for i, v := range m {
if n[i] == "username" {
c.Username = v
} else if n[i] == "password" {
c.Password = v
} else if n[i] == "hostname" {
if ipv6 {
c.Hostname = fmt.Sprintf("[%s]", v)
continue
}
c.Hostname = v
} else if n[i] == "port" && v != "" {
c.Port, _ = strconv.Atoi(v)
} else if n[i] == "database" {
c.Database = v
}
}
return c
}