From 455d1d7f780ec6d41083179665b302893150fe7c Mon Sep 17 00:00:00 2001
From: 兔子 <i@b612.me>
Date: Wed, 23 Dec 2020 20:58:16 +0800
Subject: [PATCH] add mutx

---
 que.go | 4 ++++
 1 file changed, 4 insertions(+)

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