扫码关注官方微信

扫码下载APP
返回顶部
首页 > 资讯 > 后端开发 > GO >Golang实现带优先级的select
  • 637

0
分享到

Golang实现带优先级的select

Golang实现带优先级selectGolang带优先级selectGolang select 2023-05-14 08:05:10 637人浏览 独家记忆
摘要

目录背景解决方案一个封装背景 在 golang 里面，我们经常使用 channel 进行协程之间的通信。这里有一个经典的场景，也就是生产者消费者模式，生产者协程不断地往 Channe

背景

在 golang 里面，我们经常使用 channel 进行协程之间的通信。这里有一个经典的场景，也就是生产者消费者模式，生产者协程不断地往 Channel 里面塞元素，而消费者协程不断地消费这些元素。

写成代码就是如下：

package main

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

func main() {
	ch := make(chan int, 10)
	var wg sync.WaitGroup
	wg.Add(2)
	Go producer(ch, &wg)
	go consumer(ch, &wg)
	wg.Wait()
}

// 生产者
func producer(ch chan int, wg *sync.WaitGroup) {
	defer wg.Done()
	i := 0
	for {
		select {
		case ch <- i:
		default:
			// 丢弃
			log.Println("discard")
		}
		i++
		time.Sleep(time.Second)
	}
}

// 消费者
func consumer(ch chan int, wg *sync.WaitGroup) {
	defer wg.Done()
	consume := func(i int) {
		fmt.Println(i)
		time.Sleep(time.Millisecond * 700)
	}
	for {
		i := <-ch
		consume(i) // 消费元素
	}
}

生产者不断产生元素，消费者消费元素。生产者不会等待消费者消费完毕（不然可能影响其他任务），如果 channel 已经满了，也就是说明消费者消费不过来，生产者就会丢弃这个任务。

生产者平均一秒生成1个，消费者0.7秒消费一个。正常情况下消费者是消费得过来的，然而很多时候消费者协程还需要做一些定时任务，比如一些定时清理工作。假如这个清理工作每2秒触发一次，清理时间一般需要1.5秒，也就是如果每次都做每一秒有0.75秒会被清理工作占有了，但是它不是一定要非常及时的，可以等空闲时再进行。 如下代码：

// 消费者
func consumer(ch chan int, wg *sync.WaitGroup) {
	defer wg.Done()
	t := time.NewTicker(time.Second * 2)
	consume := func(i int) {
		fmt.Println(i)
		time.Sleep(time.Millisecond * 700)
	}
	clear := func() {
		fmt.Println("clear")
		time.Sleep(time.Millisecond * 1500)
	}
	for {
		select {
		case i := <-ch:
			consume(i) // 消费元素:
		case <-t.C:
			clear() // 清理
		}
	}
}

运行程序到第15秒的时候，生产者发现 channel满了，于是开始丢包：

0
1
clear
2
3
4
5
6
clear
7
clear
8
clear
9
clear
clear
10
clear
11
12
13
14
clear
15
clear
clear
discard
16
clear
discard
discard

解决方案

既然清理任务的优先级并不高，那么它就不应该阻塞消费元素流程，而是应该在空闲时才去执行。由于 Golang 里面，如果 select 两个 case 都同时满足，会随机选一个执行，因此第一想到的可能会使用如下代码实现优先级case：

// 消费者
func consumer(ch chan int, wg *sync.WaitGroup) {
	defer wg.Done()
	t := time.NewTicker(time.Second * 2)
	consume := func(i int) {
		fmt.Println(i)
		time.Sleep(time.Millisecond * 700)
	}
	clear := func() {
		fmt.Println("clear")
		time.Sleep(time.Millisecond * 1500)
	}
	for {
		select {
		case i := <-ch:
			consume(i) // 消费元素
			continue   // 可能还有元素，不走清理逻辑
		default:
		}

		// 没有元素才走清理逻辑
		select {
		case <-t.C:
			clear() // 清理
		default:
		}
	}
}

如果运行这个程序，可以发现它能够满足优先级的需求，先消费元素，空闲时再执行清理任务。

然而，在没有元素可以消费，也没有清理任务可以执行的时候，这里的for将会不断地循环，浪费CPU资源。

其实，可以使用下面的方法实现优先级case，它能够在没有元素就绪的时候阻塞在 select，而不是不断循环：

