From 5a0e9715f45b62cfe43e6873b8828f49ab73f710 Mon Sep 17 00:00:00 2001
From: Max Baryshnikov <mephius@gmail.com>
Date: Tue, 7 Feb 2017 22:31:06 +0300
Subject: Fixed couple of "leaks" when gc is disabled (#979)

---
 kafka/protocol/message.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'kafka/protocol/message.py')

diff --git a/kafka/protocol/message.py b/kafka/protocol/message.py
index bfad127..ec5ee6c 100644
--- a/kafka/protocol/message.py
+++ b/kafka/protocol/message.py
@@ -10,7 +10,7 @@ from .struct import Struct
 from .types import (
     Int8, Int32, Int64, Bytes, Schema, AbstractType
 )
-from ..util import crc32
+from ..util import crc32, WeakMethod
 
 
 class Message(Struct):
@@ -52,7 +52,7 @@ class Message(Struct):
         self.attributes = attributes
         self.key = key
         self.value = value
-        self.encode = self._encode_self
+        self.encode = WeakMethod(self._encode_self)
 
     @property
     def timestamp_type(self):
-- 
cgit v1.2.1