wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-02 01:20:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

clean up tests

Browse Source
This commit is contained in:
Philipp Moritz
2016-08-22 15:30:16 -07:00
parent 3d6d1e8fe5
commit e1627319b3
5 changed files with 45 additions and 189 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/python/plasma.py
+4 -12
View File
@@ -16,12 +16,9 @@ PLASMA_SEAL = 2
PLASMA_TRANSFER = 3
PLASMA_DATA = 4
PLASMA_REGISTER = 5
PLASMA_GET_MANAGER_PORT = 6
PLASMA_RETURN_MANAGER_PORT = 7
class PlasmaRequest(ctypes.Structure):
_fields_ = [("type", ctypes.c_int),
("manager_id", ctypes.c_int),
("object_id", PlasmaID),
("size", ctypes.c_int64),
("addr", Addr),
@@ -57,16 +54,11 @@ class PlasmaManager(object):
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.sock.connect((addr, port))
def register(self, manager_id, addr, port):
"""Register another object manager."""
req = PlasmaRequest(type=PLASMA_REGISTER, manager_id=manager_id,
addr=Addr(*map(int, addr.split("."))), port=port)
self.sock.send(buffer(req)[:])
def transfer(self, manager_id, object_id):
def transfer(self, addr, port, object_id):
"""Transfer local object with id object_id to manager with id manager_id."""
req = PlasmaRequest(type=PLASMA_TRANSFER, manager_id=manager_id,
object_id=make_plasma_id(object_id))
req = PlasmaRequest(type=PLASMA_TRANSFER, object_id=make_plasma_id(object_id),
addr=Addr(*map(int, addr.split("."))), port=port)
print "sending port", port
self.sock.send(buffer(req)[:])
class PlasmaClient(object):