joker/godns
1
0
Fork 0
Code Issues Pull Requests 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
Show Stats 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) {
domain = strings.ToLower(domain)
ip, ok := r.hosts[domain]
if ok {
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) {
return r.redis.Hset(r.key, domain, []byte(ip))
return r.redis.Hset(r.key, strings.ToLower(domain), []byte(ip))
}
func (r *RedisHosts) Refresh() {
@ -130,6 +131,7 @@ type FileHosts struct {
}
func (f *FileHosts) Get(domain string) ([]string, bool) {
domain = strings.ToLower(domain)
ip, ok := f.hosts[domain]
if !ok {
return nil, false
@ -172,7 +174,7 @@ func (f *FileHosts) Refresh() {
continue
}
f.hosts[domain] = ip
f.hosts[strings.ToLower(domain)] = ip
}
logger.Debug("update hosts records from %s", f.file)
}