// 消费者
func consumer(ch chan int, wg *sync.WaitGroup) {
	defer wg.Done()
	t := time.NewTicker(time.Second * 2)
	consume := func(i int) {
		fmt.Println(i)
		time.Sleep(time.Millisecond * 700)
	}
	clear := func() {
		fmt.Println("clear")
		time.Sleep(time.Millisecond * 1500)
	}
	for {
		select {
		case i := <-ch:
			consume(i) // 消费元素
		case <-t.C:
		priority:
			for { // 清理前先把元素消费完
				select {
				case i := <-ch:
					consume(i) // 消费元素
				default:
					break priority // 注：这里会跳过这个循环，而不是再次执行
				}
			}
			clear() // 清理
		}
	}
}

这里的关键是在触发清理case的时候，先去把channel里面的元素消费完，再进行清理，从而保证能够留下足够的channel缓冲区给生产者放置生产的元素。

一个封装

上面那段优先级case代码其实挺常用的，但是几乎都是模板代码，特别是需要在两个地方写consume(i)，因此我们可以封装一下这段代码，方便使用，减少出错：

// 优先级select ch1 的任务先执行完毕后才会执行 ch2 里面的任务
func PrioritySelect[T1, T2 any](ch1 <-chan T1, f1 func(T1), ch2 <-chan T2, f2 func(T2)) {
	for {
		select {
		case a := <-ch1:
			f1(a)
		case b := <-ch2:
		priority:
			for {
				select {
				case a := <-ch1:
					f1(a)
				default:
					break priority
				}
			}
			f2(b)
		}
	}
}

这样，我们的消费者代码就可以简化为：

// 消费者
func consumer(ch chan int, wg *sync.WaitGroup) {
	defer wg.Done()
	t := time.NewTicker(time.Second * 2)
	consume := func(i int) {
		fmt.Println(i)
		time.Sleep(time.Millisecond * 700)
	}
	clear := func(time.Time) {
		fmt.Println("clear")
		time.Sleep(time.Millisecond * 1500)
	}
	PrioritySelect(ch, consume, t.C, clear)
}

到此这篇关于Golang实现带优先级的select的文章就介绍到这了,更多相关Golang带优先级select内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！

您可能感兴趣的文档:

--结束END--

本文标题: Golang实现带优先级的select

本文链接: https://lsjlt.com/news/202850.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

