Explorar o código

Message

- refactoring: GetBitmap() *bitmap.Bitmap
Alexey Kim hai 1 ano
pai
achega
028dc50dde
Modificáronse 1 ficheiros con 7 adicións e 2 borrados
  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
 }