快捷導(dǎo)航

詳解如何為Go中的無限循環(huán)添加時間限制

更新時間：2024年04月21日 08:53:45 作者：王義杰

在 Go 語言的開發(fā)過程中,我們有時需要在后臺執(zhí)行長時間運行的任務(wù),例如監(jiān)聽或輪詢某些資源,這篇文章將通過一個實例詳細(xì)介紹如何為 Go 語言中的無限循環(huán)設(shè)置時間限制,保證程序的健壯性和可控性,需要的朋友可以參考下

引言

在 Go 語言的開發(fā)過程中，我們有時需要在后臺執(zhí)行長時間運行的任務(wù)，例如監(jiān)聽或輪詢某些資源。但是，如果任務(wù)執(zhí)行時間過長或出現(xiàn)意外情況導(dǎo)致死循環(huán)，我們通常希望能夠設(shè)置一個超時機制來中止循環(huán)。這篇文章將通過一個實例詳細(xì)介紹如何為 Go 語言中的無限循環(huán)設(shè)置時間限制，保證程序的健壯性和可控性。

問題描述

我們有一個用于檢查 RabbitMQ 集群節(jié)點的 Go 函數(shù)，該函數(shù)包含一個無限循環(huán)，用于不斷執(zhí)行檢查命令。現(xiàn)在的需求是，如果函數(shù)運行超過3分鐘，自動終止循環(huán)。

原始代碼

原始代碼如下：

func checkRabbitmqClusterIfForgetNode(node string) bool {
    for {
        cmd := fmt.Sprintf("docker exec -i pam_pam-rabbitmq_1 rabbitmqctl --node %s cluster_status --formatter json", node)
        res, err := common.ExecuteCommandWithoutSpace("bash", "-c", cmd)
        if err != nil {
            log.Errorf("exec cmd %v failed, response: %v error: %v", cmd, res, err)
            continue
        }
        cluster, err := mq.ParseJSON(res)
        if err != nil {
            log.Errorf("parse json %v error: %v", res, err)
            continue
        }
        if nodes := cluster.CountDiskNodes(); nodes == 2 {
            log.Infof("rabbitmq cluster node number is %v, still not forget", nodes)
            continue
        }
        return true
    }
}

添加時間限制

要為這個無限循環(huán)設(shè)置時間限制，我們可以使用 Go 語言的 time 包。具體方法是使用 time.After 函數(shù)來創(chuàng)建一個超時通道，當(dāng)達(dá)到指定時間后，超時通道會接收到一個時間信號。

改進后的代碼如下：

func checkRabbitmqClusterIfForgetNode(node string) bool {
    timeout := time.After(3 * time.Minute) // 設(shè)置超時時間為3分鐘
    for {
        select {
        case <-timeout:
            log.Info("Operation timed out")
            return false // 時間超過3分鐘后退出循環(huán)
        default:
            cmd := fmt.Sprintf("docker exec -i pam_pam-rabbitmq_1 rabbitmqctl --node %s cluster_status --formatter json", node)
            res, err := common.ExecuteCommandWithoutSpace("bash", "-c", cmd)
            if err != nil {
                log.Errorf("exec cmd %v failed, response: %v error: %v", cmd, res, err)
                continue
            }
            cluster, err := mq.ParseJSON(res)
            if err != nil {
                log.Errorf("parse json %v error: %v", res, err)
                continue
            }
            if nodes := cluster.CountDiskNodes(); nodes == 2 {
                log.Infof("rabbitmq cluster node number is %v, still not forget", nodes)
                continue
            }
            return true
        }
    }
}

在這段代碼中，我們使用了 select 語句來等待超時事件。如果 timeout 通道接收到了超時信號，則函數(shù)將打印超時信息并返回 false，這表明函數(shù)因為超時而終止。這種方式非常適合處理可能無限執(zhí)行的循環(huán)任務(wù)，確保它們在給定時間后能夠被適當(dāng)中止。