使用Go語(yǔ)言實(shí)現(xiàn)心跳機(jī)制

更新時(shí)間：2024年01月31日 15:11:47 作者：banjming

心跳最典型的應(yīng)用場(chǎng)景是是探測(cè)服務(wù)是否存活,這篇文章主要來(lái)和大家介紹一下如何使用Go語(yǔ)言實(shí)現(xiàn)一個(gè)簡(jiǎn)單的心跳程序,感興趣的可以了解下

Go 實(shí)現(xiàn)心跳

心跳最典型的應(yīng)用場(chǎng)景是是探測(cè)服務(wù)是否存活，比如在 Zookeeper 中，會(huì)使用心跳探測(cè)服務(wù)是否存貨，如果服務(wù)已經(jīng)死亡，會(huì)將服務(wù)從注冊(cè)表中刪除，避免服務(wù)請(qǐng)求路由到一個(gè)已經(jīng)宕機(jī)的服務(wù)中。

Go 中實(shí)現(xiàn)心跳機(jī)制可以通過(guò) time.NewTimeTicker()，配合 channel 使用，就可以實(shí)現(xiàn)一個(gè)簡(jiǎn)單的心跳程序：

import (
	"code.byted.org/gopkg/logs"
	"context"
	"fmt"
	"testing"
	"time"
)

func sendHeartbeat(heartbeatChan chan<- time.Time) {
	ticker := time.NewTicker(time.Second)
	defer ticker.Stop()
	for {
		select {
		case t := <-ticker.C:
			heartbeatChan <- t
		}
	}
}

func TestHeartbeat(t *testing.T) {

	heartbeatChan := make(chan time.Time)
	go doWork([]int{1, 2, 3, 5, 5}, heartbeatChan)
	for t := range heartbeatChan {
		// 上報(bào)心跳包的邏輯，可以在這里實(shí)現(xiàn)
		fmt.Println("Received heartbeat at", t)
	}
    fmt.Println("heartbeat finished")
}

func doWork(nums []int, heartbeatChan chan time.Time) {
	go sendHeartbeat(heartbeatChan)
	defer func() {
		close(heartbeatChan)
	}()
	for num := range nums {
        time.Sleep(1 * time.Second)
		fmt.Println(num)
	}
}

執(zhí)行結(jié)果如下：

=== RUN TestHeartbeat
0
Received heartbeat at 2024-01-31 11:30:46.3363 +0800 CST m=+1.002830252
Received heartbeat at 2024-01-31 11:30:47.335975 +0800 CST m=+2.002513164
1
Received heartbeat at 2024-01-31 11:30:48.336252 +0800 CST m=+3.002795914
2
Received heartbeat at 2024-01-31 11:30:49.33622 +0800 CST m=+4.002768883
3
Received heartbeat at 2024-01-31 11:30:50.336222 +0800 CST m=+5.002775378
4
heartbeat finished
--- PASS: TestHeartbeat (5.00s)
PASS

心跳程序

客戶端發(fā)送心跳請(qǐng)求, 并通過(guò)重試機(jī)制。判斷重試X次失敗認(rèn)為服務(wù)離線服務(wù)端響應(yīng)心跳請(qǐng)求，通過(guò)超時(shí)機(jī)制。超時(shí)X秒未收到心跳則判斷客戶端離線

package main

import (
	"fmt"
	"net"
	"time"
)

func main() {
	// 啟動(dòng)服務(wù)端
	go startServer()

	// 啟動(dòng)客戶端
	go startClient()

	// 保持 main goroutine 活躍，避免程序退出
	select {}
}

func startServer() {
	ln, err := net.Listen("tcp", ":8080")
	if err != nil {
		fmt.Println(err)
		return
	}
	defer ln.Close()

	for {
		conn, err := ln.Accept()
		if err != nil {
			fmt.Println(err)
			continue
		}
		go handleConnection(conn)
	}
}

func handleConnection(conn net.Conn) {
	// 使用 bufio 包中的 ReadWriter 類型，方便讀寫字符串
	rw := bufio.NewReadWriter(bufio.NewReader(conn), bufio.NewWriter(conn))

	// 啟動(dòng)一個(gè) goroutine，定時(shí)發(fā)送心跳消息
	go func() {
		for {
			time.Sleep(time.Second)
			if _, err := rw.WriteString("heartbeat\n"); err != nil {
				fmt.Println(err)
				return
			}
			if err := rw.Flush(); err != nil {
				fmt.Println(err)
				return
			}
		}
	}()

	// 循環(huán)讀取客戶端發(fā)送的消息
	for {
		line, err := rw.ReadString('\n')
		if err != nil {
			fmt.Println(err)
			return
		}
		fmt.Println("received:", line)
	}
}

func startClient() {
	conn, err := net.Dial("tcp", "localhost:8080")
	if err != nil {
		fmt.Println(err)
		return
	}
	defer conn.Close()

	//
}

到此這篇關(guān)于使用Go語(yǔ)言實(shí)現(xiàn)心跳機(jī)制的文章就介紹到這了,更多相關(guān)Go心跳內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: