Browse Source

Message

- refactoring: GetBitmap() *bitmap.Bitmap
Alexey Kim 1 year ago
parent
commit
028dc50dde
1 changed files with 7 additions and 2 deletions
  1. 7 2
      message.go

+ 7 - 2
message.go

@@ -30,6 +30,11 @@ func (m *Message) SetBitmap(flags bitmap.Bitmap) {
 	m.Flags = flags.ToBytes()
 }
 
-func (m *Message) GetBitmap() bitmap.Bitmap {
-	return bitmap.FromBytes(m.Flags)
+func (m *Message) GetBitmap() *bitmap.Bitmap {
+	if m.Flags == nil {
+		return nil
+	}
+
+	flags := bitmap.FromBytes(m.Flags)
+	return &flags
 }