From e3639eb1f32d00d5d019154013e372ac96d8be65 Mon Sep 17 00:00:00 2001
From: Ivy1996-encode <2682904957@qq.com>
Date: Sat, 20 Feb 2021 23:02:42 +0800
Subject: [PATCH] add message context support

---
 message.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/message.go b/message.go
index 5ff1a52..86b08a1 100644
--- a/message.go
+++ b/message.go
@@ -4,6 +4,7 @@ import (
 	"errors"
 	"os"
 	"strings"
+	"sync"
 	"time"
 )
 
@@ -41,6 +42,8 @@ type Message struct {
 	VoiceLength           int
 	Bot                   *Bot
 	senderInGroupUserName string
+	mu                    sync.RWMutex
+	item                  map[string]interface{}
 }
 
 // 获取消息的发送者
@@ -156,6 +159,23 @@ func (m *Message) IsSystem() bool {
 	return m.MsgType == 10000
 }
 
+// 用在多个messageHandler之间传递信息
+func (m *Message) Set(key string, value interface{}) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	if m.item == nil {
+		m.item = make(map[string]interface{})
+	}
+	m.item[key] = value
+}
+
+func (m *Message) Get(key string) (value interface{}, exist bool) {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+	value, exist = m.item[key]
+	return
+}
+
 //func (m *Message) Agree() error {
 //	if !m.IsFriendAdd() {
 //		return fmt.Errorf("the excepted message type is 37, but got %d", m.MsgType)