码迷,mamicode.com
首页 > 编程语言 > 详细

Go语言无锁队列组件的实现 (chan/interface/select)

时间:2018-12-31 12:19:27      阅读:383      评论:0      收藏:0      [点我收藏+]

标签:int   cli   mil   c++   类型   ack   cond   void   struct   

1. 背景

go代码中要实现异步很简单,go funcName()。
但是进程需要控制协程数量在合理范围内,对应大批量任务可以使用“协程池 + 无锁队列”实现。

2. golang无锁队列实现思路

  • Channel是Go中的一个核心类型,你可以把它看成一个管道,通过它并发核心单元就可以发送或者接收数据进行通讯(communication)。无锁队列使用带buff的chan存储数据。
  • interface{} (类似c++的void*, java的Object)可以与任意类型互转。无锁队列使用interface{}作为数据存储类型。
  • select可以处理多个信号, 可以用来解决channel阻塞问题。

3. 代码实现

package main

import (
    "fmt"
    "time"
)

type DataContainer struct {
    Queue chan interface{}
}

func NewDataContainer(max_queue_len int) (dc *DataContainer){
    dc = &DataContainer{}
    dc.Queue = make(chan interface{}, max_queue_len)
    return dc
}

//非阻塞push
func (dc *DataContainer) Push(data interface{}, waittime time.Duration) bool{
    click := time.After(waittime)
    select {
    case dc.Queue <- data:
        return true
    case <- click:
        return false
    }
}

//非阻塞pop
func (dc *DataContainer) Pop(waittime time.Duration) (data interface{}){
    click := time.After(waittime)
    select {
    case data =<-dc.Queue:
        return data
    case <- click:
        return nil
    }
}

//test
var MAX_WAIT_TIME = 10 *time.Millisecond
func main(){
    type dataItem struct {
        name string
        age int
    }

    datacotainer := NewDataContainer(2)
    //add
    fmt.Printf("res=%v\n", datacotainer.Push(&dataItem{"zhangsan",25}, MAX_WAIT_TIME))
    fmt.Printf("res=%v\n", datacotainer.Push(&dataItem{"lisi",30}, MAX_WAIT_TIME))
    fmt.Printf("res=%v\n", datacotainer.Push(&dataItem{"wangwu",28}, MAX_WAIT_TIME))

    //get
    var item interface{}
    item = datacotainer.Pop(MAX_WAIT_TIME)
    if item != nil{
        if tmp,ok := item.(*dataItem); ok{  //interface转为具体类型
            fmt.Printf("item name:%v, age:%v\n", tmp.name, tmp.age)
        }
    }
}

Go语言无锁队列组件的实现 (chan/interface/select)

标签:int   cli   mil   c++   类型   ack   cond   void   struct   

原文地址:https://www.cnblogs.com/xudong-bupt/p/10199028.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!