使用Go語言實現(xiàn)心跳機制

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

心跳最典型的應用場景是是探測服務是否存活,這篇文章主要來和大家介紹一下如何使用Go語言實現(xiàn)一個簡單的心跳程序,感興趣的可以了解下

Go 實現(xiàn)心跳

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

Go 中實現(xiàn)心跳機制可以通過 time.NewTimeTicker()，配合 channel 使用，就可以實現(xià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 {
		// 上報心跳包的邏輯，可以在這里實現(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ā)送心跳請求, 并通過重試機制。判斷重試X次失敗認為服務離線服務端響應心跳請求，通過超時機制。超時X秒未收到心跳則判斷客戶端離線

package main

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

func main() {
	// 啟動服務端
	go startServer()

	// 啟動客戶端
	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))

	// 啟動一個 goroutine，定時發(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語言實現(xiàn)心跳機制的文章就介紹到這了,更多相關(guān)Go心跳內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: