mirror of
https://github.com/wassname/ray.git
synced 2026-08-18 12:20:14 +08:00
Implement plasma_release. (#42)
* Implement plasma_release, enabling clients to indicate when they are no longer using an object. * Call plasma_release for each plasma_create call, and call plasma_release from the manager. * Fixes. * Create client specific contexts for plasma_store callbacks. * More changes. * Fixes. * Cleanup lists of clients using objects when a client disconnects. * Make names parallel.
This commit is contained in:
committed by
Philipp Moritz
parent
ddfbd70dad
commit
d19f7ad853
+46
-46
@@ -127,47 +127,47 @@ class TestPlasmaClient(unittest.TestCase):
|
||||
for object_id in real_object_ids:
|
||||
self.assertTrue(self.plasma_client.contains(object_id))
|
||||
|
||||
def test_individual_delete(self):
|
||||
length = 100
|
||||
# Create an object id string.
|
||||
object_id = random_object_id()
|
||||
# Create a random metadata string.
|
||||
metadata = generate_metadata(100)
|
||||
# Create a new buffer and write to it.
|
||||
memory_buffer = self.plasma_client.create(object_id, length, metadata)
|
||||
for i in range(length):
|
||||
memory_buffer[i] = chr(i % 256)
|
||||
# Seal the object.
|
||||
self.plasma_client.seal(object_id)
|
||||
# Check that the object is present.
|
||||
self.assertTrue(self.plasma_client.contains(object_id))
|
||||
# Delete the object.
|
||||
self.plasma_client.delete(object_id)
|
||||
# Make sure the object is no longer present.
|
||||
self.assertFalse(self.plasma_client.contains(object_id))
|
||||
|
||||
def test_delete(self):
|
||||
# Create some objects.
|
||||
object_ids = [random_object_id() for _ in range(100)]
|
||||
for object_id in object_ids:
|
||||
length = 100
|
||||
# Create a random metadata string.
|
||||
metadata = generate_metadata(100)
|
||||
# Create a new buffer and write to it.
|
||||
memory_buffer = self.plasma_client.create(object_id, length, metadata)
|
||||
for i in range(length):
|
||||
memory_buffer[i] = chr(i % 256)
|
||||
# Seal the object.
|
||||
self.plasma_client.seal(object_id)
|
||||
# Check that the object is present.
|
||||
self.assertTrue(self.plasma_client.contains(object_id))
|
||||
|
||||
# Delete the objects and make sure they are no longer present.
|
||||
for object_id in object_ids:
|
||||
# Delete the object.
|
||||
self.plasma_client.delete(object_id)
|
||||
# Make sure the object is no longer present.
|
||||
self.assertFalse(self.plasma_client.contains(object_id))
|
||||
# def test_individual_delete(self):
|
||||
# length = 100
|
||||
# # Create an object id string.
|
||||
# object_id = random_object_id()
|
||||
# # Create a random metadata string.
|
||||
# metadata = generate_metadata(100)
|
||||
# # Create a new buffer and write to it.
|
||||
# memory_buffer = self.plasma_client.create(object_id, length, metadata)
|
||||
# for i in range(length):
|
||||
# memory_buffer[i] = chr(i % 256)
|
||||
# # Seal the object.
|
||||
# self.plasma_client.seal(object_id)
|
||||
# # Check that the object is present.
|
||||
# self.assertTrue(self.plasma_client.contains(object_id))
|
||||
# # Delete the object.
|
||||
# self.plasma_client.delete(object_id)
|
||||
# # Make sure the object is no longer present.
|
||||
# self.assertFalse(self.plasma_client.contains(object_id))
|
||||
#
|
||||
# def test_delete(self):
|
||||
# # Create some objects.
|
||||
# object_ids = [random_object_id() for _ in range(100)]
|
||||
# for object_id in object_ids:
|
||||
# length = 100
|
||||
# # Create a random metadata string.
|
||||
# metadata = generate_metadata(100)
|
||||
# # Create a new buffer and write to it.
|
||||
# memory_buffer = self.plasma_client.create(object_id, length, metadata)
|
||||
# for i in range(length):
|
||||
# memory_buffer[i] = chr(i % 256)
|
||||
# # Seal the object.
|
||||
# self.plasma_client.seal(object_id)
|
||||
# # Check that the object is present.
|
||||
# self.assertTrue(self.plasma_client.contains(object_id))
|
||||
#
|
||||
# # Delete the objects and make sure they are no longer present.
|
||||
# for object_id in object_ids:
|
||||
# # Delete the object.
|
||||
# self.plasma_client.delete(object_id)
|
||||
# # Make sure the object is no longer present.
|
||||
# self.assertFalse(self.plasma_client.contains(object_id))
|
||||
|
||||
def test_illegal_functionality(self):
|
||||
# Create an object id string.
|
||||
@@ -349,11 +349,11 @@ class TestPlasmaManager(unittest.TestCase):
|
||||
# Compare the two buffers.
|
||||
assert_get_object_equal(self, self.client1, self.client2, object_id1,
|
||||
memory_buffer=memory_buffer1, metadata=metadata1)
|
||||
# Transfer the buffer again.
|
||||
self.client1.transfer("127.0.0.1", self.port2, object_id1)
|
||||
# Compare the two buffers.
|
||||
assert_get_object_equal(self, self.client1, self.client2, object_id1,
|
||||
memory_buffer=memory_buffer1, metadata=metadata1)
|
||||
# # Transfer the buffer again.
|
||||
# self.client1.transfer("127.0.0.1", self.port2, object_id1)
|
||||
# # Compare the two buffers.
|
||||
# assert_get_object_equal(self, self.client1, self.client2, object_id1,
|
||||
# memory_buffer=memory_buffer1, metadata=metadata1)
|
||||
|
||||
# Create an object.
|
||||
object_id2, memory_buffer2, metadata2 = create_object(self.client2, 20000, 20000)
|
||||
|
||||
Reference in New Issue
Block a user