Fix crash from float timeouts

uber · Apr 7, 2016 · 12adfdb · 12adfdb
1 parent 721789a
commit 12adfdb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -4,7 +4,7 @@ Changes by Version
 0.22.3 (unreleased)
 -------------------
 
-- Nothing changed yet.
+- Fixed a bug where type mismatch for timeouts could cause a crash.
 
 
 0.22.2 (2016-04-06)

diff --git a/tchannel/rw.py b/tchannel/rw.py
@@ -363,7 +363,8 @@ def read(self, stream):
         return struct.unpack(self._format, self.take(stream, self._width))[0]
 
     def write(self, num, stream):
-        stream.write(struct.pack(self._format, num))
+        # Cast to int just in case the value is still a float
+        stream.write(struct.pack(self._format, int(num)))
         return stream
 
     def width(self):