Chore: clean up code

This commit is contained in:
Dreamacro 2019-05-14 21:35:34 +08:00
parent 0d4a999707
commit 71a08ad8e2
2 changed files with 5 additions and 5 deletions

View File

@ -99,7 +99,7 @@ func (r *Resolver) exchange(servers []*nameserver, m *D.Msg) (msg *D.Msg, err er
if err != nil || msg.Rcode != D.RcodeSuccess {
return
}
in <- &result{Msg: msg, Error: err}
in <- msg
}(server)
}
@ -114,8 +114,7 @@ func (r *Resolver) exchange(servers []*nameserver, m *D.Msg) (msg *D.Msg, err er
return nil, errors.New("All DNS requests failed")
}
resp := elm.(*result)
msg, err = resp.Msg, resp.Error
msg = elm.(*D.Msg)
return
}

View File

@ -57,10 +57,9 @@ func (s *Server) handleFakeIP(r *D.Msg) (msg *D.Msg, err error) {
q := r.Question[0]
cache, _ := s.r.cache.GetWithExpire("fakeip:" + q.String())
cache := s.r.cache.Get("fakeip:" + q.String())
if cache != nil {
msg = cache.(*D.Msg).Copy()
setMsgTTL(msg, 1)
return
}
@ -72,6 +71,8 @@ func (s *Server) handleFakeIP(r *D.Msg) (msg *D.Msg, err error) {
putMsgToCache(s.r.cache, "fakeip:"+q.String(), msg)
putMsgToCache(s.r.cache, ip.String(), msg)
// putMsgToCache depend on msg ttl to set cache expired time, then set msg ref ttl to 1
setMsgTTL(msg, 1)
}()