Răsfoiți Sursa

Message

- refactoring: GetBitmap() *bitmap.Bitmap
Alexey Kim 1 an în urmă
părinte
comite
028dc50dde
1 a modificat fișierele cu 7 adăugiri și 2 ștergeri
  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
 }