From 3905052b68f9e39cb366513841a49e9effc09664 Mon Sep 17 00:00:00 2001 From: wdvxdr Date: Thu, 29 Apr 2021 21:14:14 +0800 Subject: [PATCH] feat: write leveldb without gzip compress --- coolq/bot.go | 23 ++++------------------- 1 file changed, 4 insertions(+), 19 deletions(-) diff --git a/coolq/bot.go b/coolq/bot.go index 2b69c3d..23ef87b 100644 --- a/coolq/bot.go +++ b/coolq/bot.go @@ -132,10 +132,7 @@ func (bot *CQBot) GetMessage(mid int32) MSG { m := MSG{} data, err := bot.db.Get(binary.ToBytes(mid), nil) if err == nil { - buff := global.NewBuffer() - defer global.PutBuffer(buff) - buff.Write(binary.GZipUncompress(data)) - err = gob.NewDecoder(buff).Decode(&m) + err = gob.NewDecoder(bytes.NewReader(data)).Decode(&m) if err == nil { return m } @@ -371,11 +368,7 @@ func (bot *CQBot) InsertGroupMessage(m *message.GroupMessage) int32 { log.Warnf("记录聊天数据时出现错误: %v", err) return -1 } - gw := binary.AcquireGzipWriter() - defer binary.ReleaseGzipWriter(gw) - _, _ = gw.Write(buf.Bytes()) - _ = gw.Close() - if err := bot.db.Put(binary.ToBytes(id), gw.Bytes(), nil); err != nil { + if err := bot.db.Put(binary.ToBytes(id), buf.Bytes(), nil); err != nil { log.Warnf("记录聊天数据时出现错误: %v", err) return -1 } @@ -401,11 +394,7 @@ func (bot *CQBot) InsertPrivateMessage(m *message.PrivateMessage) int32 { log.Warnf("记录聊天数据时出现错误: %v", err) return -1 } - gw := binary.AcquireGzipWriter() - defer binary.ReleaseGzipWriter(gw) - _, _ = gw.Write(buf.Bytes()) - _ = gw.Close() - if err := bot.db.Put(binary.ToBytes(id), gw.Bytes(), nil); err != nil { + if err := bot.db.Put(binary.ToBytes(id), buf.Bytes(), nil); err != nil { log.Warnf("记录聊天数据时出现错误: %v", err) return -1 } @@ -433,11 +422,7 @@ func (bot *CQBot) InsertTempMessage(target int64, m *message.TempMessage) int32 log.Warnf("记录聊天数据时出现错误: %v", err) return -1 } - gw := binary.AcquireGzipWriter() - defer binary.ReleaseGzipWriter(gw) - _, _ = gw.Write(buf.Bytes()) - _ = gw.Close() - if err := bot.db.Put(binary.ToBytes(id), gw.Bytes(), nil); err != nil { + if err := bot.db.Put(binary.ToBytes(id), buf.Bytes(), nil); err != nil { log.Warnf("记录聊天数据时出现错误: %v", err) return -1 }