dingdayu/dnsx

package dns

import (
    "github.com/miekg/dns"

    "github.com/xyser/dnsx/model/dao"
)

// TypeMX query mx
func TypeMX(msg *dns.Msg) error {
    domain := msg.Question[0].Name
    rrs, err := dao.GetRecordByNameAndType(domain, msg.Question[0].Qtype)
    if err != nil {
        return err
    }
    for _, rr := range rrs {
        msg.Answer = append(msg.Answer, &dns.MX{
            Hdr:        dns.RR_Header{Name: msg.Question[0].Name, Rrtype: msg.Question[0].Qtype, Class: dns.ClassINET, Ttl: rr.TTL},
            Preference: uint16(rr.Priority),
            Mx:         rr.Value,
        })
    }
    return nil
}