mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 17:02:43 +08:00
fyrestone
42 lines
1.0 KiB
Python
42 lines
1.0 KiB
Python
# This file is used by CrossLanguageInvocationTest.java to test cross-language
|
|
# invocation.
|
|
|
|
import six
|
|
|
|
import ray
|
|
|
|
|
|
@ray.remote
|
|
def py_func(value):
|
|
assert isinstance(value, bytes)
|
|
return b"Response from Python: " + value
|
|
|
|
|
|
@ray.remote
|
|
def py_func_call_java_function(value):
|
|
assert isinstance(value, bytes)
|
|
f = ray.java_function("org.ray.api.test.CrossLanguageInvocationTest",
|
|
"bytesEcho")
|
|
r = f.remote(value)
|
|
return b"[Python]py_func -> " + ray.get(r)
|
|
|
|
|
|
@ray.remote
|
|
def py_func_call_java_actor(value):
|
|
assert isinstance(value, bytes)
|
|
c = ray.java_actor_class(
|
|
"org.ray.api.test.CrossLanguageInvocationTest$TestActor")
|
|
java_actor = c.remote(b"Counter")
|
|
r = java_actor.concat.remote(value)
|
|
return ray.get(r)
|
|
|
|
|
|
@ray.remote
|
|
class Counter(object):
|
|
def __init__(self, value):
|
|
self.value = int(value)
|
|
|
|
def increase(self, delta):
|
|
self.value += int(delta)
|
|
return str(self.value).encode("utf-8") if six.PY3 else str(self.value)
|