Go語言并發(fā)編程如何處理異常

小樊
81
2024-11-08 13:47:16
欄目: 編程語言

在Go語言中,處理并發(fā)編程中的異常需要使用通道(channel)和sync.Mutexsync.RWMutex來確保數(shù)據(jù)的一致性和同步。這里有一個(gè)簡(jiǎn)單的示例,展示了如何在Go語言中處理并發(fā)編程中的異常:

package main

import (
	"errors"
	"fmt"
	"sync"
)

type Task struct {
	ID      int
	Message string
}

func worker(id int, tasks <-chan Task, results chan<- error, wg *sync.WaitGroup, mu *sync.Mutex) {
	defer wg.Done()

	for task := range tasks {
		mu.Lock()
		// 模擬任務(wù)執(zhí)行過程中可能出現(xiàn)的異常
		if task.ID == 3 {
			mu.Unlock()
			results <- errors.New("task 3 encountered an error")
			continue
		}
		mu.Unlock()

		// 模擬任務(wù)正常執(zhí)行
		fmt.Printf("Worker %d processed task %d\n", id, task.ID)
		results <- nil
	}
}

func main() {
	const numWorkers = 5
	const numTasks = 10

	tasks := make(chan Task, numTasks)
	results := make(chan error, numTasks)

	var wg sync.WaitGroup
	var mu sync.Mutex

	// 啟動(dòng)工作協(xié)程
	for i := 1; i <= numWorkers; i++ {
		wg.Add(1)
		go worker(i, tasks, results, &wg, &mu)
	}

	// 向任務(wù)通道發(fā)送任務(wù)
	for i := 1; i <= numTasks; i++ {
		tasks <- Task{ID: i}
	}
	close(tasks)

	// 等待所有工作協(xié)程完成
	wg.Wait()
	close(results)

	// 處理結(jié)果
	for err := range results {
		if err != nil {
			fmt.Println("Error:", err)
		} else {
			fmt.Println("Success")
		}
	}
}

在這個(gè)示例中,我們創(chuàng)建了一個(gè)worker函數(shù),它接收一個(gè)任務(wù)通道、一個(gè)結(jié)果通道、一個(gè)等待組和一個(gè)互斥鎖。工作協(xié)程從任務(wù)通道中獲取任務(wù),執(zhí)行任務(wù),并將結(jié)果發(fā)送到結(jié)果通道。如果在執(zhí)行任務(wù)過程中遇到異常(例如任務(wù)ID為3),我們將錯(cuò)誤發(fā)送到結(jié)果通道,并繼續(xù)處理下一個(gè)任務(wù)。

main函數(shù)中,我們啟動(dòng)了5個(gè)工作協(xié)程,并向任務(wù)通道發(fā)送了10個(gè)任務(wù)。然后,我們等待所有工作協(xié)程完成,并處理結(jié)果通道中的錯(cuò)誤和成功消息。

0