快捷導(dǎo)航

Golang關(guān)鍵字select的常用用法總結(jié)

更新時(shí)間：2023年10月26日 09:58:56 作者：林欣快滾去學(xué)習(xí)

這篇文章主要為大家詳細(xì)介紹了golang中select關(guān)鍵字的常用用法,本文給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下

一、Select解決什么問(wèn)題

在Golang中，兩個(gè)協(xié)程之間通信Channel（圖一），在接受協(xié)程中通過(guò)代碼表示即為<ch；如果協(xié)程需要監(jiān)聽(tīng)多個(gè)Channel，只要有其中一個(gè)滿足條件，就執(zhí)行相應(yīng)的邏輯（圖二），這種select的應(yīng)用場(chǎng)景之一，代碼如下：

func TestSelect(t *testing.T) {
    ch1 := make(chan int, 1)
    ch2 := make(chan int , 1)

    select {
        case <-ch1:
        fmt.Println("ch1")
        case <-ch2:
        fmt.Println("ch2")
        default:
        }
}

上述代碼，創(chuàng)建兩個(gè)通道，通過(guò)select監(jiān)聽(tīng)協(xié)程是否有數(shù)據(jù)，如果有打印相應(yīng)的值，如果沒(méi)有通過(guò)default結(jié)束程序運(yùn)行；

二、Select常用用法

循環(huán)阻塞監(jiān)測(cè)

func TestLoopSelect(t *testing.T) {
    ch1 := make(chan int, 1)
    ch2 := make(chan int, 1)

    go func() {
        // 每隔1s發(fā)送一條消息到channel中
        for range time.Tick(1 * time.Second) {
            ch1 <- 1
        }
    }()

    for {
        select {
            case <-ch1:
            fmt.Println("ch1")
            case <-ch2:
            fmt.Println("ch2")
        }
    }
}

這種寫(xiě)法特別常見(jiàn)，起一個(gè)協(xié)程，阻塞循環(huán)監(jiān)聽(tīng)多個(gè)channel，如果有數(shù)據(jù)執(zhí)行對(duì)應(yīng)的操作。比如說(shuō)

// go-zero/core/discov/internal/registry.go
func (c *cluster) watchStream(cli EtcdClient, key string, rev int64) bool {
    var rch clientv3.WatchChan
    if rev != 0 {
        rch = cli.Watch(clientv3.WithRequireLeader(c.context(cli)), makeKeyPrefix(key), clientv3.WithPrefix(),
                      clientv3.WithRev(rev+1))
    } else {
        rch = cli.Watch(clientv3.WithRequireLeader(c.context(cli)), makeKeyPrefix(key), clientv3.WithPrefix())
    }

    for {
        select {
            case wresp, ok := <-rch:
        	...
            c.handleWatchEvents(key, wresp.Events)
            case <-c.done:
            return true
        }
    }
}

上述代碼，通過(guò)for + select阻塞循環(huán)監(jiān)測(cè)注冊(cè)中心數(shù)據(jù)是否有變換，有變化的話，針對(duì)變化類(lèi)型，執(zhí)行對(duì)應(yīng)邏輯；

非阻塞監(jiān)控

func TestSelect(t *testing.T) {
    ch1 := make(chan int, 1)
    ch2 := make(chan int , 1)

    select {
        case <-ch1:
        fmt.Println("ch1")
        case <-ch2:
        fmt.Println("ch2")
        default:
        }
}

這段代碼的意思是，程序執(zhí)行到select，就檢查一下channel里面是否有數(shù)據(jù)，有就處理，沒(méi)有就退出；在grpc-go中，

// grpc-go/clientconn.go
func DialContext(ctx context.Context, target string, opts ...DialOption) (conn *ClientConn, err error) {
	cc := &ClientConn{
		target: target,
		conns:  make(map[*addrConn]struct{}),
		dopts:  defaultDialOptions(),
		czData: new(channelzData),
	}

	defer func() {
		select {
		case <-ctx.Done():
			switch {
			case ctx.Err() == err:
				conn = nil
			case err == nil || !cc.dopts.returnLastError:
				conn, err = nil, ctx.Err()
			default:
				conn, err = nil, fmt.Errorf("%v: %v", ctx.Err(), err)
			}
		default:
		}
	}()

	if cc.dopts.scChan != nil {
		// Blocking wait for the initial service config.
		select {
		case sc, ok := <-cc.dopts.scChan:
			if ok {
				cc.sc = &sc
				cc.safeConfigSelector.UpdateConfigSelector(&defaultConfigSelector{&sc})
			}
		case <-ctx.Done():
			return nil, ctx.Err()
		}
	}

}

上述代碼中，有兩個(gè)地方用到select，一個(gè)地方是非阻塞，檢查contex是否被取消；另外一個(gè)是阻塞等待；