[Golang]Select多路选择的模拟实现
2015-03-11 10:38
197 查看
有时候有这样一种应用场景:需要等待多个事件到达,然后返回尽可能多的事件;如果没有事件到达就阻塞等待。例如服务器等待客户端建立连接,或者等待客户端数据等就有这种应用需求。 在go语言里,可以利用select原语和它的非阻塞(default)分支组合实现这个功能:
// 从ch获取尽可能多的数据放到events里,并返回实际数量;如果没有数据就阻塞等待 func wait(ch chan int, events []int) int { count := 0 for count < len(events) { select { case x := <-ch: events[count] = x count++ default: if count > 0 { return count } events[count] = <-ch count++ } } return count }如果再加上退出检查:
import "errors" func wait(ch chan int, exit chan bool, events []int) (int, error) { count := 0 for count < len(events) { select { case <-exit: return 0, errors.New("exit") case x := <-ch: events[count] = x count++ default: if count > 0 { return count, nil } select { case <-exit: return 0, errors.New("exit") case x := <-ch: events[count] = x count++ } } } return count, nil }可以看到,这里的实现有很多重复代码,非常的冗长难读。我们可以利用channel以下特性改写一下: 1.读取或者写入空channel时永久阻塞 2.读取一个已经关闭的channel立即返回空值
import "errors" var ( CLOSED = make(chan int) ) func init() { close(CLOSED) } func pass(flag bool) chan int { if flag { return CLOSED } return nil } func wait(ch chan int, exit chan bool, events []int) (int, error) { count := 0 LOOP: for count < len(events) { select { case <-exit: return 0, errors.New("exit") case x := <-ch: events[count] = x count++ case <-pass(count > 0): break LOOP } } return count, nil }现在的实现就比较清晰简洁易读。
相关文章推荐
- js模拟权限选择实现代码(select操作)
- Jquery模拟的select点击选择实现代码
- js模拟权限选择实现代码(select操作)
- jQuery模拟实现的select点击选择效果【附demo源码下载】
- 模拟multiple select,实现不按ctrl单击选中以及拖动选择
- js模拟权限选择实现代码(select操作)
- Multi-select ASP.NET datagrid(DataGrid多条记录选择的实现)
- Jquery实现select选择什么,页面展现为选择的项~!
- Jquery实现select选择什么,页面展现为选择的项~!
- 利用 window_onload 实现select默认选择
- 模拟Select-Options对象实现多项数据的输入功能
- javascript模拟select实现代码
- linux高级字符设备驱动之 四 poll方法(select多路监控原理与实现)
- 可编辑Select下拉列表控件实现方法(非DIV模拟)
- 用select实现I/O多路转接
- JQuery 绑定select标签的onchange事件,弹出选择的值,并实现跳转、传参(selected的值和页面其它元素的值)以及js的select
- 基于jquery的用dl模拟实现可自定义样式的SELECT下拉列表(已封装)
- 左边select内容选择到右边select内容里面的js实现方法(可以多选和全选)
- 用dl模拟实现可自定义样式的SELECT下拉列表(已封装)
- JAVA排序算法实现代码-选择(Select)式排序