b612/starnet
1
0
Fork 0
Code Issues Pull Requests Projects Releases 9 Wiki Activity

add mutx

Browse Source
This commit is contained in:
兔子 2020-12-23 20:58:16 +08:00
parent b02c1f0a81
commit 455d1d7f78
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
que.go
View File

@ -32,6 +32,8 @@ type StarQueue struct {
duration time.Duration duration time.Duration
EncodeFunc func([]byte) []byte EncodeFunc func([]byte) []byte
DecodeFunc func([]byte) []byte DecodeFunc func([]byte) []byte
//parseMu sync.Mutex
restoreMu sync.Mutex
} }
// NewQueue 建立一个新消息队列 // NewQueue 建立一个新消息队列
@ -249,6 +251,8 @@ func bytesMerge(src ...[]byte) []byte {
// Restore 获取收到的信息 // Restore 获取收到的信息
func (que *StarQueue) Restore(n int) ([]MsgQueue, error) { func (que *StarQueue) Restore(n int) ([]MsgQueue, error) {
que.restoreMu.Lock()
defer que.restoreMu.Unlock()
var res []MsgQueue var res []MsgQueue
dura := time.Duration(0) dura := time.Duration(0)
for len(que.MsgPool) < n { for len(que.MsgPool) < n {