溫馨提示×

golang超時退出怎么設(shè)置

小億
104
2023-11-01 12:26:46
欄目: 編程語言

在Go語言中,可以使用context包來設(shè)置超時退出。下面是一個示例:

package main

import (
	"context"
	"fmt"
	"time"
)

func main() {
	// 創(chuàng)建一個父級的context
	ctx := context.Background()

	// 創(chuàng)建一個帶有超時的子級context,超時時間為3秒
	ctx, cancel := context.WithTimeout(ctx, 3*time.Second)
	defer cancel()

	// 啟動一個goroutine執(zhí)行耗時操作
	go func() {
		// 模擬一個耗時的操作,需要5秒
		time.Sleep(5 * time.Second)

		// 檢查context是否已經(jīng)超時
		if ctx.Err() != nil {
			fmt.Println("超時退出")
			return
		}

		// 執(zhí)行其他操作...
	}()

	// 等待context超時或被取消
	select {
	case <-ctx.Done():
		fmt.Println(ctx.Err())
	case <-time.After(10 * time.Second):
		fmt.Println("正常退出")
	}
}

在這個示例中,我們首先使用context.Background()創(chuàng)建一個父級的context,然后使用context.WithTimeout()創(chuàng)建一個帶有超時的子級context,超時時間為3秒。

接著,我們在一個goroutine中執(zhí)行耗時操作,這個操作需要5秒。在操作執(zhí)行完畢后,我們檢查context是否已經(jīng)超時,如果超時則打印"超時退出"并返回。

最后,我們使用select語句來等待context超時或被取消。如果context超時,則打印超時錯誤信息;如果context被取消,則打印取消錯誤信息;如果超過10秒仍未超時或被取消,則打印"正常退出"。

0