From d5bff2ad8bee23ec58a16c722cef78e9a2d4b9f0 Mon Sep 17 00:00:00 2001 From: Mrs4s <1844812067@qq.com> Date: Mon, 3 Aug 2020 13:27:18 +0800 Subject: [PATCH] fix #10 --- coolq/bot.go | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/coolq/bot.go b/coolq/bot.go index db9f3b5..a3cd444 100644 --- a/coolq/bot.go +++ b/coolq/bot.go @@ -136,16 +136,18 @@ func (bot *CQBot) InsertGroupMessage(m *message.GroupMessage) int32 { "message": ToStringMessage(m.Elements, m.GroupCode, true), } id := ToGlobalId(m.GroupCode, m.Id) - err := bot.db.Update(func(tx *nutsdb.Tx) error { - buf := new(bytes.Buffer) - if err := gob.NewEncoder(buf).Encode(val); err != nil { - return err + if bot.db != nil { + err := bot.db.Update(func(tx *nutsdb.Tx) error { + buf := new(bytes.Buffer) + if err := gob.NewEncoder(buf).Encode(val); err != nil { + return err + } + return tx.Put("group-messages", binary.ToBytes(id), binary.GZipCompress(buf.Bytes()), 0) + }) + if err != nil { + log.Warnf("记录聊天数据时出现错误: %v", err) + return -1 } - return tx.Put("group-messages", binary.ToBytes(id), binary.GZipCompress(buf.Bytes()), 0) - }) - if err != nil { - log.Warnf("记录聊天数据时出现错误: %v", err) - return -1 } return id }