go語言time.After()的作用

更新時間：2024年10月18日 10:38:18 作者：shulu

time.After 是 Go 語言中的一個函數(shù),用于返回一個定時器通道,該通道在指定時間后發(fā)送當前時間,這個功能常用于超時控制和延遲執(zhí)行,本文就來詳細的介紹一下,感興趣的可以了解學習

1.延遲執(zhí)行、定時任務

package main

import (
	"fmt"
	"time"
)

func main() {
	// 等待3秒
	timer := time.After(3 * time.Second)

	// 阻塞等待time.After的通道發(fā)送時間
	x := <-timer
	fmt.Println(x)
	// 當上面的<-timer執(zhí)行后，說明已經(jīng)等待了5秒
	fmt.Println("3秒已過")

	// 你可以繼續(xù)執(zhí)行你的邏輯...
}

2.超時控制

time.After還常用于實現(xiàn)超時控制。比如，你可能在發(fā)送一個HTTP請求時，不希望這個請求無限制地等待響應，而是想在某個超時時間后停止等待并處理超時邏輯：

超時控制應該是非常常見的使用場景。

package main

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"time"
)

func main() {
	client := &http.Client{
		Timeout: 10 * time.Second, // 設置客戶端超時時間
	}

	// 模擬一個可能會超時的請求
	resp, err := client.Get("http://example.com/may-timeout")
	if err != nil {
		fmt.Println("請求錯誤:", err)
		return
	}
	defer resp.Body.Close()

	// 使用time.After實現(xiàn)讀取響應體的超時
	done := make(chan bool, 1)
	go func() {
		// 模擬讀取響應體，實際中可能是ioutil.ReadAll(resp.Body)
		// 這里用time.Sleep模擬耗時操作
		time.Sleep(2 * time.Second)
		done <- true
	}()

	select {
	case <-done:
		// 讀取成功
		body, _ := ioutil.ReadAll(resp.Body)
		fmt.Println("讀取響應體成功:", string(body))
	case <-time.After(1 * time.Second):
		// 超時
		fmt.Println("讀取響應體超時")
	}
}

超時控制channel結(jié)合了select。

go time.After 定時器示例

package main
 
import (
	"fmt"
	"time"
)
 
// 定義 stop 函數(shù)，接受 deviceid 和 gunid 作為參數(shù)
func stop(deviceid int, gunid int) {
	fmt.Printf("Stopping with device ID: %d and gun ID: %d\n", deviceid, gunid)
}
 
func main() {
	// 設定要傳遞給 stop 函數(shù)的參數(shù)
	deviceid := 1
	gunid := 2
 
	// 使用 goroutine 異步執(zhí)行計時器
	go func() {
		// 創(chuàng)建一個通道，將在 10 秒后關(guān)閉
		timer := time.After(10 * time.Second)
 
		// 等待通道關(guān)閉，即等待 10 秒
		<-timer
 
		// 在 10 秒后執(zhí)行 stop 函數(shù)
		stop(deviceid, gunid)
	}()
 
	// 主程序可以繼續(xù)執(zhí)行其他任務
	fmt.Println("Main program continues to run...")
 
	// 為了演示，讓主程序運行一段時間，以便觀察 stop 函數(shù)的執(zhí)行
	time.Sleep(15 * time.Second)
	fmt.Println("Main program ends.")
}