猜你喜欢
  • Golang实现带优先级的select
    目录背景解决方案一个封装背景 在 Golang 里面，我们经常使用 channel 进行协程之间的通信。这里有一个经典的场景，也就是生产者消费者模式，生产者协程不断地往 Channe...
    99+
    2023-05-14
    Golang实现带优先级select Golang带优先级select Golang select
  • Golang怎么实现带优先级的select
    这篇“Golang怎么实现带优先级的select”文章的知识点大部分人都不太理解，所以小编给大家总结了以下内容，内容详细，步骤清晰，具有一定的借鉴价值，希望大家阅读完这篇文章能有所收获，下面我们一起来看看这篇“Golang怎么实现带优先级的...
    99+
    2023-07-05
  • JavaScript实现优先级队列
    目录一、优先级队列介绍二、优先级队列封装一、优先级队列介绍 我们知道，普通的队列插入一个元素，数据会被放在后端，并且需要前面所有的元素都处理完成后才会处理前面的数据。但是优先级队列，...
    99+
    2024-04-02
  • JavaScript如何实现优先级队列
    这篇文章主要讲解了“JavaScript如何实现优先级队列”，文中的讲解内容简单清晰，易于学习与理解，下面请大家跟着小编的思路慢慢深入，一起来研究和学习“JavaScript如何实现优先级队列”吧！一、优先级队列介绍我们知道，普通的队列插入...
    99+
    2023-06-21
  • FreeRTOS实时操作系统的多优先级实现
    目录如何实现任务多优先级软件通用方法和硬件指令方法如何实现任务多优先级 FreeRTOS中，数字优先级越小，逻辑优先级也越小，空闲任务优先级为0.List_t pxReadyTask...
    99+
    2024-04-02
  • C#中实现PriorityQueue优先级队列的代码
    前言 前段时间看到有大佬对.net 6.0新出的PriorityQueue(优先级队列)数据结构做了解析，但是没有源码分析，所以本着探究源码的心态，看了看并分享出来。它不像普通队列先...
    99+
    2024-04-02
  • Python自带的优先级调度器有什么用
    本篇内容主要讲解“Python自带的优先级调度器有什么用”，感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷，实用性强。下面就让小编来带大家学习“Python自带的优先级调度器有什么用”吧!Python 自带一个调度器模块sched，它能...
    99+
    2023-06-15
  • Python中实例属性的优先级
    Python中实例属性的优先级？相信很多没有经验的人对此束手无策，为此本文总结了问题出现的原因和解决方法，通过这篇文章希望你能解决这个问题。python可以做什么Python是一种编程语言，内置了许多有效的工具，Python几乎无所不能，该...
    99+
    2023-06-14
  • 深度优先和广度优先的Python实现
    #coding=utf-8 class Gragh(): def __init__(self,nodes,sides): ''' nodes 表示点 sides 表示边 ...
    99+
    2023-01-31
    广度 深度 Python
  • c++中=和==的优先级
    = 运算符优先级高于 == 运算符。这意味着赋值操作优先于相等性比较执行。因此，在赋值操作之后进行相等性比较可能会导致意外结果。 C++ 中 = 和 == 的优先级 在 C++ 中，赋...
    99+
    2024-05-10
    c++ 代码可读性
  • FreeRTOS实时操作系统的多优先级怎么实现
    这篇文章主要介绍“FreeRTOS实时操作系统的多优先级怎么实现”的相关知识，小编通过实际案例向大家展示操作过程，操作方法简单快捷，实用性强，希望这篇“FreeRTOS实时操作系统的多优先级怎么实现”文章能帮助大家解决问题。如何实现任务多优...
    99+
    2023-06-29
  • 揭示Go语言运算符优先级的奥秘：揭示顶级优先级
    深入探究Go语言运算符优先级，揭示顶级优先级是什么，需要具体代码示例 在Go语言中，运算符优先级是指不同运算符之间的执行顺序。了解运算符的优先级对于正确理解和书写代码至关重要。本文将深入探究Go语言中的运算符...
    99+
    2024-01-18
    Go语言 运算符优先级 深入探究
  • Python实现优先级队列结构的方法详解
    最简单的实现 一个队列至少满足2个方法,put和get. 借助最小堆来实现. 这里按"值越大优先级越高"的顺序. #coding=utf-8 from heapq import heappush, h...
    99+
    2022-06-04
    优先级 队列 详解
  • select 与 where、group by、order by、limit 子句执行优先级比较
    当 select 和 其他三种语句的一者或者多者同时出现时，他们之间是存在执行先后顺序的。 他们的优先级顺序是：where > group by > select > order by > limit          目录 ...
    99+
    2023-10-03
    sql 数据库 mysql
  • Android 如何实现弹窗顺序&优先级控制
    一般在项目首页中，往往会有多个对话框需要弹出，比如活动弹窗、更新弹窗、评分弹窗等等，而且这些弹窗是有优先级顺序的。这些弹窗一般是通过接口请求后返回结果再显示的，如果只有几个弹窗还好处...
    99+
    2024-04-02
  • 用Python实现数据结构之优先级队列
    如果我们给每个元素都分配一个数字来标记其优先级，不妨设较小的数字具有较高的优先级，这样我们就可以在一个集合中访问优先级最高的元素并对其进行查找和删除操作了。这样，我们就引入了优先级队列 这种数据结构 最简单的优先级队列可能就是一堆不...
    99+
    2023-01-30
    优先级 数据结构 队列
  • C#的运算符优先级实例分析
    这篇文章主要介绍了C#的运算符优先级实例分析的相关知识，内容详细易懂，操作简单快捷，具有一定借鉴价值，相信大家阅读完这篇C#的运算符优先级实例分析文章都会有所收获，下面我们一起来看看吧。实例using System;namespa...
    99+
    2023-06-17
  • Win11如何调高进程的优先级?Win11调高进程的优先级方法
    　　相信很多使用过电脑的小伙伴都知道，系统资源是有限的，因此我们在打开很多的应用程序就需要设置一个优先级，那么使用Win11系统的电脑要如何去进行这个设置呢？其实操作起来还是很简单的，下面就和小编一起来看看方法吧。 　　...
    99+
    2023-05-19
    Win11调高进程 进程优先级
  • 揭示Go语言运算符优先级，揭晓最高优先级的秘密
    ...
    99+
    2024-01-18
    Go语言 运算符优先级 最高优先级
  • 揭秘Go语言运算符优先级，揭示关键的最高优先级
    Go语言运算符优先级解析，揭秘最重要的优先级是什么，需要具体代码示例 当我们在使用Go语言进行编程时，运算符是不可避免的一部分。了解运算符的优先级是理解和正确使用它们的关键。在本文中，我们将解析Go语言中运算...
    99+
    2024-01-18
    Go语言 运算符优先级 揭秘
软考高级职称资格查询
热门wiki
近期文章
推荐阅读
热门问答
热门标签
编程网，编程工程师的家园，是目前国内优秀的开源技术社区之一，形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容，为IT开发者提供了一个发现、使用、并交流开源技术的平台。

  • 官方手机版

  • 微信公众号

  • 商务合作