瀏覽代碼

Message Flags

- prevent direct access
Alexey Kim 1 年之前
父節點
當前提交
d8cfb3fb33
共有 1 個文件被更改,包括 12 次插入1 次删除
  1. 12 1
      message.go

+ 12 - 1
message.go

@@ -26,11 +26,22 @@ func NewKafkaMessage(src *kafka.Message) *Message {
 	return m
 }
 
+func (m *Message) SetFlag(x uint32) {
+	f := m.bitmap()
+	f.Set(x)
+
+	m.Flags = f.ToBytes()
+}
+
+func (m *Message) HasFlag(x uint32) bool {
+	return m.bitmap().Contains(x)
+}
+
 func (m *Message) SetBitmap(flags bitmap.Bitmap) {
 	m.Flags = flags.ToBytes()
 }
 
-func (m *Message) GetBitmap() *bitmap.Bitmap {
+func (m *Message) bitmap() *bitmap.Bitmap {
 	var flags bitmap.Bitmap
 	if m.Flags != nil {
 		flags = bitmap.FromBytes(m.Flags)