Преглед на файлове

Message

- refactoring: GetBitmap() *bitmap.Bitmap
Alexey Kim преди 1 година
родител
ревизия
028dc50dde
променени са 1 файла, в които са добавени 7 реда и са изтрити 2 реда
  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
 }