golang用 channel简单高效队列

Posted on 2022 年 1 月 28 日 by 张子萌

需求为先进先出原则，批量取出的缓存队列。要求比较简单，尝试开源queue代码，在压力大的情况下可能崩溃。使用redis做队列，CPU使用总量增长一倍多，IO开销也不小。最后用golang自带的channel，可以实现，简单高效，资源消耗也不大。

package main

import (
	"log"
	"os"
	"strconv"
	"time"
)

// 公共通道
// 通道数要比缓存最高数量大，以保证读取慢的情况下不会溢出
var Ch = make(chan string, 100000)

func wch(num int) {
	// 写入
	func() {
		if vv := recover(); vv != nil {
			os.Exit(0)
		}
	}()
	ws := strconv.Itoa(num)
	Ch <- ws
}

func rch() {
	// 读取
	var rs string
	for {
		select {
		case rs = <-Ch:
			log.Printf("read %v\n", rs)
		default:
			log.Printf("empty")
			return
		}
		if len(Ch) <= 0 {
			return
		}
		//Chsync.Done()
	}
}

func main() {
	var i int
	for i = 0; i < 200000; i++ {
		wch(i)
		if i%1000 == 0 {
			go rch()
			//Chsync.Add(1)
		}
	}
	time.Sleep(time.Millisecond * 6000)
	//Chsync.Wait()
	log.Printf("have %v", len(Ch))
}

分享到： QQ空间新浪微博腾讯微博人人网更多

一	二	三	四	五	六	日
« 1月
			1	2	3	4
5	6	7	8	9	10	11
12	13	14	15	16	17	18
19	20	21	22	23	24	25
26	27	28	29	30	31

simonzhang的家

有朋自远方来。。。。。

golang用 channel简单高效队列

0 thoughts on “golang用 channel简单高效队列”

发表评论取消回复

0 thoughts on “golang用 channel简单高效队列”

发表评论 取消回复

发表评论取消回复