淺析Golang開發(fā)中goroutine的正確使用姿勢

更新時間：2024年03月10日 16:16:39 作者：飛翔碼農

很多初級的Gopher在學習了goroutine之后,在項目中其實使用率不高,所以這篇文章小編主要來帶大家深入了解一下goroutine的常見使用方法,希望對大家有所幫助

很多初級的Gopher在學習了goroutine之后，在項目中其實使用率不高，尤其一些跨語言過來的人，對并發(fā)編程理解不深入，可能很多人只知道go func(),或者掌控不夠，謹慎一些，盡量少使用或者不使用，用的話就是go func()，主要列一下我們這邊的主要使用方法。

goroutine在項目中的使用方法

看一下樣例代碼，實際上，我們生產環(huán)境中就是這么使用的。

package logic

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

type UserData struct {
	Age     int
	Name    string
	Postion string
}

type ServerLogic struct {
	ctx    context.Context
	cancel func()
	waiter sync.WaitGroup
	ch     chan UserData
}

func NewServerLogic(logCtx *context.Context, worker int, queue int) *ServerLogic {
	logic := &ServerLogic{}
	logic.InitWorker(worker, queue)
	return logic
}

func (this *ServerLogic) InitWorker(workers int, queue int) {
	this.ch = make(chan UserData, queue)
	this.ctx, this.cancel = context.WithCancel(context.Background())
	this.waiter.Add(workers)
	for i := 0; i < workers; i++ {
		go this.Proc()
	}
}

func (this *ServerLogic) Proc() {
	defer this.waiter.Done()
	for {
		select {
		case t := <-this.ch:
			this.Dothing(t)
		case <-this.ctx.Done():
			return
		}
	}
}

func (this *ServerLogic) Dothing(data UserData) error {
	//do code
    time.Sleep(time.Second*30)
	return nil
}

func (this *ServerLogic) Close() {
	this.cancel()
    this.waiter.Wait()
}

func (this *ServerLogic) PutData(user UserData) error {
    select {
    case this.ch<-user:
        return nil
	default:
		return fmt.Errorf("queue overflow")
    }
}

如果有人想直接使用的話，只需要把UserData struct換成自己的請求數據，把Dothing里面的代碼換成讓goroutine多任務執(zhí)行的代碼就可以在自己的項目中使用了。

PutData有請求數據就放入channel，每個goroutine不停的循環(huán)從channel里面取數據，取到數據之后就執(zhí)行相應的邏輯流程，可以看到整體的調度都是channel來控制的，通過channel的通信來傳遞數據。

不要通過共享內存來通信,要通過通信來共享內存

看看大概的代碼分析

InitWorker的時候會創(chuàng)建queue個channl，再創(chuàng)建workers個goroutine，執(zhí)行go Proc()
Proc方法，里面有for的無限循環(huán)，不停從步驟1里面創(chuàng)建的channl里面獲取UserData數據，一旦獲取數據成功，就會帶著UserData數據去執(zhí)行Dothing方法。需要注意的是，這是workers個goroutine都在執(zhí)行Proc
Dothing方法，就是讓某一個goroutine拿到UserData數據去處理數據，執(zhí)行邏輯
Close方法，給所有的goroutine發(fā)送關閉的信號，channl里面不在有數據寫入，waiter.Wait()等待現(xiàn)有的channel里面數據被消費完，goroutine就執(zhí)行完畢退出。
PutData方法，就是把請求的數據交給goroutine去執(zhí)行。具體的做法，是把數據塞到channl隊列里面，如果queue個channl隊列已滿，就拋出溢出錯誤。

當然了PutData也可以等待channl隊列里面的數據被Proc拿出，然后空出位置再塞數據到channl隊列。

func (this *ServerLogic) PutData(user UserData) error {
	timer := time.NewTimer(3*time.Second)
	select {
    case this.ch<-user:
        return nil
	case <-timer.C:
		return fmt.Errorf("put timeout")
    }
}

加一個超時器，總不能等到天荒地老把，如果超過三秒，仍然沒有空出channl位置，現(xiàn)有的隊列還沒有消費完，就拋出塞數據超時的錯誤.

看一下樣例的使用的代碼

package main

import (
	context2 "context"
	"fmt"
	"test/logic"
)

func main() {
	context := context2.Background()
	server := logic.NewServerLogic(&context, 1, 2)
	rt1 := server.PutData(logic.UserData{
		Age:     11,
		Name:    "test1",
		Postion: "golang",
	})
	fmt.Println(rt1)
	rt2 := server.PutData(logic.UserData{
		Age:     12,
		Name:    "test2",
		Postion: "golang",
	})
	fmt.Println(rt2)
	rt3 := server.PutData(logic.UserData{
		Age:     13,
		Name:    "test3",
		Postion: "golang",
	})
	fmt.Println(rt3)

	server.Close()
	fmt.Println("end")
}

等待了大概三十多秒之后的結果,打印結果其實跟預想的是一樣的。

<nil>
<nil>
queue overflow
end

NewServerLogic(&context, 1, 2)代碼中，我們要求創(chuàng)建了1個goroutine，大小為2的channl隊列。
所以第一個PutData和第二個PutData是塞數據成功的。等到第三次PutData的時候，因為我們channl隊列的大小是2，已經被占滿了，所以第三次就會提示溢出錯誤。

使用goroutine另一種方法

我看項目中還有一些其他人的使用方法，區(qū)別只是退出的時候沒有使用context的cancel方法，而是使用了channel去通知退出goroutine，內部的原理其實是一樣的?？匆幌孪旅娴拇a。

func(this *ServerLogic)InitWorker(workers int, queue int)
{
    this.quit = make(chan bool)
    this.ch = make(chan UserData, queue)
    this.waiter.Add(workers)
    for i :=0;i< workers; i++ {
        go this.Proc()
    }
}

func(this *ServerLogic)Proc()
{
    defer this.waiter.Done()
    for {
        select {
            case t := ←this.ch:
            this.Dothing(t)
            case ←this.quit:
                 return   
        }
    }
}

func(this *ServerLogic)Close(){
    close(this.quit)
    this.waiter.Wait()
}

只有關閉這里是不一樣的，其他的基本一致。執(zhí)行退出的時候在Close()方法中，close(this.quit)會給quit channel寫入數據，Proc()方法會循環(huán)從channel和quit里面取數據，一旦從this.quit里面取出了數據，說明系統(tǒng)讓關閉goroutine，然后Proc方法就終止。

go func()行不行

有人說，扯這么多，為啥go func()不行，我在項目里面使用go func()運行的好好，而且golang的HTTP庫里也是使用的go c.serve(ctx)。

我的理解是主要看使用場景，如果你的服務對結果要求不是100%的成功，對并發(fā)的要求很高，那就可以使用go func()，go c.serve(ctx)也是類似，TCP本身就是不可靠的連接，HTTP也允許有極少量的失敗狀態(tài)。

如果你的服務里面只是想讓多個goroutine處理你的數據，不希望這個goroutine太多影響你的主干服務，或者你為了提高數據處理效率，想讓多個goroutine去請求第三方的服務，這樣的話，就應該創(chuàng)建若干個goroutine去并發(fā)處理你的任務，也不建議直接go func()，goroutine數量不可控，會影響其他的主干服務或者占用服務器資源，如果請求第三方的服務，可能會因為并發(fā)太高被限制，或者把第三方服務打掛。我們就遇到過這種情況。

總之，使用場景很重要，不是一概而論的。

到此這篇關于淺析Golang開發(fā)中goroutine的正確使用姿勢的文章就介紹到這了,更多相關Go goroutine內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家

您可能感興趣的文章: