-
Notifications
You must be signed in to change notification settings - Fork 0
/
mysql_reader.go
102 lines (81 loc) · 2.26 KB
/
mysql_reader.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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package api
import (
"database/sql"
"encoding/json"
)
import _ "github.com/go-sql-driver/mysql"
type MySqlReader struct {
db *sql.DB
}
func NewMySqlReader() (MySqlReader, error) {
username := GetEnvOrDefault("DB_USERNAME", "")
password := GetEnvOrDefault("DB_PASSWORD", "")
hostname := GetEnvOrDefault("DB_HOST", "localhost")
database := GetEnvOrDefault("DB_NAME", "zonedns")
return BuildMySqlReader(username, password, hostname, database)
}
func BuildMySqlReader(username string, password string, hostname string, database string) (MySqlReader, error) {
db, err := buildMySqlConnection(username, password, hostname, database)
return MySqlReader{db: db}, err
}
func (m *MySqlReader) fetchZoneById(zoneId int64) (z Zone, err error) {
row := m.db.QueryRow("SELECT z_id, z_name, z_a, z_aaaa FROM ZNS_zones WHERE z_id=?", zoneId)
var a []byte
var aaaa []byte
err = row.Scan(&z.id, &z.Name, &a, &aaaa)
if err != nil {
return
}
err = json.Unmarshal(a, &z.A)
if err != nil {
return
}
err = json.Unmarshal(aaaa, &z.AAAA)
if err != nil {
return
}
return
}
func (m *MySqlReader) fetchDomainById(domainId int64) (d Domain, err error) {
// TODO: add MX here?
row := m.db.QueryRow("SELECT d_id, d_zid, d_name, d_txt FROM ZNS_domains WHERE d_id=?", domainId)
err = row.Scan(&d.id, &d.ZoneID, &d.Name, &d.Txt)
return
}
/*
This methods retrieves list of Zones from the database server
*/
func (m *MySqlReader) FetchZones() (z []Zone, err error) {
rows, err := m.db.Query("SELECT z_id, z_name, z_a, z_aaaa from ZNS_zones ORDER BY z_id")
if err != nil {
return nil, err
}
defer rows.Close()
var a []byte
var aaaa []byte
for rows.Next() {
var zone Zone
err = rows.Scan(&zone.id, &zone.Name, &a, &aaaa)
if err != nil {
return
}
err = json.Unmarshal(a, &zone.A)
if err != nil {
return
}
err = json.Unmarshal(aaaa, &zone.AAAA)
if err != nil {
return
}
z = append(z, zone)
}
return
}
/*
This method retrieves Zone id from the database server. This ZoneID can be converted to A or AAAA record.
*/
func (m *MySqlReader) LookupDomain(domain string) (d Domain, err error) {
row := m.db.QueryRow("SELECT d_id, d_zid, d_name, d_txt from ZNS_domains WHERE d_name=? LIMIT 1", domain)
err = row.Scan(&d.id, &d.ZoneID, &d.Name, &d.Txt)
return
}