joker/godns
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add ToLower to all domain fields #24

Browse Source
This commit is contained in:
kenshinx
2016-02-15 12:34:24 +08:00
parent fde6624777
commit 4275900fb2
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
hosts.go
View File

@ -95,6 +95,7 @@ type RedisHosts struct {
} }
func (r *RedisHosts) Get(domain string) ([]string, bool) { func (r *RedisHosts) Get(domain string) ([]string, bool) {
domain = strings.ToLower(domain)
ip, ok := r.hosts[domain] ip, ok := r.hosts[domain]
if ok { if ok {
return strings.Split(ip, ","), true return strings.Split(ip, ","), true
@ -112,7 +113,7 @@ func (r *RedisHosts) Get(domain string) ([]string, bool) {
} }
func (r *RedisHosts) Set(domain, ip string) (bool, error) { func (r *RedisHosts) Set(domain, ip string) (bool, error) {
return r.redis.Hset(r.key, domain, []byte(ip)) return r.redis.Hset(r.key, strings.ToLower(domain), []byte(ip))
} }
func (r *RedisHosts) Refresh() { func (r *RedisHosts) Refresh() {
@ -130,6 +131,7 @@ type FileHosts struct {
} }
func (f *FileHosts) Get(domain string) ([]string, bool) { func (f *FileHosts) Get(domain string) ([]string, bool) {
domain = strings.ToLower(domain)
ip, ok := f.hosts[domain] ip, ok := f.hosts[domain]
if !ok { if !ok {
return nil, false return nil, false
@ -172,7 +174,7 @@ func (f *FileHosts) Refresh() {
continue continue
} }
f.hosts[domain] = ip f.hosts[strings.ToLower(domain)] = ip
} }
logger.Debug("update hosts records from %s", f.file) logger.Debug("update hosts records from %s", f.file)
} }