浏览代码

Message

- GetBitmap: prevent nil
Alexey Kim 1 年之前
父节点
当前提交
8296246895
共有 1 个文件被更改,包括 3 次插入3 次删除
  1. 3 3
      message.go

+ 3 - 3
message.go

@@ -31,10 +31,10 @@ func (m *Message) SetBitmap(flags bitmap.Bitmap) {
 }
 
 func (m *Message) GetBitmap() *bitmap.Bitmap {
-	if m.Flags == nil {
-		return nil
+	var flags bitmap.Bitmap
+	if m.Flags != nil {
+		flags = bitmap.FromBytes(m.Flags)
 	}
 
-	flags := bitmap.FromBytes(m.Flags)
 	return &flags
 }