shkong/go-cqhttp
1
0
Fork 0
mirror of https://github.com/Mrs4s/go-cqhttp.git synced 2025-05-06 20:13:50 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #863 from wdvxdr1123/dev.leveldb

Browse Source 
breaking change: write leveldb without gzip compress
This commit is contained in:
Mrs4s 2021-05-03 14:57:53 +08:00 committed by GitHub
commit 700ecdad06
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
coolq/bot.go
View File

@ -132,10 +132,7 @@ func (bot *CQBot) GetMessage(mid int32) MSG {
m := MSG{} m := MSG{}
data, err := bot.db.Get(binary.ToBytes(mid), nil) data, err := bot.db.Get(binary.ToBytes(mid), nil)
if err == nil { if err == nil {
buff := global.NewBuffer() err = gob.NewDecoder(bytes.NewReader(data)).Decode(&m)
defer global.PutBuffer(buff)
buff.Write(binary.GZipUncompress(data))
err = gob.NewDecoder(buff).Decode(&m)
if err == nil { if err == nil {
return m return m
} }
@ -371,11 +368,7 @@ func (bot *CQBot) InsertGroupMessage(m *message.GroupMessage) int32 {
log.Warnf("记录聊天数据时出现错误: %v", err) log.Warnf("记录聊天数据时出现错误: %v", err)
return -1 return -1
} }
gw := binary.AcquireGzipWriter() if err := bot.db.Put(binary.ToBytes(id), buf.Bytes(), nil); err != nil {
defer binary.ReleaseGzipWriter(gw)
_, _ = gw.Write(buf.Bytes())
_ = gw.Close()
if err := bot.db.Put(binary.ToBytes(id), gw.Bytes(), nil); err != nil {
log.Warnf("记录聊天数据时出现错误: %v", err) log.Warnf("记录聊天数据时出现错误: %v", err)
return -1 return -1
} }
@ -401,11 +394,7 @@ func (bot *CQBot) InsertPrivateMessage(m *message.PrivateMessage) int32 {
log.Warnf("记录聊天数据时出现错误: %v", err) log.Warnf("记录聊天数据时出现错误: %v", err)
return -1 return -1
} }
gw := binary.AcquireGzipWriter() if err := bot.db.Put(binary.ToBytes(id), buf.Bytes(), nil); err != nil {
defer binary.ReleaseGzipWriter(gw)
_, _ = gw.Write(buf.Bytes())
_ = gw.Close()
if err := bot.db.Put(binary.ToBytes(id), gw.Bytes(), nil); err != nil {
log.Warnf("记录聊天数据时出现错误: %v", err) log.Warnf("记录聊天数据时出现错误: %v", err)
return -1 return -1
} }
@ -433,11 +422,7 @@ func (bot *CQBot) InsertTempMessage(target int64, m *message.TempMessage) int32
log.Warnf("记录聊天数据时出现错误: %v", err) log.Warnf("记录聊天数据时出现错误: %v", err)
return -1 return -1
} }
gw := binary.AcquireGzipWriter() if err := bot.db.Put(binary.ToBytes(id), buf.Bytes(), nil); err != nil {
defer binary.ReleaseGzipWriter(gw)
_, _ = gw.Write(buf.Bytes())
_ = gw.Close()
if err := bot.db.Put(binary.ToBytes(id), gw.Bytes(), nil); err != nil {
log.Warnf("记录聊天数据时出现错误: %v", err) log.Warnf("记录聊天数据时出现错误: %v", err)
return -1 return -1
} }