Special-case handling of memoryviews when sending (fixes #14)

Thanks to memeplex.
nanomsg · Dec 15, 2015 · e0c4734 · e0c4734
1 parent e481f9e
commit e0c4734
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/nnpy/socket.py b/nnpy/socket.py
@@ -47,10 +47,12 @@ def connect(self, addr):
         assert rc > 0, rc
 
     def send(self, data, flags=0):
-        data = data.encode() if isinstance(data, str) else data
-        l = len(data)
-        buf = ffi.new('char[%i]' % l, data)
-        rc = nanomsg.nn_send(self.sock, buf, l, flags)
+        if isinstance(data, memoryview):
+            buf = ffi.from_buffer(data)
+        else:
+            data = data.encode() if isinstance(data, str) else data
+            buf = ffi.new('char[%i]' % len(data), data)
+        rc = nanomsg.nn_send(self.sock, buf, len(data), flags)
         assert rc > 0, rc
 
     def recv(self, flags=0):