disable refcounting when not used

Author: Noam Preil

arrow/ipc/message.go

@@ -86,8 +86,8 @@ func NewMessage(meta, body *memory.Buffer) *Message {
 		meta: meta,
 		body: body,
 	}
-	m.Refcount.Buffers = []**memory.Buffer{&m.meta, &m.body}
-	m.Derived = []unsafe.Pointer{unsafe.Pointer(&m.msg)}
+	m.ReferenceBuffer(&m.meta, &m.body)
+	m.ReferenceDerived(unsafe.Pointer(&m.msg))
 	m.Retain()
 	return m
 }
@@ -102,8 +102,8 @@ func newMessageFromFB(meta *flatbuf.Message, body *memory.Buffer) *Message {
 		meta: memory.NewBufferBytes(meta.Table().Bytes),
 		body: body,
 	}
-	m.Refcount.Buffers = []**memory.Buffer{&m.meta, &m.body}
-	m.Derived = []unsafe.Pointer{unsafe.Pointer(&m.msg)}
+	m.ReferenceBuffer(&m.meta, &m.body)
+	m.ReferenceDerived(unsafe.Pointer(&m.msg))
 	m.Retain()
 	return m
 }

arrow/memory/refcount.go

@@ -1,3 +1,5 @@
+//go:build refcounting
+
 package memory
 
 import (
@@ -14,6 +16,37 @@ type Refcount struct {
 	Derived []unsafe.Pointer
 }
 
+func (r *Refcount) ReferenceDependency(d ...**Refcount) {
+	if r.Dependencies == nil {
+		r.Dependencies = d
+	} else {
+		for _, d := range d {
+			r.Dependencies = append(r.Dependencies, d)
+		}
+	}
+}
+
+func (r *Refcount) ReferenceBuffer(b...**Buffer) {
+	if r.Buffers == nil {
+		r.Buffers = b
+	} else {
+		for _, b := range b {
+			r.Buffers = append(r.Buffers, b)
+		}
+	}
+}
+
+
+func (r *Refcount) ReferenceDerived(p ...unsafe.Pointer) {
+	if r.Derived == nil {
+		r.Derived = p
+	} else {
+		for _, p := range p {
+			r.Derived = append(r.Derived, p)
+		}
+	}
+}
+
 func (r *Refcount) Retain() {
 	r.count.Add(1)
 }

arrow/memory/refcount_disabled.go

@@ -0,0 +1,11 @@
+//go:build !refcounting
+
+package memory
+
+type Refcount struct{}
+
+func (r *Refcount) ReferenceDependency(d ...any) {}
+func (r *Refcount) ReferenceBuffer(b ...any) {}
+func (r *Refcount) ReferenceDerived(p ...any) {}
+func (r *Refcount) Retain() {}
+func (r *Refcount) Release() {}