前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >聊聊nacos-sdk-go的PushReceiver

聊聊nacos-sdk-go的PushReceiver

作者头像
code4it
发布2020-07-02 14:14:40
4110
发布2020-07-02 14:14:40
举报

本文主要研究一下nacos-sdk-go的PushReceiver

PushReceiver

nacos-sdk-go-v0.3.2/clients/naming_client/push_receiver.go

type PushReceiver struct {
    port        int
    host        string
    hostReactor *HostReactor
}
  • PushReceiver定义了port、host、hostReactor属性

NewPushRecevier

nacos-sdk-go-v0.3.2/clients/naming_client/push_receiver.go

func NewPushRecevier(hostReactor *HostReactor) *PushReceiver {
    pr := PushReceiver{
        hostReactor: hostReactor,
    }
    go pr.startServer()
    return &pr
}
  • NewPushRecevier方法创建PushReceiver,并异步执行pr.startServer()

startServer

nacos-sdk-go-v0.3.2/clients/naming_client/push_receiver.go

func (us *PushReceiver) startServer() {
    var conn *net.UDPConn

    for i := 0; i < 3; i++ {
        r := rand.New(rand.NewSource(time.Now().UnixNano()))
        port := r.Intn(1000) + 54951
        us.port = port
        conn1, ok := us.tryListen()

        if ok {
            conn = conn1
            log.Println("[INFO] udp server start, port: " + strconv.Itoa(port))
            break
        }

        if !ok && i == 2 {
            log.Panicf("failed to start udp server after trying 3 times.")
            //os.Exit(1)  //It is weird dangerous to invoke the os.Exit() as a Middleware.
        }
    }

    defer conn.Close()
    for {
        us.handleClient(conn)
    }
}
  • startServer方法随机执行一个端口,然后执行us.tryListen(),三次不成功则退出,成功则执行us.handleClient(conn)

tryListen

nacos-sdk-go-v0.3.2/clients/naming_client/push_receiver.go

func (us *PushReceiver) tryListen() (*net.UDPConn, bool) {
    addr, err := net.ResolveUDPAddr("udp", us.host+":"+strconv.Itoa(us.port))
    if err != nil {
        log.Printf("[ERROR]: Can't resolve address,err: %s \n", err.Error())
        return nil, false
    }

    conn, err := net.ListenUDP("udp", addr)
    if err != nil {
        log.Printf("Error listening %s:%d,err:%s \n", us.host, us.port, err.Error())
        return nil, false
    }

    return conn, true
}
  • tryListen方法先执行net.ResolveUDPAddr,然后执行net.ListenUDP("udp", addr)

handleClient

nacos-sdk-go-v0.3.2/clients/naming_client/push_receiver.go

func (us *PushReceiver) handleClient(conn *net.UDPConn) {
    data := make([]byte, 4024)
    n, remoteAddr, err := conn.ReadFromUDP(data)
    if err != nil {
        log.Printf("[ERROR]:failed to read UDP msg because of %s \n", err.Error())
        return
    }

    s := utils.TryDecompressData(data[:n])
    log.Println("[INFO] receive push: "+s+" from: ", remoteAddr)

    var pushData PushData
    err1 := json.Unmarshal([]byte(s), &pushData)
    if err1 != nil {
        log.Printf("[ERROR] failed to process push data.err:%s \n", err1.Error())
        return
    }
    ack := make(map[string]string)

    if pushData.PushType == "dom" || pushData.PushType == "service" {
        us.hostReactor.ProcessServiceJson(pushData.Data)

        ack["type"] = "push-ack"
        ack["lastRefTime"] = strconv.FormatInt(pushData.LastRefTime, 10)
        ack["data"] = ""

    } else if pushData.PushType == "dump" {
        ack["type"] = "dump-ack"
        ack["lastRefTime"] = strconv.FormatInt(pushData.LastRefTime, 10)
        ack["data"] = utils.ToJsonString(us.hostReactor.serviceInfoMap)
    } else {
        ack["type"] = "unknow-ack"
        ack["lastRefTime"] = strconv.FormatInt(pushData.LastRefTime, 10)
        ack["data"] = ""
    }

    bs, _ := json.Marshal(ack)
    conn.WriteToUDP(bs, remoteAddr)
}
  • handleClient方法通过conn.ReadFromUDP(data)接受数据,然后通过utils.TryDecompressData解压,再通过json.Unmarshal([]byte(s), &pushData)解析为PushData,之后根据pushData.PushType构造ack数据,最后通过conn.WriteToUDP(bs, remoteAddr)响应回去

小结

PushReceiver定义了port、host、hostReactor属性;它提供了NewPushRecevier、startServer、handleClient等方法

doc

  • push_receiver
本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-06-29,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 码匠的流水账 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • PushReceiver
  • NewPushRecevier
  • startServer
  • tryListen
  • handleClient
  • 小结
  • doc
相关产品与服务
对象存储
对象存储(Cloud Object Storage,COS)是由腾讯云推出的无目录层次结构、无数据格式限制,可容纳海量数据且支持 HTTP/HTTPS 协议访问的分布式存储服务。腾讯云 COS 的存储桶空间无容量上限,无需分区管理,适用于 CDN 数据分发、数据万象处理或大数据计算与分析的数据湖等多种场景。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档