一个go的messageQueue

仅供参考,如下:

package messageQueue

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

type Message interface {
}

type MessageQueue struct {
	messages []Message
	mutex    *sync.Mutex
	ch       chan int
}

func (queue *MessageQueue) Create() *MessageQueue {
	queue.messages = []Message{}
	queue.mutex = &sync.Mutex{}
	queue.ch = make(chan int, 100000)
	return queue
}

func (queue *MessageQueue) Pop(waitMilliSeconds int) *Message {
	if waitMilliSeconds > 0 {
		timeout := time.After(time.Duration(waitMilliSeconds) * time.Millisecond)
		for {
			select {
			case <-queue.ch:
				fmt.Println("channel timeout")
				goto _Out
			case <-timeout:
				fmt.Println("wait timeout")
				goto _Out
			}
		}
	} else {
		<-queue.ch
	}

_Out:
	queue.mutex.Lock()
	if len(queue.messages) == 0 {
		queue.mutex.Unlock()
		return nil
	}

	message := queue.messages[0]
	queue.messages = queue.messages[1:]
	queue.mutex.Unlock()

	return &message
}

func (queue *MessageQueue) Push(message Message) {
	queue.mutex.Lock()
	queue.messages = append(queue.messages, message)
	queue.mutex.Unlock()
	queue.ch <- 1
}

func (queue *MessageQueue) IsEmpty() bool {
	queue.mutex.Lock()
	l := len(queue.messages)
	queue.mutex.Unlock()
	return l == 0
}

func (queue *MessageQueue) Size() int {
	queue.mutex.Lock()
	l := len(queue.messages)
	queue.mutex.Unlock()
	return l
}

使用参考另一篇博客。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值