diff --git a/.bazelrc b/.bazelrc index 6303c509b6bc9..80bffe32363c9 100644 --- a/.bazelrc +++ b/.bazelrc @@ -8,6 +8,11 @@ build:windows --action_env=PATH # For --compilation_mode=dbg, consider enabling checks in the standard library as well (below). build --compilation_mode=opt # Using C++ 17 on all platforms. +build:linux --host_cxxopt="-std=c++17" +build:macos --host_cxxopt="-std=c++17" +build:clang-cl --host_cxxopt="-std=c++17" +build:msvc-cl --host_cxxopt="/std:c++17" +build:windows --host_cxxopt="/std:c++17" build:linux --cxxopt="-std=c++17" build:macos --cxxopt="-std=c++17" build:clang-cl --cxxopt="-std=c++17" @@ -187,12 +192,12 @@ build:debug --strip="never" # Undefined Behavior Sanitizer build:ubsan --strip=never -build:ubsan --copt -fsanitize=undefined build:ubsan --copt -fno-sanitize=vptr build:ubsan --copt -fno-sanitize-recover=all build:ubsan --copt -g build:ubsan --linkopt -fsanitize=undefined build:ubsan --linkopt -fno-sanitize-recover=all +build:ubsan --per_file_copt="-external/com_github_grpc_grpc/.*@-fsanitize=undefined" # Import local specific llvm config options, which can be generated by # ci/env/install-llvm-dependencies.sh diff --git a/.buildkite/lint.rayci.yml b/.buildkite/lint.rayci.yml index b0f81bab31748..027b39ee0b02a 100644 --- a/.buildkite/lint.rayci.yml +++ b/.buildkite/lint.rayci.yml @@ -72,3 +72,9 @@ steps: commands: - ./ci/lint/check-documentation-style.sh depends_on: forge + + - label: ":lint-roller: lint: python proto" + commands: + - pip install -r ci/compile_py_proto/requirements_compile_py_proto.txt + - ./ci/lint/check-python-proto.sh + depends_on: forge diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000000000..b21c9229afef1 --- /dev/null +++ b/.gitattributes @@ -0,0 +1,3 @@ +/python/ray/core/generated/**/* linguist-generated=true +/python/ray/serve/generated/**/* linguist-generated=true + diff --git a/.gitignore b/.gitignore index da77afa2b054e..058ecfd573d9c 100644 --- a/.gitignore +++ b/.gitignore @@ -1,7 +1,7 @@ # The build output should clearly not be checked in *test-output.xml /bazel-* -/python/ray/core +/python/ray/core/**/* /python/ray/pickle5_files/ /python/ray/thirdparty_files/ /python/ray/_private/runtime_env/agent/thirdparty_files/ @@ -98,6 +98,10 @@ python/ray/_raylet.pyd *_pb2.py *.pb.h *.pb.cc +!/python/ray/core/generated/ +!/python/ray/core/generated/*.py +!/python/ray/serve/generated/ +!/python/ray/serve/generated/*.py # Ray cluster configuration scripts/nodes.txt diff --git a/BUILD.bazel b/BUILD.bazel index c3f6d6a141f88..196eaceb12459 100644 --- a/BUILD.bazel +++ b/BUILD.bazel @@ -2517,45 +2517,11 @@ copy_to_workspace( dstdir = "python/ray/core/", ) - -genrule( - name = "install_py_proto", - srcs = [ - ":cp_all_py_proto", - ":cp_serve_py_proto", - ], - outs = ["install_py_proto.out"], - cmd = """ - # NOTE(hchen): Protobuf doesn't allow specifying Python package name. So we use this `sed` - # command to change the import path in the generated file. - # shellcheck disable=SC2006 - files=( - `ls python/ray/core/generated/*_pb2*.py` \ - `ls python/ray/serve/generated/*_pb2*.py` \ - ) - sed -i -E 's/from src.ray.protobuf/from ./' "$${files[@]}" - # Help the generated serve files to have the correct module - serve_files=(`ls python/ray/serve/generated/*_pb2*.py`) - sed -i -E 's/'"'"'src.ray.protobuf./'"'"'ray.serve.generated./' "$${serve_files[@]}" - # TODO(sang): Build our own proto instead of creating a new proto for opencensus separately. - # https://github.com/ray-project/ray/issues/31358 - sed -i -E 's/from opencensus.proto.metrics.v1 import/from . import/' "$${files[@]}" - sed -i -E 's/from opencensus.proto.resource.v1 import/from . import/' "$${files[@]}" - # TODO(rickyx): Remove this experimental path rewrite once we move autoscaler proto out - # of experimental. - autoscale_files=(`ls python/ray/core/generated/autoscaler_pb2*.py`) - sed -i -E 's/from ..experimental/from ./' "$${autoscale_files[@]}" - echo "$${PWD}" > $@ - """, - local = 1, -) - genrule( name = "ray_pkg", srcs = [ ":cp_raylet_so", ":python_sources", - ":install_py_proto", ":cp_redis", ":cp_raylet", ":cp_gcs_server", @@ -2574,4 +2540,3 @@ genrule( """, local = 1, ) - diff --git a/WORKSPACE b/WORKSPACE index e474868b98b45..b30eb1a133421 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -64,13 +64,22 @@ pip_parse( requirements_lock = "//release:requirements_buildkite.txt", ) +pip_parse( + name = "py_deps_compile_py_proto", + python_interpreter_target = python39, + requirements_lock = "//ci/compile_py_proto:requirements_compile_py_proto.txt", +) + load("@py_deps_buildkite//:requirements.bzl", install_py_deps_buildkite = "install_deps") load("@py_deps_ray_ci//:requirements.bzl", install_py_deps_ray_ci = "install_deps") +load("@py_deps_compile_py_proto//:requirements.bzl", install_py_deps_compile_py_proto = "install_deps") install_py_deps_buildkite() install_py_deps_ray_ci() +install_py_deps_compile_py_proto() + register_toolchains("//:python_toolchain") register_execution_platforms( diff --git a/bazel/ray_deps_setup.bzl b/bazel/ray_deps_setup.bzl index cd04c53f2c91a..66f63168a6c16 100644 --- a/bazel/ray_deps_setup.bzl +++ b/bazel/ray_deps_setup.bzl @@ -86,11 +86,19 @@ def auto_http_archive( def ray_deps_setup(): # Explicitly bring in protobuf dependency to work around # https://github.com/ray-project/ray/issues/14117 + # This is copied from grpc's bazel/grpc_deps.bzl http_archive( name = "com_google_protobuf", - strip_prefix = "protobuf-3.19.4", - urls = ["https://github.com/protocolbuffers/protobuf/archive/v3.19.4.tar.gz"], - sha256 = "3bd7828aa5af4b13b99c191e8b1e884ebfa9ad371b0ce264605d347f135d2568", + sha256 = "76a33e2136f23971ce46c72fd697cd94dc9f73d56ab23b753c3e16854c90ddfd", + strip_prefix = "protobuf-2c5fa078d8e86e5f4bd34e6f4c9ea9e8d7d4d44a", + urls = [ + # https://github.com/protocolbuffers/protobuf/commits/v23.4 + "https://github.com/protocolbuffers/protobuf/archive/2c5fa078d8e86e5f4bd34e6f4c9ea9e8d7d4d44a.tar.gz", + ], + patches = [ + "@com_github_grpc_grpc//third_party:protobuf.patch", + ], + patch_args = ["-p1"], ) # NOTE(lingxuan.zlx): 3rd party dependencies could be accessed, so it suggests @@ -205,15 +213,18 @@ def ray_deps_setup(): "@com_github_ray_project_ray//thirdparty/patches:opencensus-cpp-harvest-interval.patch", "@com_github_ray_project_ray//thirdparty/patches:opencensus-cpp-shutdown-api.patch", ], - patch_args = ["-p1"], + patch_args = ["-p1"], ) # OpenCensus depends on Abseil so we have to explicitly pull it in. # This is how diamond dependencies are prevented. auto_http_archive( name = "com_google_absl", - url = "https://github.com/abseil/abseil-cpp/archive/refs/tags/20220623.1.tar.gz", - sha256 = "91ac87d30cc6d79f9ab974c51874a704de9c2647c40f6932597329a282217ba8", + sha256 = "5366d7e7fa7ba0d915014d387b66d0d002c03236448e1ba9ef98122c13b35c36", + strip_prefix = "abseil-cpp-20230125.3", + urls = [ + "https://github.com/abseil/abseil-cpp/archive/20230125.3.tar.gz", + ], ) # OpenCensus depends on jupp0r/prometheus-cpp @@ -232,14 +243,13 @@ def ray_deps_setup(): auto_http_archive( name = "com_github_grpc_grpc", # NOTE: If you update this, also update @boringssl's hash. - url = "https://github.com/grpc/grpc/archive/refs/tags/v1.46.6.tar.gz", - sha256 = "6514b3e6eab9e9c7017304512d4420387a47b1a9c5caa986643692977ed44e8a", + url = "https://github.com/grpc/grpc/archive/refs/tags/v1.57.0.tar.gz", + sha256 = "8393767af531b2d0549a4c26cf8ba1f665b16c16fb6c9238a7755e45444881dd", patches = [ "@com_github_ray_project_ray//thirdparty/patches:grpc-cython-copts.patch", - "@com_github_ray_project_ray//thirdparty/patches:grpc-python.patch", ], ) - + http_archive( name = "openssl", strip_prefix = "openssl-1.1.1f", @@ -249,7 +259,7 @@ def ray_deps_setup(): ], build_file = "@rules_foreign_cc_thirdparty//openssl:BUILD.openssl.bazel", ) - + http_archive( name = "rules_foreign_cc", sha256 = "2a4d07cd64b0719b39a7c12218a3e507672b82a97b98c6a89d38565894cf7c51", @@ -279,11 +289,11 @@ def ray_deps_setup(): # https://github.com/grpc/grpc/blob/1ff1feaa83e071d87c07827b0a317ffac673794f/bazel/grpc_deps.bzl#L189 # Ensure this rule matches the rule used by grpc's bazel/grpc_deps.bzl name = "boringssl", - sha256 = "534fa658bd845fd974b50b10f444d392dfd0d93768c4a51b61263fd37d851c40", - strip_prefix = "boringssl-b9232f9e27e5668bc0414879dcdedb2a59ea75f2", + sha256 = "0675a4f86ce5e959703425d6f9063eaadf6b61b7f3399e77a154c0e85bad46b1", + strip_prefix = "boringssl-342e805bc1f5dfdd650e3f031686d6c939b095d9", urls = [ - "https://storage.googleapis.com/grpc-bazel-mirror/github.com/google/boringssl/archive/b9232f9e27e5668bc0414879dcdedb2a59ea75f2.tar.gz", - "https://github.com/google/boringssl/archive/b9232f9e27e5668bc0414879dcdedb2a59ea75f2.tar.gz", + "https://storage.googleapis.com/grpc-bazel-mirror/github.com/google/boringssl/archive/342e805bc1f5dfdd650e3f031686d6c939b095d9.tar.gz", + "https://github.com/google/boringssl/archive/342e805bc1f5dfdd650e3f031686d6c939b095d9.tar.gz", ], ) diff --git a/ci/ci.sh b/ci/ci.sh index 984cb16964155..b146cee9977ad 100755 --- a/ci/ci.sh +++ b/ci/ci.sh @@ -344,7 +344,6 @@ build_dashboard_front_end() { } build_sphinx_docs() { - _bazel_build_protobuf install_ray ( @@ -394,10 +393,6 @@ _bazel_build_before_install() { } -_bazel_build_protobuf() { - bazel build "//:install_py_proto" -} - install_ray() { # TODO(mehrdadn): This function should be unified with the one in python/build-wheel-windows.sh. ( diff --git a/ci/compile_py_proto/BUILD.bazel b/ci/compile_py_proto/BUILD.bazel new file mode 100644 index 0000000000000..be22d7a6accc5 --- /dev/null +++ b/ci/compile_py_proto/BUILD.bazel @@ -0,0 +1,106 @@ +load("@rules_python//python:pip.bzl", "compile_pip_requirements") +load("@rules_python//python:defs.bzl", "py_binary") +load("@py_deps_compile_py_proto//:requirements.bzl", "requirement") + +compile_pip_requirements( + name = "requirements_compile_py_proto", + requirements_in = "requirements_compile_py_proto.in", + requirements_txt = "requirements_compile_py_proto.txt", + extra_args = ["--allow-unsafe"], + tags = ["team:ci"], + visibility = ["//visibility:public"], +) + +py_binary( + name = "grpc_tools_protoc", + srcs = ["grpc_tools_protoc.py"], + deps = [ + requirement("grpcio-tools"), + ], + exec_compatible_with = ["//:hermetic_python"], + tags = ["team:ci"], + visibility = ["//visibility:public"], +) + +genrule( + name = "compile_py_proto", + srcs = [ + "@io_opencensus_proto//opencensus/proto/metrics/v1:metrics.proto", + "@io_opencensus_proto//opencensus/proto/resource/v1:resource.proto", + "//src/ray/protobuf:core_protos", + "//src/ray/protobuf:serve_protos", + ], + cmd = """ + TMP=$$(mktemp -d) + + mkdir -p $${TMP}/core + $(location :grpc_tools_protoc) --python_out=$${TMP}/core --grpc_python_out=$${TMP}/core $(locations //src/ray/protobuf:core_protos) --proto_path=. --proto_path=$$(dirname $(locations @io_opencensus_proto//opencensus/proto/metrics/v1:metrics.proto))/../../../.. + mkdir -p opencensus/proto/metrics/v1 + mkdir -p opencensus/proto/resource/v1 + cp $(locations @io_opencensus_proto//opencensus/proto/metrics/v1:metrics.proto) opencensus/proto/metrics/v1/metrics.proto + cp $(locations @io_opencensus_proto//opencensus/proto/resource/v1:resource.proto) opencensus/proto/resource/v1/resource.proto + $(location :grpc_tools_protoc) --python_out=$${TMP}/core --grpc_python_out=$${TMP}/core opencensus/proto/metrics/v1/metrics.proto opencensus/proto/resource/v1/resource.proto --proto_path=. + mv $${TMP}/core/src/ray/protobuf/*.py $${TMP}/core/ + mv $${TMP}/core/src/ray/protobuf/**/*.py $${TMP}/core/ + mv $${TMP}/core/opencensus/proto/**/v1/*.py $${TMP}/core/ + + mkdir -p $${TMP}/serve + $(location :grpc_tools_protoc) --python_out=$${TMP}/serve --grpc_python_out=$${TMP}/serve $(locations //src/ray/protobuf:serve_protos) --proto_path=. + mv $${TMP}/serve/src/ray/protobuf/*.py $${TMP}/serve/ + + # NOTE(hchen): Protobuf doesn't allow specifying Python package name. So we use this `sed` + # command to change the import path in the generated file. + # shellcheck disable=SC2006 + files=( + `ls $${TMP}/*/*_pb2*.py` + ) + sed -i -E 's/from src.ray.protobuf/from ./' "$${files[@]}" + # TODO(sang): Build our own proto instead of creating a new proto for opencensus separately. + # https://github.com/ray-project/ray/issues/31358 + sed -i -E 's/from opencensus.proto.metrics.v1 import/from . import/' "$${files[@]}" + sed -i -E 's/from opencensus.proto.resource.v1 import/from . import/' "$${files[@]}" + # TODO(rickyx): Remove this experimental path rewrite once we move instance manager proto out + # of experimental. + autoscale_files=(`ls $${TMP}/core/instance_manager_pb2*.py`) + sed -i -E 's/from ..experimental/from ./' "$${autoscale_files[@]}" + + # Help the generated serve files to have the correct module + serve_files=(`ls $${TMP}/serve/*_pb2*.py`) + sed -i -E 's/'"'"'src.ray.protobuf./'"'"'ray.serve.generated./' "$${serve_files[@]}" + + pushd $${TMP}/core + tar -cvf compile_py_proto_core.tar *.py + popd + + pushd $${TMP}/serve + tar -cvf compile_py_proto_serve.tar *.py + popd + + mv $${TMP}/core/compile_py_proto_core.tar $(location :compile_py_proto_core.tar) + mv $${TMP}/serve/compile_py_proto_serve.tar $(location :compile_py_proto_serve.tar) + rm -r $${TMP} + """, + tools = [ + ":grpc_tools_protoc", + ], + outs = [ + "compile_py_proto_core.tar", + "compile_py_proto_serve.tar" + ], + visibility = ["//visibility:public"], +) + +sh_binary( + name = "install_py_proto", + srcs = [ + ":install_py_proto.sh", + ], + data = [ + ":compile_py_proto_core.tar", + ":compile_py_proto_serve.tar", + ], + args = [ + "$(location :compile_py_proto_core.tar)", + "$(location :compile_py_proto_serve.tar)", + ], +) diff --git a/ci/compile_py_proto/grpc_tools_protoc.py b/ci/compile_py_proto/grpc_tools_protoc.py new file mode 100644 index 0000000000000..7647c77c22ac9 --- /dev/null +++ b/ci/compile_py_proto/grpc_tools_protoc.py @@ -0,0 +1,7 @@ +import subprocess +import sys + +if __name__ == "__main__": + args = [sys.executable, "-m", "grpc_tools.protoc"] + args.extend(sys.argv[1:]) + subprocess.check_call(args, stdout=sys.stdout, stderr=sys.stderr) diff --git a/ci/compile_py_proto/install_py_proto.sh b/ci/compile_py_proto/install_py_proto.sh new file mode 100755 index 0000000000000..979172496c7e3 --- /dev/null +++ b/ci/compile_py_proto/install_py_proto.sh @@ -0,0 +1,9 @@ +#!/bin/bash + +set -euo pipefail + +mkdir -p "${BUILD_WORKSPACE_DIRECTORY}"/python/ray/core/generated +tar -xvf "$1" -C "${BUILD_WORKSPACE_DIRECTORY}"/python/ray/core/generated + +mkdir -p "${BUILD_WORKSPACE_DIRECTORY}"/python/ray/serve/generated +tar -xvf "$2" -C "${BUILD_WORKSPACE_DIRECTORY}"/python/ray/serve/generated diff --git a/ci/compile_py_proto/requirements_compile_py_proto.in b/ci/compile_py_proto/requirements_compile_py_proto.in new file mode 100644 index 0000000000000..394ff04f1f105 --- /dev/null +++ b/ci/compile_py_proto/requirements_compile_py_proto.in @@ -0,0 +1,4 @@ +# This uses protoc v19 and +# is compatible with python protobuf 3.19 +grpcio-tools==1.46.5 +setuptools diff --git a/ci/compile_py_proto/requirements_compile_py_proto.txt b/ci/compile_py_proto/requirements_compile_py_proto.txt new file mode 100644 index 0000000000000..117810876c077 --- /dev/null +++ b/ci/compile_py_proto/requirements_compile_py_proto.txt @@ -0,0 +1,141 @@ +# +# This file is autogenerated by pip-compile with python 3.9 +# To update, run: +# +# bazel run //ci/compile_py_proto:requirements_compile_py_proto.update +# +grpcio==1.57.0 \ + --hash=sha256:00258cbe3f5188629828363ae8ff78477ce976a6f63fb2bb5e90088396faa82e \ + --hash=sha256:092fa155b945015754bdf988be47793c377b52b88d546e45c6a9f9579ac7f7b6 \ + --hash=sha256:0f80bf37f09e1caba6a8063e56e2b87fa335add314cf2b78ebf7cb45aa7e3d06 \ + --hash=sha256:20ec6fc4ad47d1b6e12deec5045ec3cd5402d9a1597f738263e98f490fe07056 \ + --hash=sha256:2313b124e475aa9017a9844bdc5eafb2d5abdda9d456af16fc4535408c7d6da6 \ + --hash=sha256:23e7d8849a0e58b806253fd206ac105b328171e01b8f18c7d5922274958cc87e \ + --hash=sha256:2f708a6a17868ad8bf586598bee69abded4996b18adf26fd2d91191383b79019 \ + --hash=sha256:2f7349786da979a94690cc5c2b804cab4e8774a3cf59be40d037c4342c906649 \ + --hash=sha256:34950353539e7d93f61c6796a007c705d663f3be41166358e3d88c45760c7d98 \ + --hash=sha256:40b72effd4c789de94ce1be2b5f88d7b9b5f7379fe9645f198854112a6567d9a \ + --hash=sha256:4b089f7ad1eb00a104078bab8015b0ed0ebcb3b589e527ab009c53893fd4e613 \ + --hash=sha256:4faea2cfdf762a664ab90589b66f416274887641ae17817de510b8178356bf73 \ + --hash=sha256:5371bcd861e679d63b8274f73ac281751d34bd54eccdbfcd6aa00e692a82cd7b \ + --hash=sha256:5613a2fecc82f95d6c51d15b9a72705553aa0d7c932fad7aed7afb51dc982ee5 \ + --hash=sha256:57b183e8b252825c4dd29114d6c13559be95387aafc10a7be645462a0fc98bbb \ + --hash=sha256:5b7a4ce8f862fe32b2a10b57752cf3169f5fe2915acfe7e6a1e155db3da99e79 \ + --hash=sha256:5e5b58e32ae14658085c16986d11e99abd002ddbf51c8daae8a0671fffb3467f \ + --hash=sha256:60fe15288a0a65d5c1cb5b4a62b1850d07336e3ba728257a810317be14f0c527 \ + --hash=sha256:6907b1cf8bb29b058081d2aad677b15757a44ef2d4d8d9130271d2ad5e33efca \ + --hash=sha256:76c44efa4ede1f42a9d5b2fed1fe9377e73a109bef8675fb0728eb80b0b8e8f2 \ + --hash=sha256:7a635589201b18510ff988161b7b573f50c6a48fae9cb567657920ca82022b37 \ + --hash=sha256:7b400807fa749a9eb286e2cd893e501b110b4d356a218426cb9c825a0474ca56 \ + --hash=sha256:82640e57fb86ea1d71ea9ab54f7e942502cf98a429a200b2e743d8672171734f \ + --hash=sha256:871f9999e0211f9551f368612460442a5436d9444606184652117d6a688c9f51 \ + --hash=sha256:9338bacf172e942e62e5889b6364e56657fbf8ac68062e8b25c48843e7b202bb \ + --hash=sha256:a8a8e560e8dbbdf29288872e91efd22af71e88b0e5736b0daf7773c1fecd99f0 \ + --hash=sha256:aed90d93b731929e742967e236f842a4a2174dc5db077c8f9ad2c5996f89f63e \ + --hash=sha256:b363bbb5253e5f9c23d8a0a034dfdf1b7c9e7f12e602fc788c435171e96daccc \ + --hash=sha256:b4098b6b638d9e0ca839a81656a2fd4bc26c9486ea707e8b1437d6f9d61c3941 \ + --hash=sha256:b53333627283e7241fcc217323f225c37783b5f0472316edcaa4479a213abfa6 \ + --hash=sha256:b670c2faa92124b7397b42303e4d8eb64a4cd0b7a77e35a9e865a55d61c57ef9 \ + --hash=sha256:bb396952cfa7ad2f01061fbc7dc1ad91dd9d69243bcb8110cf4e36924785a0fe \ + --hash=sha256:c60b83c43faeb6d0a9831f0351d7787a0753f5087cc6fa218d78fdf38e5acef0 \ + --hash=sha256:c6ebecfb7a31385393203eb04ed8b6a08f5002f53df3d59e5e795edb80999652 \ + --hash=sha256:d78d8b86fcdfa1e4c21f8896614b6cc7ee01a2a758ec0c4382d662f2a62cf766 \ + --hash=sha256:d7f8df114d6b4cf5a916b98389aeaf1e3132035420a88beea4e3d977e5f267a5 \ + --hash=sha256:e1cb52fa2d67d7f7fab310b600f22ce1ff04d562d46e9e0ac3e3403c2bb4cc16 \ + --hash=sha256:e3fdf04e402f12e1de8074458549337febb3b45f21076cc02ef4ff786aff687e \ + --hash=sha256:e503cb45ed12b924b5b988ba9576dc9949b2f5283b8e33b21dcb6be74a7c58d0 \ + --hash=sha256:f19ac6ac0a256cf77d3cc926ef0b4e64a9725cc612f97228cd5dc4bd9dbab03b \ + --hash=sha256:f1fb0fd4a1e9b11ac21c30c169d169ef434c6e9344ee0ab27cfa6f605f6387b2 \ + --hash=sha256:fada6b07ec4f0befe05218181f4b85176f11d531911b64c715d1875c4736d73a \ + --hash=sha256:fd173b4cf02b20f60860dc2ffe30115c18972d7d6d2d69df97ac38dee03be5bf \ + --hash=sha256:fe752639919aad9ffb0dee0d87f29a6467d1ef764f13c4644d212a9a853a078d \ + --hash=sha256:fee387d2fab144e8a34e0e9c5ca0f45c9376b99de45628265cfa9886b1dbe62b + # via grpcio-tools +grpcio-tools==1.46.5 \ + --hash=sha256:0a93cc4bacb29b830206bc7c6b187d7c31a2e65727230017accb525b96eb9b32 \ + --hash=sha256:0b36601bb6a2e44f5acefc6834ace1c46e2d3bde18463dc688229b55ff048fd4 \ + --hash=sha256:111b818887f773362e01d4a4582eeeb3126974ded6a239481d7a1b7c85766c18 \ + --hash=sha256:143b5e07d77a3f3c8b528a053e8995b30d5cd90e196f872baf6ab8761db4671a \ + --hash=sha256:15959833579cb87a39c70090a91ec495583a9b5743d655a403d0c2d45745855d \ + --hash=sha256:1cec5e920c2cc44fb6ce2101ad5689db5824ab6c5ae9916a34c1e8c6fe70cb00 \ + --hash=sha256:235577c3feabd40ea46eba3ffe47db5668547173376db395e067b6d5f086c651 \ + --hash=sha256:29e8da883104531c6b63458677628c1a19b0ba7b31cc48693b5787fecc39db9a \ + --hash=sha256:37a721695bcee591a9e05fa694c47adc46d8dd2d117f073ed543d1ac9bb8a376 \ + --hash=sha256:39ae72bc3dfabc826b5ea19224a2a1c9add50655a6c17173cb60b56b0456e46a \ + --hash=sha256:407d14c13c4c5c0640976fe55e1084195c48181c9035ce28749549c2d77fd110 \ + --hash=sha256:41a97716f40d2a8da64d5c3ea8bc1b74c8f23fc55e3171abe079251013e7ceb4 \ + --hash=sha256:48e6f37589ff030355d20ea24492a500efa66b8c552fb31b1a9068a8e2c514f7 \ + --hash=sha256:50fd071bf2657f8b874ea23c072375b2432d9e720f1e260c9d433f4a32df76fe \ + --hash=sha256:5398152ac4bfbd380172acdec59e0b90781736b6d9c2684b862e1806a25a28d2 \ + --hash=sha256:54054bc1a402b79a2a8bb2b5d7693bfc4c34d4fa96ad47faddbfbe67ec1be10d \ + --hash=sha256:5d73045090e568e0f0bd96659f6e62b9fd7f9d9f7149900e9b455c361b108c7e \ + --hash=sha256:600c0db6da81c6e17da4e53aa0cc8ab4bf4ca1af31c25dad46a49fe6f0d9fb67 \ + --hash=sha256:6025d1b004afb950388c34dfc36bff7017b8d62ab51c258630925a9aa6dcf145 \ + --hash=sha256:646556aa166cb2dd0c286afea2ae32a6638e9433381478b3439c6db5d09902af \ + --hash=sha256:67be06a956879ac719ad2ed4a6d7d66c28681ffac7a99bd593d34f60a58a8855 \ + --hash=sha256:6879967e126851487bbdc685e5d619bb972b0a56f106bd9086c27ecb843b526e \ + --hash=sha256:696d770181d9c20a8a61d22c9bd8306e9caa6a3ac6f7e232c2a605fac5f1b263 \ + --hash=sha256:697773b0044d0fa77a1e1cb1a5cc7fa9b781798bbc2a29a482786a758794f1d5 \ + --hash=sha256:6f1c5b68bd5183b740c5e34416fe3eb4c41335f804970f38a637153bf346d4ab \ + --hash=sha256:7191236150432849d5598382b28706b79f66cb48717b1e7560c46b5c304f859d \ + --hash=sha256:87b83e1b8387692d6a370ed6e47aca018cc0354cf83079250d251e8a7e4c1171 \ + --hash=sha256:8fe0fbfd01b33a29638e1df489082b77c23a680b7bb74fb2f8edf19ea3fd62a7 \ + --hash=sha256:926c18139c03975e3e4d0eb5aadd5bd563ae8426b6fc8b07979609c421dbb886 \ + --hash=sha256:935dff1cb820e3aff9b82c82c9a8ebcf7bb7aa92046a12184cd17471bae686f9 \ + --hash=sha256:99a2303c4f92eab0ed694573936dcc4af388c093f021a92f607b136f7285d4c1 \ + --hash=sha256:9a575c1a3b91b45c6d537e9e4dff7cd68ccc9b1c0f1c97d0857a2b53806f84aa \ + --hash=sha256:a0e36d9d878e5d5f92842bbba8883bae555e59382342fd610b6cb9a0a65737fc \ + --hash=sha256:a3a496528373d9ab47d3889e6d01d4417af2128864117910f749a12a9a88e3ac \ + --hash=sha256:a3fbe0d8b8eda9230d2c9e7a36ce9ac5ee8cd023e82ce7a2cee0829cb200b5b0 \ + --hash=sha256:a7c46e695214f9980b2cec7ece96318c32148593bfd4764a68ebe3be329e4a0e \ + --hash=sha256:be44ff6c42839551ac0b930e05c5a8c4f226f22802b41b378e371322d34c759e \ + --hash=sha256:bf4701d2907825b24e8571cdec754500913adc6e97c55f71f5a827e453fe0378 \ + --hash=sha256:c0e82728e018c230f0c5682be3dea41bfa0935059d37c2f26de54696aec8d469 \ + --hash=sha256:d1b94427c00160e5e9190e74308ee20987cce60f360f8a5f7bf948458617cc46 \ + --hash=sha256:d283f0ec6db8b2736ff82aafdfca19d482fb97c1a55dd99f07857b4cef91e14e \ + --hash=sha256:d58b0de05c9b7fb28601ced5dda5b1aadbda0f9f70896836e5b41c4e775250ea \ + --hash=sha256:d8e6293476562642f5e852fa897545abcf504f0a0178da342be80bc76eb144e0 \ + --hash=sha256:e1f33864875cbb931d00d8ffa1a1f602d7c49bd0bfb2d5b9fd9c419f765146d2 \ + --hash=sha256:e8acb28a773958791392298951ba879adc1cb5ad0884d579f33068da706bdf52 \ + --hash=sha256:eaca97015fd6b85d70ca262914b7ffe5a6d58a023d7b9e32e758edde484e6673 \ + --hash=sha256:ef47d4356f49519755f1d8db9b75a3e8269e0aeea49c8a0377c93966adc488a7 \ + --hash=sha256:ef5b1a5437fe0e0132ab117a1efaf04b99753fe454339c218b530bfa5e6b9c5a \ + --hash=sha256:f27c6418301e374ecfd8075499ba997e60239cda7dba43bfef8a02075d0a341a \ + --hash=sha256:f2961860fe6f5d19dfa7f2fb220ebcc39ca595d10e25e551e6a6212d16b2b9f9 \ + --hash=sha256:f3a270cb12455d4ebefe499a48daa154c2a28bd66bcbb8ee20591dd158507f28 \ + --hash=sha256:f6d617ed6eb3a3189fc00693f8a547e5134dc0bea189843c5c73a382b7808024 \ + --hash=sha256:faeb18bde6ed322c3fe8f6b905569027729ad54fbd158cb041f460b357411ff9 \ + --hash=sha256:fb2d0be5a06b27ba8d6a7790054794c759d8fcd7bfd429b24856e6368e8f51ad + # via -r ci/compile_py_proto/requirements_compile_py_proto.in +protobuf==3.20.3 \ + --hash=sha256:03038ac1cfbc41aa21f6afcbcd357281d7521b4157926f30ebecc8d4ea59dcb7 \ + --hash=sha256:28545383d61f55b57cf4df63eebd9827754fd2dc25f80c5253f9184235db242c \ + --hash=sha256:2e3427429c9cffebf259491be0af70189607f365c2f41c7c3764af6f337105f2 \ + --hash=sha256:398a9e0c3eaceb34ec1aee71894ca3299605fa8e761544934378bbc6c97de23b \ + --hash=sha256:44246bab5dd4b7fbd3c0c80b6f16686808fab0e4aca819ade6e8d294a29c7050 \ + --hash=sha256:447d43819997825d4e71bf5769d869b968ce96848b6479397e29fc24c4a5dfe9 \ + --hash=sha256:67a3598f0a2dcbc58d02dd1928544e7d88f764b47d4a286202913f0b2801c2e7 \ + --hash=sha256:74480f79a023f90dc6e18febbf7b8bac7508420f2006fabd512013c0c238f454 \ + --hash=sha256:819559cafa1a373b7096a482b504ae8a857c89593cf3a25af743ac9ecbd23480 \ + --hash=sha256:899dc660cd599d7352d6f10d83c95df430a38b410c1b66b407a6b29265d66469 \ + --hash=sha256:8c0c984a1b8fef4086329ff8dd19ac77576b384079247c770f29cc8ce3afa06c \ + --hash=sha256:9aae4406ea63d825636cc11ffb34ad3379335803216ee3a856787bcf5ccc751e \ + --hash=sha256:a7ca6d488aa8ff7f329d4c545b2dbad8ac31464f1d8b1c87ad1346717731e4db \ + --hash=sha256:b6cc7ba72a8850621bfec987cb72623e703b7fe2b9127a161ce61e61558ad905 \ + --hash=sha256:bf01b5720be110540be4286e791db73f84a2b721072a3711efff6c324cdf074b \ + --hash=sha256:c02ce36ec760252242a33967d51c289fd0e1c0e6e5cc9397e2279177716add86 \ + --hash=sha256:d9e4432ff660d67d775c66ac42a67cf2453c27cb4d738fc22cb53b5d84c135d4 \ + --hash=sha256:daa564862dd0d39c00f8086f88700fdbe8bc717e993a21e90711acfed02f2402 \ + --hash=sha256:de78575669dddf6099a8a0f46a27e82a1783c557ccc38ee620ed8cc96d3be7d7 \ + --hash=sha256:e64857f395505ebf3d2569935506ae0dfc4a15cb80dc25261176c784662cdcc4 \ + --hash=sha256:f4bd856d702e5b0d96a00ec6b307b0f51c1982c2bf9c0052cf9019e9a544ba99 \ + --hash=sha256:f4c42102bc82a51108e449cbb32b19b180022941c727bac0cfd50170341f16ee + # via grpcio-tools + +# The following packages are considered to be unsafe in a requirements file: +setuptools==68.0.0 \ + --hash=sha256:11e52c67415a381d10d6b462ced9cfb97066179f0e871399e006c4ab101fc85f \ + --hash=sha256:baf1fdb41c6da4cd2eae722e135500da913332ab3f2f5c7d33af9b492acb5235 + # via + # -r ci/compile_py_proto/requirements_compile_py_proto.in + # grpcio-tools diff --git a/ci/lint/check-python-proto.sh b/ci/lint/check-python-proto.sh new file mode 100755 index 0000000000000..af44d61d4893b --- /dev/null +++ b/ci/lint/check-python-proto.sh @@ -0,0 +1,15 @@ +#!/bin/bash + +# Checks that we manually compile protos for python when proto files are changed. + +set -euxo pipefail + +bazel run //ci/compile_py_proto:install_py_proto + +if ! git diff --quiet &>/dev/null; then + echo 'Python proto source files are not manually generated when proto files are changed.' + echo 'Run "bazel run //ci/compile_py_proto:install_py_proto" to generate.' + echo + + exit 1 +fi diff --git a/ci/lint/format.sh b/ci/lint/format.sh index e1b74f5634061..86cd2f75ca900 100755 --- a/ci/lint/format.sh +++ b/ci/lint/format.sh @@ -157,6 +157,8 @@ BLACK_EXCLUDES=( `'python/ray/thirdparty_files/*|'` `'python/ray/_private/thirdparty/*|'` `'python/ray/serve/tests/test_config_files/syntax_error\.py|'` + `'python/ray/core/generated/*|'` + `'python/ray/serve/generated/*|'` `'doc/external/*' ) @@ -164,6 +166,13 @@ GIT_LS_EXCLUDES=( ':(exclude)python/ray/cloudpickle/' ':(exclude)python/ray/_private/runtime_env/_clonevirtualenv.py' ':(exclude)doc/external/' + ':(exclude)python/ray/core/generated/' + ':(exclude)python/ray/serve/generated/' +) + +GIT_DIFF_EXCLUDES=( + ':(exclude)python/ray/core/generated/' + ':(exclude)python/ray/serve/generated/' ) JAVA_EXCLUDES=( @@ -202,7 +211,7 @@ mypy_on_each() { format_frontend() { ( echo "$(date)" "format frontend...." - local folder + local folder folder="$(pwd)/dashboard/client" local filenames # shellcheck disable=SC2207 @@ -329,16 +338,16 @@ format_changed() { # exist on both branches. MERGEBASE="$(git merge-base upstream/master HEAD)" - if ! git diff --diff-filter=ACRM --quiet --exit-code "$MERGEBASE" -- '*.py' &>/dev/null; then - git diff --name-only --diff-filter=ACRM "$MERGEBASE" -- '*.py' | xargs -P 5 \ + if ! git diff --diff-filter=ACRM --quiet --exit-code "$MERGEBASE" -- '*.py' "${GIT_DIFF_EXCLUDES[@]}" &>/dev/null; then + git diff --name-only --diff-filter=ACRM "$MERGEBASE" -- '*.py' "${GIT_DIFF_EXCLUDES[@]}" | xargs -P 5 \ isort fi - if ! git diff --diff-filter=ACRM --quiet --exit-code "$MERGEBASE" -- '*.py' &>/dev/null; then - git diff --name-only --diff-filter=ACRM "$MERGEBASE" -- '*.py' | xargs -P 5 \ + if ! git diff --diff-filter=ACRM --quiet --exit-code "$MERGEBASE" -- '*.py' "${GIT_DIFF_EXCLUDES[@]}" &>/dev/null; then + git diff --name-only --diff-filter=ACRM "$MERGEBASE" -- '*.py' "${GIT_DIFF_EXCLUDES[@]}" | xargs -P 5 \ black "${BLACK_EXCLUDES[@]}" if command -v flake8 >/dev/null; then - git diff --name-only --diff-filter=ACRM "$MERGEBASE" -- '*.py' | xargs -P 5 \ + git diff --name-only --diff-filter=ACRM "$MERGEBASE" -- '*.py' "${GIT_DIFF_EXCLUDES[@]}" | xargs -P 5 \ flake8 --config=.flake8 fi fi diff --git a/ci/ray_ci/core.tests.yml b/ci/ray_ci/core.tests.yml index c25d38aead37f..c70761dc8c8a2 100644 --- a/ci/ray_ci/core.tests.yml +++ b/ci/ray_ci/core.tests.yml @@ -10,3 +10,5 @@ flaky_tests: - //python/ray/tests:test_scheduling_performance - //python/ray/tests:test_threaded_actor - //python/ray/tests:test_unhandled_error + - //python/ray/tests:test_actor_bounded_threads + - //python/ray/tests:test_state_api_log diff --git a/cpp/src/ray/runtime/task/local_mode_task_submitter.h b/cpp/src/ray/runtime/task/local_mode_task_submitter.h index 010a9d9c2369e..210adfb5d2c9f 100644 --- a/cpp/src/ray/runtime/task/local_mode_task_submitter.h +++ b/cpp/src/ray/runtime/task/local_mode_task_submitter.h @@ -54,7 +54,8 @@ class LocalModeTaskSubmitter : public TaskSubmitter { absl::Mutex actor_contexts_mutex_; - std::unordered_map named_actors_ GUARDED_BY(named_actors_mutex_); + std::unordered_map named_actors_ + ABSL_GUARDED_BY(named_actors_mutex_); mutable absl::Mutex named_actors_mutex_; std::unique_ptr thread_pool_; diff --git a/java/dependencies.bzl b/java/dependencies.bzl index 5c7a99f36cbbc..d491174db5fd6 100644 --- a/java/dependencies.bzl +++ b/java/dependencies.bzl @@ -8,8 +8,8 @@ def gen_java_deps(): "com.github.java-json-tools:json-schema-validator:2.2.14", "com.google.code.gson:gson:2.9.1", "com.google.guava:guava:32.0.1-jre", - "com.google.protobuf:protobuf-java:3.19.6", - "com.google.protobuf:protobuf-java-util:3.19.6", + "com.google.protobuf:protobuf-java:3.23.4", + "com.google.protobuf:protobuf-java-util:3.23.4", "com.puppycrawl.tools:checkstyle:8.15", "com.sun.xml.bind:jaxb-core:2.3.0", "com.sun.xml.bind:jaxb-impl:2.3.0", diff --git a/python/ray/core/generated/agent_manager_pb2.py b/python/ray/core/generated/agent_manager_pb2.py new file mode 100644 index 0000000000000..10345e1b52ab1 --- /dev/null +++ b/python/ray/core/generated/agent_manager_pb2.py @@ -0,0 +1,32 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/agent_manager.proto +"""Generated protocol buffer code.""" +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$src/ray/protobuf/agent_manager.proto\x12\x07ray.rpc*F\n\x0e\x41gentRpcStatus\x12\x17\n\x13\x41GENT_RPC_STATUS_OK\x10\x00\x12\x1b\n\x17\x41GENT_RPC_STATUS_FAILED\x10\x01\x42\x03\xf8\x01\x01\x62\x06proto3') + +_AGENTRPCSTATUS = DESCRIPTOR.enum_types_by_name['AgentRpcStatus'] +AgentRpcStatus = enum_type_wrapper.EnumTypeWrapper(_AGENTRPCSTATUS) +AGENT_RPC_STATUS_OK = 0 +AGENT_RPC_STATUS_FAILED = 1 + + +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _AGENTRPCSTATUS._serialized_start=49 + _AGENTRPCSTATUS._serialized_end=119 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/agent_manager_pb2_grpc.py b/python/ray/core/generated/agent_manager_pb2_grpc.py new file mode 100644 index 0000000000000..2daafffebfc81 --- /dev/null +++ b/python/ray/core/generated/agent_manager_pb2_grpc.py @@ -0,0 +1,4 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + diff --git a/python/ray/core/generated/autoscaler_pb2.py b/python/ray/core/generated/autoscaler_pb2.py new file mode 100644 index 0000000000000..ecf758a4ecac6 --- /dev/null +++ b/python/ray/core/generated/autoscaler_pb2.py @@ -0,0 +1,327 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/autoscaler.proto +"""Generated protocol buffer code.""" +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!src/ray/protobuf/autoscaler.proto\x12\x12ray.rpc.autoscaler\"A\n\x16\x41ntiAffinityConstraint\x12\x12\n\nlabel_name\x18\x01 \x01(\t\x12\x13\n\x0blabel_value\x18\x02 \x01(\t\"=\n\x12\x41\x66\x66inityConstraint\x12\x12\n\nlabel_name\x18\x01 \x01(\t\x12\x13\n\x0blabel_value\x18\x02 \x01(\t\"\xbb\x01\n\x13PlacementConstraint\x12\x46\n\ranti_affinity\x18\x01 \x01(\x0b\x32*.ray.rpc.autoscaler.AntiAffinityConstraintH\x00\x88\x01\x01\x12=\n\x08\x61\x66\x66inity\x18\x02 \x01(\x0b\x32&.ray.rpc.autoscaler.AffinityConstraintH\x01\x88\x01\x01\x42\x10\n\x0e_anti_affinityB\x0b\n\t_affinity\"\xe5\x01\n\x0fResourceRequest\x12R\n\x10resources_bundle\x18\x01 \x03(\x0b\x32\x38.ray.rpc.autoscaler.ResourceRequest.ResourcesBundleEntry\x12\x46\n\x15placement_constraints\x18\x02 \x03(\x0b\x32\'.ray.rpc.autoscaler.PlacementConstraint\x1a\x36\n\x14ResourcesBundleEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\"]\n\x16ResourceRequestByCount\x12\x34\n\x07request\x18\x01 \x01(\x0b\x32#.ray.rpc.autoscaler.ResourceRequest\x12\r\n\x05\x63ount\x18\x02 \x01(\x03\"]\n\x13GangResourceRequest\x12\x35\n\x08requests\x18\x01 \x03(\x0b\x32#.ray.rpc.autoscaler.ResourceRequest\x12\x0f\n\x07\x64\x65tails\x18\x02 \x01(\t\"\\\n\x19\x43lusterResourceConstraint\x12?\n\x0bmin_bundles\x18\x01 \x03(\x0b\x32*.ray.rpc.autoscaler.ResourceRequestByCount\"\xfa\x04\n\tNodeState\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\x12\x13\n\x0binstance_id\x18\x02 \x01(\t\x12\x1a\n\x12ray_node_type_name\x18\x03 \x01(\t\x12R\n\x13\x61vailable_resources\x18\x04 \x03(\x0b\x32\x35.ray.rpc.autoscaler.NodeState.AvailableResourcesEntry\x12J\n\x0ftotal_resources\x18\x05 \x03(\x0b\x32\x31.ray.rpc.autoscaler.NodeState.TotalResourcesEntry\x12H\n\x0e\x64ynamic_labels\x18\x06 \x03(\x0b\x32\x30.ray.rpc.autoscaler.NodeState.DynamicLabelsEntry\x12\x1a\n\x12node_state_version\x18\x07 \x01(\x03\x12.\n\x06status\x18\x08 \x01(\x0e\x32\x1e.ray.rpc.autoscaler.NodeStatus\x12\x18\n\x10idle_duration_ms\x18\t \x01(\x03\x12\x17\n\x0fnode_ip_address\x18\n \x01(\t\x12\x1a\n\x12instance_type_name\x18\x0b \x01(\t\x1a\x39\n\x17\x41vailableResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x1a\x35\n\x13TotalResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x1a\x34\n\x12\x44ynamicLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"R\n\x1eGetClusterResourceStateRequest\x12\x30\n(last_seen_cluster_resource_state_version\x18\x01 \x01(\x03\"\xb1\x03\n\x14\x43lusterResourceState\x12&\n\x1e\x63luster_resource_state_version\x18\x01 \x01(\x03\x12*\n\"last_seen_autoscaler_state_version\x18\x02 \x01(\x03\x12\x32\n\x0bnode_states\x18\x03 \x03(\x0b\x32\x1d.ray.rpc.autoscaler.NodeState\x12M\n\x19pending_resource_requests\x18\x04 \x03(\x0b\x32*.ray.rpc.autoscaler.ResourceRequestByCount\x12O\n\x1epending_gang_resource_requests\x18\x05 \x03(\x0b\x32\'.ray.rpc.autoscaler.GangResourceRequest\x12S\n\x1c\x63luster_resource_constraints\x18\x06 \x03(\x0b\x32-.ray.rpc.autoscaler.ClusterResourceConstraint\x12\x1c\n\x14\x63luster_session_name\x18\x07 \x01(\t\"h\n\x1cGetClusterResourceStateReply\x12H\n\x16\x63luster_resource_state\x18\x01 \x01(\x0b\x32(.ray.rpc.autoscaler.ClusterResourceState\"s\n\x16PendingInstanceRequest\x12\x1a\n\x12instance_type_name\x18\x01 \x01(\t\x12\x1a\n\x12ray_node_type_name\x18\x02 \x01(\t\x12\r\n\x05\x63ount\x18\x03 \x01(\x05\x12\x12\n\nrequest_ts\x18\x04 \x01(\x03\"\x93\x01\n\x15\x46\x61iledInstanceRequest\x12\x1a\n\x12instance_type_name\x18\x01 \x01(\t\x12\x1a\n\x12ray_node_type_name\x18\x02 \x01(\t\x12\r\n\x05\x63ount\x18\x03 \x01(\x05\x12\x0e\n\x06reason\x18\x04 \x01(\t\x12\x10\n\x08start_ts\x18\x05 \x01(\x03\x12\x11\n\tfailed_ts\x18\x06 \x01(\x03\"\x83\x01\n\x0fPendingInstance\x12\x1a\n\x12instance_type_name\x18\x01 \x01(\t\x12\x1a\n\x12ray_node_type_name\x18\x02 \x01(\t\x12\x13\n\x0binstance_id\x18\x03 \x01(\t\x12\x12\n\nip_address\x18\x04 \x01(\t\x12\x0f\n\x07\x64\x65tails\x18\x05 \x01(\t\"\xc1\x04\n\x10\x41utoscalingState\x12\x30\n(last_seen_cluster_resource_state_version\x18\x01 \x01(\x03\x12 \n\x18\x61utoscaler_state_version\x18\x02 \x01(\x03\x12M\n\x19pending_instance_requests\x18\x03 \x03(\x0b\x32*.ray.rpc.autoscaler.PendingInstanceRequest\x12I\n\x1cinfeasible_resource_requests\x18\x04 \x03(\x0b\x32#.ray.rpc.autoscaler.ResourceRequest\x12R\n!infeasible_gang_resource_requests\x18\x05 \x03(\x0b\x32\'.ray.rpc.autoscaler.GangResourceRequest\x12^\n\'infeasible_cluster_resource_constraints\x18\x06 \x03(\x0b\x32-.ray.rpc.autoscaler.ClusterResourceConstraint\x12>\n\x11pending_instances\x18\x07 \x03(\x0b\x32#.ray.rpc.autoscaler.PendingInstance\x12K\n\x18\x66\x61iled_instance_requests\x18\x08 \x03(\x0b\x32).ray.rpc.autoscaler.FailedInstanceRequest\"`\n\x1dReportAutoscalingStateRequest\x12?\n\x11\x61utoscaling_state\x18\x01 \x01(\x0b\x32$.ray.rpc.autoscaler.AutoscalingState\"\x1d\n\x1bReportAutoscalingStateReply\"}\n\'RequestClusterResourceConstraintRequest\x12R\n\x1b\x63luster_resource_constraint\x18\x01 \x01(\x0b\x32-.ray.rpc.autoscaler.ClusterResourceConstraint\"\'\n%RequestClusterResourceConstraintReply\"\x19\n\x17GetClusterStatusRequest\"\xa2\x01\n\x15GetClusterStatusReply\x12?\n\x11\x61utoscaling_state\x18\x01 \x01(\x0b\x32$.ray.rpc.autoscaler.AutoscalingState\x12H\n\x16\x63luster_resource_state\x18\x02 \x01(\x0b\x32(.ray.rpc.autoscaler.ClusterResourceState\"p\n\x10\x44rainNodeRequest\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\x12\x33\n\x06reason\x18\x02 \x01(\x0e\x32#.ray.rpc.autoscaler.DrainNodeReason\x12\x16\n\x0ereason_message\x18\x03 \x01(\t\"%\n\x0e\x44rainNodeReply\x12\x13\n\x0bis_accepted\x18\x01 \x01(\x08*L\n\nNodeStatus\x12\x0f\n\x0bUNSPECIFIED\x10\x00\x12\x0b\n\x07RUNNING\x10\x01\x12\x08\n\x04\x44\x45\x41\x44\x10\x02\x12\x08\n\x04IDLE\x10\x03\x12\x0c\n\x08\x44RAINING\x10\x04*~\n\x0f\x44rainNodeReason\x12!\n\x1d\x44RAIN_NODE_REASON_UNSPECIFIED\x10\x00\x12&\n\"DRAIN_NODE_REASON_IDLE_TERMINATION\x10\x01\x12 \n\x1c\x44RAIN_NODE_REASON_PREEMPTION\x10\x02\x32\xf7\x04\n\x16\x41utoscalerStateService\x12\x7f\n\x17GetClusterResourceState\x12\x32.ray.rpc.autoscaler.GetClusterResourceStateRequest\x1a\x30.ray.rpc.autoscaler.GetClusterResourceStateReply\x12|\n\x16ReportAutoscalingState\x12\x31.ray.rpc.autoscaler.ReportAutoscalingStateRequest\x1a/.ray.rpc.autoscaler.ReportAutoscalingStateReply\x12\x9a\x01\n RequestClusterResourceConstraint\x12;.ray.rpc.autoscaler.RequestClusterResourceConstraintRequest\x1a\x39.ray.rpc.autoscaler.RequestClusterResourceConstraintReply\x12j\n\x10GetClusterStatus\x12+.ray.rpc.autoscaler.GetClusterStatusRequest\x1a).ray.rpc.autoscaler.GetClusterStatusReply\x12U\n\tDrainNode\x12$.ray.rpc.autoscaler.DrainNodeRequest\x1a\".ray.rpc.autoscaler.DrainNodeReplyB\x03\xf8\x01\x01\x62\x06proto3') + +_NODESTATUS = DESCRIPTOR.enum_types_by_name['NodeStatus'] +NodeStatus = enum_type_wrapper.EnumTypeWrapper(_NODESTATUS) +_DRAINNODEREASON = DESCRIPTOR.enum_types_by_name['DrainNodeReason'] +DrainNodeReason = enum_type_wrapper.EnumTypeWrapper(_DRAINNODEREASON) +UNSPECIFIED = 0 +RUNNING = 1 +DEAD = 2 +IDLE = 3 +DRAINING = 4 +DRAIN_NODE_REASON_UNSPECIFIED = 0 +DRAIN_NODE_REASON_IDLE_TERMINATION = 1 +DRAIN_NODE_REASON_PREEMPTION = 2 + + +_ANTIAFFINITYCONSTRAINT = DESCRIPTOR.message_types_by_name['AntiAffinityConstraint'] +_AFFINITYCONSTRAINT = DESCRIPTOR.message_types_by_name['AffinityConstraint'] +_PLACEMENTCONSTRAINT = DESCRIPTOR.message_types_by_name['PlacementConstraint'] +_RESOURCEREQUEST = DESCRIPTOR.message_types_by_name['ResourceRequest'] +_RESOURCEREQUEST_RESOURCESBUNDLEENTRY = _RESOURCEREQUEST.nested_types_by_name['ResourcesBundleEntry'] +_RESOURCEREQUESTBYCOUNT = DESCRIPTOR.message_types_by_name['ResourceRequestByCount'] +_GANGRESOURCEREQUEST = DESCRIPTOR.message_types_by_name['GangResourceRequest'] +_CLUSTERRESOURCECONSTRAINT = DESCRIPTOR.message_types_by_name['ClusterResourceConstraint'] +_NODESTATE = DESCRIPTOR.message_types_by_name['NodeState'] +_NODESTATE_AVAILABLERESOURCESENTRY = _NODESTATE.nested_types_by_name['AvailableResourcesEntry'] +_NODESTATE_TOTALRESOURCESENTRY = _NODESTATE.nested_types_by_name['TotalResourcesEntry'] +_NODESTATE_DYNAMICLABELSENTRY = _NODESTATE.nested_types_by_name['DynamicLabelsEntry'] +_GETCLUSTERRESOURCESTATEREQUEST = DESCRIPTOR.message_types_by_name['GetClusterResourceStateRequest'] +_CLUSTERRESOURCESTATE = DESCRIPTOR.message_types_by_name['ClusterResourceState'] +_GETCLUSTERRESOURCESTATEREPLY = DESCRIPTOR.message_types_by_name['GetClusterResourceStateReply'] +_PENDINGINSTANCEREQUEST = DESCRIPTOR.message_types_by_name['PendingInstanceRequest'] +_FAILEDINSTANCEREQUEST = DESCRIPTOR.message_types_by_name['FailedInstanceRequest'] +_PENDINGINSTANCE = DESCRIPTOR.message_types_by_name['PendingInstance'] +_AUTOSCALINGSTATE = DESCRIPTOR.message_types_by_name['AutoscalingState'] +_REPORTAUTOSCALINGSTATEREQUEST = DESCRIPTOR.message_types_by_name['ReportAutoscalingStateRequest'] +_REPORTAUTOSCALINGSTATEREPLY = DESCRIPTOR.message_types_by_name['ReportAutoscalingStateReply'] +_REQUESTCLUSTERRESOURCECONSTRAINTREQUEST = DESCRIPTOR.message_types_by_name['RequestClusterResourceConstraintRequest'] +_REQUESTCLUSTERRESOURCECONSTRAINTREPLY = DESCRIPTOR.message_types_by_name['RequestClusterResourceConstraintReply'] +_GETCLUSTERSTATUSREQUEST = DESCRIPTOR.message_types_by_name['GetClusterStatusRequest'] +_GETCLUSTERSTATUSREPLY = DESCRIPTOR.message_types_by_name['GetClusterStatusReply'] +_DRAINNODEREQUEST = DESCRIPTOR.message_types_by_name['DrainNodeRequest'] +_DRAINNODEREPLY = DESCRIPTOR.message_types_by_name['DrainNodeReply'] +AntiAffinityConstraint = _reflection.GeneratedProtocolMessageType('AntiAffinityConstraint', (_message.Message,), { + 'DESCRIPTOR' : _ANTIAFFINITYCONSTRAINT, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.AntiAffinityConstraint) + }) +_sym_db.RegisterMessage(AntiAffinityConstraint) + +AffinityConstraint = _reflection.GeneratedProtocolMessageType('AffinityConstraint', (_message.Message,), { + 'DESCRIPTOR' : _AFFINITYCONSTRAINT, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.AffinityConstraint) + }) +_sym_db.RegisterMessage(AffinityConstraint) + +PlacementConstraint = _reflection.GeneratedProtocolMessageType('PlacementConstraint', (_message.Message,), { + 'DESCRIPTOR' : _PLACEMENTCONSTRAINT, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.PlacementConstraint) + }) +_sym_db.RegisterMessage(PlacementConstraint) + +ResourceRequest = _reflection.GeneratedProtocolMessageType('ResourceRequest', (_message.Message,), { + + 'ResourcesBundleEntry' : _reflection.GeneratedProtocolMessageType('ResourcesBundleEntry', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCEREQUEST_RESOURCESBUNDLEENTRY, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.ResourceRequest.ResourcesBundleEntry) + }) + , + 'DESCRIPTOR' : _RESOURCEREQUEST, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.ResourceRequest) + }) +_sym_db.RegisterMessage(ResourceRequest) +_sym_db.RegisterMessage(ResourceRequest.ResourcesBundleEntry) + +ResourceRequestByCount = _reflection.GeneratedProtocolMessageType('ResourceRequestByCount', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCEREQUESTBYCOUNT, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.ResourceRequestByCount) + }) +_sym_db.RegisterMessage(ResourceRequestByCount) + +GangResourceRequest = _reflection.GeneratedProtocolMessageType('GangResourceRequest', (_message.Message,), { + 'DESCRIPTOR' : _GANGRESOURCEREQUEST, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.GangResourceRequest) + }) +_sym_db.RegisterMessage(GangResourceRequest) + +ClusterResourceConstraint = _reflection.GeneratedProtocolMessageType('ClusterResourceConstraint', (_message.Message,), { + 'DESCRIPTOR' : _CLUSTERRESOURCECONSTRAINT, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.ClusterResourceConstraint) + }) +_sym_db.RegisterMessage(ClusterResourceConstraint) + +NodeState = _reflection.GeneratedProtocolMessageType('NodeState', (_message.Message,), { + + 'AvailableResourcesEntry' : _reflection.GeneratedProtocolMessageType('AvailableResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _NODESTATE_AVAILABLERESOURCESENTRY, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.NodeState.AvailableResourcesEntry) + }) + , + + 'TotalResourcesEntry' : _reflection.GeneratedProtocolMessageType('TotalResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _NODESTATE_TOTALRESOURCESENTRY, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.NodeState.TotalResourcesEntry) + }) + , + + 'DynamicLabelsEntry' : _reflection.GeneratedProtocolMessageType('DynamicLabelsEntry', (_message.Message,), { + 'DESCRIPTOR' : _NODESTATE_DYNAMICLABELSENTRY, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.NodeState.DynamicLabelsEntry) + }) + , + 'DESCRIPTOR' : _NODESTATE, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.NodeState) + }) +_sym_db.RegisterMessage(NodeState) +_sym_db.RegisterMessage(NodeState.AvailableResourcesEntry) +_sym_db.RegisterMessage(NodeState.TotalResourcesEntry) +_sym_db.RegisterMessage(NodeState.DynamicLabelsEntry) + +GetClusterResourceStateRequest = _reflection.GeneratedProtocolMessageType('GetClusterResourceStateRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETCLUSTERRESOURCESTATEREQUEST, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.GetClusterResourceStateRequest) + }) +_sym_db.RegisterMessage(GetClusterResourceStateRequest) + +ClusterResourceState = _reflection.GeneratedProtocolMessageType('ClusterResourceState', (_message.Message,), { + 'DESCRIPTOR' : _CLUSTERRESOURCESTATE, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.ClusterResourceState) + }) +_sym_db.RegisterMessage(ClusterResourceState) + +GetClusterResourceStateReply = _reflection.GeneratedProtocolMessageType('GetClusterResourceStateReply', (_message.Message,), { + 'DESCRIPTOR' : _GETCLUSTERRESOURCESTATEREPLY, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.GetClusterResourceStateReply) + }) +_sym_db.RegisterMessage(GetClusterResourceStateReply) + +PendingInstanceRequest = _reflection.GeneratedProtocolMessageType('PendingInstanceRequest', (_message.Message,), { + 'DESCRIPTOR' : _PENDINGINSTANCEREQUEST, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.PendingInstanceRequest) + }) +_sym_db.RegisterMessage(PendingInstanceRequest) + +FailedInstanceRequest = _reflection.GeneratedProtocolMessageType('FailedInstanceRequest', (_message.Message,), { + 'DESCRIPTOR' : _FAILEDINSTANCEREQUEST, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.FailedInstanceRequest) + }) +_sym_db.RegisterMessage(FailedInstanceRequest) + +PendingInstance = _reflection.GeneratedProtocolMessageType('PendingInstance', (_message.Message,), { + 'DESCRIPTOR' : _PENDINGINSTANCE, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.PendingInstance) + }) +_sym_db.RegisterMessage(PendingInstance) + +AutoscalingState = _reflection.GeneratedProtocolMessageType('AutoscalingState', (_message.Message,), { + 'DESCRIPTOR' : _AUTOSCALINGSTATE, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.AutoscalingState) + }) +_sym_db.RegisterMessage(AutoscalingState) + +ReportAutoscalingStateRequest = _reflection.GeneratedProtocolMessageType('ReportAutoscalingStateRequest', (_message.Message,), { + 'DESCRIPTOR' : _REPORTAUTOSCALINGSTATEREQUEST, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.ReportAutoscalingStateRequest) + }) +_sym_db.RegisterMessage(ReportAutoscalingStateRequest) + +ReportAutoscalingStateReply = _reflection.GeneratedProtocolMessageType('ReportAutoscalingStateReply', (_message.Message,), { + 'DESCRIPTOR' : _REPORTAUTOSCALINGSTATEREPLY, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.ReportAutoscalingStateReply) + }) +_sym_db.RegisterMessage(ReportAutoscalingStateReply) + +RequestClusterResourceConstraintRequest = _reflection.GeneratedProtocolMessageType('RequestClusterResourceConstraintRequest', (_message.Message,), { + 'DESCRIPTOR' : _REQUESTCLUSTERRESOURCECONSTRAINTREQUEST, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.RequestClusterResourceConstraintRequest) + }) +_sym_db.RegisterMessage(RequestClusterResourceConstraintRequest) + +RequestClusterResourceConstraintReply = _reflection.GeneratedProtocolMessageType('RequestClusterResourceConstraintReply', (_message.Message,), { + 'DESCRIPTOR' : _REQUESTCLUSTERRESOURCECONSTRAINTREPLY, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.RequestClusterResourceConstraintReply) + }) +_sym_db.RegisterMessage(RequestClusterResourceConstraintReply) + +GetClusterStatusRequest = _reflection.GeneratedProtocolMessageType('GetClusterStatusRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETCLUSTERSTATUSREQUEST, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.GetClusterStatusRequest) + }) +_sym_db.RegisterMessage(GetClusterStatusRequest) + +GetClusterStatusReply = _reflection.GeneratedProtocolMessageType('GetClusterStatusReply', (_message.Message,), { + 'DESCRIPTOR' : _GETCLUSTERSTATUSREPLY, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.GetClusterStatusReply) + }) +_sym_db.RegisterMessage(GetClusterStatusReply) + +DrainNodeRequest = _reflection.GeneratedProtocolMessageType('DrainNodeRequest', (_message.Message,), { + 'DESCRIPTOR' : _DRAINNODEREQUEST, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.DrainNodeRequest) + }) +_sym_db.RegisterMessage(DrainNodeRequest) + +DrainNodeReply = _reflection.GeneratedProtocolMessageType('DrainNodeReply', (_message.Message,), { + 'DESCRIPTOR' : _DRAINNODEREPLY, + '__module__' : 'src.ray.protobuf.autoscaler_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.DrainNodeReply) + }) +_sym_db.RegisterMessage(DrainNodeReply) + +_AUTOSCALERSTATESERVICE = DESCRIPTOR.services_by_name['AutoscalerStateService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _RESOURCEREQUEST_RESOURCESBUNDLEENTRY._options = None + _RESOURCEREQUEST_RESOURCESBUNDLEENTRY._serialized_options = b'8\001' + _NODESTATE_AVAILABLERESOURCESENTRY._options = None + _NODESTATE_AVAILABLERESOURCESENTRY._serialized_options = b'8\001' + _NODESTATE_TOTALRESOURCESENTRY._options = None + _NODESTATE_TOTALRESOURCESENTRY._serialized_options = b'8\001' + _NODESTATE_DYNAMICLABELSENTRY._options = None + _NODESTATE_DYNAMICLABELSENTRY._serialized_options = b'8\001' + _NODESTATUS._serialized_start=3779 + _NODESTATUS._serialized_end=3855 + _DRAINNODEREASON._serialized_start=3857 + _DRAINNODEREASON._serialized_end=3983 + _ANTIAFFINITYCONSTRAINT._serialized_start=57 + _ANTIAFFINITYCONSTRAINT._serialized_end=122 + _AFFINITYCONSTRAINT._serialized_start=124 + _AFFINITYCONSTRAINT._serialized_end=185 + _PLACEMENTCONSTRAINT._serialized_start=188 + _PLACEMENTCONSTRAINT._serialized_end=375 + _RESOURCEREQUEST._serialized_start=378 + _RESOURCEREQUEST._serialized_end=607 + _RESOURCEREQUEST_RESOURCESBUNDLEENTRY._serialized_start=553 + _RESOURCEREQUEST_RESOURCESBUNDLEENTRY._serialized_end=607 + _RESOURCEREQUESTBYCOUNT._serialized_start=609 + _RESOURCEREQUESTBYCOUNT._serialized_end=702 + _GANGRESOURCEREQUEST._serialized_start=704 + _GANGRESOURCEREQUEST._serialized_end=797 + _CLUSTERRESOURCECONSTRAINT._serialized_start=799 + _CLUSTERRESOURCECONSTRAINT._serialized_end=891 + _NODESTATE._serialized_start=894 + _NODESTATE._serialized_end=1528 + _NODESTATE_AVAILABLERESOURCESENTRY._serialized_start=1362 + _NODESTATE_AVAILABLERESOURCESENTRY._serialized_end=1419 + _NODESTATE_TOTALRESOURCESENTRY._serialized_start=1421 + _NODESTATE_TOTALRESOURCESENTRY._serialized_end=1474 + _NODESTATE_DYNAMICLABELSENTRY._serialized_start=1476 + _NODESTATE_DYNAMICLABELSENTRY._serialized_end=1528 + _GETCLUSTERRESOURCESTATEREQUEST._serialized_start=1530 + _GETCLUSTERRESOURCESTATEREQUEST._serialized_end=1612 + _CLUSTERRESOURCESTATE._serialized_start=1615 + _CLUSTERRESOURCESTATE._serialized_end=2048 + _GETCLUSTERRESOURCESTATEREPLY._serialized_start=2050 + _GETCLUSTERRESOURCESTATEREPLY._serialized_end=2154 + _PENDINGINSTANCEREQUEST._serialized_start=2156 + _PENDINGINSTANCEREQUEST._serialized_end=2271 + _FAILEDINSTANCEREQUEST._serialized_start=2274 + _FAILEDINSTANCEREQUEST._serialized_end=2421 + _PENDINGINSTANCE._serialized_start=2424 + _PENDINGINSTANCE._serialized_end=2555 + _AUTOSCALINGSTATE._serialized_start=2558 + _AUTOSCALINGSTATE._serialized_end=3135 + _REPORTAUTOSCALINGSTATEREQUEST._serialized_start=3137 + _REPORTAUTOSCALINGSTATEREQUEST._serialized_end=3233 + _REPORTAUTOSCALINGSTATEREPLY._serialized_start=3235 + _REPORTAUTOSCALINGSTATEREPLY._serialized_end=3264 + _REQUESTCLUSTERRESOURCECONSTRAINTREQUEST._serialized_start=3266 + _REQUESTCLUSTERRESOURCECONSTRAINTREQUEST._serialized_end=3391 + _REQUESTCLUSTERRESOURCECONSTRAINTREPLY._serialized_start=3393 + _REQUESTCLUSTERRESOURCECONSTRAINTREPLY._serialized_end=3432 + _GETCLUSTERSTATUSREQUEST._serialized_start=3434 + _GETCLUSTERSTATUSREQUEST._serialized_end=3459 + _GETCLUSTERSTATUSREPLY._serialized_start=3462 + _GETCLUSTERSTATUSREPLY._serialized_end=3624 + _DRAINNODEREQUEST._serialized_start=3626 + _DRAINNODEREQUEST._serialized_end=3738 + _DRAINNODEREPLY._serialized_start=3740 + _DRAINNODEREPLY._serialized_end=3777 + _AUTOSCALERSTATESERVICE._serialized_start=3986 + _AUTOSCALERSTATESERVICE._serialized_end=4617 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/autoscaler_pb2_grpc.py b/python/ray/core/generated/autoscaler_pb2_grpc.py new file mode 100644 index 0000000000000..30ec13628ab4f --- /dev/null +++ b/python/ray/core/generated/autoscaler_pb2_grpc.py @@ -0,0 +1,209 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import autoscaler_pb2 as src_dot_ray_dot_protobuf_dot_autoscaler__pb2 + + +class AutoscalerStateServiceStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.GetClusterResourceState = channel.unary_unary( + '/ray.rpc.autoscaler.AutoscalerStateService/GetClusterResourceState', + request_serializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.GetClusterResourceStateRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.GetClusterResourceStateReply.FromString, + ) + self.ReportAutoscalingState = channel.unary_unary( + '/ray.rpc.autoscaler.AutoscalerStateService/ReportAutoscalingState', + request_serializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.ReportAutoscalingStateRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.ReportAutoscalingStateReply.FromString, + ) + self.RequestClusterResourceConstraint = channel.unary_unary( + '/ray.rpc.autoscaler.AutoscalerStateService/RequestClusterResourceConstraint', + request_serializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.RequestClusterResourceConstraintRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.RequestClusterResourceConstraintReply.FromString, + ) + self.GetClusterStatus = channel.unary_unary( + '/ray.rpc.autoscaler.AutoscalerStateService/GetClusterStatus', + request_serializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.GetClusterStatusRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.GetClusterStatusReply.FromString, + ) + self.DrainNode = channel.unary_unary( + '/ray.rpc.autoscaler.AutoscalerStateService/DrainNode', + request_serializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.DrainNodeRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.DrainNodeReply.FromString, + ) + + +class AutoscalerStateServiceServicer(object): + """Missing associated documentation comment in .proto file.""" + + def GetClusterResourceState(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ReportAutoscalingState(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def RequestClusterResourceConstraint(self, request, context): + """Request additional cluster resources constraints. This is the RPC call made + when users call ray.autoscaler.sdk.request_resources. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetClusterStatus(self, request, context): + """Obtain the current cluster state, including the autoscaler state (e.g. pending + requests), and the cluster resource state (e.g. ray node states). + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def DrainNode(self, request, context): + """Request to drain a specified node. + The request is accepted if the node is already dead or + the acceptance condition is met. + If the return status is not OK, then the request might fail + and it's safe to retry since it's idempotent. + Caller should not read `is_accepted` from the reply + if the return status is not OK. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_AutoscalerStateServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'GetClusterResourceState': grpc.unary_unary_rpc_method_handler( + servicer.GetClusterResourceState, + request_deserializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.GetClusterResourceStateRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.GetClusterResourceStateReply.SerializeToString, + ), + 'ReportAutoscalingState': grpc.unary_unary_rpc_method_handler( + servicer.ReportAutoscalingState, + request_deserializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.ReportAutoscalingStateRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.ReportAutoscalingStateReply.SerializeToString, + ), + 'RequestClusterResourceConstraint': grpc.unary_unary_rpc_method_handler( + servicer.RequestClusterResourceConstraint, + request_deserializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.RequestClusterResourceConstraintRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.RequestClusterResourceConstraintReply.SerializeToString, + ), + 'GetClusterStatus': grpc.unary_unary_rpc_method_handler( + servicer.GetClusterStatus, + request_deserializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.GetClusterStatusRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.GetClusterStatusReply.SerializeToString, + ), + 'DrainNode': grpc.unary_unary_rpc_method_handler( + servicer.DrainNode, + request_deserializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.DrainNodeRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_autoscaler__pb2.DrainNodeReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.autoscaler.AutoscalerStateService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class AutoscalerStateService(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def GetClusterResourceState(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.autoscaler.AutoscalerStateService/GetClusterResourceState', + src_dot_ray_dot_protobuf_dot_autoscaler__pb2.GetClusterResourceStateRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_autoscaler__pb2.GetClusterResourceStateReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ReportAutoscalingState(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.autoscaler.AutoscalerStateService/ReportAutoscalingState', + src_dot_ray_dot_protobuf_dot_autoscaler__pb2.ReportAutoscalingStateRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_autoscaler__pb2.ReportAutoscalingStateReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def RequestClusterResourceConstraint(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.autoscaler.AutoscalerStateService/RequestClusterResourceConstraint', + src_dot_ray_dot_protobuf_dot_autoscaler__pb2.RequestClusterResourceConstraintRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_autoscaler__pb2.RequestClusterResourceConstraintReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetClusterStatus(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.autoscaler.AutoscalerStateService/GetClusterStatus', + src_dot_ray_dot_protobuf_dot_autoscaler__pb2.GetClusterStatusRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_autoscaler__pb2.GetClusterStatusReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def DrainNode(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.autoscaler.AutoscalerStateService/DrainNode', + src_dot_ray_dot_protobuf_dot_autoscaler__pb2.DrainNodeRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_autoscaler__pb2.DrainNodeReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/common_pb2.py b/python/ray/core/generated/common_pb2.py new file mode 100644 index 0000000000000..04102ce7c7d9a --- /dev/null +++ b/python/ray/core/generated/common_pb2.py @@ -0,0 +1,751 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/common.proto +"""Generated protocol buffer code.""" +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from . import runtime_env_common_pb2 as src_dot_ray_dot_protobuf_dot_runtime__env__common__pb2 + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1dsrc/ray/protobuf/common.proto\x12\x07ray.rpc\x1a)src/ray/protobuf/runtime_env_common.proto\"\x19\n\x07LabelIn\x12\x0e\n\x06values\x18\x01 \x03(\t\"\x1c\n\nLabelNotIn\x12\x0e\n\x06values\x18\x01 \x03(\t\"\r\n\x0bLabelExists\"\x13\n\x11LabelDoesNotExist\"\xde\x01\n\rLabelOperator\x12$\n\x08label_in\x18\x01 \x01(\x0b\x32\x10.ray.rpc.LabelInH\x00\x12+\n\x0clabel_not_in\x18\x02 \x01(\x0b\x32\x13.ray.rpc.LabelNotInH\x00\x12,\n\x0clabel_exists\x18\x03 \x01(\x0b\x32\x14.ray.rpc.LabelExistsH\x00\x12:\n\x14label_does_not_exist\x18\x04 \x01(\x0b\x32\x1a.ray.rpc.LabelDoesNotExistH\x00\x42\x10\n\x0elabel_operator\"M\n\x14LabelMatchExpression\x12\x0b\n\x03key\x18\x01 \x01(\t\x12(\n\x08operator\x18\x02 \x01(\x0b\x32\x16.ray.rpc.LabelOperator\"K\n\x15LabelMatchExpressions\x12\x32\n\x0b\x65xpressions\x18\x01 \x03(\x0b\x32\x1d.ray.rpc.LabelMatchExpression\"y\n\x1bNodeLabelSchedulingStrategy\x12,\n\x04hard\x18\x01 \x01(\x0b\x32\x1e.ray.rpc.LabelMatchExpressions\x12,\n\x04soft\x18\x02 \x01(\x0b\x32\x1e.ray.rpc.LabelMatchExpressions\"z\n\x1eNodeAffinitySchedulingStrategy\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\x12\x0c\n\x04soft\x18\x02 \x01(\x08\x12\x1c\n\x14spill_on_unavailable\x18\x03 \x01(\x08\x12\x1b\n\x13\x66\x61il_on_unavailable\x18\x04 \x01(\x08\"\x91\x01\n PlacementGroupSchedulingStrategy\x12\x1a\n\x12placement_group_id\x18\x01 \x01(\x0c\x12$\n\x1cplacement_group_bundle_index\x18\x02 \x01(\x03\x12+\n#placement_group_capture_child_tasks\x18\x03 \x01(\x08\"\x1b\n\x19\x44\x65\x66\x61ultSchedulingStrategy\"\x1a\n\x18SpreadSchedulingStrategy\"\xbf\x03\n\x12SchedulingStrategy\x12I\n\x1b\x64\x65\x66\x61ult_scheduling_strategy\x18\x01 \x01(\x0b\x32\".ray.rpc.DefaultSchedulingStrategyH\x00\x12X\n#placement_group_scheduling_strategy\x18\x02 \x01(\x0b\x32).ray.rpc.PlacementGroupSchedulingStrategyH\x00\x12G\n\x1aspread_scheduling_strategy\x18\x03 \x01(\x0b\x32!.ray.rpc.SpreadSchedulingStrategyH\x00\x12T\n!node_affinity_scheduling_strategy\x18\x04 \x01(\x0b\x32\'.ray.rpc.NodeAffinitySchedulingStrategyH\x00\x12N\n\x1enode_label_scheduling_strategy\x18\x05 \x01(\x0b\x32$.ray.rpc.NodeLabelSchedulingStrategyH\x00\x42\x15\n\x13scheduling_strategy\"Q\n\x07\x41\x64\x64ress\x12\x11\n\traylet_id\x18\x01 \x01(\x0c\x12\x12\n\nip_address\x18\x02 \x01(\t\x12\x0c\n\x04port\x18\x03 \x01(\x05\x12\x11\n\tworker_id\x18\x04 \x01(\x0c\"V\n\x16JavaFunctionDescriptor\x12\x12\n\nclass_name\x18\x01 \x01(\t\x12\x15\n\rfunction_name\x18\x02 \x01(\t\x12\x11\n\tsignature\x18\x03 \x01(\t\"q\n\x18PythonFunctionDescriptor\x12\x13\n\x0bmodule_name\x18\x01 \x01(\t\x12\x12\n\nclass_name\x18\x02 \x01(\t\x12\x15\n\rfunction_name\x18\x03 \x01(\t\x12\x15\n\rfunction_hash\x18\x04 \x01(\t\"R\n\x15\x43ppFunctionDescriptor\x12\x15\n\rfunction_name\x18\x01 \x01(\t\x12\x0e\n\x06\x63\x61ller\x18\x02 \x01(\t\x12\x12\n\nclass_name\x18\x03 \x01(\t\"\xfc\x01\n\x12\x46unctionDescriptor\x12\x43\n\x18java_function_descriptor\x18\x01 \x01(\x0b\x32\x1f.ray.rpc.JavaFunctionDescriptorH\x00\x12G\n\x1apython_function_descriptor\x18\x02 \x01(\x0b\x32!.ray.rpc.PythonFunctionDescriptorH\x00\x12\x41\n\x17\x63pp_function_descriptor\x18\x03 \x01(\x0b\x32\x1e.ray.rpc.CppFunctionDescriptorH\x00\x42\x15\n\x13\x66unction_descriptor\"t\n\x10\x43oncurrencyGroup\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x17\n\x0fmax_concurrency\x18\x02 \x01(\x05\x12\x39\n\x14\x66unction_descriptors\x18\x03 \x03(\x0b\x32\x1b.ray.rpc.FunctionDescriptor\"\xd8\x01\n\x0cRayErrorInfo\x12\x34\n\x10\x61\x63tor_died_error\x18\x02 \x01(\x0b\x32\x18.ray.rpc.ActorDeathCauseH\x00\x12J\n\x1eruntime_env_setup_failed_error\x18\x03 \x01(\x0b\x32 .ray.rpc.RuntimeEnvFailedContextH\x00\x12\x15\n\rerror_message\x18\x05 \x01(\t\x12&\n\nerror_type\x18\x0b \x01(\x0e\x32\x12.ray.rpc.ErrorTypeB\x07\n\x05\x65rror\"\xc6\x01\n\x17OutOfMemoryErrorContext\x12\x0f\n\x07task_id\x18\x01 \x01(\x0c\x12\x11\n\ttask_name\x18\x02 \x01(\t\x12\x17\n\x0fnode_ip_address\x18\x03 \x01(\t\x12\x19\n\x11memory_used_bytes\x18\x04 \x01(\x04\x12\x1a\n\x12memory_total_bytes\x18\x05 \x01(\x04\x12\x1d\n\x15memory_usage_fraction\x18\x06 \x01(\x01\x12\x18\n\x10memory_threshold\x18\x07 \x01(\x01\"@\n\x14NodeDiedErrorContext\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\x12\x17\n\x0fnode_ip_address\x18\x02 \x01(\t\"u\n\x0cRayException\x12#\n\x08language\x18\x01 \x01(\x0e\x32\x11.ray.rpc.Language\x12\x1c\n\x14serialized_exception\x18\x02 \x01(\x0c\x12\"\n\x1a\x66ormatted_exception_string\x18\x03 \x01(\t\"\xdf\x02\n\x0f\x41\x63torDeathCause\x12>\n\x1d\x63reation_task_failure_context\x18\x01 \x01(\x0b\x32\x15.ray.rpc.RayExceptionH\x00\x12\x46\n\x1aruntime_env_failed_context\x18\x02 \x01(\x0b\x32 .ray.rpc.RuntimeEnvFailedContextH\x00\x12\x42\n\x18\x61\x63tor_died_error_context\x18\x03 \x01(\x0b\x32\x1e.ray.rpc.ActorDiedErrorContextH\x00\x12I\n\x1b\x61\x63tor_unschedulable_context\x18\x04 \x01(\x0b\x32\".ray.rpc.ActorUnschedulableContextH\x00\x12*\n\x0boom_context\x18\x05 \x01(\x0b\x32\x13.ray.rpc.OomContextH\x00\x42\t\n\x07\x63ontext\"0\n\x17RuntimeEnvFailedContext\x12\x15\n\rerror_message\x18\x01 \x01(\t\"2\n\x19\x41\x63torUnschedulableContext\x12\x15\n\rerror_message\x18\x01 \x01(\t\"\xe2\x01\n\x15\x41\x63torDiedErrorContext\x12\x15\n\rerror_message\x18\x01 \x01(\t\x12\x10\n\x08owner_id\x18\x02 \x01(\x0c\x12\x18\n\x10owner_ip_address\x18\x03 \x01(\t\x12\x17\n\x0fnode_ip_address\x18\x04 \x01(\t\x12\x0b\n\x03pid\x18\x05 \x01(\r\x12\x0c\n\x04name\x18\x06 \x01(\t\x12\x15\n\rray_namespace\x18\x07 \x01(\t\x12\x12\n\nclass_name\x18\x08 \x01(\t\x12\x10\n\x08\x61\x63tor_id\x18\t \x01(\x0c\x12\x15\n\rnever_started\x18\n \x01(\x08\"=\n\nOomContext\x12\x15\n\rerror_message\x18\x01 \x01(\t\x12\x18\n\x10\x66\x61il_immediately\x18\x02 \x01(\x08\"\xf8\x02\n\tJobConfig\x12\x13\n\x0bjvm_options\x18\x02 \x03(\t\x12\x18\n\x10\x63ode_search_path\x18\x03 \x03(\t\x12\x31\n\x10runtime_env_info\x18\x04 \x01(\x0b\x32\x17.ray.rpc.RuntimeEnvInfo\x12\x15\n\rray_namespace\x18\x05 \x01(\t\x12\x32\n\x08metadata\x18\x06 \x03(\x0b\x32 .ray.rpc.JobConfig.MetadataEntry\x12@\n\x16\x64\x65\x66\x61ult_actor_lifetime\x18\x07 \x01(\x0e\x32 .ray.rpc.JobConfig.ActorLifetime\x12\x1a\n\x12py_driver_sys_path\x18\x08 \x03(\t\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"/\n\rActorLifetime\x12\x0c\n\x08\x44\x45TACHED\x10\x00\x12\x10\n\x0cNON_DETACHED\x10\x01\"M\n\x1eStreamingGeneratorReturnIdInfo\x12\x11\n\tobject_id\x18\x01 \x01(\x0c\x12\x18\n\x10is_plasma_object\x18\x02 \x01(\x08\"\xf9\n\n\x08TaskSpec\x12\x1f\n\x04type\x18\x01 \x01(\x0e\x32\x11.ray.rpc.TaskType\x12\x0c\n\x04name\x18\x02 \x01(\t\x12#\n\x08language\x18\x03 \x01(\x0e\x32\x11.ray.rpc.Language\x12\x38\n\x13\x66unction_descriptor\x18\x04 \x01(\x0b\x32\x1b.ray.rpc.FunctionDescriptor\x12\x0e\n\x06job_id\x18\x05 \x01(\x0c\x12\x0f\n\x07task_id\x18\x06 \x01(\x0c\x12\x16\n\x0eparent_task_id\x18\x07 \x01(\x0c\x12\x16\n\x0eparent_counter\x18\x08 \x01(\x04\x12\x11\n\tcaller_id\x18\t \x01(\x0c\x12(\n\x0e\x63\x61ller_address\x18\n \x01(\x0b\x32\x10.ray.rpc.Address\x12\x1e\n\x04\x61rgs\x18\x0b \x03(\x0b\x32\x10.ray.rpc.TaskArg\x12\x13\n\x0bnum_returns\x18\x0c \x01(\x04\x12\x44\n\x12required_resources\x18\r \x03(\x0b\x32(.ray.rpc.TaskSpec.RequiredResourcesEntry\x12W\n\x1crequired_placement_resources\x18\x0e \x03(\x0b\x32\x31.ray.rpc.TaskSpec.RequiredPlacementResourcesEntry\x12@\n\x18\x61\x63tor_creation_task_spec\x18\x0f \x01(\x0b\x32\x1e.ray.rpc.ActorCreationTaskSpec\x12/\n\x0f\x61\x63tor_task_spec\x18\x10 \x01(\x0b\x32\x16.ray.rpc.ActorTaskSpec\x12\x13\n\x0bmax_retries\x18\x11 \x01(\x05\x12\x16\n\x0eskip_execution\x18\x15 \x01(\x08\x12\x1b\n\x13\x64\x65\x62ugger_breakpoint\x18\x16 \x01(\x0c\x12\x31\n\x10runtime_env_info\x18\x17 \x01(\x0b\x32\x17.ray.rpc.RuntimeEnvInfo\x12\x1e\n\x16\x63oncurrency_group_name\x18\x18 \x01(\t\x12\x18\n\x10retry_exceptions\x18\x19 \x01(\x08\x12,\n$serialized_retry_exception_allowlist\x18\x1a \x01(\x0c\x12\r\n\x05\x64\x65pth\x18\x1b \x01(\x03\x12\x38\n\x13scheduling_strategy\x18\x1c \x01(\x0b\x32\x1b.ray.rpc.SchedulingStrategy\x12\x16\n\x0e\x61ttempt_number\x18\x1d \x01(\x04\x12\x17\n\x0freturns_dynamic\x18\x1e \x01(\x08\x12\x1a\n\x12\x64ynamic_return_ids\x18\x1f \x03(\x0c\x12+\n\njob_config\x18 \x01(\x0b\x32\x12.ray.rpc.JobConfigH\x00\x88\x01\x01\x12\x19\n\x11submitter_task_id\x18! \x01(\x0c\x12\x1b\n\x13streaming_generator\x18\" \x01(\x08\x12*\n\"dependency_resolution_timestamp_ms\x18# \x01(\x03\x12 \n\x18lease_grant_timestamp_ms\x18$ \x01(\x03\x12,\n\x1fnum_streaming_generator_returns\x18% \x01(\x04H\x01\x88\x01\x01\x1a\x38\n\x16RequiredResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x1a\x41\n\x1fRequiredPlacementResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x42\r\n\x0b_job_configB\"\n _num_streaming_generator_returns\"\x9d\x04\n\rTaskInfoEntry\x12\x1f\n\x04type\x18\x01 \x01(\x0e\x32\x11.ray.rpc.TaskType\x12\x0c\n\x04name\x18\x02 \x01(\t\x12#\n\x08language\x18\x03 \x01(\x0e\x32\x11.ray.rpc.Language\x12\x1a\n\x12\x66unc_or_class_name\x18\x04 \x01(\t\x12-\n\x10scheduling_state\x18\x05 \x01(\x0e\x32\x13.ray.rpc.TaskStatus\x12\x0e\n\x06job_id\x18\x06 \x01(\x0c\x12\x0f\n\x07task_id\x18\x07 \x01(\x0c\x12\x16\n\x0eparent_task_id\x18\x08 \x01(\x0c\x12I\n\x12required_resources\x18\r \x03(\x0b\x32-.ray.rpc.TaskInfoEntry.RequiredResourcesEntry\x12\x31\n\x10runtime_env_info\x18\x17 \x01(\x0b\x32\x17.ray.rpc.RuntimeEnvInfo\x12\x14\n\x07node_id\x18\x18 \x01(\x0cH\x00\x88\x01\x01\x12\x15\n\x08\x61\x63tor_id\x18\x19 \x01(\x0cH\x01\x88\x01\x01\x12\x1f\n\x12placement_group_id\x18\x1a \x01(\x0cH\x02\x88\x01\x01\x1a\x38\n\x16RequiredResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x42\n\n\x08_node_idB\x0b\n\t_actor_idB\x15\n\x13_placement_group_id\"\x86\x02\n\x06\x42undle\x12\x33\n\tbundle_id\x18\x01 \x01(\x0b\x32 .ray.rpc.Bundle.BundleIdentifier\x12:\n\x0eunit_resources\x18\x02 \x03(\x0b\x32\".ray.rpc.Bundle.UnitResourcesEntry\x12\x0f\n\x07node_id\x18\x03 \x01(\x0c\x1a\x44\n\x10\x42undleIdentifier\x12\x1a\n\x12placement_group_id\x18\x01 \x01(\x0c\x12\x14\n\x0c\x62undle_index\x18\x02 \x01(\x05\x1a\x34\n\x12UnitResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\"\xae\x02\n\x12PlacementGroupSpec\x12\x1a\n\x12placement_group_id\x18\x01 \x01(\x0c\x12\x0c\n\x04name\x18\x02 \x01(\t\x12 \n\x07\x62undles\x18\x03 \x03(\x0b\x32\x0f.ray.rpc.Bundle\x12,\n\x08strategy\x18\x04 \x01(\x0e\x32\x1a.ray.rpc.PlacementStrategy\x12\x16\n\x0e\x63reator_job_id\x18\x05 \x01(\x0c\x12\x18\n\x10\x63reator_actor_id\x18\x06 \x01(\x0c\x12\x18\n\x10\x63reator_job_dead\x18\x07 \x01(\x08\x12\x1a\n\x12\x63reator_actor_dead\x18\x08 \x01(\x08\x12\x13\n\x0bis_detached\x18\t \x01(\x08\x12!\n\x19max_cpu_fraction_per_node\x18\n \x01(\x01\"`\n\x0fObjectReference\x12\x11\n\tobject_id\x18\x01 \x01(\x0c\x12\'\n\rowner_address\x18\x02 \x01(\x0b\x32\x10.ray.rpc.Address\x12\x11\n\tcall_site\x18\x03 \x01(\t\"\xe9\x01\n\x14ObjectReferenceCount\x12+\n\treference\x18\x01 \x01(\x0b\x32\x18.ray.rpc.ObjectReference\x12\x15\n\rhas_local_ref\x18\x02 \x01(\x08\x12#\n\tborrowers\x18\x03 \x03(\x0b\x32\x10.ray.rpc.Address\x12\x33\n\x11stored_in_objects\x18\x04 \x03(\x0b\x32\x18.ray.rpc.ObjectReference\x12!\n\x19\x63ontained_in_borrowed_ids\x18\x05 \x03(\x0c\x12\x10\n\x08\x63ontains\x18\x06 \x03(\x0c\"\x8e\x01\n\x07TaskArg\x12,\n\nobject_ref\x18\x01 \x01(\x0b\x32\x18.ray.rpc.ObjectReference\x12\x0c\n\x04\x64\x61ta\x18\x02 \x01(\x0c\x12\x10\n\x08metadata\x18\x03 \x01(\x0c\x12\x35\n\x13nested_inlined_refs\x18\x04 \x03(\x0b\x32\x18.ray.rpc.ObjectReference\"\x99\x01\n\x0cReturnObject\x12\x11\n\tobject_id\x18\x01 \x01(\x0c\x12\x11\n\tin_plasma\x18\x02 \x01(\x08\x12\x0c\n\x04\x64\x61ta\x18\x03 \x01(\x0c\x12\x10\n\x08metadata\x18\x04 \x01(\x0c\x12\x35\n\x13nested_inlined_refs\x18\x05 \x03(\x0b\x32\x18.ray.rpc.ObjectReference\x12\x0c\n\x04size\x18\x06 \x01(\x03\"\x8f\x03\n\x15\x41\x63torCreationTaskSpec\x12\x10\n\x08\x61\x63tor_id\x18\x02 \x01(\x0c\x12\x1a\n\x12max_actor_restarts\x18\x03 \x01(\x03\x12\x18\n\x10max_task_retries\x18\x04 \x01(\x03\x12\x1e\n\x16\x64ynamic_worker_options\x18\x05 \x03(\t\x12\x17\n\x0fmax_concurrency\x18\x06 \x01(\x05\x12\x13\n\x0bis_detached\x18\x07 \x01(\x08\x12\x0c\n\x04name\x18\x08 \x01(\t\x12\x15\n\rray_namespace\x18\t \x01(\t\x12\x12\n\nis_asyncio\x18\n \x01(\x08\x12\x16\n\x0e\x65xtension_data\x18\x0b \x01(\t\x12\x1f\n\x17serialized_actor_handle\x18\x0c \x01(\x0c\x12\x35\n\x12\x63oncurrency_groups\x18\r \x03(\x0b\x32\x19.ray.rpc.ConcurrencyGroup\x12\x1c\n\x14\x65xecute_out_of_order\x18\x0e \x01(\x08\x12\x19\n\x11max_pending_calls\x18\x0f \x01(\x05\"`\n\rActorTaskSpec\x12\x10\n\x08\x61\x63tor_id\x18\x02 \x01(\x0c\x12&\n\x1e\x61\x63tor_creation_dummy_object_id\x18\x04 \x01(\x0c\x12\x15\n\ractor_counter\x18\x05 \x01(\x04\",\n\x04Task\x12$\n\ttask_spec\x18\x01 \x01(\x0b\x32\x11.ray.rpc.TaskSpec\"-\n\nResourceId\x12\r\n\x05index\x18\x01 \x01(\x03\x12\x10\n\x08quantity\x18\x02 \x01(\x01\"K\n\x10ResourceMapEntry\x12\x0c\n\x04name\x18\x01 \x01(\t\x12)\n\x0cresource_ids\x18\x02 \x03(\x0b\x32\x13.ray.rpc.ResourceId\"\xc4\x02\n\x08ViewData\x12\x11\n\tview_name\x18\x01 \x01(\t\x12+\n\x08measures\x18\x02 \x03(\x0b\x32\x19.ray.rpc.ViewData.Measure\x1a\xf7\x01\n\x07Measure\x12\x0c\n\x04tags\x18\x01 \x01(\t\x12\x11\n\tint_value\x18\x02 \x01(\x03\x12\x14\n\x0c\x64ouble_value\x18\x03 \x01(\x01\x12\x18\n\x10\x64istribution_min\x18\x04 \x01(\x01\x12\x19\n\x11\x64istribution_mean\x18\x05 \x01(\x01\x12\x18\n\x10\x64istribution_max\x18\x06 \x01(\x01\x12\x1a\n\x12\x64istribution_count\x18\x07 \x01(\x01\x12&\n\x1e\x64istribution_bucket_boundaries\x18\x08 \x03(\x01\x12\"\n\x1a\x64istribution_bucket_counts\x18\t \x03(\x01\"\xfd\x01\n\rObjectRefInfo\x12\x11\n\tobject_id\x18\x01 \x01(\x0c\x12\x11\n\tcall_site\x18\x02 \x01(\t\x12\x13\n\x0bobject_size\x18\x03 \x01(\x03\x12\x17\n\x0flocal_ref_count\x18\x04 \x01(\x03\x12 \n\x18submitted_task_ref_count\x18\x05 \x01(\x03\x12\x1a\n\x12\x63ontained_in_owned\x18\x06 \x03(\x0c\x12\x18\n\x10pinned_in_memory\x18\x07 \x01(\x08\x12(\n\x0btask_status\x18\x08 \x01(\x0e\x32\x13.ray.rpc.TaskStatus\x12\x16\n\x0e\x61ttempt_number\x18\t \x01(\x04\"\x8a\x01\n\x13ResourceAllocations\x12\x41\n\x0eresource_slots\x18\x01 \x03(\x0b\x32).ray.rpc.ResourceAllocations.ResourceSlot\x1a\x30\n\x0cResourceSlot\x12\x0c\n\x04slot\x18\x01 \x01(\x03\x12\x12\n\nallocation\x18\x02 \x01(\x01\"\xcc\x06\n\x0f\x43oreWorkerStats\x12\x19\n\x11num_pending_tasks\x18\x02 \x01(\x05\x12 \n\x18num_object_refs_in_scope\x18\x03 \x01(\x05\x12\x12\n\nip_address\x18\x07 \x01(\t\x12\x0c\n\x04port\x18\x08 \x01(\x03\x12\x10\n\x08\x61\x63tor_id\x18\t \x01(\x0c\x12\x43\n\x0eused_resources\x18\n \x03(\x0b\x32+.ray.rpc.CoreWorkerStats.UsedResourcesEntry\x12\x41\n\rwebui_display\x18\x0b \x03(\x0b\x32*.ray.rpc.CoreWorkerStats.WebuiDisplayEntry\x12\x15\n\rnum_in_plasma\x18\x0c \x01(\x05\x12\x19\n\x11num_local_objects\x18\r \x01(\x05\x12 \n\x18used_object_store_memory\x18\x0e \x01(\x03\x12\x19\n\x11task_queue_length\x18\x0f \x01(\x05\x12\x1a\n\x12num_executed_tasks\x18\x10 \x01(\x05\x12\x13\n\x0b\x61\x63tor_title\x18\x11 \x01(\t\x12+\n\x0bobject_refs\x18\x12 \x03(\x0b\x32\x16.ray.rpc.ObjectRefInfo\x12\x0e\n\x06job_id\x18\x13 \x01(\x0c\x12\x11\n\tworker_id\x18\x14 \x01(\x0c\x12#\n\x08language\x18\x15 \x01(\x0e\x32\x11.ray.rpc.Language\x12\x0b\n\x03pid\x18\x16 \x01(\r\x12(\n\x0bworker_type\x18\x17 \x01(\x0e\x32\x13.ray.rpc.WorkerType\x12\x15\n\robjects_total\x18\x18 \x01(\x03\x12\x19\n\x11num_owned_objects\x18\x19 \x01(\x03\x12\x18\n\x10num_owned_actors\x18\x1a \x01(\x03\x12\x19\n\x11num_running_tasks\x18\x1b \x01(\x03\x1aR\n\x12UsedResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12+\n\x05value\x18\x02 \x01(\x0b\x32\x1c.ray.rpc.ResourceAllocations:\x02\x38\x01\x1a\x33\n\x11WebuiDisplayEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01J\x04\x08\x01\x10\x02\"!\n\x11NodeResourceUsage\x12\x0c\n\x04json\x18\x01 \x01(\t\"\xc3\x01\n\x0bMetricPoint\x12\x13\n\x0bmetric_name\x18\x01 \x01(\t\x12\x11\n\ttimestamp\x18\x02 \x01(\x03\x12\r\n\x05value\x18\x03 \x01(\x01\x12,\n\x04tags\x18\x04 \x03(\x0b\x32\x1e.ray.rpc.MetricPoint.TagsEntry\x12\x13\n\x0b\x64\x65scription\x18\x05 \x01(\t\x12\r\n\x05units\x18\x06 \x01(\t\x1a+\n\tTagsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"5\n\x0eNamedActorInfo\x12\x15\n\rray_namespace\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t*)\n\x08Language\x12\n\n\x06PYTHON\x10\x00\x12\x08\n\x04JAVA\x10\x01\x12\x07\n\x03\x43PP\x10\x02*J\n\nWorkerType\x12\n\n\x06WORKER\x10\x00\x12\n\n\x06\x44RIVER\x10\x01\x12\x10\n\x0cSPILL_WORKER\x10\x02\x12\x12\n\x0eRESTORE_WORKER\x10\x03*U\n\x08TaskType\x12\x0f\n\x0bNORMAL_TASK\x10\x00\x12\x17\n\x13\x41\x43TOR_CREATION_TASK\x10\x01\x12\x0e\n\nACTOR_TASK\x10\x02\x12\x0f\n\x0b\x44RIVER_TASK\x10\x03*\xa4\x05\n\tErrorType\x12\x0f\n\x0bWORKER_DIED\x10\x00\x12\x0e\n\nACTOR_DIED\x10\x01\x12\x1c\n\x18OBJECT_UNRECONSTRUCTABLE\x10\x02\x12\x1c\n\x18TASK_EXECUTION_EXCEPTION\x10\x03\x12\x14\n\x10OBJECT_IN_PLASMA\x10\x04\x12\x12\n\x0eTASK_CANCELLED\x10\x05\x12\x19\n\x15\x41\x43TOR_CREATION_FAILED\x10\x06\x12\x1c\n\x18RUNTIME_ENV_SETUP_FAILED\x10\x07\x12\x0f\n\x0bOBJECT_LOST\x10\x08\x12\x0e\n\nOWNER_DIED\x10\t\x12\x12\n\x0eOBJECT_DELETED\x10\n\x12 \n\x1c\x44\x45PENDENCY_RESOLUTION_FAILED\x10\x0b\x12\x32\n.OBJECT_UNRECONSTRUCTABLE_MAX_ATTEMPTS_EXCEEDED\x10\x0c\x12,\n(OBJECT_UNRECONSTRUCTABLE_LINEAGE_EVICTED\x10\r\x12\x1a\n\x16OBJECT_FETCH_TIMED_OUT\x10\x0e\x12\x15\n\x11LOCAL_RAYLET_DIED\x10\x0f\x12 \n\x1cTASK_PLACEMENT_GROUP_REMOVED\x10\x10\x12!\n\x1d\x41\x43TOR_PLACEMENT_GROUP_REMOVED\x10\x11\x12\x1c\n\x18TASK_UNSCHEDULABLE_ERROR\x10\x12\x12\x1d\n\x19\x41\x43TOR_UNSCHEDULABLE_ERROR\x10\x13\x12\x15\n\x11OUT_OF_DISK_ERROR\x10\x14\x12\x10\n\x0cOBJECT_FREED\x10\x15\x12\x11\n\rOUT_OF_MEMORY\x10\x16\x12\r\n\tNODE_DIED\x10\x17\x12\x1e\n\x1a\x45ND_OF_STREAMING_GENERATOR\x10\x18*\xf4\x01\n\nTaskStatus\x12\x07\n\x03NIL\x10\x00\x12\x16\n\x12PENDING_ARGS_AVAIL\x10\x01\x12\x1b\n\x17PENDING_NODE_ASSIGNMENT\x10\x02\x12\x1f\n\x1bPENDING_OBJ_STORE_MEM_AVAIL\x10\x03\x12\x16\n\x12PENDING_ARGS_FETCH\x10\x04\x12\x17\n\x13SUBMITTED_TO_WORKER\x10\x05\x12\x0b\n\x07RUNNING\x10\x06\x12\x16\n\x12RUNNING_IN_RAY_GET\x10\x07\x12\x17\n\x13RUNNING_IN_RAY_WAIT\x10\x08\x12\x0c\n\x08\x46INISHED\x10\t\x12\n\n\x06\x46\x41ILED\x10\n*|\n\x0eWorkerExitType\x12\x10\n\x0cSYSTEM_ERROR\x10\x00\x12\x18\n\x14INTENDED_SYSTEM_EXIT\x10\x01\x12\x0e\n\nUSER_ERROR\x10\x02\x12\x16\n\x12INTENDED_USER_EXIT\x10\x03\x12\x16\n\x12NODE_OUT_OF_MEMORY\x10\x04*M\n\x11PlacementStrategy\x12\x08\n\x04PACK\x10\x00\x12\n\n\x06SPREAD\x10\x01\x12\x0f\n\x0bSTRICT_PACK\x10\x02\x12\x11\n\rSTRICT_SPREAD\x10\x03\x42\x1d\n\x18io.ray.runtime.generated\xf8\x01\x01\x62\x06proto3') + +_LANGUAGE = DESCRIPTOR.enum_types_by_name['Language'] +Language = enum_type_wrapper.EnumTypeWrapper(_LANGUAGE) +_WORKERTYPE = DESCRIPTOR.enum_types_by_name['WorkerType'] +WorkerType = enum_type_wrapper.EnumTypeWrapper(_WORKERTYPE) +_TASKTYPE = DESCRIPTOR.enum_types_by_name['TaskType'] +TaskType = enum_type_wrapper.EnumTypeWrapper(_TASKTYPE) +_ERRORTYPE = DESCRIPTOR.enum_types_by_name['ErrorType'] +ErrorType = enum_type_wrapper.EnumTypeWrapper(_ERRORTYPE) +_TASKSTATUS = DESCRIPTOR.enum_types_by_name['TaskStatus'] +TaskStatus = enum_type_wrapper.EnumTypeWrapper(_TASKSTATUS) +_WORKEREXITTYPE = DESCRIPTOR.enum_types_by_name['WorkerExitType'] +WorkerExitType = enum_type_wrapper.EnumTypeWrapper(_WORKEREXITTYPE) +_PLACEMENTSTRATEGY = DESCRIPTOR.enum_types_by_name['PlacementStrategy'] +PlacementStrategy = enum_type_wrapper.EnumTypeWrapper(_PLACEMENTSTRATEGY) +PYTHON = 0 +JAVA = 1 +CPP = 2 +WORKER = 0 +DRIVER = 1 +SPILL_WORKER = 2 +RESTORE_WORKER = 3 +NORMAL_TASK = 0 +ACTOR_CREATION_TASK = 1 +ACTOR_TASK = 2 +DRIVER_TASK = 3 +WORKER_DIED = 0 +ACTOR_DIED = 1 +OBJECT_UNRECONSTRUCTABLE = 2 +TASK_EXECUTION_EXCEPTION = 3 +OBJECT_IN_PLASMA = 4 +TASK_CANCELLED = 5 +ACTOR_CREATION_FAILED = 6 +RUNTIME_ENV_SETUP_FAILED = 7 +OBJECT_LOST = 8 +OWNER_DIED = 9 +OBJECT_DELETED = 10 +DEPENDENCY_RESOLUTION_FAILED = 11 +OBJECT_UNRECONSTRUCTABLE_MAX_ATTEMPTS_EXCEEDED = 12 +OBJECT_UNRECONSTRUCTABLE_LINEAGE_EVICTED = 13 +OBJECT_FETCH_TIMED_OUT = 14 +LOCAL_RAYLET_DIED = 15 +TASK_PLACEMENT_GROUP_REMOVED = 16 +ACTOR_PLACEMENT_GROUP_REMOVED = 17 +TASK_UNSCHEDULABLE_ERROR = 18 +ACTOR_UNSCHEDULABLE_ERROR = 19 +OUT_OF_DISK_ERROR = 20 +OBJECT_FREED = 21 +OUT_OF_MEMORY = 22 +NODE_DIED = 23 +END_OF_STREAMING_GENERATOR = 24 +NIL = 0 +PENDING_ARGS_AVAIL = 1 +PENDING_NODE_ASSIGNMENT = 2 +PENDING_OBJ_STORE_MEM_AVAIL = 3 +PENDING_ARGS_FETCH = 4 +SUBMITTED_TO_WORKER = 5 +RUNNING = 6 +RUNNING_IN_RAY_GET = 7 +RUNNING_IN_RAY_WAIT = 8 +FINISHED = 9 +FAILED = 10 +SYSTEM_ERROR = 0 +INTENDED_SYSTEM_EXIT = 1 +USER_ERROR = 2 +INTENDED_USER_EXIT = 3 +NODE_OUT_OF_MEMORY = 4 +PACK = 0 +SPREAD = 1 +STRICT_PACK = 2 +STRICT_SPREAD = 3 + + +_LABELIN = DESCRIPTOR.message_types_by_name['LabelIn'] +_LABELNOTIN = DESCRIPTOR.message_types_by_name['LabelNotIn'] +_LABELEXISTS = DESCRIPTOR.message_types_by_name['LabelExists'] +_LABELDOESNOTEXIST = DESCRIPTOR.message_types_by_name['LabelDoesNotExist'] +_LABELOPERATOR = DESCRIPTOR.message_types_by_name['LabelOperator'] +_LABELMATCHEXPRESSION = DESCRIPTOR.message_types_by_name['LabelMatchExpression'] +_LABELMATCHEXPRESSIONS = DESCRIPTOR.message_types_by_name['LabelMatchExpressions'] +_NODELABELSCHEDULINGSTRATEGY = DESCRIPTOR.message_types_by_name['NodeLabelSchedulingStrategy'] +_NODEAFFINITYSCHEDULINGSTRATEGY = DESCRIPTOR.message_types_by_name['NodeAffinitySchedulingStrategy'] +_PLACEMENTGROUPSCHEDULINGSTRATEGY = DESCRIPTOR.message_types_by_name['PlacementGroupSchedulingStrategy'] +_DEFAULTSCHEDULINGSTRATEGY = DESCRIPTOR.message_types_by_name['DefaultSchedulingStrategy'] +_SPREADSCHEDULINGSTRATEGY = DESCRIPTOR.message_types_by_name['SpreadSchedulingStrategy'] +_SCHEDULINGSTRATEGY = DESCRIPTOR.message_types_by_name['SchedulingStrategy'] +_ADDRESS = DESCRIPTOR.message_types_by_name['Address'] +_JAVAFUNCTIONDESCRIPTOR = DESCRIPTOR.message_types_by_name['JavaFunctionDescriptor'] +_PYTHONFUNCTIONDESCRIPTOR = DESCRIPTOR.message_types_by_name['PythonFunctionDescriptor'] +_CPPFUNCTIONDESCRIPTOR = DESCRIPTOR.message_types_by_name['CppFunctionDescriptor'] +_FUNCTIONDESCRIPTOR = DESCRIPTOR.message_types_by_name['FunctionDescriptor'] +_CONCURRENCYGROUP = DESCRIPTOR.message_types_by_name['ConcurrencyGroup'] +_RAYERRORINFO = DESCRIPTOR.message_types_by_name['RayErrorInfo'] +_OUTOFMEMORYERRORCONTEXT = DESCRIPTOR.message_types_by_name['OutOfMemoryErrorContext'] +_NODEDIEDERRORCONTEXT = DESCRIPTOR.message_types_by_name['NodeDiedErrorContext'] +_RAYEXCEPTION = DESCRIPTOR.message_types_by_name['RayException'] +_ACTORDEATHCAUSE = DESCRIPTOR.message_types_by_name['ActorDeathCause'] +_RUNTIMEENVFAILEDCONTEXT = DESCRIPTOR.message_types_by_name['RuntimeEnvFailedContext'] +_ACTORUNSCHEDULABLECONTEXT = DESCRIPTOR.message_types_by_name['ActorUnschedulableContext'] +_ACTORDIEDERRORCONTEXT = DESCRIPTOR.message_types_by_name['ActorDiedErrorContext'] +_OOMCONTEXT = DESCRIPTOR.message_types_by_name['OomContext'] +_JOBCONFIG = DESCRIPTOR.message_types_by_name['JobConfig'] +_JOBCONFIG_METADATAENTRY = _JOBCONFIG.nested_types_by_name['MetadataEntry'] +_STREAMINGGENERATORRETURNIDINFO = DESCRIPTOR.message_types_by_name['StreamingGeneratorReturnIdInfo'] +_TASKSPEC = DESCRIPTOR.message_types_by_name['TaskSpec'] +_TASKSPEC_REQUIREDRESOURCESENTRY = _TASKSPEC.nested_types_by_name['RequiredResourcesEntry'] +_TASKSPEC_REQUIREDPLACEMENTRESOURCESENTRY = _TASKSPEC.nested_types_by_name['RequiredPlacementResourcesEntry'] +_TASKINFOENTRY = DESCRIPTOR.message_types_by_name['TaskInfoEntry'] +_TASKINFOENTRY_REQUIREDRESOURCESENTRY = _TASKINFOENTRY.nested_types_by_name['RequiredResourcesEntry'] +_BUNDLE = DESCRIPTOR.message_types_by_name['Bundle'] +_BUNDLE_BUNDLEIDENTIFIER = _BUNDLE.nested_types_by_name['BundleIdentifier'] +_BUNDLE_UNITRESOURCESENTRY = _BUNDLE.nested_types_by_name['UnitResourcesEntry'] +_PLACEMENTGROUPSPEC = DESCRIPTOR.message_types_by_name['PlacementGroupSpec'] +_OBJECTREFERENCE = DESCRIPTOR.message_types_by_name['ObjectReference'] +_OBJECTREFERENCECOUNT = DESCRIPTOR.message_types_by_name['ObjectReferenceCount'] +_TASKARG = DESCRIPTOR.message_types_by_name['TaskArg'] +_RETURNOBJECT = DESCRIPTOR.message_types_by_name['ReturnObject'] +_ACTORCREATIONTASKSPEC = DESCRIPTOR.message_types_by_name['ActorCreationTaskSpec'] +_ACTORTASKSPEC = DESCRIPTOR.message_types_by_name['ActorTaskSpec'] +_TASK = DESCRIPTOR.message_types_by_name['Task'] +_RESOURCEID = DESCRIPTOR.message_types_by_name['ResourceId'] +_RESOURCEMAPENTRY = DESCRIPTOR.message_types_by_name['ResourceMapEntry'] +_VIEWDATA = DESCRIPTOR.message_types_by_name['ViewData'] +_VIEWDATA_MEASURE = _VIEWDATA.nested_types_by_name['Measure'] +_OBJECTREFINFO = DESCRIPTOR.message_types_by_name['ObjectRefInfo'] +_RESOURCEALLOCATIONS = DESCRIPTOR.message_types_by_name['ResourceAllocations'] +_RESOURCEALLOCATIONS_RESOURCESLOT = _RESOURCEALLOCATIONS.nested_types_by_name['ResourceSlot'] +_COREWORKERSTATS = DESCRIPTOR.message_types_by_name['CoreWorkerStats'] +_COREWORKERSTATS_USEDRESOURCESENTRY = _COREWORKERSTATS.nested_types_by_name['UsedResourcesEntry'] +_COREWORKERSTATS_WEBUIDISPLAYENTRY = _COREWORKERSTATS.nested_types_by_name['WebuiDisplayEntry'] +_NODERESOURCEUSAGE = DESCRIPTOR.message_types_by_name['NodeResourceUsage'] +_METRICPOINT = DESCRIPTOR.message_types_by_name['MetricPoint'] +_METRICPOINT_TAGSENTRY = _METRICPOINT.nested_types_by_name['TagsEntry'] +_NAMEDACTORINFO = DESCRIPTOR.message_types_by_name['NamedActorInfo'] +_JOBCONFIG_ACTORLIFETIME = _JOBCONFIG.enum_types_by_name['ActorLifetime'] +LabelIn = _reflection.GeneratedProtocolMessageType('LabelIn', (_message.Message,), { + 'DESCRIPTOR' : _LABELIN, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.LabelIn) + }) +_sym_db.RegisterMessage(LabelIn) + +LabelNotIn = _reflection.GeneratedProtocolMessageType('LabelNotIn', (_message.Message,), { + 'DESCRIPTOR' : _LABELNOTIN, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.LabelNotIn) + }) +_sym_db.RegisterMessage(LabelNotIn) + +LabelExists = _reflection.GeneratedProtocolMessageType('LabelExists', (_message.Message,), { + 'DESCRIPTOR' : _LABELEXISTS, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.LabelExists) + }) +_sym_db.RegisterMessage(LabelExists) + +LabelDoesNotExist = _reflection.GeneratedProtocolMessageType('LabelDoesNotExist', (_message.Message,), { + 'DESCRIPTOR' : _LABELDOESNOTEXIST, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.LabelDoesNotExist) + }) +_sym_db.RegisterMessage(LabelDoesNotExist) + +LabelOperator = _reflection.GeneratedProtocolMessageType('LabelOperator', (_message.Message,), { + 'DESCRIPTOR' : _LABELOPERATOR, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.LabelOperator) + }) +_sym_db.RegisterMessage(LabelOperator) + +LabelMatchExpression = _reflection.GeneratedProtocolMessageType('LabelMatchExpression', (_message.Message,), { + 'DESCRIPTOR' : _LABELMATCHEXPRESSION, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.LabelMatchExpression) + }) +_sym_db.RegisterMessage(LabelMatchExpression) + +LabelMatchExpressions = _reflection.GeneratedProtocolMessageType('LabelMatchExpressions', (_message.Message,), { + 'DESCRIPTOR' : _LABELMATCHEXPRESSIONS, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.LabelMatchExpressions) + }) +_sym_db.RegisterMessage(LabelMatchExpressions) + +NodeLabelSchedulingStrategy = _reflection.GeneratedProtocolMessageType('NodeLabelSchedulingStrategy', (_message.Message,), { + 'DESCRIPTOR' : _NODELABELSCHEDULINGSTRATEGY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NodeLabelSchedulingStrategy) + }) +_sym_db.RegisterMessage(NodeLabelSchedulingStrategy) + +NodeAffinitySchedulingStrategy = _reflection.GeneratedProtocolMessageType('NodeAffinitySchedulingStrategy', (_message.Message,), { + 'DESCRIPTOR' : _NODEAFFINITYSCHEDULINGSTRATEGY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NodeAffinitySchedulingStrategy) + }) +_sym_db.RegisterMessage(NodeAffinitySchedulingStrategy) + +PlacementGroupSchedulingStrategy = _reflection.GeneratedProtocolMessageType('PlacementGroupSchedulingStrategy', (_message.Message,), { + 'DESCRIPTOR' : _PLACEMENTGROUPSCHEDULINGSTRATEGY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PlacementGroupSchedulingStrategy) + }) +_sym_db.RegisterMessage(PlacementGroupSchedulingStrategy) + +DefaultSchedulingStrategy = _reflection.GeneratedProtocolMessageType('DefaultSchedulingStrategy', (_message.Message,), { + 'DESCRIPTOR' : _DEFAULTSCHEDULINGSTRATEGY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DefaultSchedulingStrategy) + }) +_sym_db.RegisterMessage(DefaultSchedulingStrategy) + +SpreadSchedulingStrategy = _reflection.GeneratedProtocolMessageType('SpreadSchedulingStrategy', (_message.Message,), { + 'DESCRIPTOR' : _SPREADSCHEDULINGSTRATEGY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.SpreadSchedulingStrategy) + }) +_sym_db.RegisterMessage(SpreadSchedulingStrategy) + +SchedulingStrategy = _reflection.GeneratedProtocolMessageType('SchedulingStrategy', (_message.Message,), { + 'DESCRIPTOR' : _SCHEDULINGSTRATEGY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.SchedulingStrategy) + }) +_sym_db.RegisterMessage(SchedulingStrategy) + +Address = _reflection.GeneratedProtocolMessageType('Address', (_message.Message,), { + 'DESCRIPTOR' : _ADDRESS, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.Address) + }) +_sym_db.RegisterMessage(Address) + +JavaFunctionDescriptor = _reflection.GeneratedProtocolMessageType('JavaFunctionDescriptor', (_message.Message,), { + 'DESCRIPTOR' : _JAVAFUNCTIONDESCRIPTOR, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.JavaFunctionDescriptor) + }) +_sym_db.RegisterMessage(JavaFunctionDescriptor) + +PythonFunctionDescriptor = _reflection.GeneratedProtocolMessageType('PythonFunctionDescriptor', (_message.Message,), { + 'DESCRIPTOR' : _PYTHONFUNCTIONDESCRIPTOR, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PythonFunctionDescriptor) + }) +_sym_db.RegisterMessage(PythonFunctionDescriptor) + +CppFunctionDescriptor = _reflection.GeneratedProtocolMessageType('CppFunctionDescriptor', (_message.Message,), { + 'DESCRIPTOR' : _CPPFUNCTIONDESCRIPTOR, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CppFunctionDescriptor) + }) +_sym_db.RegisterMessage(CppFunctionDescriptor) + +FunctionDescriptor = _reflection.GeneratedProtocolMessageType('FunctionDescriptor', (_message.Message,), { + 'DESCRIPTOR' : _FUNCTIONDESCRIPTOR, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.FunctionDescriptor) + }) +_sym_db.RegisterMessage(FunctionDescriptor) + +ConcurrencyGroup = _reflection.GeneratedProtocolMessageType('ConcurrencyGroup', (_message.Message,), { + 'DESCRIPTOR' : _CONCURRENCYGROUP, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ConcurrencyGroup) + }) +_sym_db.RegisterMessage(ConcurrencyGroup) + +RayErrorInfo = _reflection.GeneratedProtocolMessageType('RayErrorInfo', (_message.Message,), { + 'DESCRIPTOR' : _RAYERRORINFO, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RayErrorInfo) + }) +_sym_db.RegisterMessage(RayErrorInfo) + +OutOfMemoryErrorContext = _reflection.GeneratedProtocolMessageType('OutOfMemoryErrorContext', (_message.Message,), { + 'DESCRIPTOR' : _OUTOFMEMORYERRORCONTEXT, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.OutOfMemoryErrorContext) + }) +_sym_db.RegisterMessage(OutOfMemoryErrorContext) + +NodeDiedErrorContext = _reflection.GeneratedProtocolMessageType('NodeDiedErrorContext', (_message.Message,), { + 'DESCRIPTOR' : _NODEDIEDERRORCONTEXT, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NodeDiedErrorContext) + }) +_sym_db.RegisterMessage(NodeDiedErrorContext) + +RayException = _reflection.GeneratedProtocolMessageType('RayException', (_message.Message,), { + 'DESCRIPTOR' : _RAYEXCEPTION, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RayException) + }) +_sym_db.RegisterMessage(RayException) + +ActorDeathCause = _reflection.GeneratedProtocolMessageType('ActorDeathCause', (_message.Message,), { + 'DESCRIPTOR' : _ACTORDEATHCAUSE, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ActorDeathCause) + }) +_sym_db.RegisterMessage(ActorDeathCause) + +RuntimeEnvFailedContext = _reflection.GeneratedProtocolMessageType('RuntimeEnvFailedContext', (_message.Message,), { + 'DESCRIPTOR' : _RUNTIMEENVFAILEDCONTEXT, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RuntimeEnvFailedContext) + }) +_sym_db.RegisterMessage(RuntimeEnvFailedContext) + +ActorUnschedulableContext = _reflection.GeneratedProtocolMessageType('ActorUnschedulableContext', (_message.Message,), { + 'DESCRIPTOR' : _ACTORUNSCHEDULABLECONTEXT, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ActorUnschedulableContext) + }) +_sym_db.RegisterMessage(ActorUnschedulableContext) + +ActorDiedErrorContext = _reflection.GeneratedProtocolMessageType('ActorDiedErrorContext', (_message.Message,), { + 'DESCRIPTOR' : _ACTORDIEDERRORCONTEXT, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ActorDiedErrorContext) + }) +_sym_db.RegisterMessage(ActorDiedErrorContext) + +OomContext = _reflection.GeneratedProtocolMessageType('OomContext', (_message.Message,), { + 'DESCRIPTOR' : _OOMCONTEXT, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.OomContext) + }) +_sym_db.RegisterMessage(OomContext) + +JobConfig = _reflection.GeneratedProtocolMessageType('JobConfig', (_message.Message,), { + + 'MetadataEntry' : _reflection.GeneratedProtocolMessageType('MetadataEntry', (_message.Message,), { + 'DESCRIPTOR' : _JOBCONFIG_METADATAENTRY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.JobConfig.MetadataEntry) + }) + , + 'DESCRIPTOR' : _JOBCONFIG, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.JobConfig) + }) +_sym_db.RegisterMessage(JobConfig) +_sym_db.RegisterMessage(JobConfig.MetadataEntry) + +StreamingGeneratorReturnIdInfo = _reflection.GeneratedProtocolMessageType('StreamingGeneratorReturnIdInfo', (_message.Message,), { + 'DESCRIPTOR' : _STREAMINGGENERATORRETURNIDINFO, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.StreamingGeneratorReturnIdInfo) + }) +_sym_db.RegisterMessage(StreamingGeneratorReturnIdInfo) + +TaskSpec = _reflection.GeneratedProtocolMessageType('TaskSpec', (_message.Message,), { + + 'RequiredResourcesEntry' : _reflection.GeneratedProtocolMessageType('RequiredResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _TASKSPEC_REQUIREDRESOURCESENTRY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TaskSpec.RequiredResourcesEntry) + }) + , + + 'RequiredPlacementResourcesEntry' : _reflection.GeneratedProtocolMessageType('RequiredPlacementResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _TASKSPEC_REQUIREDPLACEMENTRESOURCESENTRY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TaskSpec.RequiredPlacementResourcesEntry) + }) + , + 'DESCRIPTOR' : _TASKSPEC, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TaskSpec) + }) +_sym_db.RegisterMessage(TaskSpec) +_sym_db.RegisterMessage(TaskSpec.RequiredResourcesEntry) +_sym_db.RegisterMessage(TaskSpec.RequiredPlacementResourcesEntry) + +TaskInfoEntry = _reflection.GeneratedProtocolMessageType('TaskInfoEntry', (_message.Message,), { + + 'RequiredResourcesEntry' : _reflection.GeneratedProtocolMessageType('RequiredResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _TASKINFOENTRY_REQUIREDRESOURCESENTRY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TaskInfoEntry.RequiredResourcesEntry) + }) + , + 'DESCRIPTOR' : _TASKINFOENTRY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TaskInfoEntry) + }) +_sym_db.RegisterMessage(TaskInfoEntry) +_sym_db.RegisterMessage(TaskInfoEntry.RequiredResourcesEntry) + +Bundle = _reflection.GeneratedProtocolMessageType('Bundle', (_message.Message,), { + + 'BundleIdentifier' : _reflection.GeneratedProtocolMessageType('BundleIdentifier', (_message.Message,), { + 'DESCRIPTOR' : _BUNDLE_BUNDLEIDENTIFIER, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.Bundle.BundleIdentifier) + }) + , + + 'UnitResourcesEntry' : _reflection.GeneratedProtocolMessageType('UnitResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _BUNDLE_UNITRESOURCESENTRY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.Bundle.UnitResourcesEntry) + }) + , + 'DESCRIPTOR' : _BUNDLE, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.Bundle) + }) +_sym_db.RegisterMessage(Bundle) +_sym_db.RegisterMessage(Bundle.BundleIdentifier) +_sym_db.RegisterMessage(Bundle.UnitResourcesEntry) + +PlacementGroupSpec = _reflection.GeneratedProtocolMessageType('PlacementGroupSpec', (_message.Message,), { + 'DESCRIPTOR' : _PLACEMENTGROUPSPEC, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PlacementGroupSpec) + }) +_sym_db.RegisterMessage(PlacementGroupSpec) + +ObjectReference = _reflection.GeneratedProtocolMessageType('ObjectReference', (_message.Message,), { + 'DESCRIPTOR' : _OBJECTREFERENCE, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ObjectReference) + }) +_sym_db.RegisterMessage(ObjectReference) + +ObjectReferenceCount = _reflection.GeneratedProtocolMessageType('ObjectReferenceCount', (_message.Message,), { + 'DESCRIPTOR' : _OBJECTREFERENCECOUNT, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ObjectReferenceCount) + }) +_sym_db.RegisterMessage(ObjectReferenceCount) + +TaskArg = _reflection.GeneratedProtocolMessageType('TaskArg', (_message.Message,), { + 'DESCRIPTOR' : _TASKARG, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TaskArg) + }) +_sym_db.RegisterMessage(TaskArg) + +ReturnObject = _reflection.GeneratedProtocolMessageType('ReturnObject', (_message.Message,), { + 'DESCRIPTOR' : _RETURNOBJECT, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReturnObject) + }) +_sym_db.RegisterMessage(ReturnObject) + +ActorCreationTaskSpec = _reflection.GeneratedProtocolMessageType('ActorCreationTaskSpec', (_message.Message,), { + 'DESCRIPTOR' : _ACTORCREATIONTASKSPEC, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ActorCreationTaskSpec) + }) +_sym_db.RegisterMessage(ActorCreationTaskSpec) + +ActorTaskSpec = _reflection.GeneratedProtocolMessageType('ActorTaskSpec', (_message.Message,), { + 'DESCRIPTOR' : _ACTORTASKSPEC, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ActorTaskSpec) + }) +_sym_db.RegisterMessage(ActorTaskSpec) + +Task = _reflection.GeneratedProtocolMessageType('Task', (_message.Message,), { + 'DESCRIPTOR' : _TASK, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.Task) + }) +_sym_db.RegisterMessage(Task) + +ResourceId = _reflection.GeneratedProtocolMessageType('ResourceId', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCEID, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceId) + }) +_sym_db.RegisterMessage(ResourceId) + +ResourceMapEntry = _reflection.GeneratedProtocolMessageType('ResourceMapEntry', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCEMAPENTRY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceMapEntry) + }) +_sym_db.RegisterMessage(ResourceMapEntry) + +ViewData = _reflection.GeneratedProtocolMessageType('ViewData', (_message.Message,), { + + 'Measure' : _reflection.GeneratedProtocolMessageType('Measure', (_message.Message,), { + 'DESCRIPTOR' : _VIEWDATA_MEASURE, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ViewData.Measure) + }) + , + 'DESCRIPTOR' : _VIEWDATA, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ViewData) + }) +_sym_db.RegisterMessage(ViewData) +_sym_db.RegisterMessage(ViewData.Measure) + +ObjectRefInfo = _reflection.GeneratedProtocolMessageType('ObjectRefInfo', (_message.Message,), { + 'DESCRIPTOR' : _OBJECTREFINFO, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ObjectRefInfo) + }) +_sym_db.RegisterMessage(ObjectRefInfo) + +ResourceAllocations = _reflection.GeneratedProtocolMessageType('ResourceAllocations', (_message.Message,), { + + 'ResourceSlot' : _reflection.GeneratedProtocolMessageType('ResourceSlot', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCEALLOCATIONS_RESOURCESLOT, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceAllocations.ResourceSlot) + }) + , + 'DESCRIPTOR' : _RESOURCEALLOCATIONS, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceAllocations) + }) +_sym_db.RegisterMessage(ResourceAllocations) +_sym_db.RegisterMessage(ResourceAllocations.ResourceSlot) + +CoreWorkerStats = _reflection.GeneratedProtocolMessageType('CoreWorkerStats', (_message.Message,), { + + 'UsedResourcesEntry' : _reflection.GeneratedProtocolMessageType('UsedResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _COREWORKERSTATS_USEDRESOURCESENTRY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CoreWorkerStats.UsedResourcesEntry) + }) + , + + 'WebuiDisplayEntry' : _reflection.GeneratedProtocolMessageType('WebuiDisplayEntry', (_message.Message,), { + 'DESCRIPTOR' : _COREWORKERSTATS_WEBUIDISPLAYENTRY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CoreWorkerStats.WebuiDisplayEntry) + }) + , + 'DESCRIPTOR' : _COREWORKERSTATS, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CoreWorkerStats) + }) +_sym_db.RegisterMessage(CoreWorkerStats) +_sym_db.RegisterMessage(CoreWorkerStats.UsedResourcesEntry) +_sym_db.RegisterMessage(CoreWorkerStats.WebuiDisplayEntry) + +NodeResourceUsage = _reflection.GeneratedProtocolMessageType('NodeResourceUsage', (_message.Message,), { + 'DESCRIPTOR' : _NODERESOURCEUSAGE, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NodeResourceUsage) + }) +_sym_db.RegisterMessage(NodeResourceUsage) + +MetricPoint = _reflection.GeneratedProtocolMessageType('MetricPoint', (_message.Message,), { + + 'TagsEntry' : _reflection.GeneratedProtocolMessageType('TagsEntry', (_message.Message,), { + 'DESCRIPTOR' : _METRICPOINT_TAGSENTRY, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.MetricPoint.TagsEntry) + }) + , + 'DESCRIPTOR' : _METRICPOINT, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.MetricPoint) + }) +_sym_db.RegisterMessage(MetricPoint) +_sym_db.RegisterMessage(MetricPoint.TagsEntry) + +NamedActorInfo = _reflection.GeneratedProtocolMessageType('NamedActorInfo', (_message.Message,), { + 'DESCRIPTOR' : _NAMEDACTORINFO, + '__module__' : 'src.ray.protobuf.common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NamedActorInfo) + }) +_sym_db.RegisterMessage(NamedActorInfo) + +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\n\030io.ray.runtime.generated\370\001\001' + _JOBCONFIG_METADATAENTRY._options = None + _JOBCONFIG_METADATAENTRY._serialized_options = b'8\001' + _TASKSPEC_REQUIREDRESOURCESENTRY._options = None + _TASKSPEC_REQUIREDRESOURCESENTRY._serialized_options = b'8\001' + _TASKSPEC_REQUIREDPLACEMENTRESOURCESENTRY._options = None + _TASKSPEC_REQUIREDPLACEMENTRESOURCESENTRY._serialized_options = b'8\001' + _TASKINFOENTRY_REQUIREDRESOURCESENTRY._options = None + _TASKINFOENTRY_REQUIREDRESOURCESENTRY._serialized_options = b'8\001' + _BUNDLE_UNITRESOURCESENTRY._options = None + _BUNDLE_UNITRESOURCESENTRY._serialized_options = b'8\001' + _COREWORKERSTATS_USEDRESOURCESENTRY._options = None + _COREWORKERSTATS_USEDRESOURCESENTRY._serialized_options = b'8\001' + _COREWORKERSTATS_WEBUIDISPLAYENTRY._options = None + _COREWORKERSTATS_WEBUIDISPLAYENTRY._serialized_options = b'8\001' + _METRICPOINT_TAGSENTRY._options = None + _METRICPOINT_TAGSENTRY._serialized_options = b'8\001' + _LANGUAGE._serialized_start=9697 + _LANGUAGE._serialized_end=9738 + _WORKERTYPE._serialized_start=9740 + _WORKERTYPE._serialized_end=9814 + _TASKTYPE._serialized_start=9816 + _TASKTYPE._serialized_end=9901 + _ERRORTYPE._serialized_start=9904 + _ERRORTYPE._serialized_end=10580 + _TASKSTATUS._serialized_start=10583 + _TASKSTATUS._serialized_end=10827 + _WORKEREXITTYPE._serialized_start=10829 + _WORKEREXITTYPE._serialized_end=10953 + _PLACEMENTSTRATEGY._serialized_start=10955 + _PLACEMENTSTRATEGY._serialized_end=11032 + _LABELIN._serialized_start=85 + _LABELIN._serialized_end=110 + _LABELNOTIN._serialized_start=112 + _LABELNOTIN._serialized_end=140 + _LABELEXISTS._serialized_start=142 + _LABELEXISTS._serialized_end=155 + _LABELDOESNOTEXIST._serialized_start=157 + _LABELDOESNOTEXIST._serialized_end=176 + _LABELOPERATOR._serialized_start=179 + _LABELOPERATOR._serialized_end=401 + _LABELMATCHEXPRESSION._serialized_start=403 + _LABELMATCHEXPRESSION._serialized_end=480 + _LABELMATCHEXPRESSIONS._serialized_start=482 + _LABELMATCHEXPRESSIONS._serialized_end=557 + _NODELABELSCHEDULINGSTRATEGY._serialized_start=559 + _NODELABELSCHEDULINGSTRATEGY._serialized_end=680 + _NODEAFFINITYSCHEDULINGSTRATEGY._serialized_start=682 + _NODEAFFINITYSCHEDULINGSTRATEGY._serialized_end=804 + _PLACEMENTGROUPSCHEDULINGSTRATEGY._serialized_start=807 + _PLACEMENTGROUPSCHEDULINGSTRATEGY._serialized_end=952 + _DEFAULTSCHEDULINGSTRATEGY._serialized_start=954 + _DEFAULTSCHEDULINGSTRATEGY._serialized_end=981 + _SPREADSCHEDULINGSTRATEGY._serialized_start=983 + _SPREADSCHEDULINGSTRATEGY._serialized_end=1009 + _SCHEDULINGSTRATEGY._serialized_start=1012 + _SCHEDULINGSTRATEGY._serialized_end=1459 + _ADDRESS._serialized_start=1461 + _ADDRESS._serialized_end=1542 + _JAVAFUNCTIONDESCRIPTOR._serialized_start=1544 + _JAVAFUNCTIONDESCRIPTOR._serialized_end=1630 + _PYTHONFUNCTIONDESCRIPTOR._serialized_start=1632 + _PYTHONFUNCTIONDESCRIPTOR._serialized_end=1745 + _CPPFUNCTIONDESCRIPTOR._serialized_start=1747 + _CPPFUNCTIONDESCRIPTOR._serialized_end=1829 + _FUNCTIONDESCRIPTOR._serialized_start=1832 + _FUNCTIONDESCRIPTOR._serialized_end=2084 + _CONCURRENCYGROUP._serialized_start=2086 + _CONCURRENCYGROUP._serialized_end=2202 + _RAYERRORINFO._serialized_start=2205 + _RAYERRORINFO._serialized_end=2421 + _OUTOFMEMORYERRORCONTEXT._serialized_start=2424 + _OUTOFMEMORYERRORCONTEXT._serialized_end=2622 + _NODEDIEDERRORCONTEXT._serialized_start=2624 + _NODEDIEDERRORCONTEXT._serialized_end=2688 + _RAYEXCEPTION._serialized_start=2690 + _RAYEXCEPTION._serialized_end=2807 + _ACTORDEATHCAUSE._serialized_start=2810 + _ACTORDEATHCAUSE._serialized_end=3161 + _RUNTIMEENVFAILEDCONTEXT._serialized_start=3163 + _RUNTIMEENVFAILEDCONTEXT._serialized_end=3211 + _ACTORUNSCHEDULABLECONTEXT._serialized_start=3213 + _ACTORUNSCHEDULABLECONTEXT._serialized_end=3263 + _ACTORDIEDERRORCONTEXT._serialized_start=3266 + _ACTORDIEDERRORCONTEXT._serialized_end=3492 + _OOMCONTEXT._serialized_start=3494 + _OOMCONTEXT._serialized_end=3555 + _JOBCONFIG._serialized_start=3558 + _JOBCONFIG._serialized_end=3934 + _JOBCONFIG_METADATAENTRY._serialized_start=3838 + _JOBCONFIG_METADATAENTRY._serialized_end=3885 + _JOBCONFIG_ACTORLIFETIME._serialized_start=3887 + _JOBCONFIG_ACTORLIFETIME._serialized_end=3934 + _STREAMINGGENERATORRETURNIDINFO._serialized_start=3936 + _STREAMINGGENERATORRETURNIDINFO._serialized_end=4013 + _TASKSPEC._serialized_start=4016 + _TASKSPEC._serialized_end=5417 + _TASKSPEC_REQUIREDRESOURCESENTRY._serialized_start=5243 + _TASKSPEC_REQUIREDRESOURCESENTRY._serialized_end=5299 + _TASKSPEC_REQUIREDPLACEMENTRESOURCESENTRY._serialized_start=5301 + _TASKSPEC_REQUIREDPLACEMENTRESOURCESENTRY._serialized_end=5366 + _TASKINFOENTRY._serialized_start=5420 + _TASKINFOENTRY._serialized_end=5961 + _TASKINFOENTRY_REQUIREDRESOURCESENTRY._serialized_start=5243 + _TASKINFOENTRY_REQUIREDRESOURCESENTRY._serialized_end=5299 + _BUNDLE._serialized_start=5964 + _BUNDLE._serialized_end=6226 + _BUNDLE_BUNDLEIDENTIFIER._serialized_start=6104 + _BUNDLE_BUNDLEIDENTIFIER._serialized_end=6172 + _BUNDLE_UNITRESOURCESENTRY._serialized_start=6174 + _BUNDLE_UNITRESOURCESENTRY._serialized_end=6226 + _PLACEMENTGROUPSPEC._serialized_start=6229 + _PLACEMENTGROUPSPEC._serialized_end=6531 + _OBJECTREFERENCE._serialized_start=6533 + _OBJECTREFERENCE._serialized_end=6629 + _OBJECTREFERENCECOUNT._serialized_start=6632 + _OBJECTREFERENCECOUNT._serialized_end=6865 + _TASKARG._serialized_start=6868 + _TASKARG._serialized_end=7010 + _RETURNOBJECT._serialized_start=7013 + _RETURNOBJECT._serialized_end=7166 + _ACTORCREATIONTASKSPEC._serialized_start=7169 + _ACTORCREATIONTASKSPEC._serialized_end=7568 + _ACTORTASKSPEC._serialized_start=7570 + _ACTORTASKSPEC._serialized_end=7666 + _TASK._serialized_start=7668 + _TASK._serialized_end=7712 + _RESOURCEID._serialized_start=7714 + _RESOURCEID._serialized_end=7759 + _RESOURCEMAPENTRY._serialized_start=7761 + _RESOURCEMAPENTRY._serialized_end=7836 + _VIEWDATA._serialized_start=7839 + _VIEWDATA._serialized_end=8163 + _VIEWDATA_MEASURE._serialized_start=7916 + _VIEWDATA_MEASURE._serialized_end=8163 + _OBJECTREFINFO._serialized_start=8166 + _OBJECTREFINFO._serialized_end=8419 + _RESOURCEALLOCATIONS._serialized_start=8422 + _RESOURCEALLOCATIONS._serialized_end=8560 + _RESOURCEALLOCATIONS_RESOURCESLOT._serialized_start=8512 + _RESOURCEALLOCATIONS_RESOURCESLOT._serialized_end=8560 + _COREWORKERSTATS._serialized_start=8563 + _COREWORKERSTATS._serialized_end=9407 + _COREWORKERSTATS_USEDRESOURCESENTRY._serialized_start=9266 + _COREWORKERSTATS_USEDRESOURCESENTRY._serialized_end=9348 + _COREWORKERSTATS_WEBUIDISPLAYENTRY._serialized_start=9350 + _COREWORKERSTATS_WEBUIDISPLAYENTRY._serialized_end=9401 + _NODERESOURCEUSAGE._serialized_start=9409 + _NODERESOURCEUSAGE._serialized_end=9442 + _METRICPOINT._serialized_start=9445 + _METRICPOINT._serialized_end=9640 + _METRICPOINT_TAGSENTRY._serialized_start=9597 + _METRICPOINT_TAGSENTRY._serialized_end=9640 + _NAMEDACTORINFO._serialized_start=9642 + _NAMEDACTORINFO._serialized_end=9695 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/common_pb2_grpc.py b/python/ray/core/generated/common_pb2_grpc.py new file mode 100644 index 0000000000000..2daafffebfc81 --- /dev/null +++ b/python/ray/core/generated/common_pb2_grpc.py @@ -0,0 +1,4 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + diff --git a/python/ray/core/generated/core_worker_pb2.py b/python/ray/core/generated/core_worker_pb2.py new file mode 100644 index 0000000000000..d797c1c905e72 --- /dev/null +++ b/python/ray/core/generated/core_worker_pb2.py @@ -0,0 +1,510 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/core_worker.proto +"""Generated protocol buffer code.""" +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from . import common_pb2 as src_dot_ray_dot_protobuf_dot_common__pb2 +from . import pubsub_pb2 as src_dot_ray_dot_protobuf_dot_pubsub__pb2 + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"src/ray/protobuf/core_worker.proto\x12\x07ray.rpc\x1a\x1dsrc/ray/protobuf/common.proto\x1a\x1dsrc/ray/protobuf/pubsub.proto\"%\n\x0f\x41\x63tiveObjectIDs\x12\x12\n\nobject_ids\x18\x01 \x03(\x0c\"\x92\x03\n\x0b\x41\x63torHandle\x12\x10\n\x08\x61\x63tor_id\x18\x01 \x01(\x0c\x12\x10\n\x08owner_id\x18\x02 \x01(\x0c\x12\'\n\rowner_address\x18\x03 \x01(\x0b\x32\x10.ray.rpc.Address\x12\x17\n\x0f\x63reation_job_id\x18\x04 \x01(\x0c\x12)\n\x0e\x61\x63tor_language\x18\x05 \x01(\x0e\x32\x11.ray.rpc.Language\x12L\n\'actor_creation_task_function_descriptor\x18\x06 \x01(\x0b\x32\x1b.ray.rpc.FunctionDescriptor\x12\x14\n\x0c\x61\x63tor_cursor\x18\x07 \x01(\x0c\x12\x16\n\x0e\x65xtension_data\x18\x08 \x01(\x0c\x12\x18\n\x10max_task_retries\x18\t \x01(\x03\x12\x0c\n\x04name\x18\n \x01(\t\x12\x15\n\rray_namespace\x18\x0b \x01(\t\x12\x1c\n\x14\x65xecute_out_of_order\x18\x0c \x01(\x08\x12\x19\n\x11max_pending_calls\x18\r \x01(\x05\"\xc1\x01\n\x0fPushTaskRequest\x12\x1a\n\x12intended_worker_id\x18\x01 \x01(\x0c\x12$\n\ttask_spec\x18\x02 \x01(\x0b\x32\x11.ray.rpc.TaskSpec\x12\x17\n\x0fsequence_number\x18\x03 \x01(\x03\x12\x1e\n\x16\x63lient_processed_up_to\x18\x04 \x01(\x03\x12\x33\n\x10resource_mapping\x18\x05 \x03(\x0b\x32\x19.ray.rpc.ResourceMapEntry\"\xc4\x03\n\rPushTaskReply\x12-\n\x0ereturn_objects\x18\x01 \x03(\x0b\x32\x15.ray.rpc.ReturnObject\x12\x35\n\x16\x64ynamic_return_objects\x18\x02 \x03(\x0b\x32\x15.ray.rpc.ReturnObject\x12\x16\n\x0eworker_exiting\x18\x03 \x01(\x08\x12\x34\n\rborrowed_refs\x18\x04 \x03(\x0b\x32\x1d.ray.rpc.ObjectReferenceCount\x12\x1a\n\x12is_retryable_error\x18\x05 \x01(\x08\x12\x1c\n\x14is_application_error\x18\x06 \x01(\x08\x12$\n\x1cwas_cancelled_before_running\x18\x07 \x01(\x08\x12\x1c\n\x0f\x61\x63tor_repr_name\x18\x08 \x01(\tH\x00\x88\x01\x01\x12\x1c\n\x14task_execution_error\x18\t \x01(\t\x12O\n\x1estreaming_generator_return_ids\x18\n \x03(\x0b\x32\'.ray.rpc.StreamingGeneratorReturnIdInfoB\x12\n\x10_actor_repr_name\"P\n%DirectActorCallArgWaitCompleteRequest\x12\x1a\n\x12intended_worker_id\x18\x01 \x01(\x0c\x12\x0b\n\x03tag\x18\x02 \x01(\x03\"%\n#DirectActorCallArgWaitCompleteReply\"D\n\x16GetObjectStatusRequest\x12\x17\n\x0fowner_worker_id\x18\x01 \x01(\x0c\x12\x11\n\tobject_id\x18\x02 \x01(\x0c\"b\n\tRayObject\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\x12\x10\n\x08metadata\x18\x02 \x01(\x0c\x12\x35\n\x13nested_inlined_refs\x18\x03 \x03(\x0b\x32\x18.ray.rpc.ObjectReference\"\xd7\x01\n\x14GetObjectStatusReply\x12:\n\x06status\x18\x01 \x01(\x0e\x32*.ray.rpc.GetObjectStatusReply.ObjectStatus\x12\"\n\x06object\x18\x02 \x01(\x0b\x32\x12.ray.rpc.RayObject\x12\x10\n\x08node_ids\x18\x03 \x03(\x0c\x12\x13\n\x0bobject_size\x18\x04 \x01(\x04\"8\n\x0cObjectStatus\x12\x0b\n\x07\x43REATED\x10\x00\x12\x10\n\x0cOUT_OF_SCOPE\x10\x01\x12\t\n\x05\x46REED\x10\x02\"M\n\x1dWaitForActorOutOfScopeRequest\x12\x1a\n\x12intended_worker_id\x18\x01 \x01(\x0c\x12\x10\n\x08\x61\x63tor_id\x18\x02 \x01(\x0c\"\x1d\n\x1bWaitForActorOutOfScopeReply\"\x8f\x01\n UpdateObjectLocationBatchRequest\x12\x1a\n\x12intended_worker_id\x18\x01 \x01(\x0c\x12\x0f\n\x07node_id\x18\x02 \x01(\x0c\x12>\n\x17object_location_updates\x18\x03 \x03(\x0b\x32\x1d.ray.rpc.ObjectLocationUpdate\" \n\x1eUpdateObjectLocationBatchReply\"T\n\x1bObjectSpilledLocationUpdate\x12\x13\n\x0bspilled_url\x18\x03 \x01(\t\x12 \n\x18spilled_to_local_storage\x18\x04 \x01(\x08\"\xa2\x02\n\x14ObjectLocationUpdate\x12\x11\n\tobject_id\x18\x01 \x01(\x0c\x12H\n\x16plasma_location_update\x18\x02 \x01(\x0e\x32#.ray.rpc.ObjectPlasmaLocationUpdateH\x00\x88\x01\x01\x12J\n\x17spilled_location_update\x18\x03 \x01(\x0b\x32$.ray.rpc.ObjectSpilledLocationUpdateH\x01\x88\x01\x01\x12\x19\n\x0cgenerator_id\x18\x04 \x01(\x0cH\x02\x88\x01\x01\x42\x19\n\x17_plasma_location_updateB\x1a\n\x18_spilled_location_updateB\x0f\n\r_generator_id\"k\n\x1eGetObjectLocationsOwnerRequest\x12I\n\x17object_location_request\x18\x01 \x01(\x0b\x32(.ray.rpc.WorkerObjectLocationsSubMessage\"f\n\x1cGetObjectLocationsOwnerReply\x12\x46\n\x14object_location_info\x18\x01 \x01(\x0b\x32(.ray.rpc.WorkerObjectLocationsPubMessage\"\x84\x01\n\x10KillActorRequest\x12\x19\n\x11intended_actor_id\x18\x01 \x01(\x0c\x12\x12\n\nforce_kill\x18\x02 \x01(\x08\x12\x12\n\nno_restart\x18\x03 \x01(\x08\x12-\n\x0b\x64\x65\x61th_cause\x18\x04 \x01(\x0b\x32\x18.ray.rpc.ActorDeathCause\"\x10\n\x0eKillActorReply\"n\n\x11\x43\x61ncelTaskRequest\x12\x18\n\x10intended_task_id\x18\x01 \x01(\x0c\x12\x12\n\nforce_kill\x18\x02 \x01(\x08\x12\x11\n\trecursive\x18\x03 \x01(\x08\x12\x18\n\x10\x63\x61ller_worker_id\x18\x04 \x01(\x0c\"L\n\x0f\x43\x61ncelTaskReply\x12\x1e\n\x16requested_task_running\x18\x01 \x01(\x08\x12\x19\n\x11\x61ttempt_succeeded\x18\x02 \x01(\x08\"Z\n\x17RemoteCancelTaskRequest\x12\x18\n\x10remote_object_id\x18\x01 \x01(\x0c\x12\x12\n\nforce_kill\x18\x02 \x01(\x08\x12\x11\n\trecursive\x18\x03 \x01(\x08\"\x17\n\x15RemoteCancelTaskReply\"\x8d\x01\n\x19GetCoreWorkerStatsRequest\x12\x1a\n\x12intended_worker_id\x18\x01 \x01(\x0c\x12\x1b\n\x13include_memory_info\x18\x02 \x01(\x08\x12\x19\n\x11include_task_info\x18\x03 \x01(\x08\x12\x12\n\x05limit\x18\x04 \x01(\x03H\x00\x88\x01\x01\x42\x08\n\x06_limit\"\xb6\x01\n\x17GetCoreWorkerStatsReply\x12\x33\n\x11\x63ore_worker_stats\x18\x01 \x01(\x0b\x32\x18.ray.rpc.CoreWorkerStats\x12\x37\n\x17owned_task_info_entries\x18\x02 \x03(\x0b\x32\x16.ray.rpc.TaskInfoEntry\x12\x18\n\x10running_task_ids\x18\x03 \x03(\x0c\x12\x13\n\x0btasks_total\x18\x04 \x01(\x03\"0\n\x0eLocalGCRequest\x12\x1e\n\x16triggered_by_global_gc\x18\x01 \x01(\x08\"\x0e\n\x0cLocalGCReply\"-\n\x18PlasmaObjectReadyRequest\x12\x11\n\tobject_id\x18\x01 \x01(\x0c\"\x18\n\x16PlasmaObjectReadyReply\">\n\x14\x44\x65leteObjectsRequest\x12\x12\n\nobject_ids\x18\x01 \x03(\x0c\x12\x12\n\nlocal_only\x18\x02 \x01(\x08\"\x14\n\x12\x44\x65leteObjectsReply\"\x84\x01\n\x13SpillObjectsRequest\x12\x36\n\x14object_refs_to_spill\x18\x01 \x03(\x0b\x32\x18.ray.rpc.ObjectReference\x12\x35\n\x0e\x64\x65lete_request\x18\x02 \x01(\x0b\x32\x1d.ray.rpc.DeleteObjectsRequest\"0\n\x11SpillObjectsReply\x12\x1b\n\x13spilled_objects_url\x18\x01 \x03(\t\"Z\n\x1cRestoreSpilledObjectsRequest\x12\x1b\n\x13spilled_objects_url\x18\x01 \x03(\t\x12\x1d\n\x15object_ids_to_restore\x18\x02 \x03(\x0c\":\n\x1aRestoreSpilledObjectsReply\x12\x1c\n\x14\x62ytes_restored_total\x18\x01 \x01(\x03\":\n\x1b\x44\x65leteSpilledObjectsRequest\x12\x1b\n\x13spilled_objects_url\x18\x01 \x03(\t\"\x1b\n\x19\x44\x65leteSpilledObjectsReply\"!\n\x0b\x45xitRequest\x12\x12\n\nforce_exit\x18\x01 \x01(\x08\"\x1c\n\tExitReply\x12\x0f\n\x07success\x18\x01 \x01(\x08\"\x9f\x01\n\x18\x41ssignObjectOwnerRequest\x12\x11\n\tobject_id\x18\x01 \x01(\x0c\x12\x13\n\x0bobject_size\x18\x02 \x01(\x04\x12\x1c\n\x14\x63ontained_object_ids\x18\x03 \x03(\x0c\x12*\n\x10\x62orrower_address\x18\x04 \x01(\x0b\x32\x10.ray.rpc.Address\x12\x11\n\tcall_site\x18\x05 \x01(\t\"\x18\n\x16\x41ssignObjectOwnerReply\"\x1f\n\x1dRayletNotifyGCSRestartRequest\"\x1d\n\x1bRayletNotifyGCSRestartReply\"\x18\n\x16NumPendingTasksRequest\"1\n\x14NumPendingTasksReply\x12\x19\n\x11num_pending_tasks\x18\x01 \x01(\x03\"\xc3\x01\n!ReportGeneratorItemReturnsRequest\x12\x35\n\x16\x64ynamic_return_objects\x18\x01 \x03(\x0b\x32\x15.ray.rpc.ReturnObject\x12%\n\x0bworker_addr\x18\x02 \x01(\x0b\x32\x10.ray.rpc.Address\x12\x12\n\nitem_index\x18\x03 \x01(\x03\x12\x14\n\x0cgenerator_id\x18\x05 \x01(\x0c\x12\x16\n\x0e\x61ttempt_number\x18\x06 \x01(\x04\"!\n\x1fReportGeneratorItemReturnsReply*4\n\x1aObjectPlasmaLocationUpdate\x12\t\n\x05\x41\x44\x44\x45\x44\x10\x00\x12\x0b\n\x07REMOVED\x10\x01\x32\x80\x10\n\x11\x43oreWorkerService\x12\x66\n\x16RayletNotifyGCSRestart\x12&.ray.rpc.RayletNotifyGCSRestartRequest\x1a$.ray.rpc.RayletNotifyGCSRestartReply\x12<\n\x08PushTask\x12\x18.ray.rpc.PushTaskRequest\x1a\x16.ray.rpc.PushTaskReply\x12~\n\x1e\x44irectActorCallArgWaitComplete\x12..ray.rpc.DirectActorCallArgWaitCompleteRequest\x1a,.ray.rpc.DirectActorCallArgWaitCompleteReply\x12Q\n\x0fGetObjectStatus\x12\x1f.ray.rpc.GetObjectStatusRequest\x1a\x1d.ray.rpc.GetObjectStatusReply\x12\x66\n\x16WaitForActorOutOfScope\x12&.ray.rpc.WaitForActorOutOfScopeRequest\x1a$.ray.rpc.WaitForActorOutOfScopeReply\x12W\n\x11PubsubLongPolling\x12!.ray.rpc.PubsubLongPollingRequest\x1a\x1f.ray.rpc.PubsubLongPollingReply\x12r\n\x1aReportGeneratorItemReturns\x12*.ray.rpc.ReportGeneratorItemReturnsRequest\x1a(.ray.rpc.ReportGeneratorItemReturnsReply\x12Z\n\x12PubsubCommandBatch\x12\".ray.rpc.PubsubCommandBatchRequest\x1a .ray.rpc.PubsubCommandBatchReply\x12o\n\x19UpdateObjectLocationBatch\x12).ray.rpc.UpdateObjectLocationBatchRequest\x1a\'.ray.rpc.UpdateObjectLocationBatchReply\x12i\n\x17GetObjectLocationsOwner\x12\'.ray.rpc.GetObjectLocationsOwnerRequest\x1a%.ray.rpc.GetObjectLocationsOwnerReply\x12?\n\tKillActor\x12\x19.ray.rpc.KillActorRequest\x1a\x17.ray.rpc.KillActorReply\x12\x42\n\nCancelTask\x12\x1a.ray.rpc.CancelTaskRequest\x1a\x18.ray.rpc.CancelTaskReply\x12T\n\x10RemoteCancelTask\x12 .ray.rpc.RemoteCancelTaskRequest\x1a\x1e.ray.rpc.RemoteCancelTaskReply\x12Z\n\x12GetCoreWorkerStats\x12\".ray.rpc.GetCoreWorkerStatsRequest\x1a .ray.rpc.GetCoreWorkerStatsReply\x12\x39\n\x07LocalGC\x12\x17.ray.rpc.LocalGCRequest\x1a\x15.ray.rpc.LocalGCReply\x12K\n\rDeleteObjects\x12\x1d.ray.rpc.DeleteObjectsRequest\x1a\x1b.ray.rpc.DeleteObjectsReply\x12H\n\x0cSpillObjects\x12\x1c.ray.rpc.SpillObjectsRequest\x1a\x1a.ray.rpc.SpillObjectsReply\x12\x63\n\x15RestoreSpilledObjects\x12%.ray.rpc.RestoreSpilledObjectsRequest\x1a#.ray.rpc.RestoreSpilledObjectsReply\x12`\n\x14\x44\x65leteSpilledObjects\x12$.ray.rpc.DeleteSpilledObjectsRequest\x1a\".ray.rpc.DeleteSpilledObjectsReply\x12W\n\x11PlasmaObjectReady\x12!.ray.rpc.PlasmaObjectReadyRequest\x1a\x1f.ray.rpc.PlasmaObjectReadyReply\x12\x30\n\x04\x45xit\x12\x14.ray.rpc.ExitRequest\x1a\x12.ray.rpc.ExitReply\x12W\n\x11\x41ssignObjectOwner\x12!.ray.rpc.AssignObjectOwnerRequest\x1a\x1f.ray.rpc.AssignObjectOwnerReply\x12Q\n\x0fNumPendingTasks\x12\x1f.ray.rpc.NumPendingTasksRequest\x1a\x1d.ray.rpc.NumPendingTasksReplyB\x03\xf8\x01\x01\x62\x06proto3') + +_OBJECTPLASMALOCATIONUPDATE = DESCRIPTOR.enum_types_by_name['ObjectPlasmaLocationUpdate'] +ObjectPlasmaLocationUpdate = enum_type_wrapper.EnumTypeWrapper(_OBJECTPLASMALOCATIONUPDATE) +ADDED = 0 +REMOVED = 1 + + +_ACTIVEOBJECTIDS = DESCRIPTOR.message_types_by_name['ActiveObjectIDs'] +_ACTORHANDLE = DESCRIPTOR.message_types_by_name['ActorHandle'] +_PUSHTASKREQUEST = DESCRIPTOR.message_types_by_name['PushTaskRequest'] +_PUSHTASKREPLY = DESCRIPTOR.message_types_by_name['PushTaskReply'] +_DIRECTACTORCALLARGWAITCOMPLETEREQUEST = DESCRIPTOR.message_types_by_name['DirectActorCallArgWaitCompleteRequest'] +_DIRECTACTORCALLARGWAITCOMPLETEREPLY = DESCRIPTOR.message_types_by_name['DirectActorCallArgWaitCompleteReply'] +_GETOBJECTSTATUSREQUEST = DESCRIPTOR.message_types_by_name['GetObjectStatusRequest'] +_RAYOBJECT = DESCRIPTOR.message_types_by_name['RayObject'] +_GETOBJECTSTATUSREPLY = DESCRIPTOR.message_types_by_name['GetObjectStatusReply'] +_WAITFORACTOROUTOFSCOPEREQUEST = DESCRIPTOR.message_types_by_name['WaitForActorOutOfScopeRequest'] +_WAITFORACTOROUTOFSCOPEREPLY = DESCRIPTOR.message_types_by_name['WaitForActorOutOfScopeReply'] +_UPDATEOBJECTLOCATIONBATCHREQUEST = DESCRIPTOR.message_types_by_name['UpdateObjectLocationBatchRequest'] +_UPDATEOBJECTLOCATIONBATCHREPLY = DESCRIPTOR.message_types_by_name['UpdateObjectLocationBatchReply'] +_OBJECTSPILLEDLOCATIONUPDATE = DESCRIPTOR.message_types_by_name['ObjectSpilledLocationUpdate'] +_OBJECTLOCATIONUPDATE = DESCRIPTOR.message_types_by_name['ObjectLocationUpdate'] +_GETOBJECTLOCATIONSOWNERREQUEST = DESCRIPTOR.message_types_by_name['GetObjectLocationsOwnerRequest'] +_GETOBJECTLOCATIONSOWNERREPLY = DESCRIPTOR.message_types_by_name['GetObjectLocationsOwnerReply'] +_KILLACTORREQUEST = DESCRIPTOR.message_types_by_name['KillActorRequest'] +_KILLACTORREPLY = DESCRIPTOR.message_types_by_name['KillActorReply'] +_CANCELTASKREQUEST = DESCRIPTOR.message_types_by_name['CancelTaskRequest'] +_CANCELTASKREPLY = DESCRIPTOR.message_types_by_name['CancelTaskReply'] +_REMOTECANCELTASKREQUEST = DESCRIPTOR.message_types_by_name['RemoteCancelTaskRequest'] +_REMOTECANCELTASKREPLY = DESCRIPTOR.message_types_by_name['RemoteCancelTaskReply'] +_GETCOREWORKERSTATSREQUEST = DESCRIPTOR.message_types_by_name['GetCoreWorkerStatsRequest'] +_GETCOREWORKERSTATSREPLY = DESCRIPTOR.message_types_by_name['GetCoreWorkerStatsReply'] +_LOCALGCREQUEST = DESCRIPTOR.message_types_by_name['LocalGCRequest'] +_LOCALGCREPLY = DESCRIPTOR.message_types_by_name['LocalGCReply'] +_PLASMAOBJECTREADYREQUEST = DESCRIPTOR.message_types_by_name['PlasmaObjectReadyRequest'] +_PLASMAOBJECTREADYREPLY = DESCRIPTOR.message_types_by_name['PlasmaObjectReadyReply'] +_DELETEOBJECTSREQUEST = DESCRIPTOR.message_types_by_name['DeleteObjectsRequest'] +_DELETEOBJECTSREPLY = DESCRIPTOR.message_types_by_name['DeleteObjectsReply'] +_SPILLOBJECTSREQUEST = DESCRIPTOR.message_types_by_name['SpillObjectsRequest'] +_SPILLOBJECTSREPLY = DESCRIPTOR.message_types_by_name['SpillObjectsReply'] +_RESTORESPILLEDOBJECTSREQUEST = DESCRIPTOR.message_types_by_name['RestoreSpilledObjectsRequest'] +_RESTORESPILLEDOBJECTSREPLY = DESCRIPTOR.message_types_by_name['RestoreSpilledObjectsReply'] +_DELETESPILLEDOBJECTSREQUEST = DESCRIPTOR.message_types_by_name['DeleteSpilledObjectsRequest'] +_DELETESPILLEDOBJECTSREPLY = DESCRIPTOR.message_types_by_name['DeleteSpilledObjectsReply'] +_EXITREQUEST = DESCRIPTOR.message_types_by_name['ExitRequest'] +_EXITREPLY = DESCRIPTOR.message_types_by_name['ExitReply'] +_ASSIGNOBJECTOWNERREQUEST = DESCRIPTOR.message_types_by_name['AssignObjectOwnerRequest'] +_ASSIGNOBJECTOWNERREPLY = DESCRIPTOR.message_types_by_name['AssignObjectOwnerReply'] +_RAYLETNOTIFYGCSRESTARTREQUEST = DESCRIPTOR.message_types_by_name['RayletNotifyGCSRestartRequest'] +_RAYLETNOTIFYGCSRESTARTREPLY = DESCRIPTOR.message_types_by_name['RayletNotifyGCSRestartReply'] +_NUMPENDINGTASKSREQUEST = DESCRIPTOR.message_types_by_name['NumPendingTasksRequest'] +_NUMPENDINGTASKSREPLY = DESCRIPTOR.message_types_by_name['NumPendingTasksReply'] +_REPORTGENERATORITEMRETURNSREQUEST = DESCRIPTOR.message_types_by_name['ReportGeneratorItemReturnsRequest'] +_REPORTGENERATORITEMRETURNSREPLY = DESCRIPTOR.message_types_by_name['ReportGeneratorItemReturnsReply'] +_GETOBJECTSTATUSREPLY_OBJECTSTATUS = _GETOBJECTSTATUSREPLY.enum_types_by_name['ObjectStatus'] +ActiveObjectIDs = _reflection.GeneratedProtocolMessageType('ActiveObjectIDs', (_message.Message,), { + 'DESCRIPTOR' : _ACTIVEOBJECTIDS, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ActiveObjectIDs) + }) +_sym_db.RegisterMessage(ActiveObjectIDs) + +ActorHandle = _reflection.GeneratedProtocolMessageType('ActorHandle', (_message.Message,), { + 'DESCRIPTOR' : _ACTORHANDLE, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ActorHandle) + }) +_sym_db.RegisterMessage(ActorHandle) + +PushTaskRequest = _reflection.GeneratedProtocolMessageType('PushTaskRequest', (_message.Message,), { + 'DESCRIPTOR' : _PUSHTASKREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PushTaskRequest) + }) +_sym_db.RegisterMessage(PushTaskRequest) + +PushTaskReply = _reflection.GeneratedProtocolMessageType('PushTaskReply', (_message.Message,), { + 'DESCRIPTOR' : _PUSHTASKREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PushTaskReply) + }) +_sym_db.RegisterMessage(PushTaskReply) + +DirectActorCallArgWaitCompleteRequest = _reflection.GeneratedProtocolMessageType('DirectActorCallArgWaitCompleteRequest', (_message.Message,), { + 'DESCRIPTOR' : _DIRECTACTORCALLARGWAITCOMPLETEREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DirectActorCallArgWaitCompleteRequest) + }) +_sym_db.RegisterMessage(DirectActorCallArgWaitCompleteRequest) + +DirectActorCallArgWaitCompleteReply = _reflection.GeneratedProtocolMessageType('DirectActorCallArgWaitCompleteReply', (_message.Message,), { + 'DESCRIPTOR' : _DIRECTACTORCALLARGWAITCOMPLETEREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DirectActorCallArgWaitCompleteReply) + }) +_sym_db.RegisterMessage(DirectActorCallArgWaitCompleteReply) + +GetObjectStatusRequest = _reflection.GeneratedProtocolMessageType('GetObjectStatusRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETOBJECTSTATUSREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetObjectStatusRequest) + }) +_sym_db.RegisterMessage(GetObjectStatusRequest) + +RayObject = _reflection.GeneratedProtocolMessageType('RayObject', (_message.Message,), { + 'DESCRIPTOR' : _RAYOBJECT, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RayObject) + }) +_sym_db.RegisterMessage(RayObject) + +GetObjectStatusReply = _reflection.GeneratedProtocolMessageType('GetObjectStatusReply', (_message.Message,), { + 'DESCRIPTOR' : _GETOBJECTSTATUSREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetObjectStatusReply) + }) +_sym_db.RegisterMessage(GetObjectStatusReply) + +WaitForActorOutOfScopeRequest = _reflection.GeneratedProtocolMessageType('WaitForActorOutOfScopeRequest', (_message.Message,), { + 'DESCRIPTOR' : _WAITFORACTOROUTOFSCOPEREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WaitForActorOutOfScopeRequest) + }) +_sym_db.RegisterMessage(WaitForActorOutOfScopeRequest) + +WaitForActorOutOfScopeReply = _reflection.GeneratedProtocolMessageType('WaitForActorOutOfScopeReply', (_message.Message,), { + 'DESCRIPTOR' : _WAITFORACTOROUTOFSCOPEREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WaitForActorOutOfScopeReply) + }) +_sym_db.RegisterMessage(WaitForActorOutOfScopeReply) + +UpdateObjectLocationBatchRequest = _reflection.GeneratedProtocolMessageType('UpdateObjectLocationBatchRequest', (_message.Message,), { + 'DESCRIPTOR' : _UPDATEOBJECTLOCATIONBATCHREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.UpdateObjectLocationBatchRequest) + }) +_sym_db.RegisterMessage(UpdateObjectLocationBatchRequest) + +UpdateObjectLocationBatchReply = _reflection.GeneratedProtocolMessageType('UpdateObjectLocationBatchReply', (_message.Message,), { + 'DESCRIPTOR' : _UPDATEOBJECTLOCATIONBATCHREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.UpdateObjectLocationBatchReply) + }) +_sym_db.RegisterMessage(UpdateObjectLocationBatchReply) + +ObjectSpilledLocationUpdate = _reflection.GeneratedProtocolMessageType('ObjectSpilledLocationUpdate', (_message.Message,), { + 'DESCRIPTOR' : _OBJECTSPILLEDLOCATIONUPDATE, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ObjectSpilledLocationUpdate) + }) +_sym_db.RegisterMessage(ObjectSpilledLocationUpdate) + +ObjectLocationUpdate = _reflection.GeneratedProtocolMessageType('ObjectLocationUpdate', (_message.Message,), { + 'DESCRIPTOR' : _OBJECTLOCATIONUPDATE, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ObjectLocationUpdate) + }) +_sym_db.RegisterMessage(ObjectLocationUpdate) + +GetObjectLocationsOwnerRequest = _reflection.GeneratedProtocolMessageType('GetObjectLocationsOwnerRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETOBJECTLOCATIONSOWNERREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetObjectLocationsOwnerRequest) + }) +_sym_db.RegisterMessage(GetObjectLocationsOwnerRequest) + +GetObjectLocationsOwnerReply = _reflection.GeneratedProtocolMessageType('GetObjectLocationsOwnerReply', (_message.Message,), { + 'DESCRIPTOR' : _GETOBJECTLOCATIONSOWNERREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetObjectLocationsOwnerReply) + }) +_sym_db.RegisterMessage(GetObjectLocationsOwnerReply) + +KillActorRequest = _reflection.GeneratedProtocolMessageType('KillActorRequest', (_message.Message,), { + 'DESCRIPTOR' : _KILLACTORREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KillActorRequest) + }) +_sym_db.RegisterMessage(KillActorRequest) + +KillActorReply = _reflection.GeneratedProtocolMessageType('KillActorReply', (_message.Message,), { + 'DESCRIPTOR' : _KILLACTORREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KillActorReply) + }) +_sym_db.RegisterMessage(KillActorReply) + +CancelTaskRequest = _reflection.GeneratedProtocolMessageType('CancelTaskRequest', (_message.Message,), { + 'DESCRIPTOR' : _CANCELTASKREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CancelTaskRequest) + }) +_sym_db.RegisterMessage(CancelTaskRequest) + +CancelTaskReply = _reflection.GeneratedProtocolMessageType('CancelTaskReply', (_message.Message,), { + 'DESCRIPTOR' : _CANCELTASKREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CancelTaskReply) + }) +_sym_db.RegisterMessage(CancelTaskReply) + +RemoteCancelTaskRequest = _reflection.GeneratedProtocolMessageType('RemoteCancelTaskRequest', (_message.Message,), { + 'DESCRIPTOR' : _REMOTECANCELTASKREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RemoteCancelTaskRequest) + }) +_sym_db.RegisterMessage(RemoteCancelTaskRequest) + +RemoteCancelTaskReply = _reflection.GeneratedProtocolMessageType('RemoteCancelTaskReply', (_message.Message,), { + 'DESCRIPTOR' : _REMOTECANCELTASKREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RemoteCancelTaskReply) + }) +_sym_db.RegisterMessage(RemoteCancelTaskReply) + +GetCoreWorkerStatsRequest = _reflection.GeneratedProtocolMessageType('GetCoreWorkerStatsRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETCOREWORKERSTATSREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetCoreWorkerStatsRequest) + }) +_sym_db.RegisterMessage(GetCoreWorkerStatsRequest) + +GetCoreWorkerStatsReply = _reflection.GeneratedProtocolMessageType('GetCoreWorkerStatsReply', (_message.Message,), { + 'DESCRIPTOR' : _GETCOREWORKERSTATSREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetCoreWorkerStatsReply) + }) +_sym_db.RegisterMessage(GetCoreWorkerStatsReply) + +LocalGCRequest = _reflection.GeneratedProtocolMessageType('LocalGCRequest', (_message.Message,), { + 'DESCRIPTOR' : _LOCALGCREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.LocalGCRequest) + }) +_sym_db.RegisterMessage(LocalGCRequest) + +LocalGCReply = _reflection.GeneratedProtocolMessageType('LocalGCReply', (_message.Message,), { + 'DESCRIPTOR' : _LOCALGCREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.LocalGCReply) + }) +_sym_db.RegisterMessage(LocalGCReply) + +PlasmaObjectReadyRequest = _reflection.GeneratedProtocolMessageType('PlasmaObjectReadyRequest', (_message.Message,), { + 'DESCRIPTOR' : _PLASMAOBJECTREADYREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PlasmaObjectReadyRequest) + }) +_sym_db.RegisterMessage(PlasmaObjectReadyRequest) + +PlasmaObjectReadyReply = _reflection.GeneratedProtocolMessageType('PlasmaObjectReadyReply', (_message.Message,), { + 'DESCRIPTOR' : _PLASMAOBJECTREADYREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PlasmaObjectReadyReply) + }) +_sym_db.RegisterMessage(PlasmaObjectReadyReply) + +DeleteObjectsRequest = _reflection.GeneratedProtocolMessageType('DeleteObjectsRequest', (_message.Message,), { + 'DESCRIPTOR' : _DELETEOBJECTSREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DeleteObjectsRequest) + }) +_sym_db.RegisterMessage(DeleteObjectsRequest) + +DeleteObjectsReply = _reflection.GeneratedProtocolMessageType('DeleteObjectsReply', (_message.Message,), { + 'DESCRIPTOR' : _DELETEOBJECTSREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DeleteObjectsReply) + }) +_sym_db.RegisterMessage(DeleteObjectsReply) + +SpillObjectsRequest = _reflection.GeneratedProtocolMessageType('SpillObjectsRequest', (_message.Message,), { + 'DESCRIPTOR' : _SPILLOBJECTSREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.SpillObjectsRequest) + }) +_sym_db.RegisterMessage(SpillObjectsRequest) + +SpillObjectsReply = _reflection.GeneratedProtocolMessageType('SpillObjectsReply', (_message.Message,), { + 'DESCRIPTOR' : _SPILLOBJECTSREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.SpillObjectsReply) + }) +_sym_db.RegisterMessage(SpillObjectsReply) + +RestoreSpilledObjectsRequest = _reflection.GeneratedProtocolMessageType('RestoreSpilledObjectsRequest', (_message.Message,), { + 'DESCRIPTOR' : _RESTORESPILLEDOBJECTSREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RestoreSpilledObjectsRequest) + }) +_sym_db.RegisterMessage(RestoreSpilledObjectsRequest) + +RestoreSpilledObjectsReply = _reflection.GeneratedProtocolMessageType('RestoreSpilledObjectsReply', (_message.Message,), { + 'DESCRIPTOR' : _RESTORESPILLEDOBJECTSREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RestoreSpilledObjectsReply) + }) +_sym_db.RegisterMessage(RestoreSpilledObjectsReply) + +DeleteSpilledObjectsRequest = _reflection.GeneratedProtocolMessageType('DeleteSpilledObjectsRequest', (_message.Message,), { + 'DESCRIPTOR' : _DELETESPILLEDOBJECTSREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DeleteSpilledObjectsRequest) + }) +_sym_db.RegisterMessage(DeleteSpilledObjectsRequest) + +DeleteSpilledObjectsReply = _reflection.GeneratedProtocolMessageType('DeleteSpilledObjectsReply', (_message.Message,), { + 'DESCRIPTOR' : _DELETESPILLEDOBJECTSREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DeleteSpilledObjectsReply) + }) +_sym_db.RegisterMessage(DeleteSpilledObjectsReply) + +ExitRequest = _reflection.GeneratedProtocolMessageType('ExitRequest', (_message.Message,), { + 'DESCRIPTOR' : _EXITREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ExitRequest) + }) +_sym_db.RegisterMessage(ExitRequest) + +ExitReply = _reflection.GeneratedProtocolMessageType('ExitReply', (_message.Message,), { + 'DESCRIPTOR' : _EXITREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ExitReply) + }) +_sym_db.RegisterMessage(ExitReply) + +AssignObjectOwnerRequest = _reflection.GeneratedProtocolMessageType('AssignObjectOwnerRequest', (_message.Message,), { + 'DESCRIPTOR' : _ASSIGNOBJECTOWNERREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.AssignObjectOwnerRequest) + }) +_sym_db.RegisterMessage(AssignObjectOwnerRequest) + +AssignObjectOwnerReply = _reflection.GeneratedProtocolMessageType('AssignObjectOwnerReply', (_message.Message,), { + 'DESCRIPTOR' : _ASSIGNOBJECTOWNERREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.AssignObjectOwnerReply) + }) +_sym_db.RegisterMessage(AssignObjectOwnerReply) + +RayletNotifyGCSRestartRequest = _reflection.GeneratedProtocolMessageType('RayletNotifyGCSRestartRequest', (_message.Message,), { + 'DESCRIPTOR' : _RAYLETNOTIFYGCSRESTARTREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RayletNotifyGCSRestartRequest) + }) +_sym_db.RegisterMessage(RayletNotifyGCSRestartRequest) + +RayletNotifyGCSRestartReply = _reflection.GeneratedProtocolMessageType('RayletNotifyGCSRestartReply', (_message.Message,), { + 'DESCRIPTOR' : _RAYLETNOTIFYGCSRESTARTREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RayletNotifyGCSRestartReply) + }) +_sym_db.RegisterMessage(RayletNotifyGCSRestartReply) + +NumPendingTasksRequest = _reflection.GeneratedProtocolMessageType('NumPendingTasksRequest', (_message.Message,), { + 'DESCRIPTOR' : _NUMPENDINGTASKSREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NumPendingTasksRequest) + }) +_sym_db.RegisterMessage(NumPendingTasksRequest) + +NumPendingTasksReply = _reflection.GeneratedProtocolMessageType('NumPendingTasksReply', (_message.Message,), { + 'DESCRIPTOR' : _NUMPENDINGTASKSREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NumPendingTasksReply) + }) +_sym_db.RegisterMessage(NumPendingTasksReply) + +ReportGeneratorItemReturnsRequest = _reflection.GeneratedProtocolMessageType('ReportGeneratorItemReturnsRequest', (_message.Message,), { + 'DESCRIPTOR' : _REPORTGENERATORITEMRETURNSREQUEST, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportGeneratorItemReturnsRequest) + }) +_sym_db.RegisterMessage(ReportGeneratorItemReturnsRequest) + +ReportGeneratorItemReturnsReply = _reflection.GeneratedProtocolMessageType('ReportGeneratorItemReturnsReply', (_message.Message,), { + 'DESCRIPTOR' : _REPORTGENERATORITEMRETURNSREPLY, + '__module__' : 'src.ray.protobuf.core_worker_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportGeneratorItemReturnsReply) + }) +_sym_db.RegisterMessage(ReportGeneratorItemReturnsReply) + +_COREWORKERSERVICE = DESCRIPTOR.services_by_name['CoreWorkerService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _OBJECTPLASMALOCATIONUPDATE._serialized_start=4662 + _OBJECTPLASMALOCATIONUPDATE._serialized_end=4714 + _ACTIVEOBJECTIDS._serialized_start=109 + _ACTIVEOBJECTIDS._serialized_end=146 + _ACTORHANDLE._serialized_start=149 + _ACTORHANDLE._serialized_end=551 + _PUSHTASKREQUEST._serialized_start=554 + _PUSHTASKREQUEST._serialized_end=747 + _PUSHTASKREPLY._serialized_start=750 + _PUSHTASKREPLY._serialized_end=1202 + _DIRECTACTORCALLARGWAITCOMPLETEREQUEST._serialized_start=1204 + _DIRECTACTORCALLARGWAITCOMPLETEREQUEST._serialized_end=1284 + _DIRECTACTORCALLARGWAITCOMPLETEREPLY._serialized_start=1286 + _DIRECTACTORCALLARGWAITCOMPLETEREPLY._serialized_end=1323 + _GETOBJECTSTATUSREQUEST._serialized_start=1325 + _GETOBJECTSTATUSREQUEST._serialized_end=1393 + _RAYOBJECT._serialized_start=1395 + _RAYOBJECT._serialized_end=1493 + _GETOBJECTSTATUSREPLY._serialized_start=1496 + _GETOBJECTSTATUSREPLY._serialized_end=1711 + _GETOBJECTSTATUSREPLY_OBJECTSTATUS._serialized_start=1655 + _GETOBJECTSTATUSREPLY_OBJECTSTATUS._serialized_end=1711 + _WAITFORACTOROUTOFSCOPEREQUEST._serialized_start=1713 + _WAITFORACTOROUTOFSCOPEREQUEST._serialized_end=1790 + _WAITFORACTOROUTOFSCOPEREPLY._serialized_start=1792 + _WAITFORACTOROUTOFSCOPEREPLY._serialized_end=1821 + _UPDATEOBJECTLOCATIONBATCHREQUEST._serialized_start=1824 + _UPDATEOBJECTLOCATIONBATCHREQUEST._serialized_end=1967 + _UPDATEOBJECTLOCATIONBATCHREPLY._serialized_start=1969 + _UPDATEOBJECTLOCATIONBATCHREPLY._serialized_end=2001 + _OBJECTSPILLEDLOCATIONUPDATE._serialized_start=2003 + _OBJECTSPILLEDLOCATIONUPDATE._serialized_end=2087 + _OBJECTLOCATIONUPDATE._serialized_start=2090 + _OBJECTLOCATIONUPDATE._serialized_end=2380 + _GETOBJECTLOCATIONSOWNERREQUEST._serialized_start=2382 + _GETOBJECTLOCATIONSOWNERREQUEST._serialized_end=2489 + _GETOBJECTLOCATIONSOWNERREPLY._serialized_start=2491 + _GETOBJECTLOCATIONSOWNERREPLY._serialized_end=2593 + _KILLACTORREQUEST._serialized_start=2596 + _KILLACTORREQUEST._serialized_end=2728 + _KILLACTORREPLY._serialized_start=2730 + _KILLACTORREPLY._serialized_end=2746 + _CANCELTASKREQUEST._serialized_start=2748 + _CANCELTASKREQUEST._serialized_end=2858 + _CANCELTASKREPLY._serialized_start=2860 + _CANCELTASKREPLY._serialized_end=2936 + _REMOTECANCELTASKREQUEST._serialized_start=2938 + _REMOTECANCELTASKREQUEST._serialized_end=3028 + _REMOTECANCELTASKREPLY._serialized_start=3030 + _REMOTECANCELTASKREPLY._serialized_end=3053 + _GETCOREWORKERSTATSREQUEST._serialized_start=3056 + _GETCOREWORKERSTATSREQUEST._serialized_end=3197 + _GETCOREWORKERSTATSREPLY._serialized_start=3200 + _GETCOREWORKERSTATSREPLY._serialized_end=3382 + _LOCALGCREQUEST._serialized_start=3384 + _LOCALGCREQUEST._serialized_end=3432 + _LOCALGCREPLY._serialized_start=3434 + _LOCALGCREPLY._serialized_end=3448 + _PLASMAOBJECTREADYREQUEST._serialized_start=3450 + _PLASMAOBJECTREADYREQUEST._serialized_end=3495 + _PLASMAOBJECTREADYREPLY._serialized_start=3497 + _PLASMAOBJECTREADYREPLY._serialized_end=3521 + _DELETEOBJECTSREQUEST._serialized_start=3523 + _DELETEOBJECTSREQUEST._serialized_end=3585 + _DELETEOBJECTSREPLY._serialized_start=3587 + _DELETEOBJECTSREPLY._serialized_end=3607 + _SPILLOBJECTSREQUEST._serialized_start=3610 + _SPILLOBJECTSREQUEST._serialized_end=3742 + _SPILLOBJECTSREPLY._serialized_start=3744 + _SPILLOBJECTSREPLY._serialized_end=3792 + _RESTORESPILLEDOBJECTSREQUEST._serialized_start=3794 + _RESTORESPILLEDOBJECTSREQUEST._serialized_end=3884 + _RESTORESPILLEDOBJECTSREPLY._serialized_start=3886 + _RESTORESPILLEDOBJECTSREPLY._serialized_end=3944 + _DELETESPILLEDOBJECTSREQUEST._serialized_start=3946 + _DELETESPILLEDOBJECTSREQUEST._serialized_end=4004 + _DELETESPILLEDOBJECTSREPLY._serialized_start=4006 + _DELETESPILLEDOBJECTSREPLY._serialized_end=4033 + _EXITREQUEST._serialized_start=4035 + _EXITREQUEST._serialized_end=4068 + _EXITREPLY._serialized_start=4070 + _EXITREPLY._serialized_end=4098 + _ASSIGNOBJECTOWNERREQUEST._serialized_start=4101 + _ASSIGNOBJECTOWNERREQUEST._serialized_end=4260 + _ASSIGNOBJECTOWNERREPLY._serialized_start=4262 + _ASSIGNOBJECTOWNERREPLY._serialized_end=4286 + _RAYLETNOTIFYGCSRESTARTREQUEST._serialized_start=4288 + _RAYLETNOTIFYGCSRESTARTREQUEST._serialized_end=4319 + _RAYLETNOTIFYGCSRESTARTREPLY._serialized_start=4321 + _RAYLETNOTIFYGCSRESTARTREPLY._serialized_end=4350 + _NUMPENDINGTASKSREQUEST._serialized_start=4352 + _NUMPENDINGTASKSREQUEST._serialized_end=4376 + _NUMPENDINGTASKSREPLY._serialized_start=4378 + _NUMPENDINGTASKSREPLY._serialized_end=4427 + _REPORTGENERATORITEMRETURNSREQUEST._serialized_start=4430 + _REPORTGENERATORITEMRETURNSREQUEST._serialized_end=4625 + _REPORTGENERATORITEMRETURNSREPLY._serialized_start=4627 + _REPORTGENERATORITEMRETURNSREPLY._serialized_end=4660 + _COREWORKERSERVICE._serialized_start=4717 + _COREWORKERSERVICE._serialized_end=6765 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/core_worker_pb2_grpc.py b/python/ray/core/generated/core_worker_pb2_grpc.py new file mode 100644 index 0000000000000..be9751a6302e6 --- /dev/null +++ b/python/ray/core/generated/core_worker_pb2_grpc.py @@ -0,0 +1,821 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import core_worker_pb2 as src_dot_ray_dot_protobuf_dot_core__worker__pb2 +from . import pubsub_pb2 as src_dot_ray_dot_protobuf_dot_pubsub__pb2 + + +class CoreWorkerServiceStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.RayletNotifyGCSRestart = channel.unary_unary( + '/ray.rpc.CoreWorkerService/RayletNotifyGCSRestart', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RayletNotifyGCSRestartRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RayletNotifyGCSRestartReply.FromString, + ) + self.PushTask = channel.unary_unary( + '/ray.rpc.CoreWorkerService/PushTask', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PushTaskRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PushTaskReply.FromString, + ) + self.DirectActorCallArgWaitComplete = channel.unary_unary( + '/ray.rpc.CoreWorkerService/DirectActorCallArgWaitComplete', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DirectActorCallArgWaitCompleteRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DirectActorCallArgWaitCompleteReply.FromString, + ) + self.GetObjectStatus = channel.unary_unary( + '/ray.rpc.CoreWorkerService/GetObjectStatus', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectStatusRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectStatusReply.FromString, + ) + self.WaitForActorOutOfScope = channel.unary_unary( + '/ray.rpc.CoreWorkerService/WaitForActorOutOfScope', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.WaitForActorOutOfScopeRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.WaitForActorOutOfScopeReply.FromString, + ) + self.PubsubLongPolling = channel.unary_unary( + '/ray.rpc.CoreWorkerService/PubsubLongPolling', + request_serializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingReply.FromString, + ) + self.ReportGeneratorItemReturns = channel.unary_unary( + '/ray.rpc.CoreWorkerService/ReportGeneratorItemReturns', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ReportGeneratorItemReturnsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ReportGeneratorItemReturnsReply.FromString, + ) + self.PubsubCommandBatch = channel.unary_unary( + '/ray.rpc.CoreWorkerService/PubsubCommandBatch', + request_serializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchReply.FromString, + ) + self.UpdateObjectLocationBatch = channel.unary_unary( + '/ray.rpc.CoreWorkerService/UpdateObjectLocationBatch', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.UpdateObjectLocationBatchRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.UpdateObjectLocationBatchReply.FromString, + ) + self.GetObjectLocationsOwner = channel.unary_unary( + '/ray.rpc.CoreWorkerService/GetObjectLocationsOwner', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectLocationsOwnerRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectLocationsOwnerReply.FromString, + ) + self.KillActor = channel.unary_unary( + '/ray.rpc.CoreWorkerService/KillActor', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.KillActorRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.KillActorReply.FromString, + ) + self.CancelTask = channel.unary_unary( + '/ray.rpc.CoreWorkerService/CancelTask', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.CancelTaskRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.CancelTaskReply.FromString, + ) + self.RemoteCancelTask = channel.unary_unary( + '/ray.rpc.CoreWorkerService/RemoteCancelTask', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RemoteCancelTaskRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RemoteCancelTaskReply.FromString, + ) + self.GetCoreWorkerStats = channel.unary_unary( + '/ray.rpc.CoreWorkerService/GetCoreWorkerStats', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetCoreWorkerStatsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetCoreWorkerStatsReply.FromString, + ) + self.LocalGC = channel.unary_unary( + '/ray.rpc.CoreWorkerService/LocalGC', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.LocalGCRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.LocalGCReply.FromString, + ) + self.DeleteObjects = channel.unary_unary( + '/ray.rpc.CoreWorkerService/DeleteObjects', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteObjectsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteObjectsReply.FromString, + ) + self.SpillObjects = channel.unary_unary( + '/ray.rpc.CoreWorkerService/SpillObjects', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.SpillObjectsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.SpillObjectsReply.FromString, + ) + self.RestoreSpilledObjects = channel.unary_unary( + '/ray.rpc.CoreWorkerService/RestoreSpilledObjects', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RestoreSpilledObjectsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RestoreSpilledObjectsReply.FromString, + ) + self.DeleteSpilledObjects = channel.unary_unary( + '/ray.rpc.CoreWorkerService/DeleteSpilledObjects', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteSpilledObjectsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteSpilledObjectsReply.FromString, + ) + self.PlasmaObjectReady = channel.unary_unary( + '/ray.rpc.CoreWorkerService/PlasmaObjectReady', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PlasmaObjectReadyRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PlasmaObjectReadyReply.FromString, + ) + self.Exit = channel.unary_unary( + '/ray.rpc.CoreWorkerService/Exit', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ExitRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ExitReply.FromString, + ) + self.AssignObjectOwner = channel.unary_unary( + '/ray.rpc.CoreWorkerService/AssignObjectOwner', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.AssignObjectOwnerRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.AssignObjectOwnerReply.FromString, + ) + self.NumPendingTasks = channel.unary_unary( + '/ray.rpc.CoreWorkerService/NumPendingTasks', + request_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.NumPendingTasksRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.NumPendingTasksReply.FromString, + ) + + +class CoreWorkerServiceServicer(object): + """Missing associated documentation comment in .proto file.""" + + def RayletNotifyGCSRestart(self, request, context): + """Notify core worker GCS has restarted. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def PushTask(self, request, context): + """Push a task directly to this worker from another. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def DirectActorCallArgWaitComplete(self, request, context): + """Reply from raylet that wait for direct actor call args has completed. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetObjectStatus(self, request, context): + """Ask the object's owner about the object's current status. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def WaitForActorOutOfScope(self, request, context): + """Wait for the actor's owner to decide that the actor has gone out of scope. + Replying to this message indicates that the client should force-kill the + actor process, if still alive. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def PubsubLongPolling(self, request, context): + """/ The long polling request sent to the core worker for pubsub operations. + / It is replied once there are batch of objects that need to be published to + / the caller (subscriber). + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ReportGeneratorItemReturns(self, request, context): + """The RPC to report the intermediate task return to the caller. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def PubsubCommandBatch(self, request, context): + """/ The pubsub command batch request used by the subscriber. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def UpdateObjectLocationBatch(self, request, context): + """Update the batched object location information to the ownership-based object + directory. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetObjectLocationsOwner(self, request, context): + """Get object locations from the ownership-based object directory. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def KillActor(self, request, context): + """Request that the worker shut down without completing outstanding work. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def CancelTask(self, request, context): + """Request that a worker cancels a task. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def RemoteCancelTask(self, request, context): + """Request for a worker to issue a cancelation. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetCoreWorkerStats(self, request, context): + """Get metrics from core workers. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def LocalGC(self, request, context): + """Trigger local GC on the worker. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def DeleteObjects(self, request, context): + """Delete objects explicitly. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def SpillObjects(self, request, context): + """Spill objects to external storage. Caller: raylet; callee: I/O worker. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def RestoreSpilledObjects(self, request, context): + """Restore spilled objects from external storage. Caller: raylet; callee: I/O worker. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def DeleteSpilledObjects(self, request, context): + """Delete spilled objects from external storage. Caller: raylet; callee: I/O worker. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def PlasmaObjectReady(self, request, context): + """Notification from raylet that an object ID is available in local plasma. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Exit(self, request, context): + """Request for a worker to exit. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def AssignObjectOwner(self, request, context): + """Assign the owner of an object to the intended worker. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def NumPendingTasks(self, request, context): + """Get the number of pending tasks. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_CoreWorkerServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'RayletNotifyGCSRestart': grpc.unary_unary_rpc_method_handler( + servicer.RayletNotifyGCSRestart, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RayletNotifyGCSRestartRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RayletNotifyGCSRestartReply.SerializeToString, + ), + 'PushTask': grpc.unary_unary_rpc_method_handler( + servicer.PushTask, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PushTaskRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PushTaskReply.SerializeToString, + ), + 'DirectActorCallArgWaitComplete': grpc.unary_unary_rpc_method_handler( + servicer.DirectActorCallArgWaitComplete, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DirectActorCallArgWaitCompleteRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DirectActorCallArgWaitCompleteReply.SerializeToString, + ), + 'GetObjectStatus': grpc.unary_unary_rpc_method_handler( + servicer.GetObjectStatus, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectStatusRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectStatusReply.SerializeToString, + ), + 'WaitForActorOutOfScope': grpc.unary_unary_rpc_method_handler( + servicer.WaitForActorOutOfScope, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.WaitForActorOutOfScopeRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.WaitForActorOutOfScopeReply.SerializeToString, + ), + 'PubsubLongPolling': grpc.unary_unary_rpc_method_handler( + servicer.PubsubLongPolling, + request_deserializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingReply.SerializeToString, + ), + 'ReportGeneratorItemReturns': grpc.unary_unary_rpc_method_handler( + servicer.ReportGeneratorItemReturns, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ReportGeneratorItemReturnsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ReportGeneratorItemReturnsReply.SerializeToString, + ), + 'PubsubCommandBatch': grpc.unary_unary_rpc_method_handler( + servicer.PubsubCommandBatch, + request_deserializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchReply.SerializeToString, + ), + 'UpdateObjectLocationBatch': grpc.unary_unary_rpc_method_handler( + servicer.UpdateObjectLocationBatch, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.UpdateObjectLocationBatchRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.UpdateObjectLocationBatchReply.SerializeToString, + ), + 'GetObjectLocationsOwner': grpc.unary_unary_rpc_method_handler( + servicer.GetObjectLocationsOwner, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectLocationsOwnerRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectLocationsOwnerReply.SerializeToString, + ), + 'KillActor': grpc.unary_unary_rpc_method_handler( + servicer.KillActor, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.KillActorRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.KillActorReply.SerializeToString, + ), + 'CancelTask': grpc.unary_unary_rpc_method_handler( + servicer.CancelTask, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.CancelTaskRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.CancelTaskReply.SerializeToString, + ), + 'RemoteCancelTask': grpc.unary_unary_rpc_method_handler( + servicer.RemoteCancelTask, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RemoteCancelTaskRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RemoteCancelTaskReply.SerializeToString, + ), + 'GetCoreWorkerStats': grpc.unary_unary_rpc_method_handler( + servicer.GetCoreWorkerStats, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetCoreWorkerStatsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetCoreWorkerStatsReply.SerializeToString, + ), + 'LocalGC': grpc.unary_unary_rpc_method_handler( + servicer.LocalGC, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.LocalGCRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.LocalGCReply.SerializeToString, + ), + 'DeleteObjects': grpc.unary_unary_rpc_method_handler( + servicer.DeleteObjects, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteObjectsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteObjectsReply.SerializeToString, + ), + 'SpillObjects': grpc.unary_unary_rpc_method_handler( + servicer.SpillObjects, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.SpillObjectsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.SpillObjectsReply.SerializeToString, + ), + 'RestoreSpilledObjects': grpc.unary_unary_rpc_method_handler( + servicer.RestoreSpilledObjects, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RestoreSpilledObjectsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.RestoreSpilledObjectsReply.SerializeToString, + ), + 'DeleteSpilledObjects': grpc.unary_unary_rpc_method_handler( + servicer.DeleteSpilledObjects, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteSpilledObjectsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteSpilledObjectsReply.SerializeToString, + ), + 'PlasmaObjectReady': grpc.unary_unary_rpc_method_handler( + servicer.PlasmaObjectReady, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PlasmaObjectReadyRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.PlasmaObjectReadyReply.SerializeToString, + ), + 'Exit': grpc.unary_unary_rpc_method_handler( + servicer.Exit, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ExitRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.ExitReply.SerializeToString, + ), + 'AssignObjectOwner': grpc.unary_unary_rpc_method_handler( + servicer.AssignObjectOwner, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.AssignObjectOwnerRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.AssignObjectOwnerReply.SerializeToString, + ), + 'NumPendingTasks': grpc.unary_unary_rpc_method_handler( + servicer.NumPendingTasks, + request_deserializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.NumPendingTasksRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_core__worker__pb2.NumPendingTasksReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.CoreWorkerService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class CoreWorkerService(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def RayletNotifyGCSRestart(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/RayletNotifyGCSRestart', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.RayletNotifyGCSRestartRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.RayletNotifyGCSRestartReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def PushTask(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/PushTask', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.PushTaskRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.PushTaskReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def DirectActorCallArgWaitComplete(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/DirectActorCallArgWaitComplete', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.DirectActorCallArgWaitCompleteRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.DirectActorCallArgWaitCompleteReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetObjectStatus(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/GetObjectStatus', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectStatusRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectStatusReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def WaitForActorOutOfScope(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/WaitForActorOutOfScope', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.WaitForActorOutOfScopeRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.WaitForActorOutOfScopeReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def PubsubLongPolling(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/PubsubLongPolling', + src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ReportGeneratorItemReturns(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/ReportGeneratorItemReturns', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.ReportGeneratorItemReturnsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.ReportGeneratorItemReturnsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def PubsubCommandBatch(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/PubsubCommandBatch', + src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def UpdateObjectLocationBatch(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/UpdateObjectLocationBatch', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.UpdateObjectLocationBatchRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.UpdateObjectLocationBatchReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetObjectLocationsOwner(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/GetObjectLocationsOwner', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectLocationsOwnerRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetObjectLocationsOwnerReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def KillActor(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/KillActor', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.KillActorRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.KillActorReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def CancelTask(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/CancelTask', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.CancelTaskRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.CancelTaskReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def RemoteCancelTask(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/RemoteCancelTask', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.RemoteCancelTaskRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.RemoteCancelTaskReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetCoreWorkerStats(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/GetCoreWorkerStats', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetCoreWorkerStatsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.GetCoreWorkerStatsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def LocalGC(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/LocalGC', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.LocalGCRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.LocalGCReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def DeleteObjects(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/DeleteObjects', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteObjectsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteObjectsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def SpillObjects(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/SpillObjects', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.SpillObjectsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.SpillObjectsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def RestoreSpilledObjects(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/RestoreSpilledObjects', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.RestoreSpilledObjectsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.RestoreSpilledObjectsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def DeleteSpilledObjects(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/DeleteSpilledObjects', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteSpilledObjectsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.DeleteSpilledObjectsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def PlasmaObjectReady(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/PlasmaObjectReady', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.PlasmaObjectReadyRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.PlasmaObjectReadyReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def Exit(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/Exit', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.ExitRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.ExitReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def AssignObjectOwner(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/AssignObjectOwner', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.AssignObjectOwnerRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.AssignObjectOwnerReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def NumPendingTasks(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.CoreWorkerService/NumPendingTasks', + src_dot_ray_dot_protobuf_dot_core__worker__pb2.NumPendingTasksRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_core__worker__pb2.NumPendingTasksReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/dependency_pb2.py b/python/ray/core/generated/dependency_pb2.py new file mode 100644 index 0000000000000..5af1620af7ca2 --- /dev/null +++ b/python/ray/core/generated/dependency_pb2.py @@ -0,0 +1,35 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/dependency.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!src/ray/protobuf/dependency.proto\x12\x07ray.rpc\"\x1d\n\x0ePythonFunction\x12\x0b\n\x03key\x18\x01 \x01(\x0c\x42\x03\xf8\x01\x01\x62\x06proto3') + + + +_PYTHONFUNCTION = DESCRIPTOR.message_types_by_name['PythonFunction'] +PythonFunction = _reflection.GeneratedProtocolMessageType('PythonFunction', (_message.Message,), { + 'DESCRIPTOR' : _PYTHONFUNCTION, + '__module__' : 'src.ray.protobuf.dependency_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PythonFunction) + }) +_sym_db.RegisterMessage(PythonFunction) + +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _PYTHONFUNCTION._serialized_start=46 + _PYTHONFUNCTION._serialized_end=75 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/dependency_pb2_grpc.py b/python/ray/core/generated/dependency_pb2_grpc.py new file mode 100644 index 0000000000000..2daafffebfc81 --- /dev/null +++ b/python/ray/core/generated/dependency_pb2_grpc.py @@ -0,0 +1,4 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + diff --git a/python/ray/core/generated/event_pb2.py b/python/ray/core/generated/event_pb2.py new file mode 100644 index 0000000000000..be217c053d393 --- /dev/null +++ b/python/ray/core/generated/event_pb2.py @@ -0,0 +1,77 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/event.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1csrc/ray/protobuf/event.proto\x12\x07ray.rpc\"\x82\x04\n\x05\x45vent\x12\x10\n\x08\x65vent_id\x18\x01 \x01(\t\x12.\n\x0bsource_type\x18\x02 \x01(\x0e\x32\x19.ray.rpc.Event.SourceType\x12\x17\n\x0fsource_hostname\x18\x03 \x01(\t\x12\x12\n\nsource_pid\x18\x04 \x01(\x05\x12)\n\x08severity\x18\x05 \x01(\x0e\x32\x17.ray.rpc.Event.Severity\x12\r\n\x05label\x18\x06 \x01(\t\x12\x0f\n\x07message\x18\x07 \x01(\t\x12\x11\n\ttimestamp\x18\x08 \x01(\x03\x12\x37\n\rcustom_fields\x18\t \x03(\x0b\x32 .ray.rpc.Event.CustomFieldsEntry\x1a\x33\n\x11\x43ustomFieldsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"o\n\nSourceType\x12\n\n\x06\x43OMMON\x10\x00\x12\x0f\n\x0b\x43ORE_WORKER\x10\x01\x12\x07\n\x03GCS\x10\x02\x12\n\n\x06RAYLET\x10\x03\x12\x15\n\x11\x43LUSTER_LIFECYCLE\x10\x04\x12\x0e\n\nAUTOSCALER\x10\x05\x12\x08\n\x04JOBS\x10\x06\"M\n\x08Severity\x12\x08\n\x04INFO\x10\x00\x12\x0b\n\x07WARNING\x10\x01\x12\t\n\x05\x45RROR\x10\x02\x12\t\n\x05\x46\x41TAL\x10\x03\x12\t\n\x05\x44\x45\x42UG\x10\x04\x12\t\n\x05TRACE\x10\x05\")\n\x11ReportEventsReply\x12\x14\n\x0csend_success\x18\x01 \x01(\x08\",\n\x13ReportEventsRequest\x12\x15\n\revent_strings\x18\x01 \x03(\t2^\n\x12ReportEventService\x12H\n\x0cReportEvents\x12\x1c.ray.rpc.ReportEventsRequest\x1a\x1a.ray.rpc.ReportEventsReplyB\x03\xf8\x01\x01\x62\x06proto3') + + + +_EVENT = DESCRIPTOR.message_types_by_name['Event'] +_EVENT_CUSTOMFIELDSENTRY = _EVENT.nested_types_by_name['CustomFieldsEntry'] +_REPORTEVENTSREPLY = DESCRIPTOR.message_types_by_name['ReportEventsReply'] +_REPORTEVENTSREQUEST = DESCRIPTOR.message_types_by_name['ReportEventsRequest'] +_EVENT_SOURCETYPE = _EVENT.enum_types_by_name['SourceType'] +_EVENT_SEVERITY = _EVENT.enum_types_by_name['Severity'] +Event = _reflection.GeneratedProtocolMessageType('Event', (_message.Message,), { + + 'CustomFieldsEntry' : _reflection.GeneratedProtocolMessageType('CustomFieldsEntry', (_message.Message,), { + 'DESCRIPTOR' : _EVENT_CUSTOMFIELDSENTRY, + '__module__' : 'src.ray.protobuf.event_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.Event.CustomFieldsEntry) + }) + , + 'DESCRIPTOR' : _EVENT, + '__module__' : 'src.ray.protobuf.event_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.Event) + }) +_sym_db.RegisterMessage(Event) +_sym_db.RegisterMessage(Event.CustomFieldsEntry) + +ReportEventsReply = _reflection.GeneratedProtocolMessageType('ReportEventsReply', (_message.Message,), { + 'DESCRIPTOR' : _REPORTEVENTSREPLY, + '__module__' : 'src.ray.protobuf.event_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportEventsReply) + }) +_sym_db.RegisterMessage(ReportEventsReply) + +ReportEventsRequest = _reflection.GeneratedProtocolMessageType('ReportEventsRequest', (_message.Message,), { + 'DESCRIPTOR' : _REPORTEVENTSREQUEST, + '__module__' : 'src.ray.protobuf.event_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportEventsRequest) + }) +_sym_db.RegisterMessage(ReportEventsRequest) + +_REPORTEVENTSERVICE = DESCRIPTOR.services_by_name['ReportEventService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _EVENT_CUSTOMFIELDSENTRY._options = None + _EVENT_CUSTOMFIELDSENTRY._serialized_options = b'8\001' + _EVENT._serialized_start=42 + _EVENT._serialized_end=556 + _EVENT_CUSTOMFIELDSENTRY._serialized_start=313 + _EVENT_CUSTOMFIELDSENTRY._serialized_end=364 + _EVENT_SOURCETYPE._serialized_start=366 + _EVENT_SOURCETYPE._serialized_end=477 + _EVENT_SEVERITY._serialized_start=479 + _EVENT_SEVERITY._serialized_end=556 + _REPORTEVENTSREPLY._serialized_start=558 + _REPORTEVENTSREPLY._serialized_end=599 + _REPORTEVENTSREQUEST._serialized_start=601 + _REPORTEVENTSREQUEST._serialized_end=645 + _REPORTEVENTSERVICE._serialized_start=647 + _REPORTEVENTSERVICE._serialized_end=741 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/event_pb2_grpc.py b/python/ray/core/generated/event_pb2_grpc.py new file mode 100644 index 0000000000000..f11dd38d02beb --- /dev/null +++ b/python/ray/core/generated/event_pb2_grpc.py @@ -0,0 +1,66 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import event_pb2 as src_dot_ray_dot_protobuf_dot_event__pb2 + + +class ReportEventServiceStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.ReportEvents = channel.unary_unary( + '/ray.rpc.ReportEventService/ReportEvents', + request_serializer=src_dot_ray_dot_protobuf_dot_event__pb2.ReportEventsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_event__pb2.ReportEventsReply.FromString, + ) + + +class ReportEventServiceServicer(object): + """Missing associated documentation comment in .proto file.""" + + def ReportEvents(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_ReportEventServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'ReportEvents': grpc.unary_unary_rpc_method_handler( + servicer.ReportEvents, + request_deserializer=src_dot_ray_dot_protobuf_dot_event__pb2.ReportEventsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_event__pb2.ReportEventsReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.ReportEventService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class ReportEventService(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def ReportEvents(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ReportEventService/ReportEvents', + src_dot_ray_dot_protobuf_dot_event__pb2.ReportEventsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_event__pb2.ReportEventsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/gcs_pb2.py b/python/ray/core/generated/gcs_pb2.py new file mode 100644 index 0000000000000..9d0954254c129 --- /dev/null +++ b/python/ray/core/generated/gcs_pb2.py @@ -0,0 +1,592 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/gcs.proto +"""Generated protocol buffer code.""" +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from . import common_pb2 as src_dot_ray_dot_protobuf_dot_common__pb2 + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1asrc/ray/protobuf/gcs.proto\x12\x07ray.rpc\x1a\x1dsrc/ray/protobuf/common.proto\"T\n\x08GcsEntry\x12+\n\x0b\x63hange_mode\x18\x01 \x01(\x0e\x32\x16.ray.rpc.GcsChangeMode\x12\n\n\x02id\x18\x02 \x01(\x0c\x12\x0f\n\x07\x65ntries\x18\x03 \x03(\x0c\"\"\n\x0fObjectTableData\x12\x0f\n\x07manager\x18\x01 \x01(\x0c\"\xc0\x07\n\x0e\x41\x63torTableData\x12\x10\n\x08\x61\x63tor_id\x18\x01 \x01(\x0c\x12\x11\n\tparent_id\x18\x02 \x01(\x0c\x12&\n\x1e\x61\x63tor_creation_dummy_object_id\x18\x03 \x01(\x0c\x12\x0e\n\x06job_id\x18\x04 \x01(\x0c\x12\x31\n\x05state\x18\x06 \x01(\x0e\x32\".ray.rpc.ActorTableData.ActorState\x12\x14\n\x0cmax_restarts\x18\x07 \x01(\x03\x12\x14\n\x0cnum_restarts\x18\x08 \x01(\x04\x12!\n\x07\x61\x64\x64ress\x18\t \x01(\x0b\x32\x10.ray.rpc.Address\x12\'\n\rowner_address\x18\n \x01(\x0b\x32\x10.ray.rpc.Address\x12\x13\n\x0bis_detached\x18\x0b \x01(\x08\x12\x0c\n\x04name\x18\x0c \x01(\t\x12\x11\n\ttimestamp\x18\r \x01(\x01\x12\x33\n\x10resource_mapping\x18\x0f \x03(\x0b\x32\x19.ray.rpc.ResourceMapEntry\x12\x0b\n\x03pid\x18\x10 \x01(\r\x12\x38\n\x13\x66unction_descriptor\x18\x11 \x01(\x0b\x32\x1b.ray.rpc.FunctionDescriptor\x12\x15\n\rray_namespace\x18\x13 \x01(\t\x12\x12\n\nstart_time\x18\x14 \x01(\x04\x12\x10\n\x08\x65nd_time\x18\x15 \x01(\x04\x12\x1e\n\x16serialized_runtime_env\x18\x16 \x01(\t\x12\x12\n\nclass_name\x18\x17 \x01(\t\x12-\n\x0b\x64\x65\x61th_cause\x18\x18 \x01(\x0b\x32\x18.ray.rpc.ActorDeathCause\x12J\n\x12required_resources\x18\x1c \x03(\x0b\x32..ray.rpc.ActorTableData.RequiredResourcesEntry\x12\x14\n\x07node_id\x18\x1d \x01(\x0cH\x00\x88\x01\x01\x12\x1f\n\x12placement_group_id\x18\x1e \x01(\x0cH\x01\x88\x01\x01\x12\x11\n\trepr_name\x18\x1f \x01(\t\x1a\x38\n\x16RequiredResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\"a\n\nActorState\x12\x18\n\x14\x44\x45PENDENCIES_UNREADY\x10\x00\x12\x14\n\x10PENDING_CREATION\x10\x01\x12\t\n\x05\x41LIVE\x10\x02\x12\x0e\n\nRESTARTING\x10\x03\x12\x08\n\x04\x44\x45\x41\x44\x10\x04\x42\n\n\x08_node_idB\x15\n\x13_placement_group_id\"X\n\x0e\x45rrorTableData\x12\x0e\n\x06job_id\x18\x01 \x01(\x0c\x12\x0c\n\x04type\x18\x02 \x01(\t\x12\x15\n\rerror_message\x18\x03 \x01(\t\x12\x11\n\ttimestamp\x18\x04 \x01(\x01\"\x83\x01\n\x0cScheduleData\x12>\n\rschedule_plan\x18\x01 \x03(\x0b\x32\'.ray.rpc.ScheduleData.SchedulePlanEntry\x1a\x33\n\x11SchedulePlanEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x0c:\x02\x38\x01\"u\n\x11ProfileEventEntry\x12\x12\n\nstart_time\x18\x01 \x01(\x03\x12\x10\n\x08\x65nd_time\x18\x02 \x01(\x03\x12\x17\n\nextra_data\x18\x03 \x01(\tH\x00\x88\x01\x01\x12\x12\n\nevent_name\x18\x04 \x01(\tB\r\n\x0b_extra_data\"\x82\x01\n\rProfileEvents\x12\x16\n\x0e\x63omponent_type\x18\x01 \x01(\t\x12\x14\n\x0c\x63omponent_id\x18\x02 \x01(\x0c\x12\x17\n\x0fnode_ip_address\x18\x03 \x01(\t\x12*\n\x06\x65vents\x18\x04 \x03(\x0b\x32\x1a.ray.rpc.ProfileEventEntry\"\x89\x02\n\x0bTaskLogInfo\x12\x18\n\x0bstdout_file\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x18\n\x0bstderr_file\x18\x02 \x01(\tH\x01\x88\x01\x01\x12\x19\n\x0cstdout_start\x18\x03 \x01(\x05H\x02\x88\x01\x01\x12\x17\n\nstdout_end\x18\x04 \x01(\x05H\x03\x88\x01\x01\x12\x19\n\x0cstderr_start\x18\x05 \x01(\x05H\x04\x88\x01\x01\x12\x17\n\nstderr_end\x18\x06 \x01(\x05H\x05\x88\x01\x01\x42\x0e\n\x0c_stdout_fileB\x0e\n\x0c_stderr_fileB\x0f\n\r_stdout_startB\r\n\x0b_stdout_endB\x0f\n\r_stderr_startB\r\n\x0b_stderr_end\"\xf4\x04\n\x0fTaskStateUpdate\x12\x14\n\x07node_id\x18\x01 \x01(\x0cH\x00\x88\x01\x01\x12\"\n\x15pending_args_avail_ts\x18\x02 \x01(\x03H\x01\x88\x01\x01\x12\'\n\x1apending_node_assignment_ts\x18\x03 \x01(\x03H\x02\x88\x01\x01\x12#\n\x16submitted_to_worker_ts\x18\x04 \x01(\x03H\x03\x88\x01\x01\x12\x17\n\nrunning_ts\x18\x05 \x01(\x03H\x04\x88\x01\x01\x12\x18\n\x0b\x66inished_ts\x18\x06 \x01(\x03H\x05\x88\x01\x01\x12\x16\n\tfailed_ts\x18\x07 \x01(\x03H\x06\x88\x01\x01\x12\x16\n\tworker_id\x18\x08 \x01(\x0cH\x07\x88\x01\x01\x12.\n\nerror_info\x18\t \x01(\x0b\x32\x15.ray.rpc.RayErrorInfoH\x08\x88\x01\x01\x12\x30\n\rtask_log_info\x18\n \x01(\x0b\x32\x14.ray.rpc.TaskLogInfoH\t\x88\x01\x01\x12\x1c\n\x0f\x61\x63tor_repr_name\x18\x0b \x01(\tH\n\x88\x01\x01\x12\x17\n\nworker_pid\x18\x0c \x01(\x05H\x0b\x88\x01\x01\x42\n\n\x08_node_idB\x18\n\x16_pending_args_avail_tsB\x1d\n\x1b_pending_node_assignment_tsB\x19\n\x17_submitted_to_worker_tsB\r\n\x0b_running_tsB\x0e\n\x0c_finished_tsB\x0c\n\n_failed_tsB\x0c\n\n_worker_idB\r\n\x0b_error_infoB\x10\n\x0e_task_log_infoB\x12\n\x10_actor_repr_nameB\r\n\x0b_worker_pid\"\x93\x02\n\nTaskEvents\x12\x0f\n\x07task_id\x18\x01 \x01(\x0c\x12\x16\n\x0e\x61ttempt_number\x18\x02 \x01(\x05\x12.\n\ttask_info\x18\x03 \x01(\x0b\x32\x16.ray.rpc.TaskInfoEntryH\x00\x88\x01\x01\x12\x34\n\rstate_updates\x18\x04 \x01(\x0b\x32\x18.ray.rpc.TaskStateUpdateH\x01\x88\x01\x01\x12\x33\n\x0eprofile_events\x18\x05 \x01(\x0b\x32\x16.ray.rpc.ProfileEventsH\x02\x88\x01\x01\x12\x0e\n\x06job_id\x18\x06 \x01(\x0c\x42\x0c\n\n_task_infoB\x10\n\x0e_state_updatesB\x11\n\x0f_profile_events\"\x8d\x01\n\rTaskEventData\x12+\n\x0e\x65vents_by_task\x18\x01 \x03(\x0b\x32\x13.ray.rpc.TaskEvents\x12\'\n\x1fnum_profile_task_events_dropped\x18\x03 \x01(\x05\x12&\n\x1enum_status_task_events_dropped\x18\x04 \x01(\x05\".\n\x11ResourceTableData\x12\x19\n\x11resource_capacity\x18\x01 \x01(\x01\"\xb2\x01\n\x12\x41vailableResources\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\x12P\n\x13resources_available\x18\x02 \x03(\x0b\x32\x33.ray.rpc.AvailableResources.ResourcesAvailableEntry\x1a\x39\n\x17ResourcesAvailableEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\"\xde\x05\n\x0bGcsNodeInfo\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\x12\x1c\n\x14node_manager_address\x18\x02 \x01(\t\x12\x1a\n\x12raylet_socket_name\x18\x03 \x01(\t\x12 \n\x18object_store_socket_name\x18\x04 \x01(\t\x12\x19\n\x11node_manager_port\x18\x05 \x01(\x05\x12\x1b\n\x13object_manager_port\x18\x06 \x01(\x05\x12\x30\n\x05state\x18\x07 \x01(\x0e\x32!.ray.rpc.GcsNodeInfo.GcsNodeState\x12\x1d\n\x15node_manager_hostname\x18\x08 \x01(\t\x12\x1b\n\x13metrics_export_port\x18\t \x01(\x05\x12\x1e\n\x16runtime_env_agent_port\x18\x1b \x01(\x05\x12\x41\n\x0fresources_total\x18\x0b \x03(\x0b\x32(.ray.rpc.GcsNodeInfo.ResourcesTotalEntry\x12\x11\n\tnode_name\x18\x0c \x01(\t\x12\x13\n\x0binstance_id\x18\r \x01(\t\x12\x16\n\x0enode_type_name\x18\x0e \x01(\t\x12\x1a\n\x12instance_type_name\x18\x0f \x01(\t\x12\x15\n\rstart_time_ms\x18\x17 \x01(\x04\x12\x13\n\x0b\x65nd_time_ms\x18\x18 \x01(\x04\x12\x14\n\x0cis_head_node\x18\x19 \x01(\x08\x12\x30\n\x06labels\x18\x1a \x03(\x0b\x32 .ray.rpc.GcsNodeInfo.LabelsEntry\x1a\x35\n\x13ResourcesTotalEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"#\n\x0cGcsNodeState\x12\t\n\x05\x41LIVE\x10\x00\x12\x08\n\x04\x44\x45\x41\x44\x10\x01\"\xd5\x05\n\x0bJobsAPIInfo\x12\x0e\n\x06status\x18\x01 \x01(\t\x12\x12\n\nentrypoint\x18\x02 \x01(\t\x12\x14\n\x07message\x18\x03 \x01(\tH\x00\x88\x01\x01\x12\x17\n\nerror_type\x18\x04 \x01(\tH\x01\x88\x01\x01\x12\x17\n\nstart_time\x18\x05 \x01(\x04H\x02\x88\x01\x01\x12\x15\n\x08\x65nd_time\x18\x06 \x01(\x04H\x03\x88\x01\x01\x12\x34\n\x08metadata\x18\x07 \x03(\x0b\x32\".ray.rpc.JobsAPIInfo.MetadataEntry\x12\x1d\n\x10runtime_env_json\x18\x08 \x01(\tH\x04\x88\x01\x01\x12 \n\x13\x65ntrypoint_num_cpus\x18\t \x01(\x01H\x05\x88\x01\x01\x12 \n\x13\x65ntrypoint_num_gpus\x18\n \x01(\x01H\x06\x88\x01\x01\x12K\n\x14\x65ntrypoint_resources\x18\x0b \x03(\x0b\x32-.ray.rpc.JobsAPIInfo.EntrypointResourcesEntry\x12&\n\x19\x64river_agent_http_address\x18\x0c \x01(\tH\x07\x88\x01\x01\x12\x1b\n\x0e\x64river_node_id\x18\r \x01(\tH\x08\x88\x01\x01\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a:\n\x18\x45ntrypointResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x42\n\n\x08_messageB\r\n\x0b_error_typeB\r\n\x0b_start_timeB\x0b\n\t_end_timeB\x13\n\x11_runtime_env_jsonB\x16\n\x14_entrypoint_num_cpusB\x16\n\x14_entrypoint_num_gpusB\x1c\n\x1a_driver_agent_http_addressB\x11\n\x0f_driver_node_id\"\x96\x04\n\x0fWorkerTableData\x12\x10\n\x08is_alive\x18\x01 \x01(\x08\x12(\n\x0eworker_address\x18\x02 \x01(\x0b\x32\x10.ray.rpc.Address\x12\x11\n\ttimestamp\x18\x03 \x01(\x03\x12(\n\x0bworker_type\x18\x05 \x01(\x0e\x32\x13.ray.rpc.WorkerType\x12=\n\x0bworker_info\x18\x06 \x03(\x0b\x32(.ray.rpc.WorkerTableData.WorkerInfoEntry\x12\x36\n\x17\x63reation_task_exception\x18\x12 \x01(\x0b\x32\x15.ray.rpc.RayException\x12/\n\texit_type\x18\x13 \x01(\x0e\x32\x17.ray.rpc.WorkerExitTypeH\x00\x88\x01\x01\x12\x18\n\x0b\x65xit_detail\x18\x14 \x01(\tH\x01\x88\x01\x01\x12\x0b\n\x03pid\x18\x15 \x01(\r\x12\x15\n\rstart_time_ms\x18\x17 \x01(\x04\x12\x13\n\x0b\x65nd_time_ms\x18\x18 \x01(\x04\x12\x1d\n\x15worker_launch_time_ms\x18\x19 \x01(\x04\x12\x1f\n\x17worker_launched_time_ms\x18\x1a \x01(\x04\x1a\x31\n\x0fWorkerInfoEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x0c:\x02\x38\x01\x42\x0c\n\n_exit_typeB\x0e\n\x0c_exit_detail\"7\n\x0fWorkerDeltaData\x12\x11\n\traylet_id\x18\x01 \x01(\x0c\x12\x11\n\tworker_id\x18\x02 \x01(\x0c\"\x87\x01\n\x0bResourceMap\x12.\n\x05items\x18\x01 \x03(\x0b\x32\x1f.ray.rpc.ResourceMap.ItemsEntry\x1aH\n\nItemsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12)\n\x05value\x18\x02 \x01(\x0b\x32\x1a.ray.rpc.ResourceTableData:\x02\x38\x01\"\x1e\n\x0cStoredConfig\x12\x0e\n\x06\x63onfig\x18\x01 \x01(\t\"\xc7\x01\n\x12NodeResourceChange\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\x12L\n\x11updated_resources\x18\x02 \x03(\x0b\x32\x31.ray.rpc.NodeResourceChange.UpdatedResourcesEntry\x12\x19\n\x11\x64\x65leted_resources\x18\x03 \x03(\t\x1a\x37\n\x15UpdatedResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\")\n\rPubSubMessage\x12\n\n\x02id\x18\x01 \x01(\x0c\x12\x0c\n\x04\x64\x61ta\x18\x02 \x01(\x0c\"\x82\x01\n\x0eResourceUpdate\x12-\n\x06\x63hange\x18\x01 \x01(\x0b\x32\x1b.ray.rpc.NodeResourceChangeH\x00\x12&\n\x04\x64\x61ta\x18\x02 \x01(\x0b\x32\x16.ray.rpc.ResourcesDataH\x00\x42\x19\n\x17resource_change_or_data\"T\n\x1aResourceUsageBroadcastData\x12\x0e\n\x06seq_no\x18\x01 \x01(\x03\x12&\n\x05\x62\x61tch\x18\x02 \x03(\x0b\x32\x17.ray.rpc.ResourceUpdate\"\xd2\x01\n\x0eResourceDemand\x12\x31\n\x05shape\x18\x01 \x03(\x0b\x32\".ray.rpc.ResourceDemand.ShapeEntry\x12!\n\x19num_ready_requests_queued\x18\x02 \x01(\x04\x12&\n\x1enum_infeasible_requests_queued\x18\x03 \x01(\x04\x12\x14\n\x0c\x62\x61\x63klog_size\x18\x04 \x01(\x03\x1a,\n\nShapeEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\"A\n\x0cResourceLoad\x12\x31\n\x10resource_demands\x18\x01 \x03(\x0b\x32\x17.ray.rpc.ResourceDemand\"\xff\x06\n\rResourcesData\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\x12K\n\x13resources_available\x18\x02 \x03(\x0b\x32..ray.rpc.ResourcesData.ResourcesAvailableEntry\x12\x43\n\x0fresources_total\x18\x04 \x03(\x0b\x32*.ray.rpc.ResourcesData.ResourcesTotalEntry\x12?\n\rresource_load\x18\x05 \x03(\x0b\x32(.ray.rpc.ResourcesData.ResourceLoadEntry\x12\x1d\n\x15resource_load_changed\x18\x06 \x01(\x08\x12\x35\n\x16resource_load_by_shape\x18\x07 \x01(\x0b\x32\x15.ray.rpc.ResourceLoad\x12\x18\n\x10should_global_gc\x18\x08 \x01(\x08\x12\x1c\n\x14node_manager_address\x18\t \x01(\t\x12\x1b\n\x13object_pulls_queued\x18\n \x01(\x08\x12N\n\x15resources_normal_task\x18\x0b \x03(\x0b\x32/.ray.rpc.ResourcesData.ResourcesNormalTaskEntry\x12%\n\x1dresources_normal_task_changed\x18\x0c \x01(\x08\x12\'\n\x1fresources_normal_task_timestamp\x18\r \x01(\x03\x12\'\n\x1f\x63luster_full_of_actors_detected\x18\x0e \x01(\x08\x12\x18\n\x10idle_duration_ms\x18\x0f \x01(\x03\x12\x13\n\x0bis_draining\x18\x10 \x01(\x08\x1a\x39\n\x17ResourcesAvailableEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x1a\x35\n\x13ResourcesTotalEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x1a\x33\n\x11ResourceLoadEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x1a:\n\x18ResourcesNormalTaskEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01J\x04\x08\x03\x10\x04\"\xb1\x01\n\x16ResourceUsageBatchData\x12%\n\x05\x62\x61tch\x18\x01 \x03(\x0b\x32\x16.ray.rpc.ResourcesData\x12\x35\n\x16resource_load_by_shape\x18\x02 \x01(\x0b\x32\x15.ray.rpc.ResourceLoad\x12\x39\n\x14placement_group_load\x18\x03 \x01(\x0b\x32\x1b.ray.rpc.PlacementGroupLoad\"T\n\x12PlacementGroupLoad\x12>\n\x14placement_group_data\x18\x01 \x03(\x0b\x32 .ray.rpc.PlacementGroupTableData\"\xbf\x03\n\x13PlacementGroupStats\x12$\n\x1c\x63reation_request_received_ns\x18\x02 \x01(\x03\x12\"\n\x1ascheduling_started_time_ns\x18\x03 \x01(\x03\x12\x1d\n\x15scheduling_latency_us\x18\x04 \x01(\x03\x12&\n\x1e\x65nd_to_end_creation_latency_us\x18\x05 \x01(\x03\x12\x1a\n\x12scheduling_attempt\x18\x06 \x01(\r\x12\x1e\n\x16highest_retry_delay_ms\x18\x07 \x01(\x01\x12\x46\n\x10scheduling_state\x18\x08 \x01(\x0e\x32,.ray.rpc.PlacementGroupStats.SchedulingState\"\x92\x01\n\x0fSchedulingState\x12\n\n\x06QUEUED\x10\x00\x12\x0b\n\x07REMOVED\x10\x01\x12\x16\n\x12SCHEDULING_STARTED\x10\x02\x12\x10\n\x0cNO_RESOURCES\x10\x03\x12\x0e\n\nINFEASIBLE\x10\x04\x12\x1e\n\x1a\x46\x41ILED_TO_COMMIT_RESOURCES\x10\x05\x12\x0c\n\x08\x46INISHED\x10\x06\"\xf8\x04\n\x17PlacementGroupTableData\x12\x1a\n\x12placement_group_id\x18\x01 \x01(\x0c\x12\x0c\n\x04name\x18\x02 \x01(\t\x12 \n\x07\x62undles\x18\x03 \x03(\x0b\x32\x0f.ray.rpc.Bundle\x12,\n\x08strategy\x18\x04 \x01(\x0e\x32\x1a.ray.rpc.PlacementStrategy\x12\x43\n\x05state\x18\x05 \x01(\x0e\x32\x34.ray.rpc.PlacementGroupTableData.PlacementGroupState\x12\x16\n\x0e\x63reator_job_id\x18\x06 \x01(\x0c\x12\x18\n\x10\x63reator_actor_id\x18\x07 \x01(\x0c\x12\x18\n\x10\x63reator_job_dead\x18\x08 \x01(\x08\x12\x1a\n\x12\x63reator_actor_dead\x18\t \x01(\x08\x12\x13\n\x0bis_detached\x18\n \x01(\x08\x12\x15\n\rray_namespace\x18\x0b \x01(\t\x12+\n\x05stats\x18\x0c \x01(\x0b\x32\x1c.ray.rpc.PlacementGroupStats\x12!\n\x19max_cpu_fraction_per_node\x18\r \x01(\x01\x12-\n%placement_group_creation_timestamp_ms\x18\x0e \x01(\x03\x12;\n3placement_group_final_bundle_placement_timestamp_ms\x18\x0f \x01(\x03\"N\n\x13PlacementGroupState\x12\x0b\n\x07PENDING\x10\x00\x12\x0b\n\x07\x43REATED\x10\x01\x12\x0b\n\x07REMOVED\x10\x02\x12\x10\n\x0cRESCHEDULING\x10\x03\"\xcd\x02\n\x0cJobTableData\x12\x0e\n\x06job_id\x18\x01 \x01(\x0c\x12\x0f\n\x07is_dead\x18\x02 \x01(\x08\x12\x11\n\ttimestamp\x18\x03 \x01(\x03\x12\x19\n\x11\x64river_ip_address\x18\x04 \x01(\t\x12\x12\n\ndriver_pid\x18\x05 \x01(\x03\x12\"\n\x06\x63onfig\x18\x06 \x01(\x0b\x32\x12.ray.rpc.JobConfig\x12\x12\n\nstart_time\x18\x07 \x01(\x04\x12\x10\n\x08\x65nd_time\x18\x08 \x01(\x04\x12\x12\n\nentrypoint\x18\t \x01(\t\x12+\n\x08job_info\x18\n \x01(\x0b\x32\x14.ray.rpc.JobsAPIInfoH\x00\x88\x01\x01\x12\x18\n\x10is_running_tasks\x18\x0b \x01(\x08\x12(\n\x0e\x64river_address\x18\x0c \x01(\x0b\x32\x10.ray.rpc.AddressB\x0b\n\t_job_info*\xdc\x02\n\x0bTablePrefix\x12\x14\n\x10TABLE_PREFIX_MIN\x10\x00\x12\n\n\x06UNUSED\x10\x01\x12\x08\n\x04TASK\x10\x02\x12\x0f\n\x0bRAYLET_TASK\x10\x03\x12\x08\n\x04NODE\x10\x04\x12\n\n\x06OBJECT\x10\x05\x12\t\n\x05\x41\x43TOR\x10\x06\x12\x0c\n\x08\x46UNCTION\x10\x07\x12\x17\n\x13TASK_RECONSTRUCTION\x10\x08\x12\x18\n\x14RESOURCE_USAGE_BATCH\x10\t\x12\x07\n\x03JOB\x10\n\x12\x0e\n\nTASK_LEASE\x10\x0c\x12\x11\n\rNODE_RESOURCE\x10\r\x12\x10\n\x0c\x44IRECT_ACTOR\x10\x0e\x12\x0b\n\x07WORKERS\x10\x0f\x12\x13\n\x0fINTERNAL_CONFIG\x10\x10\x12\x1c\n\x18PLACEMENT_GROUP_SCHEDULE\x10\x11\x12\x13\n\x0fPLACEMENT_GROUP\x10\x12\x12\x06\n\x02KV\x10\x13\x12\x13\n\x0f\x41\x43TOR_TASK_SPEC\x10\x14*\xbe\x02\n\x0bTablePubsub\x12\x14\n\x10TABLE_PUBSUB_MIN\x10\x00\x12\x0e\n\nNO_PUBLISH\x10\x01\x12\x0f\n\x0bTASK_PUBSUB\x10\x02\x12\x16\n\x12RAYLET_TASK_PUBSUB\x10\x03\x12\x0f\n\x0bNODE_PUBSUB\x10\x04\x12\x11\n\rOBJECT_PUBSUB\x10\x05\x12\x10\n\x0c\x41\x43TOR_PUBSUB\x10\x06\x12\x1f\n\x1bRESOURCE_USAGE_BATCH_PUBSUB\x10\x07\x12\x15\n\x11TASK_LEASE_PUBSUB\x10\x08\x12\x0e\n\nJOB_PUBSUB\x10\t\x12\x18\n\x14NODE_RESOURCE_PUBSUB\x10\n\x12\x17\n\x13\x44IRECT_ACTOR_PUBSUB\x10\x0b\x12\x19\n\x15WORKER_FAILURE_PUBSUB\x10\x0c\x12\x14\n\x10TABLE_PUBSUB_MAX\x10\r*.\n\rGcsChangeMode\x12\x11\n\rAPPEND_OR_ADD\x10\x00\x12\n\n\x06REMOVE\x10\x01\x42\x1d\n\x18io.ray.runtime.generated\xf8\x01\x01\x62\x06proto3') + +_TABLEPREFIX = DESCRIPTOR.enum_types_by_name['TablePrefix'] +TablePrefix = enum_type_wrapper.EnumTypeWrapper(_TABLEPREFIX) +_TABLEPUBSUB = DESCRIPTOR.enum_types_by_name['TablePubsub'] +TablePubsub = enum_type_wrapper.EnumTypeWrapper(_TABLEPUBSUB) +_GCSCHANGEMODE = DESCRIPTOR.enum_types_by_name['GcsChangeMode'] +GcsChangeMode = enum_type_wrapper.EnumTypeWrapper(_GCSCHANGEMODE) +TABLE_PREFIX_MIN = 0 +UNUSED = 1 +TASK = 2 +RAYLET_TASK = 3 +NODE = 4 +OBJECT = 5 +ACTOR = 6 +FUNCTION = 7 +TASK_RECONSTRUCTION = 8 +RESOURCE_USAGE_BATCH = 9 +JOB = 10 +TASK_LEASE = 12 +NODE_RESOURCE = 13 +DIRECT_ACTOR = 14 +WORKERS = 15 +INTERNAL_CONFIG = 16 +PLACEMENT_GROUP_SCHEDULE = 17 +PLACEMENT_GROUP = 18 +KV = 19 +ACTOR_TASK_SPEC = 20 +TABLE_PUBSUB_MIN = 0 +NO_PUBLISH = 1 +TASK_PUBSUB = 2 +RAYLET_TASK_PUBSUB = 3 +NODE_PUBSUB = 4 +OBJECT_PUBSUB = 5 +ACTOR_PUBSUB = 6 +RESOURCE_USAGE_BATCH_PUBSUB = 7 +TASK_LEASE_PUBSUB = 8 +JOB_PUBSUB = 9 +NODE_RESOURCE_PUBSUB = 10 +DIRECT_ACTOR_PUBSUB = 11 +WORKER_FAILURE_PUBSUB = 12 +TABLE_PUBSUB_MAX = 13 +APPEND_OR_ADD = 0 +REMOVE = 1 + + +_GCSENTRY = DESCRIPTOR.message_types_by_name['GcsEntry'] +_OBJECTTABLEDATA = DESCRIPTOR.message_types_by_name['ObjectTableData'] +_ACTORTABLEDATA = DESCRIPTOR.message_types_by_name['ActorTableData'] +_ACTORTABLEDATA_REQUIREDRESOURCESENTRY = _ACTORTABLEDATA.nested_types_by_name['RequiredResourcesEntry'] +_ERRORTABLEDATA = DESCRIPTOR.message_types_by_name['ErrorTableData'] +_SCHEDULEDATA = DESCRIPTOR.message_types_by_name['ScheduleData'] +_SCHEDULEDATA_SCHEDULEPLANENTRY = _SCHEDULEDATA.nested_types_by_name['SchedulePlanEntry'] +_PROFILEEVENTENTRY = DESCRIPTOR.message_types_by_name['ProfileEventEntry'] +_PROFILEEVENTS = DESCRIPTOR.message_types_by_name['ProfileEvents'] +_TASKLOGINFO = DESCRIPTOR.message_types_by_name['TaskLogInfo'] +_TASKSTATEUPDATE = DESCRIPTOR.message_types_by_name['TaskStateUpdate'] +_TASKEVENTS = DESCRIPTOR.message_types_by_name['TaskEvents'] +_TASKEVENTDATA = DESCRIPTOR.message_types_by_name['TaskEventData'] +_RESOURCETABLEDATA = DESCRIPTOR.message_types_by_name['ResourceTableData'] +_AVAILABLERESOURCES = DESCRIPTOR.message_types_by_name['AvailableResources'] +_AVAILABLERESOURCES_RESOURCESAVAILABLEENTRY = _AVAILABLERESOURCES.nested_types_by_name['ResourcesAvailableEntry'] +_GCSNODEINFO = DESCRIPTOR.message_types_by_name['GcsNodeInfo'] +_GCSNODEINFO_RESOURCESTOTALENTRY = _GCSNODEINFO.nested_types_by_name['ResourcesTotalEntry'] +_GCSNODEINFO_LABELSENTRY = _GCSNODEINFO.nested_types_by_name['LabelsEntry'] +_JOBSAPIINFO = DESCRIPTOR.message_types_by_name['JobsAPIInfo'] +_JOBSAPIINFO_METADATAENTRY = _JOBSAPIINFO.nested_types_by_name['MetadataEntry'] +_JOBSAPIINFO_ENTRYPOINTRESOURCESENTRY = _JOBSAPIINFO.nested_types_by_name['EntrypointResourcesEntry'] +_WORKERTABLEDATA = DESCRIPTOR.message_types_by_name['WorkerTableData'] +_WORKERTABLEDATA_WORKERINFOENTRY = _WORKERTABLEDATA.nested_types_by_name['WorkerInfoEntry'] +_WORKERDELTADATA = DESCRIPTOR.message_types_by_name['WorkerDeltaData'] +_RESOURCEMAP = DESCRIPTOR.message_types_by_name['ResourceMap'] +_RESOURCEMAP_ITEMSENTRY = _RESOURCEMAP.nested_types_by_name['ItemsEntry'] +_STOREDCONFIG = DESCRIPTOR.message_types_by_name['StoredConfig'] +_NODERESOURCECHANGE = DESCRIPTOR.message_types_by_name['NodeResourceChange'] +_NODERESOURCECHANGE_UPDATEDRESOURCESENTRY = _NODERESOURCECHANGE.nested_types_by_name['UpdatedResourcesEntry'] +_PUBSUBMESSAGE = DESCRIPTOR.message_types_by_name['PubSubMessage'] +_RESOURCEUPDATE = DESCRIPTOR.message_types_by_name['ResourceUpdate'] +_RESOURCEUSAGEBROADCASTDATA = DESCRIPTOR.message_types_by_name['ResourceUsageBroadcastData'] +_RESOURCEDEMAND = DESCRIPTOR.message_types_by_name['ResourceDemand'] +_RESOURCEDEMAND_SHAPEENTRY = _RESOURCEDEMAND.nested_types_by_name['ShapeEntry'] +_RESOURCELOAD = DESCRIPTOR.message_types_by_name['ResourceLoad'] +_RESOURCESDATA = DESCRIPTOR.message_types_by_name['ResourcesData'] +_RESOURCESDATA_RESOURCESAVAILABLEENTRY = _RESOURCESDATA.nested_types_by_name['ResourcesAvailableEntry'] +_RESOURCESDATA_RESOURCESTOTALENTRY = _RESOURCESDATA.nested_types_by_name['ResourcesTotalEntry'] +_RESOURCESDATA_RESOURCELOADENTRY = _RESOURCESDATA.nested_types_by_name['ResourceLoadEntry'] +_RESOURCESDATA_RESOURCESNORMALTASKENTRY = _RESOURCESDATA.nested_types_by_name['ResourcesNormalTaskEntry'] +_RESOURCEUSAGEBATCHDATA = DESCRIPTOR.message_types_by_name['ResourceUsageBatchData'] +_PLACEMENTGROUPLOAD = DESCRIPTOR.message_types_by_name['PlacementGroupLoad'] +_PLACEMENTGROUPSTATS = DESCRIPTOR.message_types_by_name['PlacementGroupStats'] +_PLACEMENTGROUPTABLEDATA = DESCRIPTOR.message_types_by_name['PlacementGroupTableData'] +_JOBTABLEDATA = DESCRIPTOR.message_types_by_name['JobTableData'] +_ACTORTABLEDATA_ACTORSTATE = _ACTORTABLEDATA.enum_types_by_name['ActorState'] +_GCSNODEINFO_GCSNODESTATE = _GCSNODEINFO.enum_types_by_name['GcsNodeState'] +_PLACEMENTGROUPSTATS_SCHEDULINGSTATE = _PLACEMENTGROUPSTATS.enum_types_by_name['SchedulingState'] +_PLACEMENTGROUPTABLEDATA_PLACEMENTGROUPSTATE = _PLACEMENTGROUPTABLEDATA.enum_types_by_name['PlacementGroupState'] +GcsEntry = _reflection.GeneratedProtocolMessageType('GcsEntry', (_message.Message,), { + 'DESCRIPTOR' : _GCSENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GcsEntry) + }) +_sym_db.RegisterMessage(GcsEntry) + +ObjectTableData = _reflection.GeneratedProtocolMessageType('ObjectTableData', (_message.Message,), { + 'DESCRIPTOR' : _OBJECTTABLEDATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ObjectTableData) + }) +_sym_db.RegisterMessage(ObjectTableData) + +ActorTableData = _reflection.GeneratedProtocolMessageType('ActorTableData', (_message.Message,), { + + 'RequiredResourcesEntry' : _reflection.GeneratedProtocolMessageType('RequiredResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _ACTORTABLEDATA_REQUIREDRESOURCESENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ActorTableData.RequiredResourcesEntry) + }) + , + 'DESCRIPTOR' : _ACTORTABLEDATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ActorTableData) + }) +_sym_db.RegisterMessage(ActorTableData) +_sym_db.RegisterMessage(ActorTableData.RequiredResourcesEntry) + +ErrorTableData = _reflection.GeneratedProtocolMessageType('ErrorTableData', (_message.Message,), { + 'DESCRIPTOR' : _ERRORTABLEDATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ErrorTableData) + }) +_sym_db.RegisterMessage(ErrorTableData) + +ScheduleData = _reflection.GeneratedProtocolMessageType('ScheduleData', (_message.Message,), { + + 'SchedulePlanEntry' : _reflection.GeneratedProtocolMessageType('SchedulePlanEntry', (_message.Message,), { + 'DESCRIPTOR' : _SCHEDULEDATA_SCHEDULEPLANENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ScheduleData.SchedulePlanEntry) + }) + , + 'DESCRIPTOR' : _SCHEDULEDATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ScheduleData) + }) +_sym_db.RegisterMessage(ScheduleData) +_sym_db.RegisterMessage(ScheduleData.SchedulePlanEntry) + +ProfileEventEntry = _reflection.GeneratedProtocolMessageType('ProfileEventEntry', (_message.Message,), { + 'DESCRIPTOR' : _PROFILEEVENTENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ProfileEventEntry) + }) +_sym_db.RegisterMessage(ProfileEventEntry) + +ProfileEvents = _reflection.GeneratedProtocolMessageType('ProfileEvents', (_message.Message,), { + 'DESCRIPTOR' : _PROFILEEVENTS, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ProfileEvents) + }) +_sym_db.RegisterMessage(ProfileEvents) + +TaskLogInfo = _reflection.GeneratedProtocolMessageType('TaskLogInfo', (_message.Message,), { + 'DESCRIPTOR' : _TASKLOGINFO, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TaskLogInfo) + }) +_sym_db.RegisterMessage(TaskLogInfo) + +TaskStateUpdate = _reflection.GeneratedProtocolMessageType('TaskStateUpdate', (_message.Message,), { + 'DESCRIPTOR' : _TASKSTATEUPDATE, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TaskStateUpdate) + }) +_sym_db.RegisterMessage(TaskStateUpdate) + +TaskEvents = _reflection.GeneratedProtocolMessageType('TaskEvents', (_message.Message,), { + 'DESCRIPTOR' : _TASKEVENTS, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TaskEvents) + }) +_sym_db.RegisterMessage(TaskEvents) + +TaskEventData = _reflection.GeneratedProtocolMessageType('TaskEventData', (_message.Message,), { + 'DESCRIPTOR' : _TASKEVENTDATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TaskEventData) + }) +_sym_db.RegisterMessage(TaskEventData) + +ResourceTableData = _reflection.GeneratedProtocolMessageType('ResourceTableData', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCETABLEDATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceTableData) + }) +_sym_db.RegisterMessage(ResourceTableData) + +AvailableResources = _reflection.GeneratedProtocolMessageType('AvailableResources', (_message.Message,), { + + 'ResourcesAvailableEntry' : _reflection.GeneratedProtocolMessageType('ResourcesAvailableEntry', (_message.Message,), { + 'DESCRIPTOR' : _AVAILABLERESOURCES_RESOURCESAVAILABLEENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.AvailableResources.ResourcesAvailableEntry) + }) + , + 'DESCRIPTOR' : _AVAILABLERESOURCES, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.AvailableResources) + }) +_sym_db.RegisterMessage(AvailableResources) +_sym_db.RegisterMessage(AvailableResources.ResourcesAvailableEntry) + +GcsNodeInfo = _reflection.GeneratedProtocolMessageType('GcsNodeInfo', (_message.Message,), { + + 'ResourcesTotalEntry' : _reflection.GeneratedProtocolMessageType('ResourcesTotalEntry', (_message.Message,), { + 'DESCRIPTOR' : _GCSNODEINFO_RESOURCESTOTALENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GcsNodeInfo.ResourcesTotalEntry) + }) + , + + 'LabelsEntry' : _reflection.GeneratedProtocolMessageType('LabelsEntry', (_message.Message,), { + 'DESCRIPTOR' : _GCSNODEINFO_LABELSENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GcsNodeInfo.LabelsEntry) + }) + , + 'DESCRIPTOR' : _GCSNODEINFO, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GcsNodeInfo) + }) +_sym_db.RegisterMessage(GcsNodeInfo) +_sym_db.RegisterMessage(GcsNodeInfo.ResourcesTotalEntry) +_sym_db.RegisterMessage(GcsNodeInfo.LabelsEntry) + +JobsAPIInfo = _reflection.GeneratedProtocolMessageType('JobsAPIInfo', (_message.Message,), { + + 'MetadataEntry' : _reflection.GeneratedProtocolMessageType('MetadataEntry', (_message.Message,), { + 'DESCRIPTOR' : _JOBSAPIINFO_METADATAENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.JobsAPIInfo.MetadataEntry) + }) + , + + 'EntrypointResourcesEntry' : _reflection.GeneratedProtocolMessageType('EntrypointResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _JOBSAPIINFO_ENTRYPOINTRESOURCESENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.JobsAPIInfo.EntrypointResourcesEntry) + }) + , + 'DESCRIPTOR' : _JOBSAPIINFO, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.JobsAPIInfo) + }) +_sym_db.RegisterMessage(JobsAPIInfo) +_sym_db.RegisterMessage(JobsAPIInfo.MetadataEntry) +_sym_db.RegisterMessage(JobsAPIInfo.EntrypointResourcesEntry) + +WorkerTableData = _reflection.GeneratedProtocolMessageType('WorkerTableData', (_message.Message,), { + + 'WorkerInfoEntry' : _reflection.GeneratedProtocolMessageType('WorkerInfoEntry', (_message.Message,), { + 'DESCRIPTOR' : _WORKERTABLEDATA_WORKERINFOENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WorkerTableData.WorkerInfoEntry) + }) + , + 'DESCRIPTOR' : _WORKERTABLEDATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WorkerTableData) + }) +_sym_db.RegisterMessage(WorkerTableData) +_sym_db.RegisterMessage(WorkerTableData.WorkerInfoEntry) + +WorkerDeltaData = _reflection.GeneratedProtocolMessageType('WorkerDeltaData', (_message.Message,), { + 'DESCRIPTOR' : _WORKERDELTADATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WorkerDeltaData) + }) +_sym_db.RegisterMessage(WorkerDeltaData) + +ResourceMap = _reflection.GeneratedProtocolMessageType('ResourceMap', (_message.Message,), { + + 'ItemsEntry' : _reflection.GeneratedProtocolMessageType('ItemsEntry', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCEMAP_ITEMSENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceMap.ItemsEntry) + }) + , + 'DESCRIPTOR' : _RESOURCEMAP, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceMap) + }) +_sym_db.RegisterMessage(ResourceMap) +_sym_db.RegisterMessage(ResourceMap.ItemsEntry) + +StoredConfig = _reflection.GeneratedProtocolMessageType('StoredConfig', (_message.Message,), { + 'DESCRIPTOR' : _STOREDCONFIG, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.StoredConfig) + }) +_sym_db.RegisterMessage(StoredConfig) + +NodeResourceChange = _reflection.GeneratedProtocolMessageType('NodeResourceChange', (_message.Message,), { + + 'UpdatedResourcesEntry' : _reflection.GeneratedProtocolMessageType('UpdatedResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _NODERESOURCECHANGE_UPDATEDRESOURCESENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NodeResourceChange.UpdatedResourcesEntry) + }) + , + 'DESCRIPTOR' : _NODERESOURCECHANGE, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NodeResourceChange) + }) +_sym_db.RegisterMessage(NodeResourceChange) +_sym_db.RegisterMessage(NodeResourceChange.UpdatedResourcesEntry) + +PubSubMessage = _reflection.GeneratedProtocolMessageType('PubSubMessage', (_message.Message,), { + 'DESCRIPTOR' : _PUBSUBMESSAGE, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PubSubMessage) + }) +_sym_db.RegisterMessage(PubSubMessage) + +ResourceUpdate = _reflection.GeneratedProtocolMessageType('ResourceUpdate', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCEUPDATE, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceUpdate) + }) +_sym_db.RegisterMessage(ResourceUpdate) + +ResourceUsageBroadcastData = _reflection.GeneratedProtocolMessageType('ResourceUsageBroadcastData', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCEUSAGEBROADCASTDATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceUsageBroadcastData) + }) +_sym_db.RegisterMessage(ResourceUsageBroadcastData) + +ResourceDemand = _reflection.GeneratedProtocolMessageType('ResourceDemand', (_message.Message,), { + + 'ShapeEntry' : _reflection.GeneratedProtocolMessageType('ShapeEntry', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCEDEMAND_SHAPEENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceDemand.ShapeEntry) + }) + , + 'DESCRIPTOR' : _RESOURCEDEMAND, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceDemand) + }) +_sym_db.RegisterMessage(ResourceDemand) +_sym_db.RegisterMessage(ResourceDemand.ShapeEntry) + +ResourceLoad = _reflection.GeneratedProtocolMessageType('ResourceLoad', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCELOAD, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceLoad) + }) +_sym_db.RegisterMessage(ResourceLoad) + +ResourcesData = _reflection.GeneratedProtocolMessageType('ResourcesData', (_message.Message,), { + + 'ResourcesAvailableEntry' : _reflection.GeneratedProtocolMessageType('ResourcesAvailableEntry', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCESDATA_RESOURCESAVAILABLEENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourcesData.ResourcesAvailableEntry) + }) + , + + 'ResourcesTotalEntry' : _reflection.GeneratedProtocolMessageType('ResourcesTotalEntry', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCESDATA_RESOURCESTOTALENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourcesData.ResourcesTotalEntry) + }) + , + + 'ResourceLoadEntry' : _reflection.GeneratedProtocolMessageType('ResourceLoadEntry', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCESDATA_RESOURCELOADENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourcesData.ResourceLoadEntry) + }) + , + + 'ResourcesNormalTaskEntry' : _reflection.GeneratedProtocolMessageType('ResourcesNormalTaskEntry', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCESDATA_RESOURCESNORMALTASKENTRY, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourcesData.ResourcesNormalTaskEntry) + }) + , + 'DESCRIPTOR' : _RESOURCESDATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourcesData) + }) +_sym_db.RegisterMessage(ResourcesData) +_sym_db.RegisterMessage(ResourcesData.ResourcesAvailableEntry) +_sym_db.RegisterMessage(ResourcesData.ResourcesTotalEntry) +_sym_db.RegisterMessage(ResourcesData.ResourceLoadEntry) +_sym_db.RegisterMessage(ResourcesData.ResourcesNormalTaskEntry) + +ResourceUsageBatchData = _reflection.GeneratedProtocolMessageType('ResourceUsageBatchData', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCEUSAGEBATCHDATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceUsageBatchData) + }) +_sym_db.RegisterMessage(ResourceUsageBatchData) + +PlacementGroupLoad = _reflection.GeneratedProtocolMessageType('PlacementGroupLoad', (_message.Message,), { + 'DESCRIPTOR' : _PLACEMENTGROUPLOAD, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PlacementGroupLoad) + }) +_sym_db.RegisterMessage(PlacementGroupLoad) + +PlacementGroupStats = _reflection.GeneratedProtocolMessageType('PlacementGroupStats', (_message.Message,), { + 'DESCRIPTOR' : _PLACEMENTGROUPSTATS, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PlacementGroupStats) + }) +_sym_db.RegisterMessage(PlacementGroupStats) + +PlacementGroupTableData = _reflection.GeneratedProtocolMessageType('PlacementGroupTableData', (_message.Message,), { + 'DESCRIPTOR' : _PLACEMENTGROUPTABLEDATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PlacementGroupTableData) + }) +_sym_db.RegisterMessage(PlacementGroupTableData) + +JobTableData = _reflection.GeneratedProtocolMessageType('JobTableData', (_message.Message,), { + 'DESCRIPTOR' : _JOBTABLEDATA, + '__module__' : 'src.ray.protobuf.gcs_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.JobTableData) + }) +_sym_db.RegisterMessage(JobTableData) + +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\n\030io.ray.runtime.generated\370\001\001' + _ACTORTABLEDATA_REQUIREDRESOURCESENTRY._options = None + _ACTORTABLEDATA_REQUIREDRESOURCESENTRY._serialized_options = b'8\001' + _SCHEDULEDATA_SCHEDULEPLANENTRY._options = None + _SCHEDULEDATA_SCHEDULEPLANENTRY._serialized_options = b'8\001' + _AVAILABLERESOURCES_RESOURCESAVAILABLEENTRY._options = None + _AVAILABLERESOURCES_RESOURCESAVAILABLEENTRY._serialized_options = b'8\001' + _GCSNODEINFO_RESOURCESTOTALENTRY._options = None + _GCSNODEINFO_RESOURCESTOTALENTRY._serialized_options = b'8\001' + _GCSNODEINFO_LABELSENTRY._options = None + _GCSNODEINFO_LABELSENTRY._serialized_options = b'8\001' + _JOBSAPIINFO_METADATAENTRY._options = None + _JOBSAPIINFO_METADATAENTRY._serialized_options = b'8\001' + _JOBSAPIINFO_ENTRYPOINTRESOURCESENTRY._options = None + _JOBSAPIINFO_ENTRYPOINTRESOURCESENTRY._serialized_options = b'8\001' + _WORKERTABLEDATA_WORKERINFOENTRY._options = None + _WORKERTABLEDATA_WORKERINFOENTRY._serialized_options = b'8\001' + _RESOURCEMAP_ITEMSENTRY._options = None + _RESOURCEMAP_ITEMSENTRY._serialized_options = b'8\001' + _NODERESOURCECHANGE_UPDATEDRESOURCESENTRY._options = None + _NODERESOURCECHANGE_UPDATEDRESOURCESENTRY._serialized_options = b'8\001' + _RESOURCEDEMAND_SHAPEENTRY._options = None + _RESOURCEDEMAND_SHAPEENTRY._serialized_options = b'8\001' + _RESOURCESDATA_RESOURCESAVAILABLEENTRY._options = None + _RESOURCESDATA_RESOURCESAVAILABLEENTRY._serialized_options = b'8\001' + _RESOURCESDATA_RESOURCESTOTALENTRY._options = None + _RESOURCESDATA_RESOURCESTOTALENTRY._serialized_options = b'8\001' + _RESOURCESDATA_RESOURCELOADENTRY._options = None + _RESOURCESDATA_RESOURCELOADENTRY._serialized_options = b'8\001' + _RESOURCESDATA_RESOURCESNORMALTASKENTRY._options = None + _RESOURCESDATA_RESOURCESNORMALTASKENTRY._serialized_options = b'8\001' + _TABLEPREFIX._serialized_start=8740 + _TABLEPREFIX._serialized_end=9088 + _TABLEPUBSUB._serialized_start=9091 + _TABLEPUBSUB._serialized_end=9409 + _GCSCHANGEMODE._serialized_start=9411 + _GCSCHANGEMODE._serialized_end=9457 + _GCSENTRY._serialized_start=70 + _GCSENTRY._serialized_end=154 + _OBJECTTABLEDATA._serialized_start=156 + _OBJECTTABLEDATA._serialized_end=190 + _ACTORTABLEDATA._serialized_start=193 + _ACTORTABLEDATA._serialized_end=1153 + _ACTORTABLEDATA_REQUIREDRESOURCESENTRY._serialized_start=963 + _ACTORTABLEDATA_REQUIREDRESOURCESENTRY._serialized_end=1019 + _ACTORTABLEDATA_ACTORSTATE._serialized_start=1021 + _ACTORTABLEDATA_ACTORSTATE._serialized_end=1118 + _ERRORTABLEDATA._serialized_start=1155 + _ERRORTABLEDATA._serialized_end=1243 + _SCHEDULEDATA._serialized_start=1246 + _SCHEDULEDATA._serialized_end=1377 + _SCHEDULEDATA_SCHEDULEPLANENTRY._serialized_start=1326 + _SCHEDULEDATA_SCHEDULEPLANENTRY._serialized_end=1377 + _PROFILEEVENTENTRY._serialized_start=1379 + _PROFILEEVENTENTRY._serialized_end=1496 + _PROFILEEVENTS._serialized_start=1499 + _PROFILEEVENTS._serialized_end=1629 + _TASKLOGINFO._serialized_start=1632 + _TASKLOGINFO._serialized_end=1897 + _TASKSTATEUPDATE._serialized_start=1900 + _TASKSTATEUPDATE._serialized_end=2528 + _TASKEVENTS._serialized_start=2531 + _TASKEVENTS._serialized_end=2806 + _TASKEVENTDATA._serialized_start=2809 + _TASKEVENTDATA._serialized_end=2950 + _RESOURCETABLEDATA._serialized_start=2952 + _RESOURCETABLEDATA._serialized_end=2998 + _AVAILABLERESOURCES._serialized_start=3001 + _AVAILABLERESOURCES._serialized_end=3179 + _AVAILABLERESOURCES_RESOURCESAVAILABLEENTRY._serialized_start=3122 + _AVAILABLERESOURCES_RESOURCESAVAILABLEENTRY._serialized_end=3179 + _GCSNODEINFO._serialized_start=3182 + _GCSNODEINFO._serialized_end=3916 + _GCSNODEINFO_RESOURCESTOTALENTRY._serialized_start=3779 + _GCSNODEINFO_RESOURCESTOTALENTRY._serialized_end=3832 + _GCSNODEINFO_LABELSENTRY._serialized_start=3834 + _GCSNODEINFO_LABELSENTRY._serialized_end=3879 + _GCSNODEINFO_GCSNODESTATE._serialized_start=3881 + _GCSNODEINFO_GCSNODESTATE._serialized_end=3916 + _JOBSAPIINFO._serialized_start=3919 + _JOBSAPIINFO._serialized_end=4644 + _JOBSAPIINFO_METADATAENTRY._serialized_start=4364 + _JOBSAPIINFO_METADATAENTRY._serialized_end=4411 + _JOBSAPIINFO_ENTRYPOINTRESOURCESENTRY._serialized_start=4413 + _JOBSAPIINFO_ENTRYPOINTRESOURCESENTRY._serialized_end=4471 + _WORKERTABLEDATA._serialized_start=4647 + _WORKERTABLEDATA._serialized_end=5181 + _WORKERTABLEDATA_WORKERINFOENTRY._serialized_start=5102 + _WORKERTABLEDATA_WORKERINFOENTRY._serialized_end=5151 + _WORKERDELTADATA._serialized_start=5183 + _WORKERDELTADATA._serialized_end=5238 + _RESOURCEMAP._serialized_start=5241 + _RESOURCEMAP._serialized_end=5376 + _RESOURCEMAP_ITEMSENTRY._serialized_start=5304 + _RESOURCEMAP_ITEMSENTRY._serialized_end=5376 + _STOREDCONFIG._serialized_start=5378 + _STOREDCONFIG._serialized_end=5408 + _NODERESOURCECHANGE._serialized_start=5411 + _NODERESOURCECHANGE._serialized_end=5610 + _NODERESOURCECHANGE_UPDATEDRESOURCESENTRY._serialized_start=5555 + _NODERESOURCECHANGE_UPDATEDRESOURCESENTRY._serialized_end=5610 + _PUBSUBMESSAGE._serialized_start=5612 + _PUBSUBMESSAGE._serialized_end=5653 + _RESOURCEUPDATE._serialized_start=5656 + _RESOURCEUPDATE._serialized_end=5786 + _RESOURCEUSAGEBROADCASTDATA._serialized_start=5788 + _RESOURCEUSAGEBROADCASTDATA._serialized_end=5872 + _RESOURCEDEMAND._serialized_start=5875 + _RESOURCEDEMAND._serialized_end=6085 + _RESOURCEDEMAND_SHAPEENTRY._serialized_start=6041 + _RESOURCEDEMAND_SHAPEENTRY._serialized_end=6085 + _RESOURCELOAD._serialized_start=6087 + _RESOURCELOAD._serialized_end=6152 + _RESOURCESDATA._serialized_start=6155 + _RESOURCESDATA._serialized_end=7050 + _RESOURCESDATA_RESOURCESAVAILABLEENTRY._serialized_start=3122 + _RESOURCESDATA_RESOURCESAVAILABLEENTRY._serialized_end=3179 + _RESOURCESDATA_RESOURCESTOTALENTRY._serialized_start=3779 + _RESOURCESDATA_RESOURCESTOTALENTRY._serialized_end=3832 + _RESOURCESDATA_RESOURCELOADENTRY._serialized_start=6933 + _RESOURCESDATA_RESOURCELOADENTRY._serialized_end=6984 + _RESOURCESDATA_RESOURCESNORMALTASKENTRY._serialized_start=6986 + _RESOURCESDATA_RESOURCESNORMALTASKENTRY._serialized_end=7044 + _RESOURCEUSAGEBATCHDATA._serialized_start=7053 + _RESOURCEUSAGEBATCHDATA._serialized_end=7230 + _PLACEMENTGROUPLOAD._serialized_start=7232 + _PLACEMENTGROUPLOAD._serialized_end=7316 + _PLACEMENTGROUPSTATS._serialized_start=7319 + _PLACEMENTGROUPSTATS._serialized_end=7766 + _PLACEMENTGROUPSTATS_SCHEDULINGSTATE._serialized_start=7620 + _PLACEMENTGROUPSTATS_SCHEDULINGSTATE._serialized_end=7766 + _PLACEMENTGROUPTABLEDATA._serialized_start=7769 + _PLACEMENTGROUPTABLEDATA._serialized_end=8401 + _PLACEMENTGROUPTABLEDATA_PLACEMENTGROUPSTATE._serialized_start=8323 + _PLACEMENTGROUPTABLEDATA_PLACEMENTGROUPSTATE._serialized_end=8401 + _JOBTABLEDATA._serialized_start=8404 + _JOBTABLEDATA._serialized_end=8737 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/gcs_pb2_grpc.py b/python/ray/core/generated/gcs_pb2_grpc.py new file mode 100644 index 0000000000000..2daafffebfc81 --- /dev/null +++ b/python/ray/core/generated/gcs_pb2_grpc.py @@ -0,0 +1,4 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + diff --git a/python/ray/core/generated/gcs_service_pb2.py b/python/ray/core/generated/gcs_service_pb2.py new file mode 100644 index 0000000000000..ae87b6414c470 --- /dev/null +++ b/python/ray/core/generated/gcs_service_pb2.py @@ -0,0 +1,1050 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/gcs_service.proto +"""Generated protocol buffer code.""" +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from . import common_pb2 as src_dot_ray_dot_protobuf_dot_common__pb2 +from . import gcs_pb2 as src_dot_ray_dot_protobuf_dot_gcs__pb2 +from . import pubsub_pb2 as src_dot_ray_dot_protobuf_dot_pubsub__pb2 + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"src/ray/protobuf/gcs_service.proto\x12\x07ray.rpc\x1a\x1dsrc/ray/protobuf/common.proto\x1a\x1asrc/ray/protobuf/gcs.proto\x1a\x1dsrc/ray/protobuf/pubsub.proto\"4\n\rAddJobRequest\x12#\n\x04\x64\x61ta\x18\x01 \x01(\x0b\x32\x15.ray.rpc.JobTableData\"1\n\x0b\x41\x64\x64JobReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"(\n\x16MarkJobFinishedRequest\x12\x0e\n\x06job_id\x18\x01 \x01(\x0c\":\n\x14MarkJobFinishedReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"4\n\x14GetAllJobInfoRequest\x12\x12\n\x05limit\x18\x01 \x01(\x05H\x00\x88\x01\x01\x42\x08\n\x06_limit\"f\n\x12GetAllJobInfoReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12,\n\rjob_info_list\x18\x02 \x03(\x0b\x32\x15.ray.rpc.JobTableData\"C\n\x15ReportJobErrorRequest\x12*\n\tjob_error\x18\x01 \x01(\x0b\x32\x17.ray.rpc.ErrorTableData\"9\n\x13ReportJobErrorReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"\x15\n\x13GetNextJobIDRequest\"G\n\x11GetNextJobIDReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x0e\n\x06job_id\x18\x02 \x01(\x05\"5\n\x13GetActorInfoRequest\x12\x10\n\x08\x61\x63tor_id\x18\x01 \x01(\x0c\x12\x0c\n\x04name\x18\x02 \x01(\t\"j\n\x11GetActorInfoReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x31\n\x10\x61\x63tor_table_data\x18\x02 \x01(\x0b\x32\x17.ray.rpc.ActorTableData\"?\n\x18GetNamedActorInfoRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x15\n\rray_namespace\x18\x02 \x01(\t\"\x95\x01\n\x16GetNamedActorInfoReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x31\n\x10\x61\x63tor_table_data\x18\x02 \x01(\x0b\x32\x17.ray.rpc.ActorTableData\x12$\n\ttask_spec\x18\x03 \x01(\x0b\x32\x11.ray.rpc.TaskSpec\"G\n\x16ListNamedActorsRequest\x12\x16\n\x0e\x61ll_namespaces\x18\x01 \x01(\x08\x12\x15\n\rray_namespace\x18\x02 \x01(\t\"n\n\x14ListNamedActorsReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x32\n\x11named_actors_list\x18\x02 \x03(\x0b\x32\x17.ray.rpc.NamedActorInfo\"\xab\x02\n\x16GetAllActorInfoRequest\x12\x16\n\x0eshow_dead_jobs\x18\x01 \x01(\x08\x12\x12\n\x05limit\x18\x02 \x01(\x03H\x00\x88\x01\x01\x12=\n\x07\x66ilters\x18\x03 \x01(\x0b\x32\'.ray.rpc.GetAllActorInfoRequest.FiltersH\x01\x88\x01\x01\x1a\x8f\x01\n\x07\x46ilters\x12\x15\n\x08\x61\x63tor_id\x18\x01 \x01(\x0cH\x00\x88\x01\x01\x12\x13\n\x06job_id\x18\x02 \x01(\x0cH\x01\x88\x01\x01\x12\x36\n\x05state\x18\x03 \x01(\x0e\x32\".ray.rpc.ActorTableData.ActorStateH\x02\x88\x01\x01\x42\x0b\n\t_actor_idB\t\n\x07_job_idB\x08\n\x06_stateB\x08\n\x06_limitB\n\n\x08_filters\"\x92\x01\n\x14GetAllActorInfoReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x31\n\x10\x61\x63tor_table_data\x18\x02 \x03(\x0b\x32\x17.ray.rpc.ActorTableData\x12\r\n\x05total\x18\x03 \x01(\x03\x12\x14\n\x0cnum_filtered\x18\x04 \x01(\x03\"R\n\x16KillActorViaGcsRequest\x12\x10\n\x08\x61\x63tor_id\x18\x01 \x01(\x0c\x12\x12\n\nforce_kill\x18\x02 \x01(\x08\x12\x12\n\nno_restart\x18\x03 \x01(\x08\":\n\x14KillActorViaGcsReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"\x15\n\x13GetClusterIdRequest\"K\n\x11GetClusterIdReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x12\n\ncluster_id\x18\x02 \x01(\x0c\">\n\x13RegisterNodeRequest\x12\'\n\tnode_info\x18\x01 \x01(\x0b\x32\x14.ray.rpc.GcsNodeInfo\"7\n\x11RegisterNodeReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"\x17\n\x15GetAllNodeInfoRequest\"g\n\x13GetAllNodeInfoReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12,\n\x0enode_info_list\x18\x02 \x03(\x0b\x32\x14.ray.rpc.GcsNodeInfo\"+\n\x11\x43heckAliveRequest\x12\x16\n\x0eraylet_address\x18\x01 \x03(\t\"`\n\x0f\x43heckAliveReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x13\n\x0bray_version\x18\x02 \x01(\t\x12\x14\n\x0craylet_alive\x18\x03 \x03(\x08\"\x1a\n\x18GetInternalConfigRequest\"L\n\x16GetInternalConfigReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x0e\n\x06\x63onfig\x18\x02 \x01(\t\"&\n\x13GetResourcesRequest\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\"\xc3\x01\n\x11GetResourcesReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12<\n\tresources\x18\x02 \x03(\x0b\x32).ray.rpc.GetResourcesReply.ResourcesEntry\x1aL\n\x0eResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12)\n\x05value\x18\x02 \x01(\x0b\x32\x1a.ray.rpc.ResourceTableData:\x02\x38\x01\":\n\x14\x44\x65leteResourcesReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"!\n\x1fGetAllAvailableResourcesRequest\"x\n\x1dGetAllAvailableResourcesReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x33\n\x0eresources_list\x18\x02 \x03(\x0b\x32\x1b.ray.rpc.AvailableResources\"G\n\x1aReportResourceUsageRequest\x12)\n\tresources\x18\x01 \x01(\x0b\x32\x16.ray.rpc.ResourcesData\">\n\x18ReportResourceUsageReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"N\n\x1aReportWorkerFailureRequest\x12\x30\n\x0eworker_failure\x18\x01 \x01(\x0b\x32\x18.ray.rpc.WorkerTableData\">\n\x18ReportWorkerFailureReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\")\n\x14GetWorkerInfoRequest\x12\x11\n\tworker_id\x18\x01 \x01(\x0c\"m\n\x12GetWorkerInfoReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x33\n\x11worker_table_data\x18\x02 \x01(\x0b\x32\x18.ray.rpc.WorkerTableData\"7\n\x17GetAllWorkerInfoRequest\x12\x12\n\x05limit\x18\x01 \x01(\x03H\x00\x88\x01\x01\x42\x08\n\x06_limit\"\x7f\n\x15GetAllWorkerInfoReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x33\n\x11worker_table_data\x18\x02 \x03(\x0b\x32\x18.ray.rpc.WorkerTableData\x12\r\n\x05total\x18\x03 \x01(\x03\"E\n\x14\x41\x64\x64WorkerInfoRequest\x12-\n\x0bworker_data\x18\x01 \x01(\x0b\x32\x18.ray.rpc.WorkerTableData\"8\n\x12\x41\x64\x64WorkerInfoReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\":\n\x12\x43reateActorRequest\x12$\n\ttask_spec\x18\x01 \x01(\x0b\x32\x11.ray.rpc.TaskSpec\"\xd9\x01\n\x10\x43reateActorReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\'\n\ractor_address\x18\x02 \x01(\x0b\x32\x10.ray.rpc.Address\x12\x34\n\rborrowed_refs\x18\x03 \x03(\x0b\x32\x1d.ray.rpc.ObjectReferenceCount\x12\x32\n\x0b\x64\x65\x61th_cause\x18\x04 \x01(\x0b\x32\x18.ray.rpc.ActorDeathCauseH\x00\x88\x01\x01\x42\x0e\n\x0c_death_cause\"<\n\x14RegisterActorRequest\x12$\n\ttask_spec\x18\x01 \x01(\x0b\x32\x11.ray.rpc.TaskSpec\"8\n\x12RegisterActorReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"X\n\x1b\x43reatePlacementGroupRequest\x12\x39\n\x14placement_group_spec\x18\x01 \x01(\x0b\x32\x1b.ray.rpc.PlacementGroupSpec\"?\n\x19\x43reatePlacementGroupReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"9\n\x1bRemovePlacementGroupRequest\x12\x1a\n\x12placement_group_id\x18\x01 \x01(\x0c\"?\n\x19RemovePlacementGroupReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"6\n\x18GetPlacementGroupRequest\x12\x1a\n\x12placement_group_id\x18\x01 \x01(\x0c\"\x82\x01\n\x16GetPlacementGroupReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x44\n\x1aplacement_group_table_data\x18\x02 \x01(\x0b\x32 .ray.rpc.PlacementGroupTableData\";\n\x1bGetAllPlacementGroupRequest\x12\x12\n\x05limit\x18\x01 \x01(\x03H\x00\x88\x01\x01\x42\x08\n\x06_limit\"\x94\x01\n\x19GetAllPlacementGroupReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x44\n\x1aplacement_group_table_data\x18\x02 \x03(\x0b\x32 .ray.rpc.PlacementGroupTableData\x12\r\n\x05total\x18\x03 \x01(\x03\"A\n#WaitPlacementGroupUntilReadyRequest\x12\x1a\n\x12placement_group_id\x18\x01 \x01(\x0c\"G\n!WaitPlacementGroupUntilReadyReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"D\n\x1dGetNamedPlacementGroupRequest\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x15\n\rray_namespace\x18\x02 \x01(\t\"\x87\x01\n\x1bGetNamedPlacementGroupReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x44\n\x1aplacement_group_table_data\x18\x02 \x01(\x0b\x32 .ray.rpc.PlacementGroupTableData\" \n\rDrainNodeData\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\"C\n\x10\x44rainNodeRequest\x12/\n\x0f\x64rain_node_data\x18\x01 \x03(\x0b\x32\x16.ray.rpc.DrainNodeData\"\"\n\x0f\x44rainNodeStatus\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\"i\n\x0e\x44rainNodeReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x33\n\x11\x64rain_node_status\x18\x02 \x03(\x0b\x32\x18.ray.rpc.DrainNodeStatus\"6\n\x14InternalKVGetRequest\x12\x0b\n\x03key\x18\x01 \x01(\x0c\x12\x11\n\tnamespace\x18\x02 \x01(\x0c\"G\n\x12InternalKVGetReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\r\n\x05value\x18\x02 \x01(\x0c\"<\n\x19InternalKVMultiGetRequest\x12\x0c\n\x04keys\x18\x01 \x03(\x0c\x12\x11\n\tnamespace\x18\x02 \x01(\x0c\"+\n\rMapFieldEntry\x12\x0b\n\x03key\x18\x01 \x01(\x0c\x12\r\n\x05value\x18\x02 \x01(\x0c\"f\n\x17InternalKVMultiGetReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\'\n\x07results\x18\x02 \x03(\x0b\x32\x16.ray.rpc.MapFieldEntry\"X\n\x14InternalKVPutRequest\x12\x0b\n\x03key\x18\x01 \x01(\x0c\x12\r\n\x05value\x18\x02 \x01(\x0c\x12\x11\n\toverwrite\x18\x03 \x01(\x08\x12\x11\n\tnamespace\x18\x04 \x01(\x0c\"K\n\x12InternalKVPutReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x11\n\tadded_num\x18\x02 \x01(\x05\"M\n\x14InternalKVDelRequest\x12\x0b\n\x03key\x18\x01 \x01(\x0c\x12\x11\n\tnamespace\x18\x02 \x01(\x0c\x12\x15\n\rdel_by_prefix\x18\x03 \x01(\x08\"M\n\x12InternalKVDelReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x13\n\x0b\x64\x65leted_num\x18\x02 \x01(\x05\"9\n\x17InternalKVExistsRequest\x12\x0b\n\x03key\x18\x01 \x01(\x0c\x12\x11\n\tnamespace\x18\x02 \x01(\x0c\"K\n\x15InternalKVExistsReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x0e\n\x06\x65xists\x18\x02 \x01(\x08\":\n\x15InternalKVKeysRequest\x12\x0e\n\x06prefix\x18\x01 \x01(\x0c\x12\x11\n\tnamespace\x18\x02 \x01(\x0c\"J\n\x13InternalKVKeysReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x0f\n\x07results\x18\x02 \x03(\x0c\"<\n\x17PinRuntimeEnvURIRequest\x12\x0b\n\x03uri\x18\x01 \x01(\t\x12\x14\n\x0c\x65xpiration_s\x18\x02 \x01(\x05\";\n\x15PinRuntimeEnvURIReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\">\n\x11GcsPublishRequest\x12)\n\x0cpub_messages\x18\x01 \x03(\x0b\x32\x13.ray.rpc.PubMessage\"5\n\x0fGcsPublishReply\x12\"\n\x06status\x18\x64 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"j\n\x18GcsSubscriberPollRequest\x12\x15\n\rsubscriber_id\x18\x01 \x01(\x0c\x12!\n\x19max_processed_sequence_id\x18\x02 \x01(\x03\x12\x14\n\x0cpublisher_id\x18\x03 \x01(\x0c\"}\n\x16GcsSubscriberPollReply\x12)\n\x0cpub_messages\x18\x01 \x03(\x0b\x32\x13.ray.rpc.PubMessage\x12\x14\n\x0cpublisher_id\x18\x02 \x01(\x0c\x12\"\n\x06status\x18\x64 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"p\n GcsSubscriberCommandBatchRequest\x12\x15\n\rsubscriber_id\x18\x01 \x01(\x0c\x12\"\n\x08\x63ommands\x18\x02 \x03(\x0b\x32\x10.ray.rpc.Command\x12\x11\n\tsender_id\x18\x03 \x01(\x0c\"D\n\x1eGcsSubscriberCommandBatchReply\x12\"\n\x06status\x18\x64 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"\x1c\n\x1aGetAllResourceUsageRequest\"\xac\x01\n\x18GetAllResourceUsageReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12<\n\x13resource_usage_data\x18\x02 \x01(\x0b\x32\x1f.ray.rpc.ResourceUsageBatchData\x12.\n&cluster_full_of_actors_detected_by_gcs\x18\x03 \x01(\x08\"\x19\n\x17GetDrainingNodesRequest\"M\n\x15GetDrainingNodesReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12\x10\n\x08node_ids\x18\x02 \x03(\x0c\"*\n\tGcsStatus\x12\x0c\n\x04\x63ode\x18\x01 \x01(\x05\x12\x0f\n\x07message\x18\x02 \x01(\t\"?\n\x17\x41\x64\x64TaskEventDataRequest\x12$\n\x04\x64\x61ta\x18\x01 \x01(\x0b\x32\x16.ray.rpc.TaskEventData\";\n\x15\x41\x64\x64TaskEventDataReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\"\xab\x02\n\x14GetTaskEventsRequest\x12\x12\n\x05limit\x18\x03 \x01(\x03H\x00\x88\x01\x01\x12;\n\x07\x66ilters\x18\x04 \x01(\x0b\x32%.ray.rpc.GetTaskEventsRequest.FiltersH\x01\x88\x01\x01\x1a\xab\x01\n\x07\x46ilters\x12\x13\n\x06job_id\x18\x01 \x01(\x0cH\x00\x88\x01\x01\x12\x10\n\x08task_ids\x18\x02 \x03(\x0c\x12\x15\n\x08\x61\x63tor_id\x18\x03 \x01(\x0cH\x01\x88\x01\x01\x12\x11\n\x04name\x18\x04 \x01(\tH\x02\x88\x01\x01\x12\x1b\n\x0e\x65xclude_driver\x18\x05 \x01(\x08H\x03\x88\x01\x01\x42\t\n\x07_job_idB\x0b\n\t_actor_idB\x07\n\x05_nameB\x11\n\x0f_exclude_driverB\x08\n\x06_limitB\n\n\x08_filters\"\xb6\x01\n\x12GetTaskEventsReply\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.ray.rpc.GcsStatus\x12+\n\x0e\x65vents_by_task\x18\x02 \x03(\x0b\x32\x13.ray.rpc.TaskEvents\x12\'\n\x1fnum_profile_task_events_dropped\x18\x03 \x01(\x05\x12&\n\x1enum_status_task_events_dropped\x18\x04 \x01(\x05*C\n\x15GcsServiceFailureType\x12\x12\n\x0eRPC_DISCONNECT\x10\x00\x12\x16\n\x12GCS_SERVER_RESTART\x10\x01\x32\x85\x03\n\x11JobInfoGcsService\x12\x36\n\x06\x41\x64\x64Job\x12\x16.ray.rpc.AddJobRequest\x1a\x14.ray.rpc.AddJobReply\x12Q\n\x0fMarkJobFinished\x12\x1f.ray.rpc.MarkJobFinishedRequest\x1a\x1d.ray.rpc.MarkJobFinishedReply\x12K\n\rGetAllJobInfo\x12\x1d.ray.rpc.GetAllJobInfoRequest\x1a\x1b.ray.rpc.GetAllJobInfoReply\x12N\n\x0eReportJobError\x12\x1e.ray.rpc.ReportJobErrorRequest\x1a\x1c.ray.rpc.ReportJobErrorReply\x12H\n\x0cGetNextJobID\x12\x1c.ray.rpc.GetNextJobIDRequest\x1a\x1a.ray.rpc.GetNextJobIDReply2\xc5\x04\n\x13\x41\x63torInfoGcsService\x12K\n\rRegisterActor\x12\x1d.ray.rpc.RegisterActorRequest\x1a\x1b.ray.rpc.RegisterActorReply\x12\x45\n\x0b\x43reateActor\x12\x1b.ray.rpc.CreateActorRequest\x1a\x19.ray.rpc.CreateActorReply\x12H\n\x0cGetActorInfo\x12\x1c.ray.rpc.GetActorInfoRequest\x1a\x1a.ray.rpc.GetActorInfoReply\x12W\n\x11GetNamedActorInfo\x12!.ray.rpc.GetNamedActorInfoRequest\x1a\x1f.ray.rpc.GetNamedActorInfoReply\x12Q\n\x0fListNamedActors\x12\x1f.ray.rpc.ListNamedActorsRequest\x1a\x1d.ray.rpc.ListNamedActorsReply\x12Q\n\x0fGetAllActorInfo\x12\x1f.ray.rpc.GetAllActorInfoRequest\x1a\x1d.ray.rpc.GetAllActorInfoReply\x12Q\n\x0fKillActorViaGcs\x12\x1f.ray.rpc.KillActorViaGcsRequest\x1a\x1d.ray.rpc.KillActorViaGcsReply2\xe5\x02\n\x14WorkerInfoGcsService\x12]\n\x13ReportWorkerFailure\x12#.ray.rpc.ReportWorkerFailureRequest\x1a!.ray.rpc.ReportWorkerFailureReply\x12K\n\rGetWorkerInfo\x12\x1d.ray.rpc.GetWorkerInfoRequest\x1a\x1b.ray.rpc.GetWorkerInfoReply\x12T\n\x10GetAllWorkerInfo\x12 .ray.rpc.GetAllWorkerInfoRequest\x1a\x1e.ray.rpc.GetAllWorkerInfoReply\x12K\n\rAddWorkerInfo\x12\x1d.ray.rpc.AddWorkerInfoRequest\x1a\x1b.ray.rpc.AddWorkerInfoReply2\xff\x04\n\x1cPlacementGroupInfoGcsService\x12`\n\x14\x43reatePlacementGroup\x12$.ray.rpc.CreatePlacementGroupRequest\x1a\".ray.rpc.CreatePlacementGroupReply\x12`\n\x14RemovePlacementGroup\x12$.ray.rpc.RemovePlacementGroupRequest\x1a\".ray.rpc.RemovePlacementGroupReply\x12W\n\x11GetPlacementGroup\x12!.ray.rpc.GetPlacementGroupRequest\x1a\x1f.ray.rpc.GetPlacementGroupReply\x12\x66\n\x16GetNamedPlacementGroup\x12&.ray.rpc.GetNamedPlacementGroupRequest\x1a$.ray.rpc.GetNamedPlacementGroupReply\x12`\n\x14GetAllPlacementGroup\x12$.ray.rpc.GetAllPlacementGroupRequest\x1a\".ray.rpc.GetAllPlacementGroupReply\x12x\n\x1cWaitPlacementGroupUntilReady\x12,.ray.rpc.WaitPlacementGroupUntilReadyRequest\x1a*.ray.rpc.WaitPlacementGroupUntilReadyReply2\xff\x03\n\x14InternalKVGcsService\x12K\n\rInternalKVGet\x12\x1d.ray.rpc.InternalKVGetRequest\x1a\x1b.ray.rpc.InternalKVGetReply\x12Z\n\x12InternalKVMultiGet\x12\".ray.rpc.InternalKVMultiGetRequest\x1a .ray.rpc.InternalKVMultiGetReply\x12K\n\rInternalKVPut\x12\x1d.ray.rpc.InternalKVPutRequest\x1a\x1b.ray.rpc.InternalKVPutReply\x12K\n\rInternalKVDel\x12\x1d.ray.rpc.InternalKVDelRequest\x1a\x1b.ray.rpc.InternalKVDelReply\x12T\n\x10InternalKVExists\x12 .ray.rpc.InternalKVExistsRequest\x1a\x1e.ray.rpc.InternalKVExistsReply\x12N\n\x0eInternalKVKeys\x12\x1e.ray.rpc.InternalKVKeysRequest\x1a\x1c.ray.rpc.InternalKVKeysReply2l\n\x14RuntimeEnvGcsService\x12T\n\x10PinRuntimeEnvURI\x12 .ray.rpc.PinRuntimeEnvURIRequest\x1a\x1e.ray.rpc.PinRuntimeEnvURIReply2\xa8\x02\n\x18InternalPubSubGcsService\x12\x42\n\nGcsPublish\x12\x1a.ray.rpc.GcsPublishRequest\x1a\x18.ray.rpc.GcsPublishReply\x12W\n\x11GcsSubscriberPoll\x12!.ray.rpc.GcsSubscriberPollRequest\x1a\x1f.ray.rpc.GcsSubscriberPollReply\x12o\n\x19GcsSubscriberCommandBatch\x12).ray.rpc.GcsSubscriberCommandBatchRequest\x1a\'.ray.rpc.GcsSubscriberCommandBatchReply2\xe8\x03\n\x1aNodeResourceInfoGcsService\x12H\n\x0cGetResources\x12\x1c.ray.rpc.GetResourcesRequest\x1a\x1a.ray.rpc.GetResourcesReply\x12l\n\x18GetAllAvailableResources\x12(.ray.rpc.GetAllAvailableResourcesRequest\x1a&.ray.rpc.GetAllAvailableResourcesReply\x12]\n\x13ReportResourceUsage\x12#.ray.rpc.ReportResourceUsageRequest\x1a!.ray.rpc.ReportResourceUsageReply\x12]\n\x13GetAllResourceUsage\x12#.ray.rpc.GetAllResourceUsageRequest\x1a!.ray.rpc.GetAllResourceUsageReply\x12T\n\x10GetDrainingNodes\x12 .ray.rpc.GetDrainingNodesRequest\x1a\x1e.ray.rpc.GetDrainingNodesReply2\xd6\x03\n\x12NodeInfoGcsService\x12H\n\x0cGetClusterId\x12\x1c.ray.rpc.GetClusterIdRequest\x1a\x1a.ray.rpc.GetClusterIdReply\x12H\n\x0cRegisterNode\x12\x1c.ray.rpc.RegisterNodeRequest\x1a\x1a.ray.rpc.RegisterNodeReply\x12?\n\tDrainNode\x12\x19.ray.rpc.DrainNodeRequest\x1a\x17.ray.rpc.DrainNodeReply\x12N\n\x0eGetAllNodeInfo\x12\x1e.ray.rpc.GetAllNodeInfoRequest\x1a\x1c.ray.rpc.GetAllNodeInfoReply\x12W\n\x11GetInternalConfig\x12!.ray.rpc.GetInternalConfigRequest\x1a\x1f.ray.rpc.GetInternalConfigReply\x12\x42\n\nCheckAlive\x12\x1a.ray.rpc.CheckAliveRequest\x1a\x18.ray.rpc.CheckAliveReply2\xb7\x01\n\x12TaskInfoGcsService\x12T\n\x10\x41\x64\x64TaskEventData\x12 .ray.rpc.AddTaskEventDataRequest\x1a\x1e.ray.rpc.AddTaskEventDataReply\x12K\n\rGetTaskEvents\x12\x1d.ray.rpc.GetTaskEventsRequest\x1a\x1b.ray.rpc.GetTaskEventsReplyB\x03\xf8\x01\x01\x62\x06proto3') + +_GCSSERVICEFAILURETYPE = DESCRIPTOR.enum_types_by_name['GcsServiceFailureType'] +GcsServiceFailureType = enum_type_wrapper.EnumTypeWrapper(_GCSSERVICEFAILURETYPE) +RPC_DISCONNECT = 0 +GCS_SERVER_RESTART = 1 + + +_ADDJOBREQUEST = DESCRIPTOR.message_types_by_name['AddJobRequest'] +_ADDJOBREPLY = DESCRIPTOR.message_types_by_name['AddJobReply'] +_MARKJOBFINISHEDREQUEST = DESCRIPTOR.message_types_by_name['MarkJobFinishedRequest'] +_MARKJOBFINISHEDREPLY = DESCRIPTOR.message_types_by_name['MarkJobFinishedReply'] +_GETALLJOBINFOREQUEST = DESCRIPTOR.message_types_by_name['GetAllJobInfoRequest'] +_GETALLJOBINFOREPLY = DESCRIPTOR.message_types_by_name['GetAllJobInfoReply'] +_REPORTJOBERRORREQUEST = DESCRIPTOR.message_types_by_name['ReportJobErrorRequest'] +_REPORTJOBERRORREPLY = DESCRIPTOR.message_types_by_name['ReportJobErrorReply'] +_GETNEXTJOBIDREQUEST = DESCRIPTOR.message_types_by_name['GetNextJobIDRequest'] +_GETNEXTJOBIDREPLY = DESCRIPTOR.message_types_by_name['GetNextJobIDReply'] +_GETACTORINFOREQUEST = DESCRIPTOR.message_types_by_name['GetActorInfoRequest'] +_GETACTORINFOREPLY = DESCRIPTOR.message_types_by_name['GetActorInfoReply'] +_GETNAMEDACTORINFOREQUEST = DESCRIPTOR.message_types_by_name['GetNamedActorInfoRequest'] +_GETNAMEDACTORINFOREPLY = DESCRIPTOR.message_types_by_name['GetNamedActorInfoReply'] +_LISTNAMEDACTORSREQUEST = DESCRIPTOR.message_types_by_name['ListNamedActorsRequest'] +_LISTNAMEDACTORSREPLY = DESCRIPTOR.message_types_by_name['ListNamedActorsReply'] +_GETALLACTORINFOREQUEST = DESCRIPTOR.message_types_by_name['GetAllActorInfoRequest'] +_GETALLACTORINFOREQUEST_FILTERS = _GETALLACTORINFOREQUEST.nested_types_by_name['Filters'] +_GETALLACTORINFOREPLY = DESCRIPTOR.message_types_by_name['GetAllActorInfoReply'] +_KILLACTORVIAGCSREQUEST = DESCRIPTOR.message_types_by_name['KillActorViaGcsRequest'] +_KILLACTORVIAGCSREPLY = DESCRIPTOR.message_types_by_name['KillActorViaGcsReply'] +_GETCLUSTERIDREQUEST = DESCRIPTOR.message_types_by_name['GetClusterIdRequest'] +_GETCLUSTERIDREPLY = DESCRIPTOR.message_types_by_name['GetClusterIdReply'] +_REGISTERNODEREQUEST = DESCRIPTOR.message_types_by_name['RegisterNodeRequest'] +_REGISTERNODEREPLY = DESCRIPTOR.message_types_by_name['RegisterNodeReply'] +_GETALLNODEINFOREQUEST = DESCRIPTOR.message_types_by_name['GetAllNodeInfoRequest'] +_GETALLNODEINFOREPLY = DESCRIPTOR.message_types_by_name['GetAllNodeInfoReply'] +_CHECKALIVEREQUEST = DESCRIPTOR.message_types_by_name['CheckAliveRequest'] +_CHECKALIVEREPLY = DESCRIPTOR.message_types_by_name['CheckAliveReply'] +_GETINTERNALCONFIGREQUEST = DESCRIPTOR.message_types_by_name['GetInternalConfigRequest'] +_GETINTERNALCONFIGREPLY = DESCRIPTOR.message_types_by_name['GetInternalConfigReply'] +_GETRESOURCESREQUEST = DESCRIPTOR.message_types_by_name['GetResourcesRequest'] +_GETRESOURCESREPLY = DESCRIPTOR.message_types_by_name['GetResourcesReply'] +_GETRESOURCESREPLY_RESOURCESENTRY = _GETRESOURCESREPLY.nested_types_by_name['ResourcesEntry'] +_DELETERESOURCESREPLY = DESCRIPTOR.message_types_by_name['DeleteResourcesReply'] +_GETALLAVAILABLERESOURCESREQUEST = DESCRIPTOR.message_types_by_name['GetAllAvailableResourcesRequest'] +_GETALLAVAILABLERESOURCESREPLY = DESCRIPTOR.message_types_by_name['GetAllAvailableResourcesReply'] +_REPORTRESOURCEUSAGEREQUEST = DESCRIPTOR.message_types_by_name['ReportResourceUsageRequest'] +_REPORTRESOURCEUSAGEREPLY = DESCRIPTOR.message_types_by_name['ReportResourceUsageReply'] +_REPORTWORKERFAILUREREQUEST = DESCRIPTOR.message_types_by_name['ReportWorkerFailureRequest'] +_REPORTWORKERFAILUREREPLY = DESCRIPTOR.message_types_by_name['ReportWorkerFailureReply'] +_GETWORKERINFOREQUEST = DESCRIPTOR.message_types_by_name['GetWorkerInfoRequest'] +_GETWORKERINFOREPLY = DESCRIPTOR.message_types_by_name['GetWorkerInfoReply'] +_GETALLWORKERINFOREQUEST = DESCRIPTOR.message_types_by_name['GetAllWorkerInfoRequest'] +_GETALLWORKERINFOREPLY = DESCRIPTOR.message_types_by_name['GetAllWorkerInfoReply'] +_ADDWORKERINFOREQUEST = DESCRIPTOR.message_types_by_name['AddWorkerInfoRequest'] +_ADDWORKERINFOREPLY = DESCRIPTOR.message_types_by_name['AddWorkerInfoReply'] +_CREATEACTORREQUEST = DESCRIPTOR.message_types_by_name['CreateActorRequest'] +_CREATEACTORREPLY = DESCRIPTOR.message_types_by_name['CreateActorReply'] +_REGISTERACTORREQUEST = DESCRIPTOR.message_types_by_name['RegisterActorRequest'] +_REGISTERACTORREPLY = DESCRIPTOR.message_types_by_name['RegisterActorReply'] +_CREATEPLACEMENTGROUPREQUEST = DESCRIPTOR.message_types_by_name['CreatePlacementGroupRequest'] +_CREATEPLACEMENTGROUPREPLY = DESCRIPTOR.message_types_by_name['CreatePlacementGroupReply'] +_REMOVEPLACEMENTGROUPREQUEST = DESCRIPTOR.message_types_by_name['RemovePlacementGroupRequest'] +_REMOVEPLACEMENTGROUPREPLY = DESCRIPTOR.message_types_by_name['RemovePlacementGroupReply'] +_GETPLACEMENTGROUPREQUEST = DESCRIPTOR.message_types_by_name['GetPlacementGroupRequest'] +_GETPLACEMENTGROUPREPLY = DESCRIPTOR.message_types_by_name['GetPlacementGroupReply'] +_GETALLPLACEMENTGROUPREQUEST = DESCRIPTOR.message_types_by_name['GetAllPlacementGroupRequest'] +_GETALLPLACEMENTGROUPREPLY = DESCRIPTOR.message_types_by_name['GetAllPlacementGroupReply'] +_WAITPLACEMENTGROUPUNTILREADYREQUEST = DESCRIPTOR.message_types_by_name['WaitPlacementGroupUntilReadyRequest'] +_WAITPLACEMENTGROUPUNTILREADYREPLY = DESCRIPTOR.message_types_by_name['WaitPlacementGroupUntilReadyReply'] +_GETNAMEDPLACEMENTGROUPREQUEST = DESCRIPTOR.message_types_by_name['GetNamedPlacementGroupRequest'] +_GETNAMEDPLACEMENTGROUPREPLY = DESCRIPTOR.message_types_by_name['GetNamedPlacementGroupReply'] +_DRAINNODEDATA = DESCRIPTOR.message_types_by_name['DrainNodeData'] +_DRAINNODEREQUEST = DESCRIPTOR.message_types_by_name['DrainNodeRequest'] +_DRAINNODESTATUS = DESCRIPTOR.message_types_by_name['DrainNodeStatus'] +_DRAINNODEREPLY = DESCRIPTOR.message_types_by_name['DrainNodeReply'] +_INTERNALKVGETREQUEST = DESCRIPTOR.message_types_by_name['InternalKVGetRequest'] +_INTERNALKVGETREPLY = DESCRIPTOR.message_types_by_name['InternalKVGetReply'] +_INTERNALKVMULTIGETREQUEST = DESCRIPTOR.message_types_by_name['InternalKVMultiGetRequest'] +_MAPFIELDENTRY = DESCRIPTOR.message_types_by_name['MapFieldEntry'] +_INTERNALKVMULTIGETREPLY = DESCRIPTOR.message_types_by_name['InternalKVMultiGetReply'] +_INTERNALKVPUTREQUEST = DESCRIPTOR.message_types_by_name['InternalKVPutRequest'] +_INTERNALKVPUTREPLY = DESCRIPTOR.message_types_by_name['InternalKVPutReply'] +_INTERNALKVDELREQUEST = DESCRIPTOR.message_types_by_name['InternalKVDelRequest'] +_INTERNALKVDELREPLY = DESCRIPTOR.message_types_by_name['InternalKVDelReply'] +_INTERNALKVEXISTSREQUEST = DESCRIPTOR.message_types_by_name['InternalKVExistsRequest'] +_INTERNALKVEXISTSREPLY = DESCRIPTOR.message_types_by_name['InternalKVExistsReply'] +_INTERNALKVKEYSREQUEST = DESCRIPTOR.message_types_by_name['InternalKVKeysRequest'] +_INTERNALKVKEYSREPLY = DESCRIPTOR.message_types_by_name['InternalKVKeysReply'] +_PINRUNTIMEENVURIREQUEST = DESCRIPTOR.message_types_by_name['PinRuntimeEnvURIRequest'] +_PINRUNTIMEENVURIREPLY = DESCRIPTOR.message_types_by_name['PinRuntimeEnvURIReply'] +_GCSPUBLISHREQUEST = DESCRIPTOR.message_types_by_name['GcsPublishRequest'] +_GCSPUBLISHREPLY = DESCRIPTOR.message_types_by_name['GcsPublishReply'] +_GCSSUBSCRIBERPOLLREQUEST = DESCRIPTOR.message_types_by_name['GcsSubscriberPollRequest'] +_GCSSUBSCRIBERPOLLREPLY = DESCRIPTOR.message_types_by_name['GcsSubscriberPollReply'] +_GCSSUBSCRIBERCOMMANDBATCHREQUEST = DESCRIPTOR.message_types_by_name['GcsSubscriberCommandBatchRequest'] +_GCSSUBSCRIBERCOMMANDBATCHREPLY = DESCRIPTOR.message_types_by_name['GcsSubscriberCommandBatchReply'] +_GETALLRESOURCEUSAGEREQUEST = DESCRIPTOR.message_types_by_name['GetAllResourceUsageRequest'] +_GETALLRESOURCEUSAGEREPLY = DESCRIPTOR.message_types_by_name['GetAllResourceUsageReply'] +_GETDRAININGNODESREQUEST = DESCRIPTOR.message_types_by_name['GetDrainingNodesRequest'] +_GETDRAININGNODESREPLY = DESCRIPTOR.message_types_by_name['GetDrainingNodesReply'] +_GCSSTATUS = DESCRIPTOR.message_types_by_name['GcsStatus'] +_ADDTASKEVENTDATAREQUEST = DESCRIPTOR.message_types_by_name['AddTaskEventDataRequest'] +_ADDTASKEVENTDATAREPLY = DESCRIPTOR.message_types_by_name['AddTaskEventDataReply'] +_GETTASKEVENTSREQUEST = DESCRIPTOR.message_types_by_name['GetTaskEventsRequest'] +_GETTASKEVENTSREQUEST_FILTERS = _GETTASKEVENTSREQUEST.nested_types_by_name['Filters'] +_GETTASKEVENTSREPLY = DESCRIPTOR.message_types_by_name['GetTaskEventsReply'] +AddJobRequest = _reflection.GeneratedProtocolMessageType('AddJobRequest', (_message.Message,), { + 'DESCRIPTOR' : _ADDJOBREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.AddJobRequest) + }) +_sym_db.RegisterMessage(AddJobRequest) + +AddJobReply = _reflection.GeneratedProtocolMessageType('AddJobReply', (_message.Message,), { + 'DESCRIPTOR' : _ADDJOBREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.AddJobReply) + }) +_sym_db.RegisterMessage(AddJobReply) + +MarkJobFinishedRequest = _reflection.GeneratedProtocolMessageType('MarkJobFinishedRequest', (_message.Message,), { + 'DESCRIPTOR' : _MARKJOBFINISHEDREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.MarkJobFinishedRequest) + }) +_sym_db.RegisterMessage(MarkJobFinishedRequest) + +MarkJobFinishedReply = _reflection.GeneratedProtocolMessageType('MarkJobFinishedReply', (_message.Message,), { + 'DESCRIPTOR' : _MARKJOBFINISHEDREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.MarkJobFinishedReply) + }) +_sym_db.RegisterMessage(MarkJobFinishedReply) + +GetAllJobInfoRequest = _reflection.GeneratedProtocolMessageType('GetAllJobInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETALLJOBINFOREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllJobInfoRequest) + }) +_sym_db.RegisterMessage(GetAllJobInfoRequest) + +GetAllJobInfoReply = _reflection.GeneratedProtocolMessageType('GetAllJobInfoReply', (_message.Message,), { + 'DESCRIPTOR' : _GETALLJOBINFOREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllJobInfoReply) + }) +_sym_db.RegisterMessage(GetAllJobInfoReply) + +ReportJobErrorRequest = _reflection.GeneratedProtocolMessageType('ReportJobErrorRequest', (_message.Message,), { + 'DESCRIPTOR' : _REPORTJOBERRORREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportJobErrorRequest) + }) +_sym_db.RegisterMessage(ReportJobErrorRequest) + +ReportJobErrorReply = _reflection.GeneratedProtocolMessageType('ReportJobErrorReply', (_message.Message,), { + 'DESCRIPTOR' : _REPORTJOBERRORREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportJobErrorReply) + }) +_sym_db.RegisterMessage(ReportJobErrorReply) + +GetNextJobIDRequest = _reflection.GeneratedProtocolMessageType('GetNextJobIDRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETNEXTJOBIDREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetNextJobIDRequest) + }) +_sym_db.RegisterMessage(GetNextJobIDRequest) + +GetNextJobIDReply = _reflection.GeneratedProtocolMessageType('GetNextJobIDReply', (_message.Message,), { + 'DESCRIPTOR' : _GETNEXTJOBIDREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetNextJobIDReply) + }) +_sym_db.RegisterMessage(GetNextJobIDReply) + +GetActorInfoRequest = _reflection.GeneratedProtocolMessageType('GetActorInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETACTORINFOREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetActorInfoRequest) + }) +_sym_db.RegisterMessage(GetActorInfoRequest) + +GetActorInfoReply = _reflection.GeneratedProtocolMessageType('GetActorInfoReply', (_message.Message,), { + 'DESCRIPTOR' : _GETACTORINFOREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetActorInfoReply) + }) +_sym_db.RegisterMessage(GetActorInfoReply) + +GetNamedActorInfoRequest = _reflection.GeneratedProtocolMessageType('GetNamedActorInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETNAMEDACTORINFOREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetNamedActorInfoRequest) + }) +_sym_db.RegisterMessage(GetNamedActorInfoRequest) + +GetNamedActorInfoReply = _reflection.GeneratedProtocolMessageType('GetNamedActorInfoReply', (_message.Message,), { + 'DESCRIPTOR' : _GETNAMEDACTORINFOREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetNamedActorInfoReply) + }) +_sym_db.RegisterMessage(GetNamedActorInfoReply) + +ListNamedActorsRequest = _reflection.GeneratedProtocolMessageType('ListNamedActorsRequest', (_message.Message,), { + 'DESCRIPTOR' : _LISTNAMEDACTORSREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ListNamedActorsRequest) + }) +_sym_db.RegisterMessage(ListNamedActorsRequest) + +ListNamedActorsReply = _reflection.GeneratedProtocolMessageType('ListNamedActorsReply', (_message.Message,), { + 'DESCRIPTOR' : _LISTNAMEDACTORSREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ListNamedActorsReply) + }) +_sym_db.RegisterMessage(ListNamedActorsReply) + +GetAllActorInfoRequest = _reflection.GeneratedProtocolMessageType('GetAllActorInfoRequest', (_message.Message,), { + + 'Filters' : _reflection.GeneratedProtocolMessageType('Filters', (_message.Message,), { + 'DESCRIPTOR' : _GETALLACTORINFOREQUEST_FILTERS, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllActorInfoRequest.Filters) + }) + , + 'DESCRIPTOR' : _GETALLACTORINFOREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllActorInfoRequest) + }) +_sym_db.RegisterMessage(GetAllActorInfoRequest) +_sym_db.RegisterMessage(GetAllActorInfoRequest.Filters) + +GetAllActorInfoReply = _reflection.GeneratedProtocolMessageType('GetAllActorInfoReply', (_message.Message,), { + 'DESCRIPTOR' : _GETALLACTORINFOREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllActorInfoReply) + }) +_sym_db.RegisterMessage(GetAllActorInfoReply) + +KillActorViaGcsRequest = _reflection.GeneratedProtocolMessageType('KillActorViaGcsRequest', (_message.Message,), { + 'DESCRIPTOR' : _KILLACTORVIAGCSREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KillActorViaGcsRequest) + }) +_sym_db.RegisterMessage(KillActorViaGcsRequest) + +KillActorViaGcsReply = _reflection.GeneratedProtocolMessageType('KillActorViaGcsReply', (_message.Message,), { + 'DESCRIPTOR' : _KILLACTORVIAGCSREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KillActorViaGcsReply) + }) +_sym_db.RegisterMessage(KillActorViaGcsReply) + +GetClusterIdRequest = _reflection.GeneratedProtocolMessageType('GetClusterIdRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETCLUSTERIDREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetClusterIdRequest) + }) +_sym_db.RegisterMessage(GetClusterIdRequest) + +GetClusterIdReply = _reflection.GeneratedProtocolMessageType('GetClusterIdReply', (_message.Message,), { + 'DESCRIPTOR' : _GETCLUSTERIDREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetClusterIdReply) + }) +_sym_db.RegisterMessage(GetClusterIdReply) + +RegisterNodeRequest = _reflection.GeneratedProtocolMessageType('RegisterNodeRequest', (_message.Message,), { + 'DESCRIPTOR' : _REGISTERNODEREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RegisterNodeRequest) + }) +_sym_db.RegisterMessage(RegisterNodeRequest) + +RegisterNodeReply = _reflection.GeneratedProtocolMessageType('RegisterNodeReply', (_message.Message,), { + 'DESCRIPTOR' : _REGISTERNODEREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RegisterNodeReply) + }) +_sym_db.RegisterMessage(RegisterNodeReply) + +GetAllNodeInfoRequest = _reflection.GeneratedProtocolMessageType('GetAllNodeInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETALLNODEINFOREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllNodeInfoRequest) + }) +_sym_db.RegisterMessage(GetAllNodeInfoRequest) + +GetAllNodeInfoReply = _reflection.GeneratedProtocolMessageType('GetAllNodeInfoReply', (_message.Message,), { + 'DESCRIPTOR' : _GETALLNODEINFOREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllNodeInfoReply) + }) +_sym_db.RegisterMessage(GetAllNodeInfoReply) + +CheckAliveRequest = _reflection.GeneratedProtocolMessageType('CheckAliveRequest', (_message.Message,), { + 'DESCRIPTOR' : _CHECKALIVEREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CheckAliveRequest) + }) +_sym_db.RegisterMessage(CheckAliveRequest) + +CheckAliveReply = _reflection.GeneratedProtocolMessageType('CheckAliveReply', (_message.Message,), { + 'DESCRIPTOR' : _CHECKALIVEREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CheckAliveReply) + }) +_sym_db.RegisterMessage(CheckAliveReply) + +GetInternalConfigRequest = _reflection.GeneratedProtocolMessageType('GetInternalConfigRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETINTERNALCONFIGREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetInternalConfigRequest) + }) +_sym_db.RegisterMessage(GetInternalConfigRequest) + +GetInternalConfigReply = _reflection.GeneratedProtocolMessageType('GetInternalConfigReply', (_message.Message,), { + 'DESCRIPTOR' : _GETINTERNALCONFIGREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetInternalConfigReply) + }) +_sym_db.RegisterMessage(GetInternalConfigReply) + +GetResourcesRequest = _reflection.GeneratedProtocolMessageType('GetResourcesRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETRESOURCESREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetResourcesRequest) + }) +_sym_db.RegisterMessage(GetResourcesRequest) + +GetResourcesReply = _reflection.GeneratedProtocolMessageType('GetResourcesReply', (_message.Message,), { + + 'ResourcesEntry' : _reflection.GeneratedProtocolMessageType('ResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _GETRESOURCESREPLY_RESOURCESENTRY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetResourcesReply.ResourcesEntry) + }) + , + 'DESCRIPTOR' : _GETRESOURCESREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetResourcesReply) + }) +_sym_db.RegisterMessage(GetResourcesReply) +_sym_db.RegisterMessage(GetResourcesReply.ResourcesEntry) + +DeleteResourcesReply = _reflection.GeneratedProtocolMessageType('DeleteResourcesReply', (_message.Message,), { + 'DESCRIPTOR' : _DELETERESOURCESREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DeleteResourcesReply) + }) +_sym_db.RegisterMessage(DeleteResourcesReply) + +GetAllAvailableResourcesRequest = _reflection.GeneratedProtocolMessageType('GetAllAvailableResourcesRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETALLAVAILABLERESOURCESREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllAvailableResourcesRequest) + }) +_sym_db.RegisterMessage(GetAllAvailableResourcesRequest) + +GetAllAvailableResourcesReply = _reflection.GeneratedProtocolMessageType('GetAllAvailableResourcesReply', (_message.Message,), { + 'DESCRIPTOR' : _GETALLAVAILABLERESOURCESREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllAvailableResourcesReply) + }) +_sym_db.RegisterMessage(GetAllAvailableResourcesReply) + +ReportResourceUsageRequest = _reflection.GeneratedProtocolMessageType('ReportResourceUsageRequest', (_message.Message,), { + 'DESCRIPTOR' : _REPORTRESOURCEUSAGEREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportResourceUsageRequest) + }) +_sym_db.RegisterMessage(ReportResourceUsageRequest) + +ReportResourceUsageReply = _reflection.GeneratedProtocolMessageType('ReportResourceUsageReply', (_message.Message,), { + 'DESCRIPTOR' : _REPORTRESOURCEUSAGEREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportResourceUsageReply) + }) +_sym_db.RegisterMessage(ReportResourceUsageReply) + +ReportWorkerFailureRequest = _reflection.GeneratedProtocolMessageType('ReportWorkerFailureRequest', (_message.Message,), { + 'DESCRIPTOR' : _REPORTWORKERFAILUREREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportWorkerFailureRequest) + }) +_sym_db.RegisterMessage(ReportWorkerFailureRequest) + +ReportWorkerFailureReply = _reflection.GeneratedProtocolMessageType('ReportWorkerFailureReply', (_message.Message,), { + 'DESCRIPTOR' : _REPORTWORKERFAILUREREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportWorkerFailureReply) + }) +_sym_db.RegisterMessage(ReportWorkerFailureReply) + +GetWorkerInfoRequest = _reflection.GeneratedProtocolMessageType('GetWorkerInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETWORKERINFOREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetWorkerInfoRequest) + }) +_sym_db.RegisterMessage(GetWorkerInfoRequest) + +GetWorkerInfoReply = _reflection.GeneratedProtocolMessageType('GetWorkerInfoReply', (_message.Message,), { + 'DESCRIPTOR' : _GETWORKERINFOREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetWorkerInfoReply) + }) +_sym_db.RegisterMessage(GetWorkerInfoReply) + +GetAllWorkerInfoRequest = _reflection.GeneratedProtocolMessageType('GetAllWorkerInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETALLWORKERINFOREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllWorkerInfoRequest) + }) +_sym_db.RegisterMessage(GetAllWorkerInfoRequest) + +GetAllWorkerInfoReply = _reflection.GeneratedProtocolMessageType('GetAllWorkerInfoReply', (_message.Message,), { + 'DESCRIPTOR' : _GETALLWORKERINFOREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllWorkerInfoReply) + }) +_sym_db.RegisterMessage(GetAllWorkerInfoReply) + +AddWorkerInfoRequest = _reflection.GeneratedProtocolMessageType('AddWorkerInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _ADDWORKERINFOREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.AddWorkerInfoRequest) + }) +_sym_db.RegisterMessage(AddWorkerInfoRequest) + +AddWorkerInfoReply = _reflection.GeneratedProtocolMessageType('AddWorkerInfoReply', (_message.Message,), { + 'DESCRIPTOR' : _ADDWORKERINFOREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.AddWorkerInfoReply) + }) +_sym_db.RegisterMessage(AddWorkerInfoReply) + +CreateActorRequest = _reflection.GeneratedProtocolMessageType('CreateActorRequest', (_message.Message,), { + 'DESCRIPTOR' : _CREATEACTORREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CreateActorRequest) + }) +_sym_db.RegisterMessage(CreateActorRequest) + +CreateActorReply = _reflection.GeneratedProtocolMessageType('CreateActorReply', (_message.Message,), { + 'DESCRIPTOR' : _CREATEACTORREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CreateActorReply) + }) +_sym_db.RegisterMessage(CreateActorReply) + +RegisterActorRequest = _reflection.GeneratedProtocolMessageType('RegisterActorRequest', (_message.Message,), { + 'DESCRIPTOR' : _REGISTERACTORREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RegisterActorRequest) + }) +_sym_db.RegisterMessage(RegisterActorRequest) + +RegisterActorReply = _reflection.GeneratedProtocolMessageType('RegisterActorReply', (_message.Message,), { + 'DESCRIPTOR' : _REGISTERACTORREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RegisterActorReply) + }) +_sym_db.RegisterMessage(RegisterActorReply) + +CreatePlacementGroupRequest = _reflection.GeneratedProtocolMessageType('CreatePlacementGroupRequest', (_message.Message,), { + 'DESCRIPTOR' : _CREATEPLACEMENTGROUPREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CreatePlacementGroupRequest) + }) +_sym_db.RegisterMessage(CreatePlacementGroupRequest) + +CreatePlacementGroupReply = _reflection.GeneratedProtocolMessageType('CreatePlacementGroupReply', (_message.Message,), { + 'DESCRIPTOR' : _CREATEPLACEMENTGROUPREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CreatePlacementGroupReply) + }) +_sym_db.RegisterMessage(CreatePlacementGroupReply) + +RemovePlacementGroupRequest = _reflection.GeneratedProtocolMessageType('RemovePlacementGroupRequest', (_message.Message,), { + 'DESCRIPTOR' : _REMOVEPLACEMENTGROUPREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RemovePlacementGroupRequest) + }) +_sym_db.RegisterMessage(RemovePlacementGroupRequest) + +RemovePlacementGroupReply = _reflection.GeneratedProtocolMessageType('RemovePlacementGroupReply', (_message.Message,), { + 'DESCRIPTOR' : _REMOVEPLACEMENTGROUPREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RemovePlacementGroupReply) + }) +_sym_db.RegisterMessage(RemovePlacementGroupReply) + +GetPlacementGroupRequest = _reflection.GeneratedProtocolMessageType('GetPlacementGroupRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETPLACEMENTGROUPREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetPlacementGroupRequest) + }) +_sym_db.RegisterMessage(GetPlacementGroupRequest) + +GetPlacementGroupReply = _reflection.GeneratedProtocolMessageType('GetPlacementGroupReply', (_message.Message,), { + 'DESCRIPTOR' : _GETPLACEMENTGROUPREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetPlacementGroupReply) + }) +_sym_db.RegisterMessage(GetPlacementGroupReply) + +GetAllPlacementGroupRequest = _reflection.GeneratedProtocolMessageType('GetAllPlacementGroupRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETALLPLACEMENTGROUPREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllPlacementGroupRequest) + }) +_sym_db.RegisterMessage(GetAllPlacementGroupRequest) + +GetAllPlacementGroupReply = _reflection.GeneratedProtocolMessageType('GetAllPlacementGroupReply', (_message.Message,), { + 'DESCRIPTOR' : _GETALLPLACEMENTGROUPREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllPlacementGroupReply) + }) +_sym_db.RegisterMessage(GetAllPlacementGroupReply) + +WaitPlacementGroupUntilReadyRequest = _reflection.GeneratedProtocolMessageType('WaitPlacementGroupUntilReadyRequest', (_message.Message,), { + 'DESCRIPTOR' : _WAITPLACEMENTGROUPUNTILREADYREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WaitPlacementGroupUntilReadyRequest) + }) +_sym_db.RegisterMessage(WaitPlacementGroupUntilReadyRequest) + +WaitPlacementGroupUntilReadyReply = _reflection.GeneratedProtocolMessageType('WaitPlacementGroupUntilReadyReply', (_message.Message,), { + 'DESCRIPTOR' : _WAITPLACEMENTGROUPUNTILREADYREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WaitPlacementGroupUntilReadyReply) + }) +_sym_db.RegisterMessage(WaitPlacementGroupUntilReadyReply) + +GetNamedPlacementGroupRequest = _reflection.GeneratedProtocolMessageType('GetNamedPlacementGroupRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETNAMEDPLACEMENTGROUPREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetNamedPlacementGroupRequest) + }) +_sym_db.RegisterMessage(GetNamedPlacementGroupRequest) + +GetNamedPlacementGroupReply = _reflection.GeneratedProtocolMessageType('GetNamedPlacementGroupReply', (_message.Message,), { + 'DESCRIPTOR' : _GETNAMEDPLACEMENTGROUPREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetNamedPlacementGroupReply) + }) +_sym_db.RegisterMessage(GetNamedPlacementGroupReply) + +DrainNodeData = _reflection.GeneratedProtocolMessageType('DrainNodeData', (_message.Message,), { + 'DESCRIPTOR' : _DRAINNODEDATA, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DrainNodeData) + }) +_sym_db.RegisterMessage(DrainNodeData) + +DrainNodeRequest = _reflection.GeneratedProtocolMessageType('DrainNodeRequest', (_message.Message,), { + 'DESCRIPTOR' : _DRAINNODEREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DrainNodeRequest) + }) +_sym_db.RegisterMessage(DrainNodeRequest) + +DrainNodeStatus = _reflection.GeneratedProtocolMessageType('DrainNodeStatus', (_message.Message,), { + 'DESCRIPTOR' : _DRAINNODESTATUS, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DrainNodeStatus) + }) +_sym_db.RegisterMessage(DrainNodeStatus) + +DrainNodeReply = _reflection.GeneratedProtocolMessageType('DrainNodeReply', (_message.Message,), { + 'DESCRIPTOR' : _DRAINNODEREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DrainNodeReply) + }) +_sym_db.RegisterMessage(DrainNodeReply) + +InternalKVGetRequest = _reflection.GeneratedProtocolMessageType('InternalKVGetRequest', (_message.Message,), { + 'DESCRIPTOR' : _INTERNALKVGETREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InternalKVGetRequest) + }) +_sym_db.RegisterMessage(InternalKVGetRequest) + +InternalKVGetReply = _reflection.GeneratedProtocolMessageType('InternalKVGetReply', (_message.Message,), { + 'DESCRIPTOR' : _INTERNALKVGETREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InternalKVGetReply) + }) +_sym_db.RegisterMessage(InternalKVGetReply) + +InternalKVMultiGetRequest = _reflection.GeneratedProtocolMessageType('InternalKVMultiGetRequest', (_message.Message,), { + 'DESCRIPTOR' : _INTERNALKVMULTIGETREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InternalKVMultiGetRequest) + }) +_sym_db.RegisterMessage(InternalKVMultiGetRequest) + +MapFieldEntry = _reflection.GeneratedProtocolMessageType('MapFieldEntry', (_message.Message,), { + 'DESCRIPTOR' : _MAPFIELDENTRY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.MapFieldEntry) + }) +_sym_db.RegisterMessage(MapFieldEntry) + +InternalKVMultiGetReply = _reflection.GeneratedProtocolMessageType('InternalKVMultiGetReply', (_message.Message,), { + 'DESCRIPTOR' : _INTERNALKVMULTIGETREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InternalKVMultiGetReply) + }) +_sym_db.RegisterMessage(InternalKVMultiGetReply) + +InternalKVPutRequest = _reflection.GeneratedProtocolMessageType('InternalKVPutRequest', (_message.Message,), { + 'DESCRIPTOR' : _INTERNALKVPUTREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InternalKVPutRequest) + }) +_sym_db.RegisterMessage(InternalKVPutRequest) + +InternalKVPutReply = _reflection.GeneratedProtocolMessageType('InternalKVPutReply', (_message.Message,), { + 'DESCRIPTOR' : _INTERNALKVPUTREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InternalKVPutReply) + }) +_sym_db.RegisterMessage(InternalKVPutReply) + +InternalKVDelRequest = _reflection.GeneratedProtocolMessageType('InternalKVDelRequest', (_message.Message,), { + 'DESCRIPTOR' : _INTERNALKVDELREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InternalKVDelRequest) + }) +_sym_db.RegisterMessage(InternalKVDelRequest) + +InternalKVDelReply = _reflection.GeneratedProtocolMessageType('InternalKVDelReply', (_message.Message,), { + 'DESCRIPTOR' : _INTERNALKVDELREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InternalKVDelReply) + }) +_sym_db.RegisterMessage(InternalKVDelReply) + +InternalKVExistsRequest = _reflection.GeneratedProtocolMessageType('InternalKVExistsRequest', (_message.Message,), { + 'DESCRIPTOR' : _INTERNALKVEXISTSREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InternalKVExistsRequest) + }) +_sym_db.RegisterMessage(InternalKVExistsRequest) + +InternalKVExistsReply = _reflection.GeneratedProtocolMessageType('InternalKVExistsReply', (_message.Message,), { + 'DESCRIPTOR' : _INTERNALKVEXISTSREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InternalKVExistsReply) + }) +_sym_db.RegisterMessage(InternalKVExistsReply) + +InternalKVKeysRequest = _reflection.GeneratedProtocolMessageType('InternalKVKeysRequest', (_message.Message,), { + 'DESCRIPTOR' : _INTERNALKVKEYSREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InternalKVKeysRequest) + }) +_sym_db.RegisterMessage(InternalKVKeysRequest) + +InternalKVKeysReply = _reflection.GeneratedProtocolMessageType('InternalKVKeysReply', (_message.Message,), { + 'DESCRIPTOR' : _INTERNALKVKEYSREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InternalKVKeysReply) + }) +_sym_db.RegisterMessage(InternalKVKeysReply) + +PinRuntimeEnvURIRequest = _reflection.GeneratedProtocolMessageType('PinRuntimeEnvURIRequest', (_message.Message,), { + 'DESCRIPTOR' : _PINRUNTIMEENVURIREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PinRuntimeEnvURIRequest) + }) +_sym_db.RegisterMessage(PinRuntimeEnvURIRequest) + +PinRuntimeEnvURIReply = _reflection.GeneratedProtocolMessageType('PinRuntimeEnvURIReply', (_message.Message,), { + 'DESCRIPTOR' : _PINRUNTIMEENVURIREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PinRuntimeEnvURIReply) + }) +_sym_db.RegisterMessage(PinRuntimeEnvURIReply) + +GcsPublishRequest = _reflection.GeneratedProtocolMessageType('GcsPublishRequest', (_message.Message,), { + 'DESCRIPTOR' : _GCSPUBLISHREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GcsPublishRequest) + }) +_sym_db.RegisterMessage(GcsPublishRequest) + +GcsPublishReply = _reflection.GeneratedProtocolMessageType('GcsPublishReply', (_message.Message,), { + 'DESCRIPTOR' : _GCSPUBLISHREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GcsPublishReply) + }) +_sym_db.RegisterMessage(GcsPublishReply) + +GcsSubscriberPollRequest = _reflection.GeneratedProtocolMessageType('GcsSubscriberPollRequest', (_message.Message,), { + 'DESCRIPTOR' : _GCSSUBSCRIBERPOLLREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GcsSubscriberPollRequest) + }) +_sym_db.RegisterMessage(GcsSubscriberPollRequest) + +GcsSubscriberPollReply = _reflection.GeneratedProtocolMessageType('GcsSubscriberPollReply', (_message.Message,), { + 'DESCRIPTOR' : _GCSSUBSCRIBERPOLLREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GcsSubscriberPollReply) + }) +_sym_db.RegisterMessage(GcsSubscriberPollReply) + +GcsSubscriberCommandBatchRequest = _reflection.GeneratedProtocolMessageType('GcsSubscriberCommandBatchRequest', (_message.Message,), { + 'DESCRIPTOR' : _GCSSUBSCRIBERCOMMANDBATCHREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GcsSubscriberCommandBatchRequest) + }) +_sym_db.RegisterMessage(GcsSubscriberCommandBatchRequest) + +GcsSubscriberCommandBatchReply = _reflection.GeneratedProtocolMessageType('GcsSubscriberCommandBatchReply', (_message.Message,), { + 'DESCRIPTOR' : _GCSSUBSCRIBERCOMMANDBATCHREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GcsSubscriberCommandBatchReply) + }) +_sym_db.RegisterMessage(GcsSubscriberCommandBatchReply) + +GetAllResourceUsageRequest = _reflection.GeneratedProtocolMessageType('GetAllResourceUsageRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETALLRESOURCEUSAGEREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllResourceUsageRequest) + }) +_sym_db.RegisterMessage(GetAllResourceUsageRequest) + +GetAllResourceUsageReply = _reflection.GeneratedProtocolMessageType('GetAllResourceUsageReply', (_message.Message,), { + 'DESCRIPTOR' : _GETALLRESOURCEUSAGEREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetAllResourceUsageReply) + }) +_sym_db.RegisterMessage(GetAllResourceUsageReply) + +GetDrainingNodesRequest = _reflection.GeneratedProtocolMessageType('GetDrainingNodesRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETDRAININGNODESREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetDrainingNodesRequest) + }) +_sym_db.RegisterMessage(GetDrainingNodesRequest) + +GetDrainingNodesReply = _reflection.GeneratedProtocolMessageType('GetDrainingNodesReply', (_message.Message,), { + 'DESCRIPTOR' : _GETDRAININGNODESREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetDrainingNodesReply) + }) +_sym_db.RegisterMessage(GetDrainingNodesReply) + +GcsStatus = _reflection.GeneratedProtocolMessageType('GcsStatus', (_message.Message,), { + 'DESCRIPTOR' : _GCSSTATUS, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GcsStatus) + }) +_sym_db.RegisterMessage(GcsStatus) + +AddTaskEventDataRequest = _reflection.GeneratedProtocolMessageType('AddTaskEventDataRequest', (_message.Message,), { + 'DESCRIPTOR' : _ADDTASKEVENTDATAREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.AddTaskEventDataRequest) + }) +_sym_db.RegisterMessage(AddTaskEventDataRequest) + +AddTaskEventDataReply = _reflection.GeneratedProtocolMessageType('AddTaskEventDataReply', (_message.Message,), { + 'DESCRIPTOR' : _ADDTASKEVENTDATAREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.AddTaskEventDataReply) + }) +_sym_db.RegisterMessage(AddTaskEventDataReply) + +GetTaskEventsRequest = _reflection.GeneratedProtocolMessageType('GetTaskEventsRequest', (_message.Message,), { + + 'Filters' : _reflection.GeneratedProtocolMessageType('Filters', (_message.Message,), { + 'DESCRIPTOR' : _GETTASKEVENTSREQUEST_FILTERS, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetTaskEventsRequest.Filters) + }) + , + 'DESCRIPTOR' : _GETTASKEVENTSREQUEST, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetTaskEventsRequest) + }) +_sym_db.RegisterMessage(GetTaskEventsRequest) +_sym_db.RegisterMessage(GetTaskEventsRequest.Filters) + +GetTaskEventsReply = _reflection.GeneratedProtocolMessageType('GetTaskEventsReply', (_message.Message,), { + 'DESCRIPTOR' : _GETTASKEVENTSREPLY, + '__module__' : 'src.ray.protobuf.gcs_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetTaskEventsReply) + }) +_sym_db.RegisterMessage(GetTaskEventsReply) + +_JOBINFOGCSSERVICE = DESCRIPTOR.services_by_name['JobInfoGcsService'] +_ACTORINFOGCSSERVICE = DESCRIPTOR.services_by_name['ActorInfoGcsService'] +_WORKERINFOGCSSERVICE = DESCRIPTOR.services_by_name['WorkerInfoGcsService'] +_PLACEMENTGROUPINFOGCSSERVICE = DESCRIPTOR.services_by_name['PlacementGroupInfoGcsService'] +_INTERNALKVGCSSERVICE = DESCRIPTOR.services_by_name['InternalKVGcsService'] +_RUNTIMEENVGCSSERVICE = DESCRIPTOR.services_by_name['RuntimeEnvGcsService'] +_INTERNALPUBSUBGCSSERVICE = DESCRIPTOR.services_by_name['InternalPubSubGcsService'] +_NODERESOURCEINFOGCSSERVICE = DESCRIPTOR.services_by_name['NodeResourceInfoGcsService'] +_NODEINFOGCSSERVICE = DESCRIPTOR.services_by_name['NodeInfoGcsService'] +_TASKINFOGCSSERVICE = DESCRIPTOR.services_by_name['TaskInfoGcsService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _GETRESOURCESREPLY_RESOURCESENTRY._options = None + _GETRESOURCESREPLY_RESOURCESENTRY._serialized_options = b'8\001' + _GCSSERVICEFAILURETYPE._serialized_start=7931 + _GCSSERVICEFAILURETYPE._serialized_end=7998 + _ADDJOBREQUEST._serialized_start=137 + _ADDJOBREQUEST._serialized_end=189 + _ADDJOBREPLY._serialized_start=191 + _ADDJOBREPLY._serialized_end=240 + _MARKJOBFINISHEDREQUEST._serialized_start=242 + _MARKJOBFINISHEDREQUEST._serialized_end=282 + _MARKJOBFINISHEDREPLY._serialized_start=284 + _MARKJOBFINISHEDREPLY._serialized_end=342 + _GETALLJOBINFOREQUEST._serialized_start=344 + _GETALLJOBINFOREQUEST._serialized_end=396 + _GETALLJOBINFOREPLY._serialized_start=398 + _GETALLJOBINFOREPLY._serialized_end=500 + _REPORTJOBERRORREQUEST._serialized_start=502 + _REPORTJOBERRORREQUEST._serialized_end=569 + _REPORTJOBERRORREPLY._serialized_start=571 + _REPORTJOBERRORREPLY._serialized_end=628 + _GETNEXTJOBIDREQUEST._serialized_start=630 + _GETNEXTJOBIDREQUEST._serialized_end=651 + _GETNEXTJOBIDREPLY._serialized_start=653 + _GETNEXTJOBIDREPLY._serialized_end=724 + _GETACTORINFOREQUEST._serialized_start=726 + _GETACTORINFOREQUEST._serialized_end=779 + _GETACTORINFOREPLY._serialized_start=781 + _GETACTORINFOREPLY._serialized_end=887 + _GETNAMEDACTORINFOREQUEST._serialized_start=889 + _GETNAMEDACTORINFOREQUEST._serialized_end=952 + _GETNAMEDACTORINFOREPLY._serialized_start=955 + _GETNAMEDACTORINFOREPLY._serialized_end=1104 + _LISTNAMEDACTORSREQUEST._serialized_start=1106 + _LISTNAMEDACTORSREQUEST._serialized_end=1177 + _LISTNAMEDACTORSREPLY._serialized_start=1179 + _LISTNAMEDACTORSREPLY._serialized_end=1289 + _GETALLACTORINFOREQUEST._serialized_start=1292 + _GETALLACTORINFOREQUEST._serialized_end=1591 + _GETALLACTORINFOREQUEST_FILTERS._serialized_start=1426 + _GETALLACTORINFOREQUEST_FILTERS._serialized_end=1569 + _GETALLACTORINFOREPLY._serialized_start=1594 + _GETALLACTORINFOREPLY._serialized_end=1740 + _KILLACTORVIAGCSREQUEST._serialized_start=1742 + _KILLACTORVIAGCSREQUEST._serialized_end=1824 + _KILLACTORVIAGCSREPLY._serialized_start=1826 + _KILLACTORVIAGCSREPLY._serialized_end=1884 + _GETCLUSTERIDREQUEST._serialized_start=1886 + _GETCLUSTERIDREQUEST._serialized_end=1907 + _GETCLUSTERIDREPLY._serialized_start=1909 + _GETCLUSTERIDREPLY._serialized_end=1984 + _REGISTERNODEREQUEST._serialized_start=1986 + _REGISTERNODEREQUEST._serialized_end=2048 + _REGISTERNODEREPLY._serialized_start=2050 + _REGISTERNODEREPLY._serialized_end=2105 + _GETALLNODEINFOREQUEST._serialized_start=2107 + _GETALLNODEINFOREQUEST._serialized_end=2130 + _GETALLNODEINFOREPLY._serialized_start=2132 + _GETALLNODEINFOREPLY._serialized_end=2235 + _CHECKALIVEREQUEST._serialized_start=2237 + _CHECKALIVEREQUEST._serialized_end=2280 + _CHECKALIVEREPLY._serialized_start=2282 + _CHECKALIVEREPLY._serialized_end=2378 + _GETINTERNALCONFIGREQUEST._serialized_start=2380 + _GETINTERNALCONFIGREQUEST._serialized_end=2406 + _GETINTERNALCONFIGREPLY._serialized_start=2408 + _GETINTERNALCONFIGREPLY._serialized_end=2484 + _GETRESOURCESREQUEST._serialized_start=2486 + _GETRESOURCESREQUEST._serialized_end=2524 + _GETRESOURCESREPLY._serialized_start=2527 + _GETRESOURCESREPLY._serialized_end=2722 + _GETRESOURCESREPLY_RESOURCESENTRY._serialized_start=2646 + _GETRESOURCESREPLY_RESOURCESENTRY._serialized_end=2722 + _DELETERESOURCESREPLY._serialized_start=2724 + _DELETERESOURCESREPLY._serialized_end=2782 + _GETALLAVAILABLERESOURCESREQUEST._serialized_start=2784 + _GETALLAVAILABLERESOURCESREQUEST._serialized_end=2817 + _GETALLAVAILABLERESOURCESREPLY._serialized_start=2819 + _GETALLAVAILABLERESOURCESREPLY._serialized_end=2939 + _REPORTRESOURCEUSAGEREQUEST._serialized_start=2941 + _REPORTRESOURCEUSAGEREQUEST._serialized_end=3012 + _REPORTRESOURCEUSAGEREPLY._serialized_start=3014 + _REPORTRESOURCEUSAGEREPLY._serialized_end=3076 + _REPORTWORKERFAILUREREQUEST._serialized_start=3078 + _REPORTWORKERFAILUREREQUEST._serialized_end=3156 + _REPORTWORKERFAILUREREPLY._serialized_start=3158 + _REPORTWORKERFAILUREREPLY._serialized_end=3220 + _GETWORKERINFOREQUEST._serialized_start=3222 + _GETWORKERINFOREQUEST._serialized_end=3263 + _GETWORKERINFOREPLY._serialized_start=3265 + _GETWORKERINFOREPLY._serialized_end=3374 + _GETALLWORKERINFOREQUEST._serialized_start=3376 + _GETALLWORKERINFOREQUEST._serialized_end=3431 + _GETALLWORKERINFOREPLY._serialized_start=3433 + _GETALLWORKERINFOREPLY._serialized_end=3560 + _ADDWORKERINFOREQUEST._serialized_start=3562 + _ADDWORKERINFOREQUEST._serialized_end=3631 + _ADDWORKERINFOREPLY._serialized_start=3633 + _ADDWORKERINFOREPLY._serialized_end=3689 + _CREATEACTORREQUEST._serialized_start=3691 + _CREATEACTORREQUEST._serialized_end=3749 + _CREATEACTORREPLY._serialized_start=3752 + _CREATEACTORREPLY._serialized_end=3969 + _REGISTERACTORREQUEST._serialized_start=3971 + _REGISTERACTORREQUEST._serialized_end=4031 + _REGISTERACTORREPLY._serialized_start=4033 + _REGISTERACTORREPLY._serialized_end=4089 + _CREATEPLACEMENTGROUPREQUEST._serialized_start=4091 + _CREATEPLACEMENTGROUPREQUEST._serialized_end=4179 + _CREATEPLACEMENTGROUPREPLY._serialized_start=4181 + _CREATEPLACEMENTGROUPREPLY._serialized_end=4244 + _REMOVEPLACEMENTGROUPREQUEST._serialized_start=4246 + _REMOVEPLACEMENTGROUPREQUEST._serialized_end=4303 + _REMOVEPLACEMENTGROUPREPLY._serialized_start=4305 + _REMOVEPLACEMENTGROUPREPLY._serialized_end=4368 + _GETPLACEMENTGROUPREQUEST._serialized_start=4370 + _GETPLACEMENTGROUPREQUEST._serialized_end=4424 + _GETPLACEMENTGROUPREPLY._serialized_start=4427 + _GETPLACEMENTGROUPREPLY._serialized_end=4557 + _GETALLPLACEMENTGROUPREQUEST._serialized_start=4559 + _GETALLPLACEMENTGROUPREQUEST._serialized_end=4618 + _GETALLPLACEMENTGROUPREPLY._serialized_start=4621 + _GETALLPLACEMENTGROUPREPLY._serialized_end=4769 + _WAITPLACEMENTGROUPUNTILREADYREQUEST._serialized_start=4771 + _WAITPLACEMENTGROUPUNTILREADYREQUEST._serialized_end=4836 + _WAITPLACEMENTGROUPUNTILREADYREPLY._serialized_start=4838 + _WAITPLACEMENTGROUPUNTILREADYREPLY._serialized_end=4909 + _GETNAMEDPLACEMENTGROUPREQUEST._serialized_start=4911 + _GETNAMEDPLACEMENTGROUPREQUEST._serialized_end=4979 + _GETNAMEDPLACEMENTGROUPREPLY._serialized_start=4982 + _GETNAMEDPLACEMENTGROUPREPLY._serialized_end=5117 + _DRAINNODEDATA._serialized_start=5119 + _DRAINNODEDATA._serialized_end=5151 + _DRAINNODEREQUEST._serialized_start=5153 + _DRAINNODEREQUEST._serialized_end=5220 + _DRAINNODESTATUS._serialized_start=5222 + _DRAINNODESTATUS._serialized_end=5256 + _DRAINNODEREPLY._serialized_start=5258 + _DRAINNODEREPLY._serialized_end=5363 + _INTERNALKVGETREQUEST._serialized_start=5365 + _INTERNALKVGETREQUEST._serialized_end=5419 + _INTERNALKVGETREPLY._serialized_start=5421 + _INTERNALKVGETREPLY._serialized_end=5492 + _INTERNALKVMULTIGETREQUEST._serialized_start=5494 + _INTERNALKVMULTIGETREQUEST._serialized_end=5554 + _MAPFIELDENTRY._serialized_start=5556 + _MAPFIELDENTRY._serialized_end=5599 + _INTERNALKVMULTIGETREPLY._serialized_start=5601 + _INTERNALKVMULTIGETREPLY._serialized_end=5703 + _INTERNALKVPUTREQUEST._serialized_start=5705 + _INTERNALKVPUTREQUEST._serialized_end=5793 + _INTERNALKVPUTREPLY._serialized_start=5795 + _INTERNALKVPUTREPLY._serialized_end=5870 + _INTERNALKVDELREQUEST._serialized_start=5872 + _INTERNALKVDELREQUEST._serialized_end=5949 + _INTERNALKVDELREPLY._serialized_start=5951 + _INTERNALKVDELREPLY._serialized_end=6028 + _INTERNALKVEXISTSREQUEST._serialized_start=6030 + _INTERNALKVEXISTSREQUEST._serialized_end=6087 + _INTERNALKVEXISTSREPLY._serialized_start=6089 + _INTERNALKVEXISTSREPLY._serialized_end=6164 + _INTERNALKVKEYSREQUEST._serialized_start=6166 + _INTERNALKVKEYSREQUEST._serialized_end=6224 + _INTERNALKVKEYSREPLY._serialized_start=6226 + _INTERNALKVKEYSREPLY._serialized_end=6300 + _PINRUNTIMEENVURIREQUEST._serialized_start=6302 + _PINRUNTIMEENVURIREQUEST._serialized_end=6362 + _PINRUNTIMEENVURIREPLY._serialized_start=6364 + _PINRUNTIMEENVURIREPLY._serialized_end=6423 + _GCSPUBLISHREQUEST._serialized_start=6425 + _GCSPUBLISHREQUEST._serialized_end=6487 + _GCSPUBLISHREPLY._serialized_start=6489 + _GCSPUBLISHREPLY._serialized_end=6542 + _GCSSUBSCRIBERPOLLREQUEST._serialized_start=6544 + _GCSSUBSCRIBERPOLLREQUEST._serialized_end=6650 + _GCSSUBSCRIBERPOLLREPLY._serialized_start=6652 + _GCSSUBSCRIBERPOLLREPLY._serialized_end=6777 + _GCSSUBSCRIBERCOMMANDBATCHREQUEST._serialized_start=6779 + _GCSSUBSCRIBERCOMMANDBATCHREQUEST._serialized_end=6891 + _GCSSUBSCRIBERCOMMANDBATCHREPLY._serialized_start=6893 + _GCSSUBSCRIBERCOMMANDBATCHREPLY._serialized_end=6961 + _GETALLRESOURCEUSAGEREQUEST._serialized_start=6963 + _GETALLRESOURCEUSAGEREQUEST._serialized_end=6991 + _GETALLRESOURCEUSAGEREPLY._serialized_start=6994 + _GETALLRESOURCEUSAGEREPLY._serialized_end=7166 + _GETDRAININGNODESREQUEST._serialized_start=7168 + _GETDRAININGNODESREQUEST._serialized_end=7193 + _GETDRAININGNODESREPLY._serialized_start=7195 + _GETDRAININGNODESREPLY._serialized_end=7272 + _GCSSTATUS._serialized_start=7274 + _GCSSTATUS._serialized_end=7316 + _ADDTASKEVENTDATAREQUEST._serialized_start=7318 + _ADDTASKEVENTDATAREQUEST._serialized_end=7381 + _ADDTASKEVENTDATAREPLY._serialized_start=7383 + _ADDTASKEVENTDATAREPLY._serialized_end=7442 + _GETTASKEVENTSREQUEST._serialized_start=7445 + _GETTASKEVENTSREQUEST._serialized_end=7744 + _GETTASKEVENTSREQUEST_FILTERS._serialized_start=7551 + _GETTASKEVENTSREQUEST_FILTERS._serialized_end=7722 + _GETTASKEVENTSREPLY._serialized_start=7747 + _GETTASKEVENTSREPLY._serialized_end=7929 + _JOBINFOGCSSERVICE._serialized_start=8001 + _JOBINFOGCSSERVICE._serialized_end=8390 + _ACTORINFOGCSSERVICE._serialized_start=8393 + _ACTORINFOGCSSERVICE._serialized_end=8974 + _WORKERINFOGCSSERVICE._serialized_start=8977 + _WORKERINFOGCSSERVICE._serialized_end=9334 + _PLACEMENTGROUPINFOGCSSERVICE._serialized_start=9337 + _PLACEMENTGROUPINFOGCSSERVICE._serialized_end=9976 + _INTERNALKVGCSSERVICE._serialized_start=9979 + _INTERNALKVGCSSERVICE._serialized_end=10490 + _RUNTIMEENVGCSSERVICE._serialized_start=10492 + _RUNTIMEENVGCSSERVICE._serialized_end=10600 + _INTERNALPUBSUBGCSSERVICE._serialized_start=10603 + _INTERNALPUBSUBGCSSERVICE._serialized_end=10899 + _NODERESOURCEINFOGCSSERVICE._serialized_start=10902 + _NODERESOURCEINFOGCSSERVICE._serialized_end=11390 + _NODEINFOGCSSERVICE._serialized_start=11393 + _NODEINFOGCSSERVICE._serialized_end=11863 + _TASKINFOGCSSERVICE._serialized_start=11866 + _TASKINFOGCSSERVICE._serialized_end=12049 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/gcs_service_pb2_grpc.py b/python/ray/core/generated/gcs_service_pb2_grpc.py new file mode 100644 index 0000000000000..c5f8d69b0d136 --- /dev/null +++ b/python/ray/core/generated/gcs_service_pb2_grpc.py @@ -0,0 +1,1851 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import gcs_service_pb2 as src_dot_ray_dot_protobuf_dot_gcs__service__pb2 + + +class JobInfoGcsServiceStub(object): + """Service for job info access. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.AddJob = channel.unary_unary( + '/ray.rpc.JobInfoGcsService/AddJob', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddJobRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddJobReply.FromString, + ) + self.MarkJobFinished = channel.unary_unary( + '/ray.rpc.JobInfoGcsService/MarkJobFinished', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.MarkJobFinishedRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.MarkJobFinishedReply.FromString, + ) + self.GetAllJobInfo = channel.unary_unary( + '/ray.rpc.JobInfoGcsService/GetAllJobInfo', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllJobInfoRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllJobInfoReply.FromString, + ) + self.ReportJobError = channel.unary_unary( + '/ray.rpc.JobInfoGcsService/ReportJobError', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportJobErrorRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportJobErrorReply.FromString, + ) + self.GetNextJobID = channel.unary_unary( + '/ray.rpc.JobInfoGcsService/GetNextJobID', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNextJobIDRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNextJobIDReply.FromString, + ) + + +class JobInfoGcsServiceServicer(object): + """Service for job info access. + """ + + def AddJob(self, request, context): + """Add job to GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def MarkJobFinished(self, request, context): + """Mark job as finished to GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetAllJobInfo(self, request, context): + """Get information of all jobs from GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ReportJobError(self, request, context): + """Report job error. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetNextJobID(self, request, context): + """Get next job id. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_JobInfoGcsServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'AddJob': grpc.unary_unary_rpc_method_handler( + servicer.AddJob, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddJobRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddJobReply.SerializeToString, + ), + 'MarkJobFinished': grpc.unary_unary_rpc_method_handler( + servicer.MarkJobFinished, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.MarkJobFinishedRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.MarkJobFinishedReply.SerializeToString, + ), + 'GetAllJobInfo': grpc.unary_unary_rpc_method_handler( + servicer.GetAllJobInfo, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllJobInfoRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllJobInfoReply.SerializeToString, + ), + 'ReportJobError': grpc.unary_unary_rpc_method_handler( + servicer.ReportJobError, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportJobErrorRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportJobErrorReply.SerializeToString, + ), + 'GetNextJobID': grpc.unary_unary_rpc_method_handler( + servicer.GetNextJobID, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNextJobIDRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNextJobIDReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.JobInfoGcsService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class JobInfoGcsService(object): + """Service for job info access. + """ + + @staticmethod + def AddJob(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.JobInfoGcsService/AddJob', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddJobRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddJobReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def MarkJobFinished(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.JobInfoGcsService/MarkJobFinished', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.MarkJobFinishedRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.MarkJobFinishedReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetAllJobInfo(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.JobInfoGcsService/GetAllJobInfo', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllJobInfoRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllJobInfoReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ReportJobError(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.JobInfoGcsService/ReportJobError', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportJobErrorRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportJobErrorReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetNextJobID(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.JobInfoGcsService/GetNextJobID', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNextJobIDRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNextJobIDReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class ActorInfoGcsServiceStub(object): + """Service for actor info access. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.RegisterActor = channel.unary_unary( + '/ray.rpc.ActorInfoGcsService/RegisterActor', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RegisterActorRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RegisterActorReply.FromString, + ) + self.CreateActor = channel.unary_unary( + '/ray.rpc.ActorInfoGcsService/CreateActor', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CreateActorRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CreateActorReply.FromString, + ) + self.GetActorInfo = channel.unary_unary( + '/ray.rpc.ActorInfoGcsService/GetActorInfo', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetActorInfoRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetActorInfoReply.FromString, + ) + self.GetNamedActorInfo = channel.unary_unary( + '/ray.rpc.ActorInfoGcsService/GetNamedActorInfo', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNamedActorInfoRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNamedActorInfoReply.FromString, + ) + self.ListNamedActors = channel.unary_unary( + '/ray.rpc.ActorInfoGcsService/ListNamedActors', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ListNamedActorsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ListNamedActorsReply.FromString, + ) + self.GetAllActorInfo = channel.unary_unary( + '/ray.rpc.ActorInfoGcsService/GetAllActorInfo', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllActorInfoRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllActorInfoReply.FromString, + ) + self.KillActorViaGcs = channel.unary_unary( + '/ray.rpc.ActorInfoGcsService/KillActorViaGcs', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.KillActorViaGcsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.KillActorViaGcsReply.FromString, + ) + + +class ActorInfoGcsServiceServicer(object): + """Service for actor info access. + """ + + def RegisterActor(self, request, context): + """Register actor to gcs service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def CreateActor(self, request, context): + """Create actor which local dependencies are resolved. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetActorInfo(self, request, context): + """Get actor data from GCS Service by actor id. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetNamedActorInfo(self, request, context): + """Get actor data from GCS Service by name. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ListNamedActors(self, request, context): + """Get all named actor names from GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetAllActorInfo(self, request, context): + """Get information of all actor from GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def KillActorViaGcs(self, request, context): + """Kill actor via GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_ActorInfoGcsServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'RegisterActor': grpc.unary_unary_rpc_method_handler( + servicer.RegisterActor, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RegisterActorRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RegisterActorReply.SerializeToString, + ), + 'CreateActor': grpc.unary_unary_rpc_method_handler( + servicer.CreateActor, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CreateActorRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CreateActorReply.SerializeToString, + ), + 'GetActorInfo': grpc.unary_unary_rpc_method_handler( + servicer.GetActorInfo, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetActorInfoRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetActorInfoReply.SerializeToString, + ), + 'GetNamedActorInfo': grpc.unary_unary_rpc_method_handler( + servicer.GetNamedActorInfo, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNamedActorInfoRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNamedActorInfoReply.SerializeToString, + ), + 'ListNamedActors': grpc.unary_unary_rpc_method_handler( + servicer.ListNamedActors, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ListNamedActorsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ListNamedActorsReply.SerializeToString, + ), + 'GetAllActorInfo': grpc.unary_unary_rpc_method_handler( + servicer.GetAllActorInfo, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllActorInfoRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllActorInfoReply.SerializeToString, + ), + 'KillActorViaGcs': grpc.unary_unary_rpc_method_handler( + servicer.KillActorViaGcs, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.KillActorViaGcsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.KillActorViaGcsReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.ActorInfoGcsService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class ActorInfoGcsService(object): + """Service for actor info access. + """ + + @staticmethod + def RegisterActor(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ActorInfoGcsService/RegisterActor', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RegisterActorRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RegisterActorReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def CreateActor(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ActorInfoGcsService/CreateActor', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CreateActorRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CreateActorReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetActorInfo(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ActorInfoGcsService/GetActorInfo', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetActorInfoRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetActorInfoReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetNamedActorInfo(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ActorInfoGcsService/GetNamedActorInfo', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNamedActorInfoRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNamedActorInfoReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ListNamedActors(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ActorInfoGcsService/ListNamedActors', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ListNamedActorsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ListNamedActorsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetAllActorInfo(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ActorInfoGcsService/GetAllActorInfo', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllActorInfoRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllActorInfoReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def KillActorViaGcs(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ActorInfoGcsService/KillActorViaGcs', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.KillActorViaGcsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.KillActorViaGcsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class WorkerInfoGcsServiceStub(object): + """Service for worker info access. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.ReportWorkerFailure = channel.unary_unary( + '/ray.rpc.WorkerInfoGcsService/ReportWorkerFailure', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportWorkerFailureRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportWorkerFailureReply.FromString, + ) + self.GetWorkerInfo = channel.unary_unary( + '/ray.rpc.WorkerInfoGcsService/GetWorkerInfo', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetWorkerInfoRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetWorkerInfoReply.FromString, + ) + self.GetAllWorkerInfo = channel.unary_unary( + '/ray.rpc.WorkerInfoGcsService/GetAllWorkerInfo', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllWorkerInfoRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllWorkerInfoReply.FromString, + ) + self.AddWorkerInfo = channel.unary_unary( + '/ray.rpc.WorkerInfoGcsService/AddWorkerInfo', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddWorkerInfoRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddWorkerInfoReply.FromString, + ) + + +class WorkerInfoGcsServiceServicer(object): + """Service for worker info access. + """ + + def ReportWorkerFailure(self, request, context): + """Report a worker failure to GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetWorkerInfo(self, request, context): + """Get worker information from GCS Service by worker id. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetAllWorkerInfo(self, request, context): + """Get information of all workers from GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def AddWorkerInfo(self, request, context): + """Add worker information to GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_WorkerInfoGcsServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'ReportWorkerFailure': grpc.unary_unary_rpc_method_handler( + servicer.ReportWorkerFailure, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportWorkerFailureRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportWorkerFailureReply.SerializeToString, + ), + 'GetWorkerInfo': grpc.unary_unary_rpc_method_handler( + servicer.GetWorkerInfo, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetWorkerInfoRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetWorkerInfoReply.SerializeToString, + ), + 'GetAllWorkerInfo': grpc.unary_unary_rpc_method_handler( + servicer.GetAllWorkerInfo, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllWorkerInfoRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllWorkerInfoReply.SerializeToString, + ), + 'AddWorkerInfo': grpc.unary_unary_rpc_method_handler( + servicer.AddWorkerInfo, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddWorkerInfoRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddWorkerInfoReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.WorkerInfoGcsService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class WorkerInfoGcsService(object): + """Service for worker info access. + """ + + @staticmethod + def ReportWorkerFailure(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.WorkerInfoGcsService/ReportWorkerFailure', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportWorkerFailureRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportWorkerFailureReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetWorkerInfo(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.WorkerInfoGcsService/GetWorkerInfo', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetWorkerInfoRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetWorkerInfoReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetAllWorkerInfo(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.WorkerInfoGcsService/GetAllWorkerInfo', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllWorkerInfoRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllWorkerInfoReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def AddWorkerInfo(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.WorkerInfoGcsService/AddWorkerInfo', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddWorkerInfoRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddWorkerInfoReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class PlacementGroupInfoGcsServiceStub(object): + """Service for placement group info access. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.CreatePlacementGroup = channel.unary_unary( + '/ray.rpc.PlacementGroupInfoGcsService/CreatePlacementGroup', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CreatePlacementGroupRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CreatePlacementGroupReply.FromString, + ) + self.RemovePlacementGroup = channel.unary_unary( + '/ray.rpc.PlacementGroupInfoGcsService/RemovePlacementGroup', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RemovePlacementGroupRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RemovePlacementGroupReply.FromString, + ) + self.GetPlacementGroup = channel.unary_unary( + '/ray.rpc.PlacementGroupInfoGcsService/GetPlacementGroup', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetPlacementGroupRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetPlacementGroupReply.FromString, + ) + self.GetNamedPlacementGroup = channel.unary_unary( + '/ray.rpc.PlacementGroupInfoGcsService/GetNamedPlacementGroup', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNamedPlacementGroupRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNamedPlacementGroupReply.FromString, + ) + self.GetAllPlacementGroup = channel.unary_unary( + '/ray.rpc.PlacementGroupInfoGcsService/GetAllPlacementGroup', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllPlacementGroupRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllPlacementGroupReply.FromString, + ) + self.WaitPlacementGroupUntilReady = channel.unary_unary( + '/ray.rpc.PlacementGroupInfoGcsService/WaitPlacementGroupUntilReady', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.WaitPlacementGroupUntilReadyRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.WaitPlacementGroupUntilReadyReply.FromString, + ) + + +class PlacementGroupInfoGcsServiceServicer(object): + """Service for placement group info access. + """ + + def CreatePlacementGroup(self, request, context): + """Create placement group via gcs service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def RemovePlacementGroup(self, request, context): + """Remove placement group via gcs service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetPlacementGroup(self, request, context): + """Get placement group information via gcs service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetNamedPlacementGroup(self, request, context): + """Get named placement group information via gcs service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetAllPlacementGroup(self, request, context): + """Get information of all placement group from GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def WaitPlacementGroupUntilReady(self, request, context): + """Wait for placement group until ready. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_PlacementGroupInfoGcsServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'CreatePlacementGroup': grpc.unary_unary_rpc_method_handler( + servicer.CreatePlacementGroup, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CreatePlacementGroupRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CreatePlacementGroupReply.SerializeToString, + ), + 'RemovePlacementGroup': grpc.unary_unary_rpc_method_handler( + servicer.RemovePlacementGroup, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RemovePlacementGroupRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RemovePlacementGroupReply.SerializeToString, + ), + 'GetPlacementGroup': grpc.unary_unary_rpc_method_handler( + servicer.GetPlacementGroup, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetPlacementGroupRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetPlacementGroupReply.SerializeToString, + ), + 'GetNamedPlacementGroup': grpc.unary_unary_rpc_method_handler( + servicer.GetNamedPlacementGroup, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNamedPlacementGroupRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNamedPlacementGroupReply.SerializeToString, + ), + 'GetAllPlacementGroup': grpc.unary_unary_rpc_method_handler( + servicer.GetAllPlacementGroup, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllPlacementGroupRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllPlacementGroupReply.SerializeToString, + ), + 'WaitPlacementGroupUntilReady': grpc.unary_unary_rpc_method_handler( + servicer.WaitPlacementGroupUntilReady, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.WaitPlacementGroupUntilReadyRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.WaitPlacementGroupUntilReadyReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.PlacementGroupInfoGcsService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class PlacementGroupInfoGcsService(object): + """Service for placement group info access. + """ + + @staticmethod + def CreatePlacementGroup(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.PlacementGroupInfoGcsService/CreatePlacementGroup', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CreatePlacementGroupRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CreatePlacementGroupReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def RemovePlacementGroup(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.PlacementGroupInfoGcsService/RemovePlacementGroup', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RemovePlacementGroupRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RemovePlacementGroupReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetPlacementGroup(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.PlacementGroupInfoGcsService/GetPlacementGroup', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetPlacementGroupRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetPlacementGroupReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetNamedPlacementGroup(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.PlacementGroupInfoGcsService/GetNamedPlacementGroup', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNamedPlacementGroupRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetNamedPlacementGroupReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetAllPlacementGroup(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.PlacementGroupInfoGcsService/GetAllPlacementGroup', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllPlacementGroupRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllPlacementGroupReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def WaitPlacementGroupUntilReady(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.PlacementGroupInfoGcsService/WaitPlacementGroupUntilReady', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.WaitPlacementGroupUntilReadyRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.WaitPlacementGroupUntilReadyReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class InternalKVGcsServiceStub(object): + """Service for KV storage + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.InternalKVGet = channel.unary_unary( + '/ray.rpc.InternalKVGcsService/InternalKVGet', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVGetRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVGetReply.FromString, + ) + self.InternalKVMultiGet = channel.unary_unary( + '/ray.rpc.InternalKVGcsService/InternalKVMultiGet', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVMultiGetRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVMultiGetReply.FromString, + ) + self.InternalKVPut = channel.unary_unary( + '/ray.rpc.InternalKVGcsService/InternalKVPut', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVPutRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVPutReply.FromString, + ) + self.InternalKVDel = channel.unary_unary( + '/ray.rpc.InternalKVGcsService/InternalKVDel', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVDelRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVDelReply.FromString, + ) + self.InternalKVExists = channel.unary_unary( + '/ray.rpc.InternalKVGcsService/InternalKVExists', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVExistsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVExistsReply.FromString, + ) + self.InternalKVKeys = channel.unary_unary( + '/ray.rpc.InternalKVGcsService/InternalKVKeys', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVKeysRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVKeysReply.FromString, + ) + + +class InternalKVGcsServiceServicer(object): + """Service for KV storage + """ + + def InternalKVGet(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def InternalKVMultiGet(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def InternalKVPut(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def InternalKVDel(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def InternalKVExists(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def InternalKVKeys(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_InternalKVGcsServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'InternalKVGet': grpc.unary_unary_rpc_method_handler( + servicer.InternalKVGet, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVGetRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVGetReply.SerializeToString, + ), + 'InternalKVMultiGet': grpc.unary_unary_rpc_method_handler( + servicer.InternalKVMultiGet, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVMultiGetRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVMultiGetReply.SerializeToString, + ), + 'InternalKVPut': grpc.unary_unary_rpc_method_handler( + servicer.InternalKVPut, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVPutRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVPutReply.SerializeToString, + ), + 'InternalKVDel': grpc.unary_unary_rpc_method_handler( + servicer.InternalKVDel, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVDelRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVDelReply.SerializeToString, + ), + 'InternalKVExists': grpc.unary_unary_rpc_method_handler( + servicer.InternalKVExists, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVExistsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVExistsReply.SerializeToString, + ), + 'InternalKVKeys': grpc.unary_unary_rpc_method_handler( + servicer.InternalKVKeys, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVKeysRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVKeysReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.InternalKVGcsService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class InternalKVGcsService(object): + """Service for KV storage + """ + + @staticmethod + def InternalKVGet(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.InternalKVGcsService/InternalKVGet', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVGetRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVGetReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def InternalKVMultiGet(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.InternalKVGcsService/InternalKVMultiGet', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVMultiGetRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVMultiGetReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def InternalKVPut(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.InternalKVGcsService/InternalKVPut', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVPutRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVPutReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def InternalKVDel(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.InternalKVGcsService/InternalKVDel', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVDelRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVDelReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def InternalKVExists(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.InternalKVGcsService/InternalKVExists', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVExistsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVExistsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def InternalKVKeys(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.InternalKVGcsService/InternalKVKeys', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVKeysRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.InternalKVKeysReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class RuntimeEnvGcsServiceStub(object): + """/ Handles pinning package URIs that are stored in the GCS. + / Future runtime_env-related RPCs should be added to this service. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.PinRuntimeEnvURI = channel.unary_unary( + '/ray.rpc.RuntimeEnvGcsService/PinRuntimeEnvURI', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.PinRuntimeEnvURIRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.PinRuntimeEnvURIReply.FromString, + ) + + +class RuntimeEnvGcsServiceServicer(object): + """/ Handles pinning package URIs that are stored in the GCS. + / Future runtime_env-related RPCs should be added to this service. + """ + + def PinRuntimeEnvURI(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_RuntimeEnvGcsServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'PinRuntimeEnvURI': grpc.unary_unary_rpc_method_handler( + servicer.PinRuntimeEnvURI, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.PinRuntimeEnvURIRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.PinRuntimeEnvURIReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.RuntimeEnvGcsService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class RuntimeEnvGcsService(object): + """/ Handles pinning package URIs that are stored in the GCS. + / Future runtime_env-related RPCs should be added to this service. + """ + + @staticmethod + def PinRuntimeEnvURI(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RuntimeEnvGcsService/PinRuntimeEnvURI', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.PinRuntimeEnvURIRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.PinRuntimeEnvURIReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class InternalPubSubGcsServiceStub(object): + """/ This supports subscribing updates from GCS with long poll, and registering / + / de-registering subscribers. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.GcsPublish = channel.unary_unary( + '/ray.rpc.InternalPubSubGcsService/GcsPublish', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsPublishRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsPublishReply.FromString, + ) + self.GcsSubscriberPoll = channel.unary_unary( + '/ray.rpc.InternalPubSubGcsService/GcsSubscriberPoll', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsSubscriberPollRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsSubscriberPollReply.FromString, + ) + self.GcsSubscriberCommandBatch = channel.unary_unary( + '/ray.rpc.InternalPubSubGcsService/GcsSubscriberCommandBatch', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsSubscriberCommandBatchRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsSubscriberCommandBatchReply.FromString, + ) + + +class InternalPubSubGcsServiceServicer(object): + """/ This supports subscribing updates from GCS with long poll, and registering / + / de-registering subscribers. + """ + + def GcsPublish(self, request, context): + """/ The request to sent to GCS to publish messages. + / Currently only supporting error info, logs and Python function messages. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GcsSubscriberPoll(self, request, context): + """/ The long polling request sent to GCS for pubsub operations. + / The long poll request will be replied once there are a batch of messages that + / need to be published to the caller (subscriber). + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GcsSubscriberCommandBatch(self, request, context): + """/ A batch of subscribe / unsubscribe requests sent by the subscriber. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_InternalPubSubGcsServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'GcsPublish': grpc.unary_unary_rpc_method_handler( + servicer.GcsPublish, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsPublishRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsPublishReply.SerializeToString, + ), + 'GcsSubscriberPoll': grpc.unary_unary_rpc_method_handler( + servicer.GcsSubscriberPoll, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsSubscriberPollRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsSubscriberPollReply.SerializeToString, + ), + 'GcsSubscriberCommandBatch': grpc.unary_unary_rpc_method_handler( + servicer.GcsSubscriberCommandBatch, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsSubscriberCommandBatchRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsSubscriberCommandBatchReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.InternalPubSubGcsService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class InternalPubSubGcsService(object): + """/ This supports subscribing updates from GCS with long poll, and registering / + / de-registering subscribers. + """ + + @staticmethod + def GcsPublish(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.InternalPubSubGcsService/GcsPublish', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsPublishRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsPublishReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GcsSubscriberPoll(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.InternalPubSubGcsService/GcsSubscriberPoll', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsSubscriberPollRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsSubscriberPollReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GcsSubscriberCommandBatch(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.InternalPubSubGcsService/GcsSubscriberCommandBatch', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsSubscriberCommandBatchRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GcsSubscriberCommandBatchReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class NodeResourceInfoGcsServiceStub(object): + """Service for node resource info access. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.GetResources = channel.unary_unary( + '/ray.rpc.NodeResourceInfoGcsService/GetResources', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetResourcesRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetResourcesReply.FromString, + ) + self.GetAllAvailableResources = channel.unary_unary( + '/ray.rpc.NodeResourceInfoGcsService/GetAllAvailableResources', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllAvailableResourcesRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllAvailableResourcesReply.FromString, + ) + self.ReportResourceUsage = channel.unary_unary( + '/ray.rpc.NodeResourceInfoGcsService/ReportResourceUsage', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportResourceUsageRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportResourceUsageReply.FromString, + ) + self.GetAllResourceUsage = channel.unary_unary( + '/ray.rpc.NodeResourceInfoGcsService/GetAllResourceUsage', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllResourceUsageRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllResourceUsageReply.FromString, + ) + self.GetDrainingNodes = channel.unary_unary( + '/ray.rpc.NodeResourceInfoGcsService/GetDrainingNodes', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetDrainingNodesRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetDrainingNodesReply.FromString, + ) + + +class NodeResourceInfoGcsServiceServicer(object): + """Service for node resource info access. + """ + + def GetResources(self, request, context): + """Get node's resources from GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetAllAvailableResources(self, request, context): + """Get available resources of all nodes. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ReportResourceUsage(self, request, context): + """Report resource usage of a node to GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetAllResourceUsage(self, request, context): + """Get resource usage of all nodes from GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetDrainingNodes(self, request, context): + """Get ids of draining nodes. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_NodeResourceInfoGcsServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'GetResources': grpc.unary_unary_rpc_method_handler( + servicer.GetResources, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetResourcesRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetResourcesReply.SerializeToString, + ), + 'GetAllAvailableResources': grpc.unary_unary_rpc_method_handler( + servicer.GetAllAvailableResources, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllAvailableResourcesRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllAvailableResourcesReply.SerializeToString, + ), + 'ReportResourceUsage': grpc.unary_unary_rpc_method_handler( + servicer.ReportResourceUsage, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportResourceUsageRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportResourceUsageReply.SerializeToString, + ), + 'GetAllResourceUsage': grpc.unary_unary_rpc_method_handler( + servicer.GetAllResourceUsage, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllResourceUsageRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllResourceUsageReply.SerializeToString, + ), + 'GetDrainingNodes': grpc.unary_unary_rpc_method_handler( + servicer.GetDrainingNodes, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetDrainingNodesRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetDrainingNodesReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.NodeResourceInfoGcsService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class NodeResourceInfoGcsService(object): + """Service for node resource info access. + """ + + @staticmethod + def GetResources(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeResourceInfoGcsService/GetResources', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetResourcesRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetResourcesReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetAllAvailableResources(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeResourceInfoGcsService/GetAllAvailableResources', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllAvailableResourcesRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllAvailableResourcesReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ReportResourceUsage(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeResourceInfoGcsService/ReportResourceUsage', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportResourceUsageRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.ReportResourceUsageReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetAllResourceUsage(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeResourceInfoGcsService/GetAllResourceUsage', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllResourceUsageRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllResourceUsageReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetDrainingNodes(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeResourceInfoGcsService/GetDrainingNodes', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetDrainingNodesRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetDrainingNodesReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class NodeInfoGcsServiceStub(object): + """Service for node info access. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.GetClusterId = channel.unary_unary( + '/ray.rpc.NodeInfoGcsService/GetClusterId', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetClusterIdRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetClusterIdReply.FromString, + ) + self.RegisterNode = channel.unary_unary( + '/ray.rpc.NodeInfoGcsService/RegisterNode', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RegisterNodeRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RegisterNodeReply.FromString, + ) + self.DrainNode = channel.unary_unary( + '/ray.rpc.NodeInfoGcsService/DrainNode', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.DrainNodeRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.DrainNodeReply.FromString, + ) + self.GetAllNodeInfo = channel.unary_unary( + '/ray.rpc.NodeInfoGcsService/GetAllNodeInfo', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllNodeInfoRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllNodeInfoReply.FromString, + ) + self.GetInternalConfig = channel.unary_unary( + '/ray.rpc.NodeInfoGcsService/GetInternalConfig', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetInternalConfigRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetInternalConfigReply.FromString, + ) + self.CheckAlive = channel.unary_unary( + '/ray.rpc.NodeInfoGcsService/CheckAlive', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CheckAliveRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CheckAliveReply.FromString, + ) + + +class NodeInfoGcsServiceServicer(object): + """Service for node info access. + """ + + def GetClusterId(self, request, context): + """Register a client to GCS Service. Must be called before any other RPC in GCSClient. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def RegisterNode(self, request, context): + """Register a node to GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def DrainNode(self, request, context): + """Drain a node from GCS Service. + When the RPC is replied, + - It is guaranteed that the requested nodes are going to be drained eventually. + - It is guaranteed that GCS has persisted the draining states. + - It is **not** guarateed that nodes receive the drain requests from GCS. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetAllNodeInfo(self, request, context): + """Get information of all nodes from GCS Service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetInternalConfig(self, request, context): + """Get cluster internal config. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def CheckAlive(self, request, context): + """Check alive. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_NodeInfoGcsServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'GetClusterId': grpc.unary_unary_rpc_method_handler( + servicer.GetClusterId, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetClusterIdRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetClusterIdReply.SerializeToString, + ), + 'RegisterNode': grpc.unary_unary_rpc_method_handler( + servicer.RegisterNode, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RegisterNodeRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RegisterNodeReply.SerializeToString, + ), + 'DrainNode': grpc.unary_unary_rpc_method_handler( + servicer.DrainNode, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.DrainNodeRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.DrainNodeReply.SerializeToString, + ), + 'GetAllNodeInfo': grpc.unary_unary_rpc_method_handler( + servicer.GetAllNodeInfo, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllNodeInfoRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllNodeInfoReply.SerializeToString, + ), + 'GetInternalConfig': grpc.unary_unary_rpc_method_handler( + servicer.GetInternalConfig, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetInternalConfigRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetInternalConfigReply.SerializeToString, + ), + 'CheckAlive': grpc.unary_unary_rpc_method_handler( + servicer.CheckAlive, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CheckAliveRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CheckAliveReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.NodeInfoGcsService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class NodeInfoGcsService(object): + """Service for node info access. + """ + + @staticmethod + def GetClusterId(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeInfoGcsService/GetClusterId', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetClusterIdRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetClusterIdReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def RegisterNode(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeInfoGcsService/RegisterNode', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RegisterNodeRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.RegisterNodeReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def DrainNode(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeInfoGcsService/DrainNode', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.DrainNodeRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.DrainNodeReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetAllNodeInfo(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeInfoGcsService/GetAllNodeInfo', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllNodeInfoRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetAllNodeInfoReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetInternalConfig(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeInfoGcsService/GetInternalConfig', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetInternalConfigRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetInternalConfigReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def CheckAlive(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeInfoGcsService/CheckAlive', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CheckAliveRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.CheckAliveReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class TaskInfoGcsServiceStub(object): + """Service for task info access. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.AddTaskEventData = channel.unary_unary( + '/ray.rpc.TaskInfoGcsService/AddTaskEventData', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddTaskEventDataRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddTaskEventDataReply.FromString, + ) + self.GetTaskEvents = channel.unary_unary( + '/ray.rpc.TaskInfoGcsService/GetTaskEvents', + request_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetTaskEventsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetTaskEventsReply.FromString, + ) + + +class TaskInfoGcsServiceServicer(object): + """Service for task info access. + """ + + def AddTaskEventData(self, request, context): + """Add task event data to GCS. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetTaskEvents(self, request, context): + """Get task events. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_TaskInfoGcsServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'AddTaskEventData': grpc.unary_unary_rpc_method_handler( + servicer.AddTaskEventData, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddTaskEventDataRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddTaskEventDataReply.SerializeToString, + ), + 'GetTaskEvents': grpc.unary_unary_rpc_method_handler( + servicer.GetTaskEvents, + request_deserializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetTaskEventsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetTaskEventsReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.TaskInfoGcsService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class TaskInfoGcsService(object): + """Service for task info access. + """ + + @staticmethod + def AddTaskEventData(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.TaskInfoGcsService/AddTaskEventData', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddTaskEventDataRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.AddTaskEventDataReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetTaskEvents(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.TaskInfoGcsService/GetTaskEvents', + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetTaskEventsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_gcs__service__pb2.GetTaskEventsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/instance_manager_pb2.py b/python/ray/core/generated/instance_manager_pb2.py new file mode 100644 index 0000000000000..414826707f882 --- /dev/null +++ b/python/ray/core/generated/instance_manager_pb2.py @@ -0,0 +1,158 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/experimental/instance_manager.proto +"""Generated protocol buffer code.""" +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n4src/ray/protobuf/experimental/instance_manager.proto\x12\x12ray.rpc.autoscaler\"\xcb\x01\n\x0cInstanceType\x12\x11\n\ttype_name\x18\x01 \x01(\t\x12\x32\n\tnode_kind\x18\x02 \x01(\x0e\x32\x1f.ray.rpc.autoscaler.RayNodeKind\x12\x42\n\tresources\x18\x03 \x03(\x0b\x32/.ray.rpc.autoscaler.InstanceType.ResourcesEntry\x1a\x30\n\x0eResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\"\"\n GetAvailableInstanceTypesRequest\"z\n!GetAvailableInstanceTypesResponse\x12\x38\n\x0einstance_types\x18\x01 \x03(\x0b\x32 .ray.rpc.autoscaler.InstanceType\x12\x1b\n\x13\x61vailable_instances\x18\x02 \x01(\x03\"\xb3\x05\n\x08Instance\x12\x13\n\x0binstance_id\x18\x0b \x01(\t\x12;\n\x06status\x18\x0c \x01(\x0e\x32+.ray.rpc.autoscaler.Instance.InstanceStatus\x12\x15\n\rinstance_type\x18\r \x01(\t\x12I\n\x0ftotal_resources\x18\x0e \x03(\x0b\x32\x30.ray.rpc.autoscaler.Instance.TotalResourcesEntry\x12%\n\x1dtimestamp_since_last_modified\x18\x0f \x01(\x03\x12\x19\n\x11\x63loud_instance_id\x18\x10 \x01(\t\x12\x13\n\x0binternal_ip\x18\x11 \x01(\t\x12\x13\n\x0b\x65xternal_ip\x18\x12 \x01(\t\x12\x0f\n\x07version\x18\x13 \x01(\x03\x12:\n\nray_status\x18\x14 \x01(\x0e\x32&.ray.rpc.autoscaler.Instance.RayStatus\x1a\x35\n\x13TotalResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\"\x90\x01\n\x0eInstanceStatus\x12\x0b\n\x07UNKNOWN\x10\x00\x12\n\n\x06QEUEUD\x10\x01\x12\r\n\tREQUESTED\x10\x02\x12\r\n\tALLOCATED\x10\x03\x12\x15\n\x11\x41LLOCATION_FAILED\x10\x04\x12\x0c\n\x08STOPPING\x10\x05\x12\x0b\n\x07STOPPED\x10\x06\x12\x15\n\x11GARBAGE_COLLECTED\x10\x07\"p\n\tRayStatus\x12\x15\n\x11RAY_STATUS_UNKOWN\x10\x00\x12\x12\n\x0eRAY_INSTALLING\x10\x01\x12\x16\n\x12RAY_INSTALL_FAILED\x10\x02\x12\x0f\n\x0bRAY_RUNNING\x10\x03\x12\x0f\n\x0bRAY_STOPPED\x10\x04\"\x9e\x01\n!UpdateInstanceManagerStateRequest\x12\x18\n\x10\x65xpected_version\x18\x01 \x01(\x03\x12<\n\x12new_nodes_to_start\x18\x02 \x03(\x0b\x32 .ray.rpc.autoscaler.InstanceType\x12!\n\x19instance_ids_to_terminate\x18\x03 \x03(\t\"Z\n\x1fUpdateInstanceManagerStateReply\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x15\n\rerror_message\x18\x02 \x01(\t\x12\x0f\n\x07version\x18\x03 \x01(\x03\"X\n\x14InstanceManagerState\x12\x0f\n\x07version\x18\x01 \x01(\x03\x12/\n\tinstances\x18\x02 \x03(\x0b\x32\x1c.ray.rpc.autoscaler.Instance\" \n\x1eGetInstanceManagerStateRequest\"W\n\x1cGetInstanceManagerStateReply\x12\x37\n\x05state\x18\x01 \x01(\x0b\x32(.ray.rpc.autoscaler.InstanceManagerState*0\n\x0bRayNodeKind\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x08\n\x04HEAD\x10\x01\x12\n\n\x06WORKER\x10\x02\x32\xaf\x03\n\x16InstanceManagerService\x12\x7f\n\x17GetInstanceManagerState\x12\x32.ray.rpc.autoscaler.GetInstanceManagerStateRequest\x1a\x30.ray.rpc.autoscaler.GetInstanceManagerStateReply\x12\x88\x01\n\x1aUpdateInstanceManagerState\x12\x35.ray.rpc.autoscaler.UpdateInstanceManagerStateRequest\x1a\x33.ray.rpc.autoscaler.UpdateInstanceManagerStateReply\x12\x88\x01\n\x19GetAvailableInstanceTypes\x12\x34.ray.rpc.autoscaler.GetAvailableInstanceTypesRequest\x1a\x35.ray.rpc.autoscaler.GetAvailableInstanceTypesResponseB\x03\xf8\x01\x01\x62\x06proto3') + +_RAYNODEKIND = DESCRIPTOR.enum_types_by_name['RayNodeKind'] +RayNodeKind = enum_type_wrapper.EnumTypeWrapper(_RAYNODEKIND) +UNKNOWN = 0 +HEAD = 1 +WORKER = 2 + + +_INSTANCETYPE = DESCRIPTOR.message_types_by_name['InstanceType'] +_INSTANCETYPE_RESOURCESENTRY = _INSTANCETYPE.nested_types_by_name['ResourcesEntry'] +_GETAVAILABLEINSTANCETYPESREQUEST = DESCRIPTOR.message_types_by_name['GetAvailableInstanceTypesRequest'] +_GETAVAILABLEINSTANCETYPESRESPONSE = DESCRIPTOR.message_types_by_name['GetAvailableInstanceTypesResponse'] +_INSTANCE = DESCRIPTOR.message_types_by_name['Instance'] +_INSTANCE_TOTALRESOURCESENTRY = _INSTANCE.nested_types_by_name['TotalResourcesEntry'] +_UPDATEINSTANCEMANAGERSTATEREQUEST = DESCRIPTOR.message_types_by_name['UpdateInstanceManagerStateRequest'] +_UPDATEINSTANCEMANAGERSTATEREPLY = DESCRIPTOR.message_types_by_name['UpdateInstanceManagerStateReply'] +_INSTANCEMANAGERSTATE = DESCRIPTOR.message_types_by_name['InstanceManagerState'] +_GETINSTANCEMANAGERSTATEREQUEST = DESCRIPTOR.message_types_by_name['GetInstanceManagerStateRequest'] +_GETINSTANCEMANAGERSTATEREPLY = DESCRIPTOR.message_types_by_name['GetInstanceManagerStateReply'] +_INSTANCE_INSTANCESTATUS = _INSTANCE.enum_types_by_name['InstanceStatus'] +_INSTANCE_RAYSTATUS = _INSTANCE.enum_types_by_name['RayStatus'] +InstanceType = _reflection.GeneratedProtocolMessageType('InstanceType', (_message.Message,), { + + 'ResourcesEntry' : _reflection.GeneratedProtocolMessageType('ResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _INSTANCETYPE_RESOURCESENTRY, + '__module__' : 'src.ray.protobuf.experimental.instance_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.InstanceType.ResourcesEntry) + }) + , + 'DESCRIPTOR' : _INSTANCETYPE, + '__module__' : 'src.ray.protobuf.experimental.instance_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.InstanceType) + }) +_sym_db.RegisterMessage(InstanceType) +_sym_db.RegisterMessage(InstanceType.ResourcesEntry) + +GetAvailableInstanceTypesRequest = _reflection.GeneratedProtocolMessageType('GetAvailableInstanceTypesRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETAVAILABLEINSTANCETYPESREQUEST, + '__module__' : 'src.ray.protobuf.experimental.instance_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.GetAvailableInstanceTypesRequest) + }) +_sym_db.RegisterMessage(GetAvailableInstanceTypesRequest) + +GetAvailableInstanceTypesResponse = _reflection.GeneratedProtocolMessageType('GetAvailableInstanceTypesResponse', (_message.Message,), { + 'DESCRIPTOR' : _GETAVAILABLEINSTANCETYPESRESPONSE, + '__module__' : 'src.ray.protobuf.experimental.instance_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.GetAvailableInstanceTypesResponse) + }) +_sym_db.RegisterMessage(GetAvailableInstanceTypesResponse) + +Instance = _reflection.GeneratedProtocolMessageType('Instance', (_message.Message,), { + + 'TotalResourcesEntry' : _reflection.GeneratedProtocolMessageType('TotalResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _INSTANCE_TOTALRESOURCESENTRY, + '__module__' : 'src.ray.protobuf.experimental.instance_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.Instance.TotalResourcesEntry) + }) + , + 'DESCRIPTOR' : _INSTANCE, + '__module__' : 'src.ray.protobuf.experimental.instance_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.Instance) + }) +_sym_db.RegisterMessage(Instance) +_sym_db.RegisterMessage(Instance.TotalResourcesEntry) + +UpdateInstanceManagerStateRequest = _reflection.GeneratedProtocolMessageType('UpdateInstanceManagerStateRequest', (_message.Message,), { + 'DESCRIPTOR' : _UPDATEINSTANCEMANAGERSTATEREQUEST, + '__module__' : 'src.ray.protobuf.experimental.instance_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.UpdateInstanceManagerStateRequest) + }) +_sym_db.RegisterMessage(UpdateInstanceManagerStateRequest) + +UpdateInstanceManagerStateReply = _reflection.GeneratedProtocolMessageType('UpdateInstanceManagerStateReply', (_message.Message,), { + 'DESCRIPTOR' : _UPDATEINSTANCEMANAGERSTATEREPLY, + '__module__' : 'src.ray.protobuf.experimental.instance_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.UpdateInstanceManagerStateReply) + }) +_sym_db.RegisterMessage(UpdateInstanceManagerStateReply) + +InstanceManagerState = _reflection.GeneratedProtocolMessageType('InstanceManagerState', (_message.Message,), { + 'DESCRIPTOR' : _INSTANCEMANAGERSTATE, + '__module__' : 'src.ray.protobuf.experimental.instance_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.InstanceManagerState) + }) +_sym_db.RegisterMessage(InstanceManagerState) + +GetInstanceManagerStateRequest = _reflection.GeneratedProtocolMessageType('GetInstanceManagerStateRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETINSTANCEMANAGERSTATEREQUEST, + '__module__' : 'src.ray.protobuf.experimental.instance_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.GetInstanceManagerStateRequest) + }) +_sym_db.RegisterMessage(GetInstanceManagerStateRequest) + +GetInstanceManagerStateReply = _reflection.GeneratedProtocolMessageType('GetInstanceManagerStateReply', (_message.Message,), { + 'DESCRIPTOR' : _GETINSTANCEMANAGERSTATEREPLY, + '__module__' : 'src.ray.protobuf.experimental.instance_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.autoscaler.GetInstanceManagerStateReply) + }) +_sym_db.RegisterMessage(GetInstanceManagerStateReply) + +_INSTANCEMANAGERSERVICE = DESCRIPTOR.services_by_name['InstanceManagerService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _INSTANCETYPE_RESOURCESENTRY._options = None + _INSTANCETYPE_RESOURCESENTRY._serialized_options = b'8\001' + _INSTANCE_TOTALRESOURCESENTRY._options = None + _INSTANCE_TOTALRESOURCESENTRY._serialized_options = b'8\001' + _RAYNODEKIND._serialized_start=1602 + _RAYNODEKIND._serialized_end=1650 + _INSTANCETYPE._serialized_start=77 + _INSTANCETYPE._serialized_end=280 + _INSTANCETYPE_RESOURCESENTRY._serialized_start=232 + _INSTANCETYPE_RESOURCESENTRY._serialized_end=280 + _GETAVAILABLEINSTANCETYPESREQUEST._serialized_start=282 + _GETAVAILABLEINSTANCETYPESREQUEST._serialized_end=316 + _GETAVAILABLEINSTANCETYPESRESPONSE._serialized_start=318 + _GETAVAILABLEINSTANCETYPESRESPONSE._serialized_end=440 + _INSTANCE._serialized_start=443 + _INSTANCE._serialized_end=1134 + _INSTANCE_TOTALRESOURCESENTRY._serialized_start=820 + _INSTANCE_TOTALRESOURCESENTRY._serialized_end=873 + _INSTANCE_INSTANCESTATUS._serialized_start=876 + _INSTANCE_INSTANCESTATUS._serialized_end=1020 + _INSTANCE_RAYSTATUS._serialized_start=1022 + _INSTANCE_RAYSTATUS._serialized_end=1134 + _UPDATEINSTANCEMANAGERSTATEREQUEST._serialized_start=1137 + _UPDATEINSTANCEMANAGERSTATEREQUEST._serialized_end=1295 + _UPDATEINSTANCEMANAGERSTATEREPLY._serialized_start=1297 + _UPDATEINSTANCEMANAGERSTATEREPLY._serialized_end=1387 + _INSTANCEMANAGERSTATE._serialized_start=1389 + _INSTANCEMANAGERSTATE._serialized_end=1477 + _GETINSTANCEMANAGERSTATEREQUEST._serialized_start=1479 + _GETINSTANCEMANAGERSTATEREQUEST._serialized_end=1511 + _GETINSTANCEMANAGERSTATEREPLY._serialized_start=1513 + _GETINSTANCEMANAGERSTATEREPLY._serialized_end=1600 + _INSTANCEMANAGERSERVICE._serialized_start=1653 + _INSTANCEMANAGERSERVICE._serialized_end=2084 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/instance_manager_pb2_grpc.py b/python/ray/core/generated/instance_manager_pb2_grpc.py new file mode 100644 index 0000000000000..afbb367808ec6 --- /dev/null +++ b/python/ray/core/generated/instance_manager_pb2_grpc.py @@ -0,0 +1,132 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import instance_manager_pb2 as src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2 + + +class InstanceManagerServiceStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.GetInstanceManagerState = channel.unary_unary( + '/ray.rpc.autoscaler.InstanceManagerService/GetInstanceManagerState', + request_serializer=src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.GetInstanceManagerStateRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.GetInstanceManagerStateReply.FromString, + ) + self.UpdateInstanceManagerState = channel.unary_unary( + '/ray.rpc.autoscaler.InstanceManagerService/UpdateInstanceManagerState', + request_serializer=src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.UpdateInstanceManagerStateRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.UpdateInstanceManagerStateReply.FromString, + ) + self.GetAvailableInstanceTypes = channel.unary_unary( + '/ray.rpc.autoscaler.InstanceManagerService/GetAvailableInstanceTypes', + request_serializer=src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.GetAvailableInstanceTypesRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.GetAvailableInstanceTypesResponse.FromString, + ) + + +class InstanceManagerServiceServicer(object): + """Missing associated documentation comment in .proto file.""" + + def GetInstanceManagerState(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def UpdateInstanceManagerState(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetAvailableInstanceTypes(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_InstanceManagerServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'GetInstanceManagerState': grpc.unary_unary_rpc_method_handler( + servicer.GetInstanceManagerState, + request_deserializer=src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.GetInstanceManagerStateRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.GetInstanceManagerStateReply.SerializeToString, + ), + 'UpdateInstanceManagerState': grpc.unary_unary_rpc_method_handler( + servicer.UpdateInstanceManagerState, + request_deserializer=src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.UpdateInstanceManagerStateRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.UpdateInstanceManagerStateReply.SerializeToString, + ), + 'GetAvailableInstanceTypes': grpc.unary_unary_rpc_method_handler( + servicer.GetAvailableInstanceTypes, + request_deserializer=src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.GetAvailableInstanceTypesRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.GetAvailableInstanceTypesResponse.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.autoscaler.InstanceManagerService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class InstanceManagerService(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def GetInstanceManagerState(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.autoscaler.InstanceManagerService/GetInstanceManagerState', + src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.GetInstanceManagerStateRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.GetInstanceManagerStateReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def UpdateInstanceManagerState(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.autoscaler.InstanceManagerService/UpdateInstanceManagerState', + src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.UpdateInstanceManagerStateRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.UpdateInstanceManagerStateReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetAvailableInstanceTypes(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.autoscaler.InstanceManagerService/GetAvailableInstanceTypes', + src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.GetAvailableInstanceTypesRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_experimental_dot_instance__manager__pb2.GetAvailableInstanceTypesResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/job_agent_pb2.py b/python/ray/core/generated/job_agent_pb2.py new file mode 100644 index 0000000000000..d0c2d37a402dd --- /dev/null +++ b/python/ray/core/generated/job_agent_pb2.py @@ -0,0 +1,49 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/job_agent.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from . import agent_manager_pb2 as src_dot_ray_dot_protobuf_dot_agent__manager__pb2 + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n src/ray/protobuf/job_agent.proto\x12\x07ray.rpc\x1a$src/ray/protobuf/agent_manager.proto\"2\n\x17InitializeJobEnvRequest\x12\x17\n\x0fjob_description\x18\x01 \x01(\t\"k\n\x15InitializeJobEnvReply\x12\'\n\x06status\x18\x01 \x01(\x0e\x32\x17.ray.rpc.AgentRpcStatus\x12\x15\n\rerror_message\x18\x02 \x01(\t\x12\x12\n\ndriver_pid\x18\x03 \x01(\x05\x32g\n\x0fJobAgentService\x12T\n\x10InitializeJobEnv\x12 .ray.rpc.InitializeJobEnvRequest\x1a\x1e.ray.rpc.InitializeJobEnvReplyB\x03\xf8\x01\x01\x62\x06proto3') + + + +_INITIALIZEJOBENVREQUEST = DESCRIPTOR.message_types_by_name['InitializeJobEnvRequest'] +_INITIALIZEJOBENVREPLY = DESCRIPTOR.message_types_by_name['InitializeJobEnvReply'] +InitializeJobEnvRequest = _reflection.GeneratedProtocolMessageType('InitializeJobEnvRequest', (_message.Message,), { + 'DESCRIPTOR' : _INITIALIZEJOBENVREQUEST, + '__module__' : 'src.ray.protobuf.job_agent_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InitializeJobEnvRequest) + }) +_sym_db.RegisterMessage(InitializeJobEnvRequest) + +InitializeJobEnvReply = _reflection.GeneratedProtocolMessageType('InitializeJobEnvReply', (_message.Message,), { + 'DESCRIPTOR' : _INITIALIZEJOBENVREPLY, + '__module__' : 'src.ray.protobuf.job_agent_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InitializeJobEnvReply) + }) +_sym_db.RegisterMessage(InitializeJobEnvReply) + +_JOBAGENTSERVICE = DESCRIPTOR.services_by_name['JobAgentService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _INITIALIZEJOBENVREQUEST._serialized_start=83 + _INITIALIZEJOBENVREQUEST._serialized_end=133 + _INITIALIZEJOBENVREPLY._serialized_start=135 + _INITIALIZEJOBENVREPLY._serialized_end=242 + _JOBAGENTSERVICE._serialized_start=244 + _JOBAGENTSERVICE._serialized_end=347 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/job_agent_pb2_grpc.py b/python/ray/core/generated/job_agent_pb2_grpc.py new file mode 100644 index 0000000000000..49244a06338c4 --- /dev/null +++ b/python/ray/core/generated/job_agent_pb2_grpc.py @@ -0,0 +1,70 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import job_agent_pb2 as src_dot_ray_dot_protobuf_dot_job__agent__pb2 + + +class JobAgentServiceStub(object): + """Service for communicating with the job_agent.py process on a remote node. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.InitializeJobEnv = channel.unary_unary( + '/ray.rpc.JobAgentService/InitializeJobEnv', + request_serializer=src_dot_ray_dot_protobuf_dot_job__agent__pb2.InitializeJobEnvRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_job__agent__pb2.InitializeJobEnvReply.FromString, + ) + + +class JobAgentServiceServicer(object): + """Service for communicating with the job_agent.py process on a remote node. + """ + + def InitializeJobEnv(self, request, context): + """Initialize job env. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_JobAgentServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'InitializeJobEnv': grpc.unary_unary_rpc_method_handler( + servicer.InitializeJobEnv, + request_deserializer=src_dot_ray_dot_protobuf_dot_job__agent__pb2.InitializeJobEnvRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_job__agent__pb2.InitializeJobEnvReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.JobAgentService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class JobAgentService(object): + """Service for communicating with the job_agent.py process on a remote node. + """ + + @staticmethod + def InitializeJobEnv(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.JobAgentService/InitializeJobEnv', + src_dot_ray_dot_protobuf_dot_job__agent__pb2.InitializeJobEnvRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_job__agent__pb2.InitializeJobEnvReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/logging_pb2.py b/python/ray/core/generated/logging_pb2.py new file mode 100644 index 0000000000000..5c575c4f64747 --- /dev/null +++ b/python/ray/core/generated/logging_pb2.py @@ -0,0 +1,35 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/logging.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1esrc/ray/protobuf/logging.proto\x12\x07ray.rpc\"{\n\x08LogBatch\x12\n\n\x02ip\x18\x01 \x01(\t\x12\x0b\n\x03pid\x18\x02 \x01(\t\x12\x0e\n\x06job_id\x18\x03 \x01(\t\x12\x10\n\x08is_error\x18\x04 \x01(\x08\x12\r\n\x05lines\x18\x05 \x03(\t\x12\x12\n\nactor_name\x18\x06 \x01(\t\x12\x11\n\ttask_name\x18\x07 \x01(\tB\x03\xf8\x01\x01\x62\x06proto3') + + + +_LOGBATCH = DESCRIPTOR.message_types_by_name['LogBatch'] +LogBatch = _reflection.GeneratedProtocolMessageType('LogBatch', (_message.Message,), { + 'DESCRIPTOR' : _LOGBATCH, + '__module__' : 'src.ray.protobuf.logging_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.LogBatch) + }) +_sym_db.RegisterMessage(LogBatch) + +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _LOGBATCH._serialized_start=43 + _LOGBATCH._serialized_end=166 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/logging_pb2_grpc.py b/python/ray/core/generated/logging_pb2_grpc.py new file mode 100644 index 0000000000000..2daafffebfc81 --- /dev/null +++ b/python/ray/core/generated/logging_pb2_grpc.py @@ -0,0 +1,4 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + diff --git a/python/ray/core/generated/metrics_pb2.py b/python/ray/core/generated/metrics_pb2.py new file mode 100644 index 0000000000000..202f8202072af --- /dev/null +++ b/python/ray/core/generated/metrics_pb2.py @@ -0,0 +1,190 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: opencensus/proto/metrics/v1/metrics.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2 +from google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2 +from . import resource_pb2 as opencensus_dot_proto_dot_resource_dot_v1_dot_resource__pb2 + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)opencensus/proto/metrics/v1/metrics.proto\x12\x1bopencensus.proto.metrics.v1\x1a\x1fgoogle/protobuf/timestamp.proto\x1a\x1egoogle/protobuf/wrappers.proto\x1a+opencensus/proto/resource/v1/resource.proto\"\xc9\x01\n\x06Metric\x12H\n\x11metric_descriptor\x18\x01 \x01(\x0b\x32-.opencensus.proto.metrics.v1.MetricDescriptor\x12;\n\ntimeseries\x18\x02 \x03(\x0b\x32\'.opencensus.proto.metrics.v1.TimeSeries\x12\x38\n\x08resource\x18\x03 \x01(\x0b\x32&.opencensus.proto.resource.v1.Resource\"\xec\x02\n\x10MetricDescriptor\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x0c\n\x04unit\x18\x03 \x01(\t\x12@\n\x04type\x18\x04 \x01(\x0e\x32\x32.opencensus.proto.metrics.v1.MetricDescriptor.Type\x12\x39\n\nlabel_keys\x18\x05 \x03(\x0b\x32%.opencensus.proto.metrics.v1.LabelKey\"\xa9\x01\n\x04Type\x12\x0f\n\x0bUNSPECIFIED\x10\x00\x12\x0f\n\x0bGAUGE_INT64\x10\x01\x12\x10\n\x0cGAUGE_DOUBLE\x10\x02\x12\x16\n\x12GAUGE_DISTRIBUTION\x10\x03\x12\x14\n\x10\x43UMULATIVE_INT64\x10\x04\x12\x15\n\x11\x43UMULATIVE_DOUBLE\x10\x05\x12\x1b\n\x17\x43UMULATIVE_DISTRIBUTION\x10\x06\x12\x0b\n\x07SUMMARY\x10\x07\",\n\x08LabelKey\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\"\xb4\x01\n\nTimeSeries\x12\x33\n\x0fstart_timestamp\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12=\n\x0clabel_values\x18\x02 \x03(\x0b\x32\'.opencensus.proto.metrics.v1.LabelValue\x12\x32\n\x06points\x18\x03 \x03(\x0b\x32\".opencensus.proto.metrics.v1.Point\".\n\nLabelValue\x12\r\n\x05value\x18\x01 \x01(\t\x12\x11\n\thas_value\x18\x02 \x01(\x08\"\x80\x02\n\x05Point\x12-\n\ttimestamp\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x15\n\x0bint64_value\x18\x02 \x01(\x03H\x00\x12\x16\n\x0c\x64ouble_value\x18\x03 \x01(\x01H\x00\x12L\n\x12\x64istribution_value\x18\x04 \x01(\x0b\x32..opencensus.proto.metrics.v1.DistributionValueH\x00\x12\x42\n\rsummary_value\x18\x05 \x01(\x0b\x32).opencensus.proto.metrics.v1.SummaryValueH\x00\x42\x07\n\x05value\"\xc2\x05\n\x11\x44istributionValue\x12\r\n\x05\x63ount\x18\x01 \x01(\x03\x12\x0b\n\x03sum\x18\x02 \x01(\x01\x12 \n\x18sum_of_squared_deviation\x18\x03 \x01(\x01\x12T\n\x0e\x62ucket_options\x18\x04 \x01(\x0b\x32<.opencensus.proto.metrics.v1.DistributionValue.BucketOptions\x12\x46\n\x07\x62uckets\x18\x05 \x03(\x0b\x32\x35.opencensus.proto.metrics.v1.DistributionValue.Bucket\x1a\x8e\x01\n\rBucketOptions\x12Y\n\x08\x65xplicit\x18\x01 \x01(\x0b\x32\x45.opencensus.proto.metrics.v1.DistributionValue.BucketOptions.ExplicitH\x00\x1a\x1a\n\x08\x45xplicit\x12\x0e\n\x06\x62ounds\x18\x01 \x03(\x01\x42\x06\n\x04type\x1a\x62\n\x06\x42ucket\x12\r\n\x05\x63ount\x18\x01 \x01(\x03\x12I\n\x08\x65xemplar\x18\x02 \x01(\x0b\x32\x37.opencensus.proto.metrics.v1.DistributionValue.Exemplar\x1a\xdb\x01\n\x08\x45xemplar\x12\r\n\x05value\x18\x01 \x01(\x01\x12-\n\ttimestamp\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12]\n\x0b\x61ttachments\x18\x03 \x03(\x0b\x32H.opencensus.proto.metrics.v1.DistributionValue.Exemplar.AttachmentsEntry\x1a\x32\n\x10\x41ttachmentsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xa8\x03\n\x0cSummaryValue\x12*\n\x05\x63ount\x18\x01 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\x12)\n\x03sum\x18\x02 \x01(\x0b\x32\x1c.google.protobuf.DoubleValue\x12\x44\n\x08snapshot\x18\x03 \x01(\x0b\x32\x32.opencensus.proto.metrics.v1.SummaryValue.Snapshot\x1a\xfa\x01\n\x08Snapshot\x12*\n\x05\x63ount\x18\x01 \x01(\x0b\x32\x1b.google.protobuf.Int64Value\x12)\n\x03sum\x18\x02 \x01(\x0b\x32\x1c.google.protobuf.DoubleValue\x12_\n\x11percentile_values\x18\x03 \x03(\x0b\x32\x44.opencensus.proto.metrics.v1.SummaryValue.Snapshot.ValueAtPercentile\x1a\x36\n\x11ValueAtPercentile\x12\x12\n\npercentile\x18\x01 \x01(\x01\x12\r\n\x05value\x18\x02 \x01(\x01\x42\x94\x01\n\x1eio.opencensus.proto.metrics.v1B\x0cMetricsProtoP\x01ZDgithub.com/census-instrumentation/opencensus-proto/gen-go/metrics/v1\xea\x02\x1bOpenCensus.Proto.Metrics.V1b\x06proto3') + + + +_METRIC = DESCRIPTOR.message_types_by_name['Metric'] +_METRICDESCRIPTOR = DESCRIPTOR.message_types_by_name['MetricDescriptor'] +_LABELKEY = DESCRIPTOR.message_types_by_name['LabelKey'] +_TIMESERIES = DESCRIPTOR.message_types_by_name['TimeSeries'] +_LABELVALUE = DESCRIPTOR.message_types_by_name['LabelValue'] +_POINT = DESCRIPTOR.message_types_by_name['Point'] +_DISTRIBUTIONVALUE = DESCRIPTOR.message_types_by_name['DistributionValue'] +_DISTRIBUTIONVALUE_BUCKETOPTIONS = _DISTRIBUTIONVALUE.nested_types_by_name['BucketOptions'] +_DISTRIBUTIONVALUE_BUCKETOPTIONS_EXPLICIT = _DISTRIBUTIONVALUE_BUCKETOPTIONS.nested_types_by_name['Explicit'] +_DISTRIBUTIONVALUE_BUCKET = _DISTRIBUTIONVALUE.nested_types_by_name['Bucket'] +_DISTRIBUTIONVALUE_EXEMPLAR = _DISTRIBUTIONVALUE.nested_types_by_name['Exemplar'] +_DISTRIBUTIONVALUE_EXEMPLAR_ATTACHMENTSENTRY = _DISTRIBUTIONVALUE_EXEMPLAR.nested_types_by_name['AttachmentsEntry'] +_SUMMARYVALUE = DESCRIPTOR.message_types_by_name['SummaryValue'] +_SUMMARYVALUE_SNAPSHOT = _SUMMARYVALUE.nested_types_by_name['Snapshot'] +_SUMMARYVALUE_SNAPSHOT_VALUEATPERCENTILE = _SUMMARYVALUE_SNAPSHOT.nested_types_by_name['ValueAtPercentile'] +_METRICDESCRIPTOR_TYPE = _METRICDESCRIPTOR.enum_types_by_name['Type'] +Metric = _reflection.GeneratedProtocolMessageType('Metric', (_message.Message,), { + 'DESCRIPTOR' : _METRIC, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.Metric) + }) +_sym_db.RegisterMessage(Metric) + +MetricDescriptor = _reflection.GeneratedProtocolMessageType('MetricDescriptor', (_message.Message,), { + 'DESCRIPTOR' : _METRICDESCRIPTOR, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.MetricDescriptor) + }) +_sym_db.RegisterMessage(MetricDescriptor) + +LabelKey = _reflection.GeneratedProtocolMessageType('LabelKey', (_message.Message,), { + 'DESCRIPTOR' : _LABELKEY, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.LabelKey) + }) +_sym_db.RegisterMessage(LabelKey) + +TimeSeries = _reflection.GeneratedProtocolMessageType('TimeSeries', (_message.Message,), { + 'DESCRIPTOR' : _TIMESERIES, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.TimeSeries) + }) +_sym_db.RegisterMessage(TimeSeries) + +LabelValue = _reflection.GeneratedProtocolMessageType('LabelValue', (_message.Message,), { + 'DESCRIPTOR' : _LABELVALUE, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.LabelValue) + }) +_sym_db.RegisterMessage(LabelValue) + +Point = _reflection.GeneratedProtocolMessageType('Point', (_message.Message,), { + 'DESCRIPTOR' : _POINT, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.Point) + }) +_sym_db.RegisterMessage(Point) + +DistributionValue = _reflection.GeneratedProtocolMessageType('DistributionValue', (_message.Message,), { + + 'BucketOptions' : _reflection.GeneratedProtocolMessageType('BucketOptions', (_message.Message,), { + + 'Explicit' : _reflection.GeneratedProtocolMessageType('Explicit', (_message.Message,), { + 'DESCRIPTOR' : _DISTRIBUTIONVALUE_BUCKETOPTIONS_EXPLICIT, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.DistributionValue.BucketOptions.Explicit) + }) + , + 'DESCRIPTOR' : _DISTRIBUTIONVALUE_BUCKETOPTIONS, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.DistributionValue.BucketOptions) + }) + , + + 'Bucket' : _reflection.GeneratedProtocolMessageType('Bucket', (_message.Message,), { + 'DESCRIPTOR' : _DISTRIBUTIONVALUE_BUCKET, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.DistributionValue.Bucket) + }) + , + + 'Exemplar' : _reflection.GeneratedProtocolMessageType('Exemplar', (_message.Message,), { + + 'AttachmentsEntry' : _reflection.GeneratedProtocolMessageType('AttachmentsEntry', (_message.Message,), { + 'DESCRIPTOR' : _DISTRIBUTIONVALUE_EXEMPLAR_ATTACHMENTSENTRY, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.DistributionValue.Exemplar.AttachmentsEntry) + }) + , + 'DESCRIPTOR' : _DISTRIBUTIONVALUE_EXEMPLAR, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.DistributionValue.Exemplar) + }) + , + 'DESCRIPTOR' : _DISTRIBUTIONVALUE, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.DistributionValue) + }) +_sym_db.RegisterMessage(DistributionValue) +_sym_db.RegisterMessage(DistributionValue.BucketOptions) +_sym_db.RegisterMessage(DistributionValue.BucketOptions.Explicit) +_sym_db.RegisterMessage(DistributionValue.Bucket) +_sym_db.RegisterMessage(DistributionValue.Exemplar) +_sym_db.RegisterMessage(DistributionValue.Exemplar.AttachmentsEntry) + +SummaryValue = _reflection.GeneratedProtocolMessageType('SummaryValue', (_message.Message,), { + + 'Snapshot' : _reflection.GeneratedProtocolMessageType('Snapshot', (_message.Message,), { + + 'ValueAtPercentile' : _reflection.GeneratedProtocolMessageType('ValueAtPercentile', (_message.Message,), { + 'DESCRIPTOR' : _SUMMARYVALUE_SNAPSHOT_VALUEATPERCENTILE, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.SummaryValue.Snapshot.ValueAtPercentile) + }) + , + 'DESCRIPTOR' : _SUMMARYVALUE_SNAPSHOT, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.SummaryValue.Snapshot) + }) + , + 'DESCRIPTOR' : _SUMMARYVALUE, + '__module__' : 'opencensus.proto.metrics.v1.metrics_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.metrics.v1.SummaryValue) + }) +_sym_db.RegisterMessage(SummaryValue) +_sym_db.RegisterMessage(SummaryValue.Snapshot) +_sym_db.RegisterMessage(SummaryValue.Snapshot.ValueAtPercentile) + +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\n\036io.opencensus.proto.metrics.v1B\014MetricsProtoP\001ZDgithub.com/census-instrumentation/opencensus-proto/gen-go/metrics/v1\352\002\033OpenCensus.Proto.Metrics.V1' + _DISTRIBUTIONVALUE_EXEMPLAR_ATTACHMENTSENTRY._options = None + _DISTRIBUTIONVALUE_EXEMPLAR_ATTACHMENTSENTRY._serialized_options = b'8\001' + _METRIC._serialized_start=185 + _METRIC._serialized_end=386 + _METRICDESCRIPTOR._serialized_start=389 + _METRICDESCRIPTOR._serialized_end=753 + _METRICDESCRIPTOR_TYPE._serialized_start=584 + _METRICDESCRIPTOR_TYPE._serialized_end=753 + _LABELKEY._serialized_start=755 + _LABELKEY._serialized_end=799 + _TIMESERIES._serialized_start=802 + _TIMESERIES._serialized_end=982 + _LABELVALUE._serialized_start=984 + _LABELVALUE._serialized_end=1030 + _POINT._serialized_start=1033 + _POINT._serialized_end=1289 + _DISTRIBUTIONVALUE._serialized_start=1292 + _DISTRIBUTIONVALUE._serialized_end=1998 + _DISTRIBUTIONVALUE_BUCKETOPTIONS._serialized_start=1534 + _DISTRIBUTIONVALUE_BUCKETOPTIONS._serialized_end=1676 + _DISTRIBUTIONVALUE_BUCKETOPTIONS_EXPLICIT._serialized_start=1642 + _DISTRIBUTIONVALUE_BUCKETOPTIONS_EXPLICIT._serialized_end=1668 + _DISTRIBUTIONVALUE_BUCKET._serialized_start=1678 + _DISTRIBUTIONVALUE_BUCKET._serialized_end=1776 + _DISTRIBUTIONVALUE_EXEMPLAR._serialized_start=1779 + _DISTRIBUTIONVALUE_EXEMPLAR._serialized_end=1998 + _DISTRIBUTIONVALUE_EXEMPLAR_ATTACHMENTSENTRY._serialized_start=1948 + _DISTRIBUTIONVALUE_EXEMPLAR_ATTACHMENTSENTRY._serialized_end=1998 + _SUMMARYVALUE._serialized_start=2001 + _SUMMARYVALUE._serialized_end=2425 + _SUMMARYVALUE_SNAPSHOT._serialized_start=2175 + _SUMMARYVALUE_SNAPSHOT._serialized_end=2425 + _SUMMARYVALUE_SNAPSHOT_VALUEATPERCENTILE._serialized_start=2371 + _SUMMARYVALUE_SNAPSHOT_VALUEATPERCENTILE._serialized_end=2425 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/metrics_pb2_grpc.py b/python/ray/core/generated/metrics_pb2_grpc.py new file mode 100644 index 0000000000000..2daafffebfc81 --- /dev/null +++ b/python/ray/core/generated/metrics_pb2_grpc.py @@ -0,0 +1,4 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + diff --git a/python/ray/core/generated/monitor_pb2.py b/python/ray/core/generated/monitor_pb2.py new file mode 100644 index 0000000000000..efb117c198b28 --- /dev/null +++ b/python/ray/core/generated/monitor_pb2.py @@ -0,0 +1,160 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/monitor.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1esrc/ray/protobuf/monitor.proto\x12\x07ray.rpc\"\x16\n\x14GetRayVersionRequest\"%\n\x12GetRayVersionReply\x12\x0f\n\x07version\x18\x01 \x01(\t\"+\n\x17\x44rainAndKillNodeRequest\x12\x10\n\x08node_ids\x18\x01 \x03(\x0c\".\n\x15\x44rainAndKillNodeReply\x12\x15\n\rdrained_nodes\x18\x02 \x03(\x0c\"}\n\x0eResourceBundle\x12\x39\n\tresources\x18\x01 \x03(\x0b\x32&.ray.rpc.ResourceBundle.ResourcesEntry\x1a\x30\n\x0eResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\"\xc2\x02\n\x0fResourceRequest\x12K\n\x15resource_request_type\x18\x01 \x01(\x0e\x32,.ray.rpc.ResourceRequest.ResourceRequestType\x12\r\n\x05\x63ount\x18\x02 \x01(\x05\x12(\n\x07\x62undles\x18\x03 \x03(\x0b\x32\x17.ray.rpc.ResourceBundle\"\xa8\x01\n\x13ResourceRequestType\x12\x14\n\x10TASK_RESERVATION\x10\x00\x12\x1d\n\x19STRICT_SPREAD_RESERVATION\x10\x01\x12\x16\n\x12SPREAD_RESERVATION\x10\x02\x12\x14\n\x10PACK_RESERVATION\x10\x03\x12\x1b\n\x17STRICT_PACK_RESERVATION\x10\x04\x12\x11\n\rMIN_RESOURCES\x10\x05\"\xac\x02\n\nNodeStatus\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\x12\x0f\n\x07\x61\x64\x64ress\x18\x02 \x01(\t\x12H\n\x13\x61vailable_resources\x18\x03 \x03(\x0b\x32+.ray.rpc.NodeStatus.AvailableResourcesEntry\x12@\n\x0ftotal_resources\x18\x04 \x03(\x0b\x32\'.ray.rpc.NodeStatus.TotalResourcesEntry\x1a\x39\n\x17\x41vailableResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x1a\x35\n\x13TotalResourcesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\"\x1c\n\x1aGetSchedulingStatusRequest\"{\n\x18GetSchedulingStatusReply\x12\x33\n\x11resource_requests\x18\x01 \x03(\x0b\x32\x18.ray.rpc.ResourceRequest\x12*\n\rnode_statuses\x18\x02 \x03(\x0b\x32\x13.ray.rpc.NodeStatus2\x95\x02\n\x11MonitorGcsService\x12K\n\rGetRayVersion\x12\x1d.ray.rpc.GetRayVersionRequest\x1a\x1b.ray.rpc.GetRayVersionReply\x12T\n\x10\x44rainAndKillNode\x12 .ray.rpc.DrainAndKillNodeRequest\x1a\x1e.ray.rpc.DrainAndKillNodeReply\x12]\n\x13GetSchedulingStatus\x12#.ray.rpc.GetSchedulingStatusRequest\x1a!.ray.rpc.GetSchedulingStatusReplyB\x03\xf8\x01\x01\x62\x06proto3') + + + +_GETRAYVERSIONREQUEST = DESCRIPTOR.message_types_by_name['GetRayVersionRequest'] +_GETRAYVERSIONREPLY = DESCRIPTOR.message_types_by_name['GetRayVersionReply'] +_DRAINANDKILLNODEREQUEST = DESCRIPTOR.message_types_by_name['DrainAndKillNodeRequest'] +_DRAINANDKILLNODEREPLY = DESCRIPTOR.message_types_by_name['DrainAndKillNodeReply'] +_RESOURCEBUNDLE = DESCRIPTOR.message_types_by_name['ResourceBundle'] +_RESOURCEBUNDLE_RESOURCESENTRY = _RESOURCEBUNDLE.nested_types_by_name['ResourcesEntry'] +_RESOURCEREQUEST = DESCRIPTOR.message_types_by_name['ResourceRequest'] +_NODESTATUS = DESCRIPTOR.message_types_by_name['NodeStatus'] +_NODESTATUS_AVAILABLERESOURCESENTRY = _NODESTATUS.nested_types_by_name['AvailableResourcesEntry'] +_NODESTATUS_TOTALRESOURCESENTRY = _NODESTATUS.nested_types_by_name['TotalResourcesEntry'] +_GETSCHEDULINGSTATUSREQUEST = DESCRIPTOR.message_types_by_name['GetSchedulingStatusRequest'] +_GETSCHEDULINGSTATUSREPLY = DESCRIPTOR.message_types_by_name['GetSchedulingStatusReply'] +_RESOURCEREQUEST_RESOURCEREQUESTTYPE = _RESOURCEREQUEST.enum_types_by_name['ResourceRequestType'] +GetRayVersionRequest = _reflection.GeneratedProtocolMessageType('GetRayVersionRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETRAYVERSIONREQUEST, + '__module__' : 'src.ray.protobuf.monitor_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetRayVersionRequest) + }) +_sym_db.RegisterMessage(GetRayVersionRequest) + +GetRayVersionReply = _reflection.GeneratedProtocolMessageType('GetRayVersionReply', (_message.Message,), { + 'DESCRIPTOR' : _GETRAYVERSIONREPLY, + '__module__' : 'src.ray.protobuf.monitor_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetRayVersionReply) + }) +_sym_db.RegisterMessage(GetRayVersionReply) + +DrainAndKillNodeRequest = _reflection.GeneratedProtocolMessageType('DrainAndKillNodeRequest', (_message.Message,), { + 'DESCRIPTOR' : _DRAINANDKILLNODEREQUEST, + '__module__' : 'src.ray.protobuf.monitor_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DrainAndKillNodeRequest) + }) +_sym_db.RegisterMessage(DrainAndKillNodeRequest) + +DrainAndKillNodeReply = _reflection.GeneratedProtocolMessageType('DrainAndKillNodeReply', (_message.Message,), { + 'DESCRIPTOR' : _DRAINANDKILLNODEREPLY, + '__module__' : 'src.ray.protobuf.monitor_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DrainAndKillNodeReply) + }) +_sym_db.RegisterMessage(DrainAndKillNodeReply) + +ResourceBundle = _reflection.GeneratedProtocolMessageType('ResourceBundle', (_message.Message,), { + + 'ResourcesEntry' : _reflection.GeneratedProtocolMessageType('ResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCEBUNDLE_RESOURCESENTRY, + '__module__' : 'src.ray.protobuf.monitor_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceBundle.ResourcesEntry) + }) + , + 'DESCRIPTOR' : _RESOURCEBUNDLE, + '__module__' : 'src.ray.protobuf.monitor_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceBundle) + }) +_sym_db.RegisterMessage(ResourceBundle) +_sym_db.RegisterMessage(ResourceBundle.ResourcesEntry) + +ResourceRequest = _reflection.GeneratedProtocolMessageType('ResourceRequest', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCEREQUEST, + '__module__' : 'src.ray.protobuf.monitor_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ResourceRequest) + }) +_sym_db.RegisterMessage(ResourceRequest) + +NodeStatus = _reflection.GeneratedProtocolMessageType('NodeStatus', (_message.Message,), { + + 'AvailableResourcesEntry' : _reflection.GeneratedProtocolMessageType('AvailableResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _NODESTATUS_AVAILABLERESOURCESENTRY, + '__module__' : 'src.ray.protobuf.monitor_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NodeStatus.AvailableResourcesEntry) + }) + , + + 'TotalResourcesEntry' : _reflection.GeneratedProtocolMessageType('TotalResourcesEntry', (_message.Message,), { + 'DESCRIPTOR' : _NODESTATUS_TOTALRESOURCESENTRY, + '__module__' : 'src.ray.protobuf.monitor_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NodeStatus.TotalResourcesEntry) + }) + , + 'DESCRIPTOR' : _NODESTATUS, + '__module__' : 'src.ray.protobuf.monitor_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NodeStatus) + }) +_sym_db.RegisterMessage(NodeStatus) +_sym_db.RegisterMessage(NodeStatus.AvailableResourcesEntry) +_sym_db.RegisterMessage(NodeStatus.TotalResourcesEntry) + +GetSchedulingStatusRequest = _reflection.GeneratedProtocolMessageType('GetSchedulingStatusRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETSCHEDULINGSTATUSREQUEST, + '__module__' : 'src.ray.protobuf.monitor_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetSchedulingStatusRequest) + }) +_sym_db.RegisterMessage(GetSchedulingStatusRequest) + +GetSchedulingStatusReply = _reflection.GeneratedProtocolMessageType('GetSchedulingStatusReply', (_message.Message,), { + 'DESCRIPTOR' : _GETSCHEDULINGSTATUSREPLY, + '__module__' : 'src.ray.protobuf.monitor_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetSchedulingStatusReply) + }) +_sym_db.RegisterMessage(GetSchedulingStatusReply) + +_MONITORGCSSERVICE = DESCRIPTOR.services_by_name['MonitorGcsService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _RESOURCEBUNDLE_RESOURCESENTRY._options = None + _RESOURCEBUNDLE_RESOURCESENTRY._serialized_options = b'8\001' + _NODESTATUS_AVAILABLERESOURCESENTRY._options = None + _NODESTATUS_AVAILABLERESOURCESENTRY._serialized_options = b'8\001' + _NODESTATUS_TOTALRESOURCESENTRY._options = None + _NODESTATUS_TOTALRESOURCESENTRY._serialized_options = b'8\001' + _GETRAYVERSIONREQUEST._serialized_start=43 + _GETRAYVERSIONREQUEST._serialized_end=65 + _GETRAYVERSIONREPLY._serialized_start=67 + _GETRAYVERSIONREPLY._serialized_end=104 + _DRAINANDKILLNODEREQUEST._serialized_start=106 + _DRAINANDKILLNODEREQUEST._serialized_end=149 + _DRAINANDKILLNODEREPLY._serialized_start=151 + _DRAINANDKILLNODEREPLY._serialized_end=197 + _RESOURCEBUNDLE._serialized_start=199 + _RESOURCEBUNDLE._serialized_end=324 + _RESOURCEBUNDLE_RESOURCESENTRY._serialized_start=276 + _RESOURCEBUNDLE_RESOURCESENTRY._serialized_end=324 + _RESOURCEREQUEST._serialized_start=327 + _RESOURCEREQUEST._serialized_end=649 + _RESOURCEREQUEST_RESOURCEREQUESTTYPE._serialized_start=481 + _RESOURCEREQUEST_RESOURCEREQUESTTYPE._serialized_end=649 + _NODESTATUS._serialized_start=652 + _NODESTATUS._serialized_end=952 + _NODESTATUS_AVAILABLERESOURCESENTRY._serialized_start=840 + _NODESTATUS_AVAILABLERESOURCESENTRY._serialized_end=897 + _NODESTATUS_TOTALRESOURCESENTRY._serialized_start=899 + _NODESTATUS_TOTALRESOURCESENTRY._serialized_end=952 + _GETSCHEDULINGSTATUSREQUEST._serialized_start=954 + _GETSCHEDULINGSTATUSREQUEST._serialized_end=982 + _GETSCHEDULINGSTATUSREPLY._serialized_start=984 + _GETSCHEDULINGSTATUSREPLY._serialized_end=1107 + _MONITORGCSSERVICE._serialized_start=1110 + _MONITORGCSSERVICE._serialized_end=1387 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/monitor_pb2_grpc.py b/python/ray/core/generated/monitor_pb2_grpc.py new file mode 100644 index 0000000000000..79f6f5a569ebe --- /dev/null +++ b/python/ray/core/generated/monitor_pb2_grpc.py @@ -0,0 +1,141 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import monitor_pb2 as src_dot_ray_dot_protobuf_dot_monitor__pb2 + + +class MonitorGcsServiceStub(object): + """This service provides a stable interface for a monitor/autoscaler process to interact + with Ray. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.GetRayVersion = channel.unary_unary( + '/ray.rpc.MonitorGcsService/GetRayVersion', + request_serializer=src_dot_ray_dot_protobuf_dot_monitor__pb2.GetRayVersionRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_monitor__pb2.GetRayVersionReply.FromString, + ) + self.DrainAndKillNode = channel.unary_unary( + '/ray.rpc.MonitorGcsService/DrainAndKillNode', + request_serializer=src_dot_ray_dot_protobuf_dot_monitor__pb2.DrainAndKillNodeRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_monitor__pb2.DrainAndKillNodeReply.FromString, + ) + self.GetSchedulingStatus = channel.unary_unary( + '/ray.rpc.MonitorGcsService/GetSchedulingStatus', + request_serializer=src_dot_ray_dot_protobuf_dot_monitor__pb2.GetSchedulingStatusRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_monitor__pb2.GetSchedulingStatusReply.FromString, + ) + + +class MonitorGcsServiceServicer(object): + """This service provides a stable interface for a monitor/autoscaler process to interact + with Ray. + """ + + def GetRayVersion(self, request, context): + """Get the ray version of the service. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def DrainAndKillNode(self, request, context): + """Request that GCS drain and kill a node. This call is idempotent, and could + need to be retried if the head node fails. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetSchedulingStatus(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_MonitorGcsServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'GetRayVersion': grpc.unary_unary_rpc_method_handler( + servicer.GetRayVersion, + request_deserializer=src_dot_ray_dot_protobuf_dot_monitor__pb2.GetRayVersionRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_monitor__pb2.GetRayVersionReply.SerializeToString, + ), + 'DrainAndKillNode': grpc.unary_unary_rpc_method_handler( + servicer.DrainAndKillNode, + request_deserializer=src_dot_ray_dot_protobuf_dot_monitor__pb2.DrainAndKillNodeRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_monitor__pb2.DrainAndKillNodeReply.SerializeToString, + ), + 'GetSchedulingStatus': grpc.unary_unary_rpc_method_handler( + servicer.GetSchedulingStatus, + request_deserializer=src_dot_ray_dot_protobuf_dot_monitor__pb2.GetSchedulingStatusRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_monitor__pb2.GetSchedulingStatusReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.MonitorGcsService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class MonitorGcsService(object): + """This service provides a stable interface for a monitor/autoscaler process to interact + with Ray. + """ + + @staticmethod + def GetRayVersion(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.MonitorGcsService/GetRayVersion', + src_dot_ray_dot_protobuf_dot_monitor__pb2.GetRayVersionRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_monitor__pb2.GetRayVersionReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def DrainAndKillNode(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.MonitorGcsService/DrainAndKillNode', + src_dot_ray_dot_protobuf_dot_monitor__pb2.DrainAndKillNodeRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_monitor__pb2.DrainAndKillNodeReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetSchedulingStatus(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.MonitorGcsService/GetSchedulingStatus', + src_dot_ray_dot_protobuf_dot_monitor__pb2.GetSchedulingStatusRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_monitor__pb2.GetSchedulingStatusReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/node_manager_pb2.py b/python/ray/core/generated/node_manager_pb2.py new file mode 100644 index 0000000000000..a0edaccdac5f0 --- /dev/null +++ b/python/ray/core/generated/node_manager_pb2.py @@ -0,0 +1,514 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/node_manager.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from . import common_pb2 as src_dot_ray_dot_protobuf_dot_common__pb2 +from . import gcs_pb2 as src_dot_ray_dot_protobuf_dot_gcs__pb2 +from . import autoscaler_pb2 as src_dot_ray_dot_protobuf_dot_autoscaler__pb2 + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#src/ray/protobuf/node_manager.proto\x12\x07ray.rpc\x1a\x1dsrc/ray/protobuf/common.proto\x1a\x1asrc/ray/protobuf/gcs.proto\x1a!src/ray/protobuf/autoscaler.proto\"U\n\x13WorkerBacklogReport\x12(\n\rresource_spec\x18\x01 \x01(\x0b\x32\x11.ray.rpc.TaskSpec\x12\x14\n\x0c\x62\x61\x63klog_size\x18\x02 \x01(\x03\"f\n\x1aReportWorkerBacklogRequest\x12\x11\n\tworker_id\x18\x01 \x01(\x0c\x12\x35\n\x0f\x62\x61\x63klog_reports\x18\x02 \x03(\x0b\x32\x1c.ray.rpc.WorkerBacklogReport\"\x1a\n\x18ReportWorkerBacklogReply\"\x9b\x01\n\x19RequestWorkerLeaseRequest\x12(\n\rresource_spec\x18\x01 \x01(\x0b\x32\x11.ray.rpc.TaskSpec\x12\x14\n\x0c\x62\x61\x63klog_size\x18\x02 \x01(\x03\x12\x17\n\x0fgrant_or_reject\x18\x03 \x01(\x08\x12%\n\x1dis_selected_based_on_locality\x18\x04 \x01(\x08\"\xf6\x04\n\x17RequestWorkerLeaseReply\x12(\n\x0eworker_address\x18\x01 \x01(\x0b\x32\x10.ray.rpc.Address\x12\x31\n\x17retry_at_raylet_address\x18\x02 \x01(\x0b\x32\x10.ray.rpc.Address\x12\x33\n\x10resource_mapping\x18\x03 \x03(\x0b\x32\x19.ray.rpc.ResourceMapEntry\x12\x10\n\x08\x63\x61nceled\x18\x04 \x01(\x08\x12\x12\n\nworker_pid\x18\x06 \x01(\r\x12\x10\n\x08rejected\x18\x07 \x01(\x08\x12.\n\x0eresources_data\x18\x08 \x01(\x0b\x32\x16.ray.rpc.ResourcesData\x12L\n\x0c\x66\x61ilure_type\x18\t \x01(\x0e\x32\x36.ray.rpc.RequestWorkerLeaseReply.SchedulingFailureType\x12\"\n\x1ascheduling_failure_message\x18\n \x01(\t\"\xee\x01\n\x15SchedulingFailureType\x12\x0e\n\nNOT_FAILED\x10\x00\x12\x15\n\x11SCHEDULING_FAILED\x10\x01\x12\x30\n,SCHEDULING_CANCELLED_PLACEMENT_GROUP_REMOVED\x10\x02\x12\x31\n-SCHEDULING_CANCELLED_RUNTIME_ENV_SETUP_FAILED\x10\x03\x12!\n\x1dSCHEDULING_CANCELLED_INTENDED\x10\x04\x12&\n\"SCHEDULING_CANCELLED_UNSCHEDULABLE\x10\x05\"F\n\x1dPrepareBundleResourcesRequest\x12%\n\x0c\x62undle_specs\x18\x01 \x03(\x0b\x32\x0f.ray.rpc.Bundle\".\n\x1bPrepareBundleResourcesReply\x12\x0f\n\x07success\x18\x01 \x01(\x08\"E\n\x1c\x43ommitBundleResourcesRequest\x12%\n\x0c\x62undle_specs\x18\x01 \x03(\x0b\x32\x0f.ray.rpc.Bundle\"\x1c\n\x1a\x43ommitBundleResourcesReply\"D\n\x1c\x43\x61ncelResourceReserveRequest\x12$\n\x0b\x62undle_spec\x18\x01 \x01(\x0b\x32\x0f.ray.rpc.Bundle\"\x1c\n\x1a\x43\x61ncelResourceReserveReply\"p\n\x13ReturnWorkerRequest\x12\x13\n\x0bworker_port\x18\x01 \x01(\x05\x12\x11\n\tworker_id\x18\x02 \x01(\x0c\x12\x19\n\x11\x64isconnect_worker\x18\x03 \x01(\x08\x12\x16\n\x0eworker_exiting\x18\x04 \x01(\x08\"\x13\n\x11ReturnWorkerReply\"8\n\x1bReleaseUnusedWorkersRequest\x12\x19\n\x11worker_ids_in_use\x18\x01 \x03(\x0c\"\x1b\n\x19ReleaseUnusedWorkersReply\")\n\x15ShutdownRayletRequest\x12\x10\n\x08graceful\x18\x01 \x01(\x08\"\x15\n\x13ShutdownRayletReply\"+\n\x18\x43\x61ncelWorkerLeaseRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\x0c\")\n\x16\x43\x61ncelWorkerLeaseReply\x12\x0f\n\x07success\x18\x01 \x01(\x08\"~\n\x13PinObjectIDsRequest\x12\'\n\rowner_address\x18\x01 \x01(\x0b\x32\x10.ray.rpc.Address\x12\x12\n\nobject_ids\x18\x02 \x03(\x0c\x12\x19\n\x0cgenerator_id\x18\x03 \x01(\x0cH\x00\x88\x01\x01\x42\x0f\n\r_generator_id\"&\n\x11PinObjectIDsReply\x12\x11\n\tsuccesses\x18\x01 \x03(\x08\"2\n\x13GetNodeStatsRequest\x12\x1b\n\x13include_memory_info\x18\x01 \x01(\x08\"\xd0\x03\n\x10ObjectStoreStats\x12\x1a\n\x12spill_time_total_s\x18\x01 \x01(\x01\x12\x1b\n\x13spilled_bytes_total\x18\x02 \x01(\x03\x12\x1d\n\x15spilled_objects_total\x18\x03 \x01(\x03\x12\x1c\n\x14restore_time_total_s\x18\x04 \x01(\x01\x12\x1c\n\x14restored_bytes_total\x18\x05 \x01(\x03\x12\x1e\n\x16restored_objects_total\x18\x06 \x01(\x03\x12\x1f\n\x17object_store_bytes_used\x18\x07 \x01(\x03\x12 \n\x18object_store_bytes_avail\x18\x08 \x01(\x03\x12\'\n\x1fobject_store_bytes_primary_copy\x18\t \x01(\x03\x12#\n\x1bobject_store_bytes_fallback\x18\n \x01(\x03\x12\x19\n\x11num_local_objects\x18\x0b \x01(\x03\x12\x16\n\x0e\x63onsumed_bytes\x18\x0c \x01(\x03\x12\x1b\n\x13object_pulls_queued\x18\r \x01(\x08\x12\'\n\x1fnum_object_store_primary_copies\x18\x0e \x01(\x03\"\x8e\x01\n\x11GetNodeStatsReply\x12\x34\n\x12\x63ore_workers_stats\x18\x01 \x03(\x0b\x32\x18.ray.rpc.CoreWorkerStats\x12\x13\n\x0bnum_workers\x18\x03 \x01(\r\x12.\n\x0bstore_stats\x18\x06 \x01(\x0b\x32\x19.ray.rpc.ObjectStoreStats\"\x11\n\x0fGlobalGCRequest\"\x0f\n\rGlobalGCReply\"<\n\x1d\x46ormatGlobalMemoryInfoRequest\x12\x1b\n\x13include_memory_info\x18\x01 \x01(\x08\"e\n\x1b\x46ormatGlobalMemoryInfoReply\x12\x16\n\x0ememory_summary\x18\x01 \x01(\t\x12.\n\x0bstore_stats\x18\x02 \x01(\x0b\x32\x19.ray.rpc.ObjectStoreStats\"1\n\x1cRequestObjectSpillageRequest\x12\x11\n\tobject_id\x18\x01 \x01(\x0c\"Z\n\x1aRequestObjectSpillageReply\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x12\n\nobject_url\x18\x02 \x01(\t\x12\x17\n\x0fspilled_node_id\x18\x03 \x01(\x0c\"F\n\x1bReleaseUnusedBundlesRequest\x12\'\n\x0e\x62undles_in_use\x18\x01 \x03(\x0b\x32\x0f.ray.rpc.Bundle\"\x1b\n\x19ReleaseUnusedBundlesReply\"\x18\n\x16GetSystemConfigRequest\"-\n\x14GetSystemConfigReply\x12\x15\n\rsystem_config\x18\x01 \x01(\t\"E\n\x1aUpdateResourceUsageRequest\x12\'\n\x1fserialized_resource_usage_batch\x18\x01 \x01(\x0c\"\x1a\n\x18UpdateResourceUsageReply\"3\n\x13GetTasksInfoRequest\x12\x12\n\x05limit\x18\x01 \x01(\x03H\x00\x88\x01\x01\x42\x08\n\x06_limit\"u\n\x11GetTasksInfoReply\x12\x37\n\x17owned_task_info_entries\x18\x01 \x03(\x0b\x32\x16.ray.rpc.TaskInfoEntry\x12\x18\n\x10running_task_ids\x18\x02 \x03(\x0c\x12\r\n\x05total\x18\x03 \x01(\x03\"5\n\x15GetObjectsInfoRequest\x12\x12\n\x05limit\x18\x01 \x01(\x03H\x00\x88\x01\x01\x42\x08\n\x06_limit\"Z\n\x13GetObjectsInfoReply\x12\x34\n\x12\x63ore_workers_stats\x18\x01 \x03(\x0b\x32\x18.ray.rpc.CoreWorkerStats\x12\r\n\x05total\x18\x02 \x01(\x03\"\x18\n\x16GetResourceLoadRequest\"A\n\x14GetResourceLoadReply\x12)\n\tresources\x18\x01 \x01(\x0b\x32\x16.ray.rpc.ResourcesData\"\x19\n\x17NotifyGCSRestartRequest\"\x17\n\x15NotifyGCSRestartReply\"-\n\x1aGetTaskFailureCauseRequest\x12\x0f\n\x07task_id\x18\x01 \x01(\x0c\"~\n\x18GetTaskFailureCauseReply\x12\x31\n\rfailure_cause\x18\x01 \x01(\x0b\x32\x15.ray.rpc.RayErrorInfoH\x00\x88\x01\x01\x12\x1d\n\x15\x66\x61il_task_immediately\x18\x02 \x01(\x08\x42\x10\n\x0e_failure_cause\"a\n\x12\x44rainRayletRequest\x12\x33\n\x06reason\x18\x01 \x01(\x0e\x32#.ray.rpc.autoscaler.DrainNodeReason\x12\x16\n\x0ereason_message\x18\x02 \x01(\t\"\'\n\x10\x44rainRayletReply\x12\x13\n\x0bis_accepted\x18\x01 \x01(\x08\x32\xf2\x0f\n\x12NodeManagerService\x12]\n\x13UpdateResourceUsage\x12#.ray.rpc.UpdateResourceUsageRequest\x1a!.ray.rpc.UpdateResourceUsageReply\x12T\n\x10NotifyGCSRestart\x12 .ray.rpc.NotifyGCSRestartRequest\x1a\x1e.ray.rpc.NotifyGCSRestartReply\x12Q\n\x0fGetResourceLoad\x12\x1f.ray.rpc.GetResourceLoadRequest\x1a\x1d.ray.rpc.GetResourceLoadReply\x12Z\n\x12RequestWorkerLease\x12\".ray.rpc.RequestWorkerLeaseRequest\x1a .ray.rpc.RequestWorkerLeaseReply\x12]\n\x13ReportWorkerBacklog\x12#.ray.rpc.ReportWorkerBacklogRequest\x1a!.ray.rpc.ReportWorkerBacklogReply\x12H\n\x0cReturnWorker\x12\x1c.ray.rpc.ReturnWorkerRequest\x1a\x1a.ray.rpc.ReturnWorkerReply\x12`\n\x14ReleaseUnusedWorkers\x12$.ray.rpc.ReleaseUnusedWorkersRequest\x1a\".ray.rpc.ReleaseUnusedWorkersReply\x12N\n\x0eShutdownRaylet\x12\x1e.ray.rpc.ShutdownRayletRequest\x1a\x1c.ray.rpc.ShutdownRayletReply\x12\x45\n\x0b\x44rainRaylet\x12\x1b.ray.rpc.DrainRayletRequest\x1a\x19.ray.rpc.DrainRayletReply\x12\x66\n\x16PrepareBundleResources\x12&.ray.rpc.PrepareBundleResourcesRequest\x1a$.ray.rpc.PrepareBundleResourcesReply\x12\x63\n\x15\x43ommitBundleResources\x12%.ray.rpc.CommitBundleResourcesRequest\x1a#.ray.rpc.CommitBundleResourcesReply\x12\x63\n\x15\x43\x61ncelResourceReserve\x12%.ray.rpc.CancelResourceReserveRequest\x1a#.ray.rpc.CancelResourceReserveReply\x12W\n\x11\x43\x61ncelWorkerLease\x12!.ray.rpc.CancelWorkerLeaseRequest\x1a\x1f.ray.rpc.CancelWorkerLeaseReply\x12H\n\x0cPinObjectIDs\x12\x1c.ray.rpc.PinObjectIDsRequest\x1a\x1a.ray.rpc.PinObjectIDsReply\x12H\n\x0cGetNodeStats\x12\x1c.ray.rpc.GetNodeStatsRequest\x1a\x1a.ray.rpc.GetNodeStatsReply\x12<\n\x08GlobalGC\x12\x18.ray.rpc.GlobalGCRequest\x1a\x16.ray.rpc.GlobalGCReply\x12\x66\n\x16\x46ormatGlobalMemoryInfo\x12&.ray.rpc.FormatGlobalMemoryInfoRequest\x1a$.ray.rpc.FormatGlobalMemoryInfoReply\x12\x63\n\x15RequestObjectSpillage\x12%.ray.rpc.RequestObjectSpillageRequest\x1a#.ray.rpc.RequestObjectSpillageReply\x12`\n\x14ReleaseUnusedBundles\x12$.ray.rpc.ReleaseUnusedBundlesRequest\x1a\".ray.rpc.ReleaseUnusedBundlesReply\x12Q\n\x0fGetSystemConfig\x12\x1f.ray.rpc.GetSystemConfigRequest\x1a\x1d.ray.rpc.GetSystemConfigReply\x12H\n\x0cGetTasksInfo\x12\x1c.ray.rpc.GetTasksInfoRequest\x1a\x1a.ray.rpc.GetTasksInfoReply\x12N\n\x0eGetObjectsInfo\x12\x1e.ray.rpc.GetObjectsInfoRequest\x1a\x1c.ray.rpc.GetObjectsInfoReply\x12]\n\x13GetTaskFailureCause\x12#.ray.rpc.GetTaskFailureCauseRequest\x1a!.ray.rpc.GetTaskFailureCauseReplyB\x03\xf8\x01\x01\x62\x06proto3') + + + +_WORKERBACKLOGREPORT = DESCRIPTOR.message_types_by_name['WorkerBacklogReport'] +_REPORTWORKERBACKLOGREQUEST = DESCRIPTOR.message_types_by_name['ReportWorkerBacklogRequest'] +_REPORTWORKERBACKLOGREPLY = DESCRIPTOR.message_types_by_name['ReportWorkerBacklogReply'] +_REQUESTWORKERLEASEREQUEST = DESCRIPTOR.message_types_by_name['RequestWorkerLeaseRequest'] +_REQUESTWORKERLEASEREPLY = DESCRIPTOR.message_types_by_name['RequestWorkerLeaseReply'] +_PREPAREBUNDLERESOURCESREQUEST = DESCRIPTOR.message_types_by_name['PrepareBundleResourcesRequest'] +_PREPAREBUNDLERESOURCESREPLY = DESCRIPTOR.message_types_by_name['PrepareBundleResourcesReply'] +_COMMITBUNDLERESOURCESREQUEST = DESCRIPTOR.message_types_by_name['CommitBundleResourcesRequest'] +_COMMITBUNDLERESOURCESREPLY = DESCRIPTOR.message_types_by_name['CommitBundleResourcesReply'] +_CANCELRESOURCERESERVEREQUEST = DESCRIPTOR.message_types_by_name['CancelResourceReserveRequest'] +_CANCELRESOURCERESERVEREPLY = DESCRIPTOR.message_types_by_name['CancelResourceReserveReply'] +_RETURNWORKERREQUEST = DESCRIPTOR.message_types_by_name['ReturnWorkerRequest'] +_RETURNWORKERREPLY = DESCRIPTOR.message_types_by_name['ReturnWorkerReply'] +_RELEASEUNUSEDWORKERSREQUEST = DESCRIPTOR.message_types_by_name['ReleaseUnusedWorkersRequest'] +_RELEASEUNUSEDWORKERSREPLY = DESCRIPTOR.message_types_by_name['ReleaseUnusedWorkersReply'] +_SHUTDOWNRAYLETREQUEST = DESCRIPTOR.message_types_by_name['ShutdownRayletRequest'] +_SHUTDOWNRAYLETREPLY = DESCRIPTOR.message_types_by_name['ShutdownRayletReply'] +_CANCELWORKERLEASEREQUEST = DESCRIPTOR.message_types_by_name['CancelWorkerLeaseRequest'] +_CANCELWORKERLEASEREPLY = DESCRIPTOR.message_types_by_name['CancelWorkerLeaseReply'] +_PINOBJECTIDSREQUEST = DESCRIPTOR.message_types_by_name['PinObjectIDsRequest'] +_PINOBJECTIDSREPLY = DESCRIPTOR.message_types_by_name['PinObjectIDsReply'] +_GETNODESTATSREQUEST = DESCRIPTOR.message_types_by_name['GetNodeStatsRequest'] +_OBJECTSTORESTATS = DESCRIPTOR.message_types_by_name['ObjectStoreStats'] +_GETNODESTATSREPLY = DESCRIPTOR.message_types_by_name['GetNodeStatsReply'] +_GLOBALGCREQUEST = DESCRIPTOR.message_types_by_name['GlobalGCRequest'] +_GLOBALGCREPLY = DESCRIPTOR.message_types_by_name['GlobalGCReply'] +_FORMATGLOBALMEMORYINFOREQUEST = DESCRIPTOR.message_types_by_name['FormatGlobalMemoryInfoRequest'] +_FORMATGLOBALMEMORYINFOREPLY = DESCRIPTOR.message_types_by_name['FormatGlobalMemoryInfoReply'] +_REQUESTOBJECTSPILLAGEREQUEST = DESCRIPTOR.message_types_by_name['RequestObjectSpillageRequest'] +_REQUESTOBJECTSPILLAGEREPLY = DESCRIPTOR.message_types_by_name['RequestObjectSpillageReply'] +_RELEASEUNUSEDBUNDLESREQUEST = DESCRIPTOR.message_types_by_name['ReleaseUnusedBundlesRequest'] +_RELEASEUNUSEDBUNDLESREPLY = DESCRIPTOR.message_types_by_name['ReleaseUnusedBundlesReply'] +_GETSYSTEMCONFIGREQUEST = DESCRIPTOR.message_types_by_name['GetSystemConfigRequest'] +_GETSYSTEMCONFIGREPLY = DESCRIPTOR.message_types_by_name['GetSystemConfigReply'] +_UPDATERESOURCEUSAGEREQUEST = DESCRIPTOR.message_types_by_name['UpdateResourceUsageRequest'] +_UPDATERESOURCEUSAGEREPLY = DESCRIPTOR.message_types_by_name['UpdateResourceUsageReply'] +_GETTASKSINFOREQUEST = DESCRIPTOR.message_types_by_name['GetTasksInfoRequest'] +_GETTASKSINFOREPLY = DESCRIPTOR.message_types_by_name['GetTasksInfoReply'] +_GETOBJECTSINFOREQUEST = DESCRIPTOR.message_types_by_name['GetObjectsInfoRequest'] +_GETOBJECTSINFOREPLY = DESCRIPTOR.message_types_by_name['GetObjectsInfoReply'] +_GETRESOURCELOADREQUEST = DESCRIPTOR.message_types_by_name['GetResourceLoadRequest'] +_GETRESOURCELOADREPLY = DESCRIPTOR.message_types_by_name['GetResourceLoadReply'] +_NOTIFYGCSRESTARTREQUEST = DESCRIPTOR.message_types_by_name['NotifyGCSRestartRequest'] +_NOTIFYGCSRESTARTREPLY = DESCRIPTOR.message_types_by_name['NotifyGCSRestartReply'] +_GETTASKFAILURECAUSEREQUEST = DESCRIPTOR.message_types_by_name['GetTaskFailureCauseRequest'] +_GETTASKFAILURECAUSEREPLY = DESCRIPTOR.message_types_by_name['GetTaskFailureCauseReply'] +_DRAINRAYLETREQUEST = DESCRIPTOR.message_types_by_name['DrainRayletRequest'] +_DRAINRAYLETREPLY = DESCRIPTOR.message_types_by_name['DrainRayletReply'] +_REQUESTWORKERLEASEREPLY_SCHEDULINGFAILURETYPE = _REQUESTWORKERLEASEREPLY.enum_types_by_name['SchedulingFailureType'] +WorkerBacklogReport = _reflection.GeneratedProtocolMessageType('WorkerBacklogReport', (_message.Message,), { + 'DESCRIPTOR' : _WORKERBACKLOGREPORT, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WorkerBacklogReport) + }) +_sym_db.RegisterMessage(WorkerBacklogReport) + +ReportWorkerBacklogRequest = _reflection.GeneratedProtocolMessageType('ReportWorkerBacklogRequest', (_message.Message,), { + 'DESCRIPTOR' : _REPORTWORKERBACKLOGREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportWorkerBacklogRequest) + }) +_sym_db.RegisterMessage(ReportWorkerBacklogRequest) + +ReportWorkerBacklogReply = _reflection.GeneratedProtocolMessageType('ReportWorkerBacklogReply', (_message.Message,), { + 'DESCRIPTOR' : _REPORTWORKERBACKLOGREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportWorkerBacklogReply) + }) +_sym_db.RegisterMessage(ReportWorkerBacklogReply) + +RequestWorkerLeaseRequest = _reflection.GeneratedProtocolMessageType('RequestWorkerLeaseRequest', (_message.Message,), { + 'DESCRIPTOR' : _REQUESTWORKERLEASEREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RequestWorkerLeaseRequest) + }) +_sym_db.RegisterMessage(RequestWorkerLeaseRequest) + +RequestWorkerLeaseReply = _reflection.GeneratedProtocolMessageType('RequestWorkerLeaseReply', (_message.Message,), { + 'DESCRIPTOR' : _REQUESTWORKERLEASEREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RequestWorkerLeaseReply) + }) +_sym_db.RegisterMessage(RequestWorkerLeaseReply) + +PrepareBundleResourcesRequest = _reflection.GeneratedProtocolMessageType('PrepareBundleResourcesRequest', (_message.Message,), { + 'DESCRIPTOR' : _PREPAREBUNDLERESOURCESREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PrepareBundleResourcesRequest) + }) +_sym_db.RegisterMessage(PrepareBundleResourcesRequest) + +PrepareBundleResourcesReply = _reflection.GeneratedProtocolMessageType('PrepareBundleResourcesReply', (_message.Message,), { + 'DESCRIPTOR' : _PREPAREBUNDLERESOURCESREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PrepareBundleResourcesReply) + }) +_sym_db.RegisterMessage(PrepareBundleResourcesReply) + +CommitBundleResourcesRequest = _reflection.GeneratedProtocolMessageType('CommitBundleResourcesRequest', (_message.Message,), { + 'DESCRIPTOR' : _COMMITBUNDLERESOURCESREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CommitBundleResourcesRequest) + }) +_sym_db.RegisterMessage(CommitBundleResourcesRequest) + +CommitBundleResourcesReply = _reflection.GeneratedProtocolMessageType('CommitBundleResourcesReply', (_message.Message,), { + 'DESCRIPTOR' : _COMMITBUNDLERESOURCESREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CommitBundleResourcesReply) + }) +_sym_db.RegisterMessage(CommitBundleResourcesReply) + +CancelResourceReserveRequest = _reflection.GeneratedProtocolMessageType('CancelResourceReserveRequest', (_message.Message,), { + 'DESCRIPTOR' : _CANCELRESOURCERESERVEREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CancelResourceReserveRequest) + }) +_sym_db.RegisterMessage(CancelResourceReserveRequest) + +CancelResourceReserveReply = _reflection.GeneratedProtocolMessageType('CancelResourceReserveReply', (_message.Message,), { + 'DESCRIPTOR' : _CANCELRESOURCERESERVEREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CancelResourceReserveReply) + }) +_sym_db.RegisterMessage(CancelResourceReserveReply) + +ReturnWorkerRequest = _reflection.GeneratedProtocolMessageType('ReturnWorkerRequest', (_message.Message,), { + 'DESCRIPTOR' : _RETURNWORKERREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReturnWorkerRequest) + }) +_sym_db.RegisterMessage(ReturnWorkerRequest) + +ReturnWorkerReply = _reflection.GeneratedProtocolMessageType('ReturnWorkerReply', (_message.Message,), { + 'DESCRIPTOR' : _RETURNWORKERREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReturnWorkerReply) + }) +_sym_db.RegisterMessage(ReturnWorkerReply) + +ReleaseUnusedWorkersRequest = _reflection.GeneratedProtocolMessageType('ReleaseUnusedWorkersRequest', (_message.Message,), { + 'DESCRIPTOR' : _RELEASEUNUSEDWORKERSREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReleaseUnusedWorkersRequest) + }) +_sym_db.RegisterMessage(ReleaseUnusedWorkersRequest) + +ReleaseUnusedWorkersReply = _reflection.GeneratedProtocolMessageType('ReleaseUnusedWorkersReply', (_message.Message,), { + 'DESCRIPTOR' : _RELEASEUNUSEDWORKERSREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReleaseUnusedWorkersReply) + }) +_sym_db.RegisterMessage(ReleaseUnusedWorkersReply) + +ShutdownRayletRequest = _reflection.GeneratedProtocolMessageType('ShutdownRayletRequest', (_message.Message,), { + 'DESCRIPTOR' : _SHUTDOWNRAYLETREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ShutdownRayletRequest) + }) +_sym_db.RegisterMessage(ShutdownRayletRequest) + +ShutdownRayletReply = _reflection.GeneratedProtocolMessageType('ShutdownRayletReply', (_message.Message,), { + 'DESCRIPTOR' : _SHUTDOWNRAYLETREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ShutdownRayletReply) + }) +_sym_db.RegisterMessage(ShutdownRayletReply) + +CancelWorkerLeaseRequest = _reflection.GeneratedProtocolMessageType('CancelWorkerLeaseRequest', (_message.Message,), { + 'DESCRIPTOR' : _CANCELWORKERLEASEREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CancelWorkerLeaseRequest) + }) +_sym_db.RegisterMessage(CancelWorkerLeaseRequest) + +CancelWorkerLeaseReply = _reflection.GeneratedProtocolMessageType('CancelWorkerLeaseReply', (_message.Message,), { + 'DESCRIPTOR' : _CANCELWORKERLEASEREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CancelWorkerLeaseReply) + }) +_sym_db.RegisterMessage(CancelWorkerLeaseReply) + +PinObjectIDsRequest = _reflection.GeneratedProtocolMessageType('PinObjectIDsRequest', (_message.Message,), { + 'DESCRIPTOR' : _PINOBJECTIDSREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PinObjectIDsRequest) + }) +_sym_db.RegisterMessage(PinObjectIDsRequest) + +PinObjectIDsReply = _reflection.GeneratedProtocolMessageType('PinObjectIDsReply', (_message.Message,), { + 'DESCRIPTOR' : _PINOBJECTIDSREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PinObjectIDsReply) + }) +_sym_db.RegisterMessage(PinObjectIDsReply) + +GetNodeStatsRequest = _reflection.GeneratedProtocolMessageType('GetNodeStatsRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETNODESTATSREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetNodeStatsRequest) + }) +_sym_db.RegisterMessage(GetNodeStatsRequest) + +ObjectStoreStats = _reflection.GeneratedProtocolMessageType('ObjectStoreStats', (_message.Message,), { + 'DESCRIPTOR' : _OBJECTSTORESTATS, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ObjectStoreStats) + }) +_sym_db.RegisterMessage(ObjectStoreStats) + +GetNodeStatsReply = _reflection.GeneratedProtocolMessageType('GetNodeStatsReply', (_message.Message,), { + 'DESCRIPTOR' : _GETNODESTATSREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetNodeStatsReply) + }) +_sym_db.RegisterMessage(GetNodeStatsReply) + +GlobalGCRequest = _reflection.GeneratedProtocolMessageType('GlobalGCRequest', (_message.Message,), { + 'DESCRIPTOR' : _GLOBALGCREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GlobalGCRequest) + }) +_sym_db.RegisterMessage(GlobalGCRequest) + +GlobalGCReply = _reflection.GeneratedProtocolMessageType('GlobalGCReply', (_message.Message,), { + 'DESCRIPTOR' : _GLOBALGCREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GlobalGCReply) + }) +_sym_db.RegisterMessage(GlobalGCReply) + +FormatGlobalMemoryInfoRequest = _reflection.GeneratedProtocolMessageType('FormatGlobalMemoryInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _FORMATGLOBALMEMORYINFOREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.FormatGlobalMemoryInfoRequest) + }) +_sym_db.RegisterMessage(FormatGlobalMemoryInfoRequest) + +FormatGlobalMemoryInfoReply = _reflection.GeneratedProtocolMessageType('FormatGlobalMemoryInfoReply', (_message.Message,), { + 'DESCRIPTOR' : _FORMATGLOBALMEMORYINFOREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.FormatGlobalMemoryInfoReply) + }) +_sym_db.RegisterMessage(FormatGlobalMemoryInfoReply) + +RequestObjectSpillageRequest = _reflection.GeneratedProtocolMessageType('RequestObjectSpillageRequest', (_message.Message,), { + 'DESCRIPTOR' : _REQUESTOBJECTSPILLAGEREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RequestObjectSpillageRequest) + }) +_sym_db.RegisterMessage(RequestObjectSpillageRequest) + +RequestObjectSpillageReply = _reflection.GeneratedProtocolMessageType('RequestObjectSpillageReply', (_message.Message,), { + 'DESCRIPTOR' : _REQUESTOBJECTSPILLAGEREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RequestObjectSpillageReply) + }) +_sym_db.RegisterMessage(RequestObjectSpillageReply) + +ReleaseUnusedBundlesRequest = _reflection.GeneratedProtocolMessageType('ReleaseUnusedBundlesRequest', (_message.Message,), { + 'DESCRIPTOR' : _RELEASEUNUSEDBUNDLESREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReleaseUnusedBundlesRequest) + }) +_sym_db.RegisterMessage(ReleaseUnusedBundlesRequest) + +ReleaseUnusedBundlesReply = _reflection.GeneratedProtocolMessageType('ReleaseUnusedBundlesReply', (_message.Message,), { + 'DESCRIPTOR' : _RELEASEUNUSEDBUNDLESREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReleaseUnusedBundlesReply) + }) +_sym_db.RegisterMessage(ReleaseUnusedBundlesReply) + +GetSystemConfigRequest = _reflection.GeneratedProtocolMessageType('GetSystemConfigRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETSYSTEMCONFIGREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetSystemConfigRequest) + }) +_sym_db.RegisterMessage(GetSystemConfigRequest) + +GetSystemConfigReply = _reflection.GeneratedProtocolMessageType('GetSystemConfigReply', (_message.Message,), { + 'DESCRIPTOR' : _GETSYSTEMCONFIGREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetSystemConfigReply) + }) +_sym_db.RegisterMessage(GetSystemConfigReply) + +UpdateResourceUsageRequest = _reflection.GeneratedProtocolMessageType('UpdateResourceUsageRequest', (_message.Message,), { + 'DESCRIPTOR' : _UPDATERESOURCEUSAGEREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.UpdateResourceUsageRequest) + }) +_sym_db.RegisterMessage(UpdateResourceUsageRequest) + +UpdateResourceUsageReply = _reflection.GeneratedProtocolMessageType('UpdateResourceUsageReply', (_message.Message,), { + 'DESCRIPTOR' : _UPDATERESOURCEUSAGEREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.UpdateResourceUsageReply) + }) +_sym_db.RegisterMessage(UpdateResourceUsageReply) + +GetTasksInfoRequest = _reflection.GeneratedProtocolMessageType('GetTasksInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETTASKSINFOREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetTasksInfoRequest) + }) +_sym_db.RegisterMessage(GetTasksInfoRequest) + +GetTasksInfoReply = _reflection.GeneratedProtocolMessageType('GetTasksInfoReply', (_message.Message,), { + 'DESCRIPTOR' : _GETTASKSINFOREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetTasksInfoReply) + }) +_sym_db.RegisterMessage(GetTasksInfoReply) + +GetObjectsInfoRequest = _reflection.GeneratedProtocolMessageType('GetObjectsInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETOBJECTSINFOREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetObjectsInfoRequest) + }) +_sym_db.RegisterMessage(GetObjectsInfoRequest) + +GetObjectsInfoReply = _reflection.GeneratedProtocolMessageType('GetObjectsInfoReply', (_message.Message,), { + 'DESCRIPTOR' : _GETOBJECTSINFOREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetObjectsInfoReply) + }) +_sym_db.RegisterMessage(GetObjectsInfoReply) + +GetResourceLoadRequest = _reflection.GeneratedProtocolMessageType('GetResourceLoadRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETRESOURCELOADREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetResourceLoadRequest) + }) +_sym_db.RegisterMessage(GetResourceLoadRequest) + +GetResourceLoadReply = _reflection.GeneratedProtocolMessageType('GetResourceLoadReply', (_message.Message,), { + 'DESCRIPTOR' : _GETRESOURCELOADREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetResourceLoadReply) + }) +_sym_db.RegisterMessage(GetResourceLoadReply) + +NotifyGCSRestartRequest = _reflection.GeneratedProtocolMessageType('NotifyGCSRestartRequest', (_message.Message,), { + 'DESCRIPTOR' : _NOTIFYGCSRESTARTREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NotifyGCSRestartRequest) + }) +_sym_db.RegisterMessage(NotifyGCSRestartRequest) + +NotifyGCSRestartReply = _reflection.GeneratedProtocolMessageType('NotifyGCSRestartReply', (_message.Message,), { + 'DESCRIPTOR' : _NOTIFYGCSRESTARTREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.NotifyGCSRestartReply) + }) +_sym_db.RegisterMessage(NotifyGCSRestartReply) + +GetTaskFailureCauseRequest = _reflection.GeneratedProtocolMessageType('GetTaskFailureCauseRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETTASKFAILURECAUSEREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetTaskFailureCauseRequest) + }) +_sym_db.RegisterMessage(GetTaskFailureCauseRequest) + +GetTaskFailureCauseReply = _reflection.GeneratedProtocolMessageType('GetTaskFailureCauseReply', (_message.Message,), { + 'DESCRIPTOR' : _GETTASKFAILURECAUSEREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetTaskFailureCauseReply) + }) +_sym_db.RegisterMessage(GetTaskFailureCauseReply) + +DrainRayletRequest = _reflection.GeneratedProtocolMessageType('DrainRayletRequest', (_message.Message,), { + 'DESCRIPTOR' : _DRAINRAYLETREQUEST, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DrainRayletRequest) + }) +_sym_db.RegisterMessage(DrainRayletRequest) + +DrainRayletReply = _reflection.GeneratedProtocolMessageType('DrainRayletReply', (_message.Message,), { + 'DESCRIPTOR' : _DRAINRAYLETREPLY, + '__module__' : 'src.ray.protobuf.node_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DrainRayletReply) + }) +_sym_db.RegisterMessage(DrainRayletReply) + +_NODEMANAGERSERVICE = DESCRIPTOR.services_by_name['NodeManagerService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _WORKERBACKLOGREPORT._serialized_start=142 + _WORKERBACKLOGREPORT._serialized_end=227 + _REPORTWORKERBACKLOGREQUEST._serialized_start=229 + _REPORTWORKERBACKLOGREQUEST._serialized_end=331 + _REPORTWORKERBACKLOGREPLY._serialized_start=333 + _REPORTWORKERBACKLOGREPLY._serialized_end=359 + _REQUESTWORKERLEASEREQUEST._serialized_start=362 + _REQUESTWORKERLEASEREQUEST._serialized_end=517 + _REQUESTWORKERLEASEREPLY._serialized_start=520 + _REQUESTWORKERLEASEREPLY._serialized_end=1150 + _REQUESTWORKERLEASEREPLY_SCHEDULINGFAILURETYPE._serialized_start=912 + _REQUESTWORKERLEASEREPLY_SCHEDULINGFAILURETYPE._serialized_end=1150 + _PREPAREBUNDLERESOURCESREQUEST._serialized_start=1152 + _PREPAREBUNDLERESOURCESREQUEST._serialized_end=1222 + _PREPAREBUNDLERESOURCESREPLY._serialized_start=1224 + _PREPAREBUNDLERESOURCESREPLY._serialized_end=1270 + _COMMITBUNDLERESOURCESREQUEST._serialized_start=1272 + _COMMITBUNDLERESOURCESREQUEST._serialized_end=1341 + _COMMITBUNDLERESOURCESREPLY._serialized_start=1343 + _COMMITBUNDLERESOURCESREPLY._serialized_end=1371 + _CANCELRESOURCERESERVEREQUEST._serialized_start=1373 + _CANCELRESOURCERESERVEREQUEST._serialized_end=1441 + _CANCELRESOURCERESERVEREPLY._serialized_start=1443 + _CANCELRESOURCERESERVEREPLY._serialized_end=1471 + _RETURNWORKERREQUEST._serialized_start=1473 + _RETURNWORKERREQUEST._serialized_end=1585 + _RETURNWORKERREPLY._serialized_start=1587 + _RETURNWORKERREPLY._serialized_end=1606 + _RELEASEUNUSEDWORKERSREQUEST._serialized_start=1608 + _RELEASEUNUSEDWORKERSREQUEST._serialized_end=1664 + _RELEASEUNUSEDWORKERSREPLY._serialized_start=1666 + _RELEASEUNUSEDWORKERSREPLY._serialized_end=1693 + _SHUTDOWNRAYLETREQUEST._serialized_start=1695 + _SHUTDOWNRAYLETREQUEST._serialized_end=1736 + _SHUTDOWNRAYLETREPLY._serialized_start=1738 + _SHUTDOWNRAYLETREPLY._serialized_end=1759 + _CANCELWORKERLEASEREQUEST._serialized_start=1761 + _CANCELWORKERLEASEREQUEST._serialized_end=1804 + _CANCELWORKERLEASEREPLY._serialized_start=1806 + _CANCELWORKERLEASEREPLY._serialized_end=1847 + _PINOBJECTIDSREQUEST._serialized_start=1849 + _PINOBJECTIDSREQUEST._serialized_end=1975 + _PINOBJECTIDSREPLY._serialized_start=1977 + _PINOBJECTIDSREPLY._serialized_end=2015 + _GETNODESTATSREQUEST._serialized_start=2017 + _GETNODESTATSREQUEST._serialized_end=2067 + _OBJECTSTORESTATS._serialized_start=2070 + _OBJECTSTORESTATS._serialized_end=2534 + _GETNODESTATSREPLY._serialized_start=2537 + _GETNODESTATSREPLY._serialized_end=2679 + _GLOBALGCREQUEST._serialized_start=2681 + _GLOBALGCREQUEST._serialized_end=2698 + _GLOBALGCREPLY._serialized_start=2700 + _GLOBALGCREPLY._serialized_end=2715 + _FORMATGLOBALMEMORYINFOREQUEST._serialized_start=2717 + _FORMATGLOBALMEMORYINFOREQUEST._serialized_end=2777 + _FORMATGLOBALMEMORYINFOREPLY._serialized_start=2779 + _FORMATGLOBALMEMORYINFOREPLY._serialized_end=2880 + _REQUESTOBJECTSPILLAGEREQUEST._serialized_start=2882 + _REQUESTOBJECTSPILLAGEREQUEST._serialized_end=2931 + _REQUESTOBJECTSPILLAGEREPLY._serialized_start=2933 + _REQUESTOBJECTSPILLAGEREPLY._serialized_end=3023 + _RELEASEUNUSEDBUNDLESREQUEST._serialized_start=3025 + _RELEASEUNUSEDBUNDLESREQUEST._serialized_end=3095 + _RELEASEUNUSEDBUNDLESREPLY._serialized_start=3097 + _RELEASEUNUSEDBUNDLESREPLY._serialized_end=3124 + _GETSYSTEMCONFIGREQUEST._serialized_start=3126 + _GETSYSTEMCONFIGREQUEST._serialized_end=3150 + _GETSYSTEMCONFIGREPLY._serialized_start=3152 + _GETSYSTEMCONFIGREPLY._serialized_end=3197 + _UPDATERESOURCEUSAGEREQUEST._serialized_start=3199 + _UPDATERESOURCEUSAGEREQUEST._serialized_end=3268 + _UPDATERESOURCEUSAGEREPLY._serialized_start=3270 + _UPDATERESOURCEUSAGEREPLY._serialized_end=3296 + _GETTASKSINFOREQUEST._serialized_start=3298 + _GETTASKSINFOREQUEST._serialized_end=3349 + _GETTASKSINFOREPLY._serialized_start=3351 + _GETTASKSINFOREPLY._serialized_end=3468 + _GETOBJECTSINFOREQUEST._serialized_start=3470 + _GETOBJECTSINFOREQUEST._serialized_end=3523 + _GETOBJECTSINFOREPLY._serialized_start=3525 + _GETOBJECTSINFOREPLY._serialized_end=3615 + _GETRESOURCELOADREQUEST._serialized_start=3617 + _GETRESOURCELOADREQUEST._serialized_end=3641 + _GETRESOURCELOADREPLY._serialized_start=3643 + _GETRESOURCELOADREPLY._serialized_end=3708 + _NOTIFYGCSRESTARTREQUEST._serialized_start=3710 + _NOTIFYGCSRESTARTREQUEST._serialized_end=3735 + _NOTIFYGCSRESTARTREPLY._serialized_start=3737 + _NOTIFYGCSRESTARTREPLY._serialized_end=3760 + _GETTASKFAILURECAUSEREQUEST._serialized_start=3762 + _GETTASKFAILURECAUSEREQUEST._serialized_end=3807 + _GETTASKFAILURECAUSEREPLY._serialized_start=3809 + _GETTASKFAILURECAUSEREPLY._serialized_end=3935 + _DRAINRAYLETREQUEST._serialized_start=3937 + _DRAINRAYLETREQUEST._serialized_end=4034 + _DRAINRAYLETREPLY._serialized_start=4036 + _DRAINRAYLETREPLY._serialized_end=4075 + _NODEMANAGERSERVICE._serialized_start=4078 + _NODEMANAGERSERVICE._serialized_end=6112 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/node_manager_pb2_grpc.py b/python/ray/core/generated/node_manager_pb2_grpc.py new file mode 100644 index 0000000000000..0c11ad084c093 --- /dev/null +++ b/python/ray/core/generated/node_manager_pb2_grpc.py @@ -0,0 +1,828 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import node_manager_pb2 as src_dot_ray_dot_protobuf_dot_node__manager__pb2 + + +class NodeManagerServiceStub(object): + """Service for inter-node-manager communication. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.UpdateResourceUsage = channel.unary_unary( + '/ray.rpc.NodeManagerService/UpdateResourceUsage', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.UpdateResourceUsageRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.UpdateResourceUsageReply.FromString, + ) + self.NotifyGCSRestart = channel.unary_unary( + '/ray.rpc.NodeManagerService/NotifyGCSRestart', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.NotifyGCSRestartRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.NotifyGCSRestartReply.FromString, + ) + self.GetResourceLoad = channel.unary_unary( + '/ray.rpc.NodeManagerService/GetResourceLoad', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetResourceLoadRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetResourceLoadReply.FromString, + ) + self.RequestWorkerLease = channel.unary_unary( + '/ray.rpc.NodeManagerService/RequestWorkerLease', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.RequestWorkerLeaseRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.RequestWorkerLeaseReply.FromString, + ) + self.ReportWorkerBacklog = channel.unary_unary( + '/ray.rpc.NodeManagerService/ReportWorkerBacklog', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReportWorkerBacklogRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReportWorkerBacklogReply.FromString, + ) + self.ReturnWorker = channel.unary_unary( + '/ray.rpc.NodeManagerService/ReturnWorker', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReturnWorkerRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReturnWorkerReply.FromString, + ) + self.ReleaseUnusedWorkers = channel.unary_unary( + '/ray.rpc.NodeManagerService/ReleaseUnusedWorkers', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReleaseUnusedWorkersRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReleaseUnusedWorkersReply.FromString, + ) + self.ShutdownRaylet = channel.unary_unary( + '/ray.rpc.NodeManagerService/ShutdownRaylet', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ShutdownRayletRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ShutdownRayletReply.FromString, + ) + self.DrainRaylet = channel.unary_unary( + '/ray.rpc.NodeManagerService/DrainRaylet', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.DrainRayletRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.DrainRayletReply.FromString, + ) + self.PrepareBundleResources = channel.unary_unary( + '/ray.rpc.NodeManagerService/PrepareBundleResources', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.PrepareBundleResourcesRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.PrepareBundleResourcesReply.FromString, + ) + self.CommitBundleResources = channel.unary_unary( + '/ray.rpc.NodeManagerService/CommitBundleResources', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.CommitBundleResourcesRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.CommitBundleResourcesReply.FromString, + ) + self.CancelResourceReserve = channel.unary_unary( + '/ray.rpc.NodeManagerService/CancelResourceReserve', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.CancelResourceReserveRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.CancelResourceReserveReply.FromString, + ) + self.CancelWorkerLease = channel.unary_unary( + '/ray.rpc.NodeManagerService/CancelWorkerLease', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.CancelWorkerLeaseRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.CancelWorkerLeaseReply.FromString, + ) + self.PinObjectIDs = channel.unary_unary( + '/ray.rpc.NodeManagerService/PinObjectIDs', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.PinObjectIDsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.PinObjectIDsReply.FromString, + ) + self.GetNodeStats = channel.unary_unary( + '/ray.rpc.NodeManagerService/GetNodeStats', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetNodeStatsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetNodeStatsReply.FromString, + ) + self.GlobalGC = channel.unary_unary( + '/ray.rpc.NodeManagerService/GlobalGC', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GlobalGCRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GlobalGCReply.FromString, + ) + self.FormatGlobalMemoryInfo = channel.unary_unary( + '/ray.rpc.NodeManagerService/FormatGlobalMemoryInfo', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.FormatGlobalMemoryInfoRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.FormatGlobalMemoryInfoReply.FromString, + ) + self.RequestObjectSpillage = channel.unary_unary( + '/ray.rpc.NodeManagerService/RequestObjectSpillage', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.RequestObjectSpillageRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.RequestObjectSpillageReply.FromString, + ) + self.ReleaseUnusedBundles = channel.unary_unary( + '/ray.rpc.NodeManagerService/ReleaseUnusedBundles', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReleaseUnusedBundlesRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReleaseUnusedBundlesReply.FromString, + ) + self.GetSystemConfig = channel.unary_unary( + '/ray.rpc.NodeManagerService/GetSystemConfig', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetSystemConfigRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetSystemConfigReply.FromString, + ) + self.GetTasksInfo = channel.unary_unary( + '/ray.rpc.NodeManagerService/GetTasksInfo', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetTasksInfoRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetTasksInfoReply.FromString, + ) + self.GetObjectsInfo = channel.unary_unary( + '/ray.rpc.NodeManagerService/GetObjectsInfo', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetObjectsInfoRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetObjectsInfoReply.FromString, + ) + self.GetTaskFailureCause = channel.unary_unary( + '/ray.rpc.NodeManagerService/GetTaskFailureCause', + request_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetTaskFailureCauseRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetTaskFailureCauseReply.FromString, + ) + + +class NodeManagerServiceServicer(object): + """Service for inter-node-manager communication. + """ + + def UpdateResourceUsage(self, request, context): + """Update the node's view of the cluster resource usage. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def NotifyGCSRestart(self, request, context): + """Handle the case when GCS restarted. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetResourceLoad(self, request, context): + """Get the resource load of the raylet. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def RequestWorkerLease(self, request, context): + """Request a worker from the raylet. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ReportWorkerBacklog(self, request, context): + """Report task backlog information from a worker to the raylet + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ReturnWorker(self, request, context): + """Release a worker back to its raylet. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ReleaseUnusedWorkers(self, request, context): + """This method is only used by GCS, and the purpose is to release leased workers + that may be leaked. When GCS restarts, it doesn't know which workers it has leased + in the previous lifecycle. In this case, GCS will send a list of worker ids that + are still needed. And Raylet will release other leased workers. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ShutdownRaylet(self, request, context): + """/ Shutdown the raylet (node manager). + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def DrainRaylet(self, request, context): + """Request to drain the raylet. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def PrepareBundleResources(self, request, context): + """Request a raylet to lock resources for a bundle. + This is the first phase of 2PC protocol for atomic placement group creation. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def CommitBundleResources(self, request, context): + """Commit bundle resources to a raylet. + This is the second phase of 2PC protocol for atomic placement group creation. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def CancelResourceReserve(self, request, context): + """Return resource for the raylet. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def CancelWorkerLease(self, request, context): + """Cancel a pending lease request. This only returns success if the + lease request was not yet granted. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def PinObjectIDs(self, request, context): + """Pin the provided object IDs. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetNodeStats(self, request, context): + """Get the current node stats. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GlobalGC(self, request, context): + """Trigger garbage collection in all workers across the cluster. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def FormatGlobalMemoryInfo(self, request, context): + """Get global object reference stats in formatted form. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def RequestObjectSpillage(self, request, context): + """Ask the raylet to spill an object to external storage. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ReleaseUnusedBundles(self, request, context): + """This method is only used by GCS, and the purpose is to release bundles + that may be leaked. When GCS restarts, it doesn't know which bundles it has leased + in the previous lifecycle. In this case, GCS will send a list of bundles that + are still needed. And Raylet will release other bundles. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetSystemConfig(self, request, context): + """Get the system config. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetTasksInfo(self, request, context): + """[State API] Get the all task information of the node. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetObjectsInfo(self, request, context): + """[State API] Get the all object information of the node. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetTaskFailureCause(self, request, context): + """Gets the task execution result. May contain a result if + the task completed in error. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_NodeManagerServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'UpdateResourceUsage': grpc.unary_unary_rpc_method_handler( + servicer.UpdateResourceUsage, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.UpdateResourceUsageRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.UpdateResourceUsageReply.SerializeToString, + ), + 'NotifyGCSRestart': grpc.unary_unary_rpc_method_handler( + servicer.NotifyGCSRestart, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.NotifyGCSRestartRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.NotifyGCSRestartReply.SerializeToString, + ), + 'GetResourceLoad': grpc.unary_unary_rpc_method_handler( + servicer.GetResourceLoad, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetResourceLoadRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetResourceLoadReply.SerializeToString, + ), + 'RequestWorkerLease': grpc.unary_unary_rpc_method_handler( + servicer.RequestWorkerLease, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.RequestWorkerLeaseRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.RequestWorkerLeaseReply.SerializeToString, + ), + 'ReportWorkerBacklog': grpc.unary_unary_rpc_method_handler( + servicer.ReportWorkerBacklog, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReportWorkerBacklogRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReportWorkerBacklogReply.SerializeToString, + ), + 'ReturnWorker': grpc.unary_unary_rpc_method_handler( + servicer.ReturnWorker, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReturnWorkerRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReturnWorkerReply.SerializeToString, + ), + 'ReleaseUnusedWorkers': grpc.unary_unary_rpc_method_handler( + servicer.ReleaseUnusedWorkers, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReleaseUnusedWorkersRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReleaseUnusedWorkersReply.SerializeToString, + ), + 'ShutdownRaylet': grpc.unary_unary_rpc_method_handler( + servicer.ShutdownRaylet, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ShutdownRayletRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ShutdownRayletReply.SerializeToString, + ), + 'DrainRaylet': grpc.unary_unary_rpc_method_handler( + servicer.DrainRaylet, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.DrainRayletRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.DrainRayletReply.SerializeToString, + ), + 'PrepareBundleResources': grpc.unary_unary_rpc_method_handler( + servicer.PrepareBundleResources, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.PrepareBundleResourcesRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.PrepareBundleResourcesReply.SerializeToString, + ), + 'CommitBundleResources': grpc.unary_unary_rpc_method_handler( + servicer.CommitBundleResources, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.CommitBundleResourcesRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.CommitBundleResourcesReply.SerializeToString, + ), + 'CancelResourceReserve': grpc.unary_unary_rpc_method_handler( + servicer.CancelResourceReserve, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.CancelResourceReserveRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.CancelResourceReserveReply.SerializeToString, + ), + 'CancelWorkerLease': grpc.unary_unary_rpc_method_handler( + servicer.CancelWorkerLease, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.CancelWorkerLeaseRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.CancelWorkerLeaseReply.SerializeToString, + ), + 'PinObjectIDs': grpc.unary_unary_rpc_method_handler( + servicer.PinObjectIDs, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.PinObjectIDsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.PinObjectIDsReply.SerializeToString, + ), + 'GetNodeStats': grpc.unary_unary_rpc_method_handler( + servicer.GetNodeStats, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetNodeStatsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetNodeStatsReply.SerializeToString, + ), + 'GlobalGC': grpc.unary_unary_rpc_method_handler( + servicer.GlobalGC, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GlobalGCRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GlobalGCReply.SerializeToString, + ), + 'FormatGlobalMemoryInfo': grpc.unary_unary_rpc_method_handler( + servicer.FormatGlobalMemoryInfo, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.FormatGlobalMemoryInfoRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.FormatGlobalMemoryInfoReply.SerializeToString, + ), + 'RequestObjectSpillage': grpc.unary_unary_rpc_method_handler( + servicer.RequestObjectSpillage, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.RequestObjectSpillageRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.RequestObjectSpillageReply.SerializeToString, + ), + 'ReleaseUnusedBundles': grpc.unary_unary_rpc_method_handler( + servicer.ReleaseUnusedBundles, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReleaseUnusedBundlesRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReleaseUnusedBundlesReply.SerializeToString, + ), + 'GetSystemConfig': grpc.unary_unary_rpc_method_handler( + servicer.GetSystemConfig, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetSystemConfigRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetSystemConfigReply.SerializeToString, + ), + 'GetTasksInfo': grpc.unary_unary_rpc_method_handler( + servicer.GetTasksInfo, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetTasksInfoRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetTasksInfoReply.SerializeToString, + ), + 'GetObjectsInfo': grpc.unary_unary_rpc_method_handler( + servicer.GetObjectsInfo, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetObjectsInfoRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetObjectsInfoReply.SerializeToString, + ), + 'GetTaskFailureCause': grpc.unary_unary_rpc_method_handler( + servicer.GetTaskFailureCause, + request_deserializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetTaskFailureCauseRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetTaskFailureCauseReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.NodeManagerService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class NodeManagerService(object): + """Service for inter-node-manager communication. + """ + + @staticmethod + def UpdateResourceUsage(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/UpdateResourceUsage', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.UpdateResourceUsageRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.UpdateResourceUsageReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def NotifyGCSRestart(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/NotifyGCSRestart', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.NotifyGCSRestartRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.NotifyGCSRestartReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetResourceLoad(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/GetResourceLoad', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetResourceLoadRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetResourceLoadReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def RequestWorkerLease(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/RequestWorkerLease', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.RequestWorkerLeaseRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.RequestWorkerLeaseReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ReportWorkerBacklog(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/ReportWorkerBacklog', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReportWorkerBacklogRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReportWorkerBacklogReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ReturnWorker(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/ReturnWorker', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReturnWorkerRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReturnWorkerReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ReleaseUnusedWorkers(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/ReleaseUnusedWorkers', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReleaseUnusedWorkersRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReleaseUnusedWorkersReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ShutdownRaylet(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/ShutdownRaylet', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.ShutdownRayletRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.ShutdownRayletReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def DrainRaylet(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/DrainRaylet', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.DrainRayletRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.DrainRayletReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def PrepareBundleResources(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/PrepareBundleResources', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.PrepareBundleResourcesRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.PrepareBundleResourcesReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def CommitBundleResources(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/CommitBundleResources', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.CommitBundleResourcesRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.CommitBundleResourcesReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def CancelResourceReserve(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/CancelResourceReserve', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.CancelResourceReserveRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.CancelResourceReserveReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def CancelWorkerLease(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/CancelWorkerLease', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.CancelWorkerLeaseRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.CancelWorkerLeaseReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def PinObjectIDs(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/PinObjectIDs', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.PinObjectIDsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.PinObjectIDsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetNodeStats(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/GetNodeStats', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetNodeStatsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetNodeStatsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GlobalGC(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/GlobalGC', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GlobalGCRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GlobalGCReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def FormatGlobalMemoryInfo(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/FormatGlobalMemoryInfo', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.FormatGlobalMemoryInfoRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.FormatGlobalMemoryInfoReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def RequestObjectSpillage(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/RequestObjectSpillage', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.RequestObjectSpillageRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.RequestObjectSpillageReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ReleaseUnusedBundles(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/ReleaseUnusedBundles', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReleaseUnusedBundlesRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.ReleaseUnusedBundlesReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetSystemConfig(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/GetSystemConfig', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetSystemConfigRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetSystemConfigReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetTasksInfo(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/GetTasksInfo', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetTasksInfoRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetTasksInfoReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetObjectsInfo(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/GetObjectsInfo', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetObjectsInfoRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetObjectsInfoReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetTaskFailureCause(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.NodeManagerService/GetTaskFailureCause', + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetTaskFailureCauseRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_node__manager__pb2.GetTaskFailureCauseReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/object_manager_pb2.py b/python/ray/core/generated/object_manager_pb2.py new file mode 100644 index 0000000000000..5be2603ab7548 --- /dev/null +++ b/python/ray/core/generated/object_manager_pb2.py @@ -0,0 +1,89 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/object_manager.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from . import common_pb2 as src_dot_ray_dot_protobuf_dot_common__pb2 + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%src/ray/protobuf/object_manager.proto\x12\x07ray.rpc\x1a\x1dsrc/ray/protobuf/common.proto\"\xb8\x01\n\x0bPushRequest\x12\x0f\n\x07push_id\x18\x01 \x01(\x0c\x12\x11\n\tobject_id\x18\x02 \x01(\x0c\x12\x0f\n\x07node_id\x18\x03 \x01(\x0c\x12\'\n\rowner_address\x18\x04 \x01(\x0b\x32\x10.ray.rpc.Address\x12\x13\n\x0b\x63hunk_index\x18\x05 \x01(\r\x12\x11\n\tdata_size\x18\x06 \x01(\x04\x12\x15\n\rmetadata_size\x18\x07 \x01(\x04\x12\x0c\n\x04\x64\x61ta\x18\x08 \x01(\x0c\"1\n\x0bPullRequest\x12\x0f\n\x07node_id\x18\x01 \x01(\x0c\x12\x11\n\tobject_id\x18\x02 \x01(\x0c\"(\n\x12\x46reeObjectsRequest\x12\x12\n\nobject_ids\x18\x01 \x03(\x0c\"\x0b\n\tPushReply\"\x0b\n\tPullReply\"\x12\n\x10\x46reeObjectsReply2\xc1\x01\n\x14ObjectManagerService\x12\x30\n\x04Push\x12\x14.ray.rpc.PushRequest\x1a\x12.ray.rpc.PushReply\x12\x30\n\x04Pull\x12\x14.ray.rpc.PullRequest\x1a\x12.ray.rpc.PullReply\x12\x45\n\x0b\x46reeObjects\x12\x1b.ray.rpc.FreeObjectsRequest\x1a\x19.ray.rpc.FreeObjectsReplyB\x03\xf8\x01\x01\x62\x06proto3') + + + +_PUSHREQUEST = DESCRIPTOR.message_types_by_name['PushRequest'] +_PULLREQUEST = DESCRIPTOR.message_types_by_name['PullRequest'] +_FREEOBJECTSREQUEST = DESCRIPTOR.message_types_by_name['FreeObjectsRequest'] +_PUSHREPLY = DESCRIPTOR.message_types_by_name['PushReply'] +_PULLREPLY = DESCRIPTOR.message_types_by_name['PullReply'] +_FREEOBJECTSREPLY = DESCRIPTOR.message_types_by_name['FreeObjectsReply'] +PushRequest = _reflection.GeneratedProtocolMessageType('PushRequest', (_message.Message,), { + 'DESCRIPTOR' : _PUSHREQUEST, + '__module__' : 'src.ray.protobuf.object_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PushRequest) + }) +_sym_db.RegisterMessage(PushRequest) + +PullRequest = _reflection.GeneratedProtocolMessageType('PullRequest', (_message.Message,), { + 'DESCRIPTOR' : _PULLREQUEST, + '__module__' : 'src.ray.protobuf.object_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PullRequest) + }) +_sym_db.RegisterMessage(PullRequest) + +FreeObjectsRequest = _reflection.GeneratedProtocolMessageType('FreeObjectsRequest', (_message.Message,), { + 'DESCRIPTOR' : _FREEOBJECTSREQUEST, + '__module__' : 'src.ray.protobuf.object_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.FreeObjectsRequest) + }) +_sym_db.RegisterMessage(FreeObjectsRequest) + +PushReply = _reflection.GeneratedProtocolMessageType('PushReply', (_message.Message,), { + 'DESCRIPTOR' : _PUSHREPLY, + '__module__' : 'src.ray.protobuf.object_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PushReply) + }) +_sym_db.RegisterMessage(PushReply) + +PullReply = _reflection.GeneratedProtocolMessageType('PullReply', (_message.Message,), { + 'DESCRIPTOR' : _PULLREPLY, + '__module__' : 'src.ray.protobuf.object_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PullReply) + }) +_sym_db.RegisterMessage(PullReply) + +FreeObjectsReply = _reflection.GeneratedProtocolMessageType('FreeObjectsReply', (_message.Message,), { + 'DESCRIPTOR' : _FREEOBJECTSREPLY, + '__module__' : 'src.ray.protobuf.object_manager_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.FreeObjectsReply) + }) +_sym_db.RegisterMessage(FreeObjectsReply) + +_OBJECTMANAGERSERVICE = DESCRIPTOR.services_by_name['ObjectManagerService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _PUSHREQUEST._serialized_start=82 + _PUSHREQUEST._serialized_end=266 + _PULLREQUEST._serialized_start=268 + _PULLREQUEST._serialized_end=317 + _FREEOBJECTSREQUEST._serialized_start=319 + _FREEOBJECTSREQUEST._serialized_end=359 + _PUSHREPLY._serialized_start=361 + _PUSHREPLY._serialized_end=372 + _PULLREPLY._serialized_start=374 + _PULLREPLY._serialized_end=385 + _FREEOBJECTSREPLY._serialized_start=387 + _FREEOBJECTSREPLY._serialized_end=405 + _OBJECTMANAGERSERVICE._serialized_start=408 + _OBJECTMANAGERSERVICE._serialized_end=601 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/object_manager_pb2_grpc.py b/python/ray/core/generated/object_manager_pb2_grpc.py new file mode 100644 index 0000000000000..c8b5309015f5f --- /dev/null +++ b/python/ray/core/generated/object_manager_pb2_grpc.py @@ -0,0 +1,135 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import object_manager_pb2 as src_dot_ray_dot_protobuf_dot_object__manager__pb2 + + +class ObjectManagerServiceStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.Push = channel.unary_unary( + '/ray.rpc.ObjectManagerService/Push', + request_serializer=src_dot_ray_dot_protobuf_dot_object__manager__pb2.PushRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_object__manager__pb2.PushReply.FromString, + ) + self.Pull = channel.unary_unary( + '/ray.rpc.ObjectManagerService/Pull', + request_serializer=src_dot_ray_dot_protobuf_dot_object__manager__pb2.PullRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_object__manager__pb2.PullReply.FromString, + ) + self.FreeObjects = channel.unary_unary( + '/ray.rpc.ObjectManagerService/FreeObjects', + request_serializer=src_dot_ray_dot_protobuf_dot_object__manager__pb2.FreeObjectsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_object__manager__pb2.FreeObjectsReply.FromString, + ) + + +class ObjectManagerServiceServicer(object): + """Missing associated documentation comment in .proto file.""" + + def Push(self, request, context): + """Push service used to send object chunks + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Pull(self, request, context): + """Try to pull object from remote object manager + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def FreeObjects(self, request, context): + """Tell remote object manager to free some objects + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_ObjectManagerServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'Push': grpc.unary_unary_rpc_method_handler( + servicer.Push, + request_deserializer=src_dot_ray_dot_protobuf_dot_object__manager__pb2.PushRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_object__manager__pb2.PushReply.SerializeToString, + ), + 'Pull': grpc.unary_unary_rpc_method_handler( + servicer.Pull, + request_deserializer=src_dot_ray_dot_protobuf_dot_object__manager__pb2.PullRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_object__manager__pb2.PullReply.SerializeToString, + ), + 'FreeObjects': grpc.unary_unary_rpc_method_handler( + servicer.FreeObjects, + request_deserializer=src_dot_ray_dot_protobuf_dot_object__manager__pb2.FreeObjectsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_object__manager__pb2.FreeObjectsReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.ObjectManagerService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class ObjectManagerService(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def Push(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ObjectManagerService/Push', + src_dot_ray_dot_protobuf_dot_object__manager__pb2.PushRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_object__manager__pb2.PushReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def Pull(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ObjectManagerService/Pull', + src_dot_ray_dot_protobuf_dot_object__manager__pb2.PullRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_object__manager__pb2.PullReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def FreeObjects(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ObjectManagerService/FreeObjects', + src_dot_ray_dot_protobuf_dot_object__manager__pb2.FreeObjectsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_object__manager__pb2.FreeObjectsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/pubsub_pb2.py b/python/ray/core/generated/pubsub_pb2.py new file mode 100644 index 0000000000000..c8638501bf149 --- /dev/null +++ b/python/ray/core/generated/pubsub_pb2.py @@ -0,0 +1,198 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/pubsub.proto +"""Generated protocol buffer code.""" +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from . import common_pb2 as src_dot_ray_dot_protobuf_dot_common__pb2 +from . import dependency_pb2 as src_dot_ray_dot_protobuf_dot_dependency__pb2 +from . import gcs_pb2 as src_dot_ray_dot_protobuf_dot_gcs__pb2 +from . import logging_pb2 as src_dot_ray_dot_protobuf_dot_logging__pb2 + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1dsrc/ray/protobuf/pubsub.proto\x12\x07ray.rpc\x1a\x1dsrc/ray/protobuf/common.proto\x1a!src/ray/protobuf/dependency.proto\x1a\x1asrc/ray/protobuf/gcs.proto\x1a\x1esrc/ray/protobuf/logging.proto\"\x80\x07\n\nPubMessage\x12*\n\x0c\x63hannel_type\x18\x01 \x01(\x0e\x32\x14.ray.rpc.ChannelType\x12\x0e\n\x06key_id\x18\x02 \x01(\x0c\x12N\n\x1eworker_object_eviction_message\x18\x03 \x01(\x0b\x32$.ray.rpc.WorkerObjectEvictionMessageH\x00\x12\x46\n\x1aworker_ref_removed_message\x18\x04 \x01(\x0b\x32 .ray.rpc.WorkerRefRemovedMessageH\x00\x12S\n\x1fworker_object_locations_message\x18\x05 \x01(\x0b\x32(.ray.rpc.WorkerObjectLocationsPubMessageH\x00\x12\x30\n\ractor_message\x18\x07 \x01(\x0b\x32\x17.ray.rpc.ActorTableDataH\x00\x12,\n\x0bjob_message\x18\x08 \x01(\x0b\x32\x15.ray.rpc.JobTableDataH\x00\x12\x31\n\x11node_info_message\x18\t \x01(\x0b\x32\x14.ray.rpc.GcsNodeInfoH\x00\x12<\n\x15node_resource_message\x18\n \x01(\x0b\x32\x1b.ray.rpc.NodeResourceChangeH\x00\x12\x38\n\x14worker_delta_message\x18\x0b \x01(\x0b\x32\x18.ray.rpc.WorkerDeltaDataH\x00\x12\x35\n\x12\x65rror_info_message\x18\x0c \x01(\x0b\x32\x17.ray.rpc.ErrorTableDataH\x00\x12.\n\x11log_batch_message\x18\r \x01(\x0b\x32\x11.ray.rpc.LogBatchH\x00\x12:\n\x17python_function_message\x18\x0e \x01(\x0b\x32\x17.ray.rpc.PythonFunctionH\x00\x12\x41\n\x1bnode_resource_usage_message\x18\x0f \x01(\x0b\x32\x1a.ray.rpc.NodeResourceUsageH\x00\x12\x32\n\x0f\x66\x61ilure_message\x18\x06 \x01(\x0b\x32\x17.ray.rpc.FailureMessageH\x00\x12\x13\n\x0bsequence_id\x18\x10 \x01(\x03\x42\x0f\n\rinner_message\"0\n\x1bWorkerObjectEvictionMessage\x12\x11\n\tobject_id\x18\x01 \x01(\x0c\"O\n\x17WorkerRefRemovedMessage\x12\x34\n\rborrowed_refs\x18\x01 \x03(\x0b\x32\x1d.ray.rpc.ObjectReferenceCount\"\xbe\x01\n\x1fWorkerObjectLocationsPubMessage\x12\x10\n\x08node_ids\x18\x01 \x03(\x0c\x12\x13\n\x0bobject_size\x18\x02 \x01(\x04\x12\x13\n\x0bspilled_url\x18\x03 \x01(\t\x12\x17\n\x0fspilled_node_id\x18\x04 \x01(\x0c\x12\x17\n\x0fprimary_node_id\x18\x06 \x01(\x0c\x12\x13\n\x0bref_removed\x18\x07 \x01(\x08\x12\x18\n\x10pending_creation\x18\x08 \x01(\x08\"\x10\n\x0e\x46\x61ilureMessage\"\xcd\x01\n\x07\x43ommand\x12*\n\x0c\x63hannel_type\x18\x01 \x01(\x0e\x32\x14.ray.rpc.ChannelType\x12\x0e\n\x06key_id\x18\x02 \x01(\x0c\x12:\n\x13unsubscribe_message\x18\x03 \x01(\x0b\x32\x1b.ray.rpc.UnsubscribeMessageH\x00\x12\x30\n\x11subscribe_message\x18\x04 \x01(\x0b\x32\x13.ray.rpc.SubMessageH\x00\x42\x18\n\x16\x63ommand_message_one_of\"\x14\n\x12UnsubscribeMessage\"\x95\x02\n\nSubMessage\x12Q\n\x1eworker_object_eviction_message\x18\x01 \x01(\x0b\x32\'.ray.rpc.WorkerObjectEvictionSubMessageH\x00\x12I\n\x1aworker_ref_removed_message\x18\x02 \x01(\x0b\x32#.ray.rpc.WorkerRefRemovedSubMessageH\x00\x12S\n\x1fworker_object_locations_message\x18\x03 \x01(\x0b\x32(.ray.rpc.WorkerObjectLocationsSubMessageH\x00\x42\x14\n\x12sub_message_one_of\"\xa9\x01\n\x1eWorkerObjectEvictionSubMessage\x12\x1a\n\x12intended_worker_id\x18\x01 \x01(\x0c\x12\x11\n\tobject_id\x18\x02 \x01(\x0c\x12,\n\x12subscriber_address\x18\x03 \x01(\x0b\x32\x10.ray.rpc.Address\x12\x19\n\x0cgenerator_id\x18\x04 \x01(\x0cH\x00\x88\x01\x01\x42\x0f\n\r_generator_id\"\x9c\x01\n\x1aWorkerRefRemovedSubMessage\x12\x1a\n\x12intended_worker_id\x18\x01 \x01(\x0c\x12+\n\treference\x18\x02 \x01(\x0b\x32\x18.ray.rpc.ObjectReference\x12\x17\n\x0f\x63ontained_in_id\x18\x03 \x01(\x0c\x12\x1c\n\x14subscriber_worker_id\x18\x04 \x01(\x0c\"P\n\x1fWorkerObjectLocationsSubMessage\x12\x1a\n\x12intended_worker_id\x18\x01 \x01(\x0c\x12\x11\n\tobject_id\x18\x02 \x01(\x0c\"j\n\x18PubsubLongPollingRequest\x12\x15\n\rsubscriber_id\x18\x01 \x01(\x0c\x12!\n\x19max_processed_sequence_id\x18\x02 \x01(\x03\x12\x14\n\x0cpublisher_id\x18\x03 \x01(\x0c\"Y\n\x16PubsubLongPollingReply\x12)\n\x0cpub_messages\x18\x01 \x03(\x0b\x32\x13.ray.rpc.PubMessage\x12\x14\n\x0cpublisher_id\x18\x02 \x01(\x0c\"V\n\x19PubsubCommandBatchRequest\x12\x15\n\rsubscriber_id\x18\x01 \x01(\x0c\x12\"\n\x08\x63ommands\x18\x02 \x03(\x0b\x32\x10.ray.rpc.Command\"\x19\n\x17PubsubCommandBatchReply*\xca\x02\n\x0b\x43hannelType\x12\x1a\n\x16WORKER_OBJECT_EVICTION\x10\x00\x12\x1e\n\x1aWORKER_REF_REMOVED_CHANNEL\x10\x01\x12#\n\x1fWORKER_OBJECT_LOCATIONS_CHANNEL\x10\x02\x12\x15\n\x11GCS_ACTOR_CHANNEL\x10\x03\x12\x13\n\x0fGCS_JOB_CHANNEL\x10\x04\x12\x19\n\x15GCS_NODE_INFO_CHANNEL\x10\x05\x12\x1c\n\x18GCS_WORKER_DELTA_CHANNEL\x10\x06\x12\x1a\n\x16RAY_ERROR_INFO_CHANNEL\x10\x07\x12\x13\n\x0fRAY_LOG_CHANNEL\x10\x08\x12\x1f\n\x1bRAY_PYTHON_FUNCTION_CHANNEL\x10\t\x12#\n\x1fRAY_NODE_RESOURCE_USAGE_CHANNEL\x10\n2\xc8\x01\n\x11SubscriberService\x12W\n\x11PubsubLongPolling\x12!.ray.rpc.PubsubLongPollingRequest\x1a\x1f.ray.rpc.PubsubLongPollingReply\x12Z\n\x12PubsubCommandBatch\x12\".ray.rpc.PubsubCommandBatchRequest\x1a .ray.rpc.PubsubCommandBatchReplyB\x03\xf8\x01\x01\x62\x06proto3') + +_CHANNELTYPE = DESCRIPTOR.enum_types_by_name['ChannelType'] +ChannelType = enum_type_wrapper.EnumTypeWrapper(_CHANNELTYPE) +WORKER_OBJECT_EVICTION = 0 +WORKER_REF_REMOVED_CHANNEL = 1 +WORKER_OBJECT_LOCATIONS_CHANNEL = 2 +GCS_ACTOR_CHANNEL = 3 +GCS_JOB_CHANNEL = 4 +GCS_NODE_INFO_CHANNEL = 5 +GCS_WORKER_DELTA_CHANNEL = 6 +RAY_ERROR_INFO_CHANNEL = 7 +RAY_LOG_CHANNEL = 8 +RAY_PYTHON_FUNCTION_CHANNEL = 9 +RAY_NODE_RESOURCE_USAGE_CHANNEL = 10 + + +_PUBMESSAGE = DESCRIPTOR.message_types_by_name['PubMessage'] +_WORKEROBJECTEVICTIONMESSAGE = DESCRIPTOR.message_types_by_name['WorkerObjectEvictionMessage'] +_WORKERREFREMOVEDMESSAGE = DESCRIPTOR.message_types_by_name['WorkerRefRemovedMessage'] +_WORKEROBJECTLOCATIONSPUBMESSAGE = DESCRIPTOR.message_types_by_name['WorkerObjectLocationsPubMessage'] +_FAILUREMESSAGE = DESCRIPTOR.message_types_by_name['FailureMessage'] +_COMMAND = DESCRIPTOR.message_types_by_name['Command'] +_UNSUBSCRIBEMESSAGE = DESCRIPTOR.message_types_by_name['UnsubscribeMessage'] +_SUBMESSAGE = DESCRIPTOR.message_types_by_name['SubMessage'] +_WORKEROBJECTEVICTIONSUBMESSAGE = DESCRIPTOR.message_types_by_name['WorkerObjectEvictionSubMessage'] +_WORKERREFREMOVEDSUBMESSAGE = DESCRIPTOR.message_types_by_name['WorkerRefRemovedSubMessage'] +_WORKEROBJECTLOCATIONSSUBMESSAGE = DESCRIPTOR.message_types_by_name['WorkerObjectLocationsSubMessage'] +_PUBSUBLONGPOLLINGREQUEST = DESCRIPTOR.message_types_by_name['PubsubLongPollingRequest'] +_PUBSUBLONGPOLLINGREPLY = DESCRIPTOR.message_types_by_name['PubsubLongPollingReply'] +_PUBSUBCOMMANDBATCHREQUEST = DESCRIPTOR.message_types_by_name['PubsubCommandBatchRequest'] +_PUBSUBCOMMANDBATCHREPLY = DESCRIPTOR.message_types_by_name['PubsubCommandBatchReply'] +PubMessage = _reflection.GeneratedProtocolMessageType('PubMessage', (_message.Message,), { + 'DESCRIPTOR' : _PUBMESSAGE, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PubMessage) + }) +_sym_db.RegisterMessage(PubMessage) + +WorkerObjectEvictionMessage = _reflection.GeneratedProtocolMessageType('WorkerObjectEvictionMessage', (_message.Message,), { + 'DESCRIPTOR' : _WORKEROBJECTEVICTIONMESSAGE, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WorkerObjectEvictionMessage) + }) +_sym_db.RegisterMessage(WorkerObjectEvictionMessage) + +WorkerRefRemovedMessage = _reflection.GeneratedProtocolMessageType('WorkerRefRemovedMessage', (_message.Message,), { + 'DESCRIPTOR' : _WORKERREFREMOVEDMESSAGE, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WorkerRefRemovedMessage) + }) +_sym_db.RegisterMessage(WorkerRefRemovedMessage) + +WorkerObjectLocationsPubMessage = _reflection.GeneratedProtocolMessageType('WorkerObjectLocationsPubMessage', (_message.Message,), { + 'DESCRIPTOR' : _WORKEROBJECTLOCATIONSPUBMESSAGE, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WorkerObjectLocationsPubMessage) + }) +_sym_db.RegisterMessage(WorkerObjectLocationsPubMessage) + +FailureMessage = _reflection.GeneratedProtocolMessageType('FailureMessage', (_message.Message,), { + 'DESCRIPTOR' : _FAILUREMESSAGE, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.FailureMessage) + }) +_sym_db.RegisterMessage(FailureMessage) + +Command = _reflection.GeneratedProtocolMessageType('Command', (_message.Message,), { + 'DESCRIPTOR' : _COMMAND, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.Command) + }) +_sym_db.RegisterMessage(Command) + +UnsubscribeMessage = _reflection.GeneratedProtocolMessageType('UnsubscribeMessage', (_message.Message,), { + 'DESCRIPTOR' : _UNSUBSCRIBEMESSAGE, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.UnsubscribeMessage) + }) +_sym_db.RegisterMessage(UnsubscribeMessage) + +SubMessage = _reflection.GeneratedProtocolMessageType('SubMessage', (_message.Message,), { + 'DESCRIPTOR' : _SUBMESSAGE, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.SubMessage) + }) +_sym_db.RegisterMessage(SubMessage) + +WorkerObjectEvictionSubMessage = _reflection.GeneratedProtocolMessageType('WorkerObjectEvictionSubMessage', (_message.Message,), { + 'DESCRIPTOR' : _WORKEROBJECTEVICTIONSUBMESSAGE, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WorkerObjectEvictionSubMessage) + }) +_sym_db.RegisterMessage(WorkerObjectEvictionSubMessage) + +WorkerRefRemovedSubMessage = _reflection.GeneratedProtocolMessageType('WorkerRefRemovedSubMessage', (_message.Message,), { + 'DESCRIPTOR' : _WORKERREFREMOVEDSUBMESSAGE, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WorkerRefRemovedSubMessage) + }) +_sym_db.RegisterMessage(WorkerRefRemovedSubMessage) + +WorkerObjectLocationsSubMessage = _reflection.GeneratedProtocolMessageType('WorkerObjectLocationsSubMessage', (_message.Message,), { + 'DESCRIPTOR' : _WORKEROBJECTLOCATIONSSUBMESSAGE, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WorkerObjectLocationsSubMessage) + }) +_sym_db.RegisterMessage(WorkerObjectLocationsSubMessage) + +PubsubLongPollingRequest = _reflection.GeneratedProtocolMessageType('PubsubLongPollingRequest', (_message.Message,), { + 'DESCRIPTOR' : _PUBSUBLONGPOLLINGREQUEST, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PubsubLongPollingRequest) + }) +_sym_db.RegisterMessage(PubsubLongPollingRequest) + +PubsubLongPollingReply = _reflection.GeneratedProtocolMessageType('PubsubLongPollingReply', (_message.Message,), { + 'DESCRIPTOR' : _PUBSUBLONGPOLLINGREPLY, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PubsubLongPollingReply) + }) +_sym_db.RegisterMessage(PubsubLongPollingReply) + +PubsubCommandBatchRequest = _reflection.GeneratedProtocolMessageType('PubsubCommandBatchRequest', (_message.Message,), { + 'DESCRIPTOR' : _PUBSUBCOMMANDBATCHREQUEST, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PubsubCommandBatchRequest) + }) +_sym_db.RegisterMessage(PubsubCommandBatchRequest) + +PubsubCommandBatchReply = _reflection.GeneratedProtocolMessageType('PubsubCommandBatchReply', (_message.Message,), { + 'DESCRIPTOR' : _PUBSUBCOMMANDBATCHREPLY, + '__module__' : 'src.ray.protobuf.pubsub_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PubsubCommandBatchReply) + }) +_sym_db.RegisterMessage(PubsubCommandBatchReply) + +_SUBSCRIBERSERVICE = DESCRIPTOR.services_by_name['SubscriberService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _CHANNELTYPE._serialized_start=2647 + _CHANNELTYPE._serialized_end=2977 + _PUBMESSAGE._serialized_start=169 + _PUBMESSAGE._serialized_end=1065 + _WORKEROBJECTEVICTIONMESSAGE._serialized_start=1067 + _WORKEROBJECTEVICTIONMESSAGE._serialized_end=1115 + _WORKERREFREMOVEDMESSAGE._serialized_start=1117 + _WORKERREFREMOVEDMESSAGE._serialized_end=1196 + _WORKEROBJECTLOCATIONSPUBMESSAGE._serialized_start=1199 + _WORKEROBJECTLOCATIONSPUBMESSAGE._serialized_end=1389 + _FAILUREMESSAGE._serialized_start=1391 + _FAILUREMESSAGE._serialized_end=1407 + _COMMAND._serialized_start=1410 + _COMMAND._serialized_end=1615 + _UNSUBSCRIBEMESSAGE._serialized_start=1617 + _UNSUBSCRIBEMESSAGE._serialized_end=1637 + _SUBMESSAGE._serialized_start=1640 + _SUBMESSAGE._serialized_end=1917 + _WORKEROBJECTEVICTIONSUBMESSAGE._serialized_start=1920 + _WORKEROBJECTEVICTIONSUBMESSAGE._serialized_end=2089 + _WORKERREFREMOVEDSUBMESSAGE._serialized_start=2092 + _WORKERREFREMOVEDSUBMESSAGE._serialized_end=2248 + _WORKEROBJECTLOCATIONSSUBMESSAGE._serialized_start=2250 + _WORKEROBJECTLOCATIONSSUBMESSAGE._serialized_end=2330 + _PUBSUBLONGPOLLINGREQUEST._serialized_start=2332 + _PUBSUBLONGPOLLINGREQUEST._serialized_end=2438 + _PUBSUBLONGPOLLINGREPLY._serialized_start=2440 + _PUBSUBLONGPOLLINGREPLY._serialized_end=2529 + _PUBSUBCOMMANDBATCHREQUEST._serialized_start=2531 + _PUBSUBCOMMANDBATCHREQUEST._serialized_end=2617 + _PUBSUBCOMMANDBATCHREPLY._serialized_start=2619 + _PUBSUBCOMMANDBATCHREPLY._serialized_end=2644 + _SUBSCRIBERSERVICE._serialized_start=2980 + _SUBSCRIBERSERVICE._serialized_end=3180 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/pubsub_pb2_grpc.py b/python/ray/core/generated/pubsub_pb2_grpc.py new file mode 100644 index 0000000000000..817cf84ad29a3 --- /dev/null +++ b/python/ray/core/generated/pubsub_pb2_grpc.py @@ -0,0 +1,103 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import pubsub_pb2 as src_dot_ray_dot_protobuf_dot_pubsub__pb2 + + +class SubscriberServiceStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.PubsubLongPolling = channel.unary_unary( + '/ray.rpc.SubscriberService/PubsubLongPolling', + request_serializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingReply.FromString, + ) + self.PubsubCommandBatch = channel.unary_unary( + '/ray.rpc.SubscriberService/PubsubCommandBatch', + request_serializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchReply.FromString, + ) + + +class SubscriberServiceServicer(object): + """Missing associated documentation comment in .proto file.""" + + def PubsubLongPolling(self, request, context): + """/ The long polling request sent to the publisher for pubsub operations. + / It is replied once there are batch of objects that need to be published to + / the caller (subscriber). + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def PubsubCommandBatch(self, request, context): + """/ The pubsub command batch request used by the subscriber. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_SubscriberServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'PubsubLongPolling': grpc.unary_unary_rpc_method_handler( + servicer.PubsubLongPolling, + request_deserializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingReply.SerializeToString, + ), + 'PubsubCommandBatch': grpc.unary_unary_rpc_method_handler( + servicer.PubsubCommandBatch, + request_deserializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.SubscriberService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class SubscriberService(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def PubsubLongPolling(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.SubscriberService/PubsubLongPolling', + src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubLongPollingReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def PubsubCommandBatch(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.SubscriberService/PubsubCommandBatch', + src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_pubsub__pb2.PubsubCommandBatchReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/ray_client_pb2.py b/python/ray/core/generated/ray_client_pb2.py new file mode 100644 index 0000000000000..ee67f4b66dd02 --- /dev/null +++ b/python/ray/core/generated/ray_client_pb2.py @@ -0,0 +1,561 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/ray_client.proto +"""Generated protocol buffer code.""" +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!src/ray/protobuf/ray_client.proto\x12\x07ray.rpc\"\x95\x01\n\x03\x41rg\x12$\n\x05local\x18\x01 \x01(\x0e\x32\x15.ray.rpc.Arg.Locality\x12\x14\n\x0creference_id\x18\x02 \x01(\x0c\x12\x0c\n\x04\x64\x61ta\x18\x03 \x01(\x0c\x12\x1b\n\x04type\x18\x04 \x01(\x0e\x32\r.ray.rpc.Type\"\'\n\x08Locality\x12\x0c\n\x08INTERNED\x10\x00\x12\r\n\tREFERENCE\x10\x01\"&\n\x0bTaskOptions\x12\x17\n\x0fpickled_options\x18\x01 \x01(\x0c\"\xf8\x03\n\nClientTask\x12\x30\n\x04type\x18\x01 \x01(\x0e\x32\".ray.rpc.ClientTask.RemoteExecType\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x12\n\npayload_id\x18\x03 \x01(\x0c\x12\x1a\n\x04\x61rgs\x18\x04 \x03(\x0b\x32\x0c.ray.rpc.Arg\x12/\n\x06kwargs\x18\x05 \x03(\x0b\x32\x1f.ray.rpc.ClientTask.KwargsEntry\x12\x11\n\tclient_id\x18\x06 \x01(\t\x12%\n\x07options\x18\x07 \x01(\x0b\x32\x14.ray.rpc.TaskOptions\x12.\n\x10\x62\x61seline_options\x18\x08 \x01(\x0b\x32\x14.ray.rpc.TaskOptions\x12\x11\n\tnamespace\x18\t \x01(\t\x12\x0c\n\x04\x64\x61ta\x18\n \x01(\x0c\x12\x10\n\x08\x63hunk_id\x18\x0b \x01(\x05\x12\x14\n\x0ctotal_chunks\x18\x0c \x01(\x05\x1a;\n\x0bKwargsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x1b\n\x05value\x18\x02 \x01(\x0b\x32\x0c.ray.rpc.Arg:\x02\x38\x01\"Y\n\x0eRemoteExecType\x12\x0c\n\x08\x46UNCTION\x10\x00\x12\t\n\x05\x41\x43TOR\x10\x01\x12\n\n\x06METHOD\x10\x02\x12\x11\n\rSTATIC_METHOD\x10\x03\x12\x0f\n\x0bNAMED_ACTOR\x10\x04\"D\n\x10\x43lientTaskTicket\x12\r\n\x05valid\x18\x01 \x01(\x08\x12\x12\n\nreturn_ids\x18\x02 \x03(\x0c\x12\r\n\x05\x65rror\x18\x03 \x01(\x0c\"\x7f\n\nPutRequest\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\x12\x15\n\rclient_ref_id\x18\x02 \x01(\x0c\x12\x10\n\x08\x63hunk_id\x18\x03 \x01(\x05\x12\x14\n\x0ctotal_chunks\x18\x04 \x01(\x05\x12\x12\n\ntotal_size\x18\x05 \x01(\x03\x12\x10\n\x08owner_id\x18\x06 \x01(\x0c\"7\n\x0bPutResponse\x12\n\n\x02id\x18\x01 \x01(\x0c\x12\r\n\x05valid\x18\x02 \x01(\x08\x12\r\n\x05\x65rror\x18\x03 \x01(\x0c\"h\n\nGetRequest\x12\x0b\n\x03ids\x18\x04 \x03(\x0c\x12\x0f\n\x07timeout\x18\x02 \x01(\x02\x12\x14\n\x0c\x61synchronous\x18\x03 \x01(\x08\x12\x16\n\x0estart_chunk_id\x18\x05 \x01(\x05\x12\x0e\n\x02id\x18\x01 \x01(\x0c\x42\x02\x18\x01\"u\n\x0bGetResponse\x12\r\n\x05valid\x18\x01 \x01(\x08\x12\x0c\n\x04\x64\x61ta\x18\x02 \x01(\x0c\x12\r\n\x05\x65rror\x18\x03 \x01(\x0c\x12\x10\n\x08\x63hunk_id\x18\x04 \x01(\x05\x12\x14\n\x0ctotal_chunks\x18\x05 \x01(\x05\x12\x12\n\ntotal_size\x18\x06 \x01(\x04\"Z\n\x0bWaitRequest\x12\x12\n\nobject_ids\x18\x01 \x03(\x0c\x12\x13\n\x0bnum_returns\x18\x02 \x01(\x03\x12\x0f\n\x07timeout\x18\x03 \x01(\x01\x12\x11\n\tclient_id\x18\x04 \x01(\t\"U\n\x0cWaitResponse\x12\r\n\x05valid\x18\x01 \x01(\x08\x12\x18\n\x10ready_object_ids\x18\x02 \x03(\x0c\x12\x1c\n\x14remaining_object_ids\x18\x03 \x03(\x0c\"\xad\x01\n\x0f\x43lusterInfoType\"\x99\x01\n\x08TypeEnum\x12\x12\n\x0eIS_INITIALIZED\x10\x00\x12\t\n\x05NODES\x10\x01\x12\x15\n\x11\x43LUSTER_RESOURCES\x10\x02\x12\x17\n\x13\x41VAILABLE_RESOURCES\x10\x03\x12\x13\n\x0fRUNTIME_CONTEXT\x10\x04\x12\x0c\n\x08TIMELINE\x10\x05\x12\x08\n\x04PING\x10\x06\x12\x11\n\rDASHBOARD_URL\x10\x07\"E\n\x12\x43lusterInfoRequest\x12/\n\x04type\x18\x01 \x01(\x0e\x32!.ray.rpc.ClusterInfoType.TypeEnum\"\x8a\x04\n\x13\x43lusterInfoResponse\x12/\n\x04type\x18\x01 \x01(\x0e\x32!.ray.rpc.ClusterInfoType.TypeEnum\x12\x0e\n\x04json\x18\x02 \x01(\tH\x00\x12\x44\n\x0eresource_table\x18\x03 \x01(\x0b\x32*.ray.rpc.ClusterInfoResponse.ResourceTableH\x00\x12\x46\n\x0fruntime_context\x18\x04 \x01(\x0b\x32+.ray.rpc.ClusterInfoResponse.RuntimeContextH\x00\x1a\x83\x01\n\rResourceTable\x12\x44\n\x05table\x18\x01 \x03(\x0b\x32\x35.ray.rpc.ClusterInfoResponse.ResourceTable.TableEntry\x1a,\n\nTableEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x01:\x02\x38\x01\x1a\x8c\x01\n\x0eRuntimeContext\x12\x0e\n\x06job_id\x18\x01 \x01(\x0c\x12\x0f\n\x07node_id\x18\x02 \x01(\x0c\x12\x11\n\tnamespace\x18\x03 \x01(\t\x12\x13\n\x0bruntime_env\x18\x04 \x01(\t\x12\x1c\n\x14\x63\x61pture_client_tasks\x18\x05 \x01(\x08\x12\x13\n\x0bgcs_address\x18\x06 \x01(\tB\x0f\n\rresponse_type\"\xaf\x02\n\x10TerminateRequest\x12\x11\n\tclient_id\x18\x01 \x01(\t\x12\x39\n\x05\x61\x63tor\x18\x02 \x01(\x0b\x32(.ray.rpc.TerminateRequest.ActorTerminateH\x00\x12\x44\n\x0btask_object\x18\x03 \x01(\x0b\x32-.ray.rpc.TerminateRequest.TaskObjectTerminateH\x00\x1a\x30\n\x0e\x41\x63torTerminate\x12\n\n\x02id\x18\x01 \x01(\x0c\x12\x12\n\nno_restart\x18\x02 \x01(\x08\x1a\x43\n\x13TaskObjectTerminate\x12\n\n\x02id\x18\x01 \x01(\x0c\x12\r\n\x05\x66orce\x18\x02 \x01(\x08\x12\x11\n\trecursive\x18\x03 \x01(\x08\x42\x10\n\x0eterminate_type\"\x1f\n\x11TerminateResponse\x12\n\n\x02ok\x18\x01 \x01(\x08\"D\n\x0fKVExistsRequest\x12\x0b\n\x03key\x18\x01 \x01(\x0c\x12\x16\n\tnamespace\x18\x02 \x01(\x0cH\x00\x88\x01\x01\x42\x0c\n\n_namespace\"\"\n\x10KVExistsResponse\x12\x0e\n\x06\x65xists\x18\x01 \x01(\x08\"A\n\x0cKVGetRequest\x12\x0b\n\x03key\x18\x01 \x01(\x0c\x12\x16\n\tnamespace\x18\x02 \x01(\x0cH\x00\x88\x01\x01\x42\x0c\n\n_namespace\"-\n\rKVGetResponse\x12\x12\n\x05value\x18\x01 \x01(\x0cH\x00\x88\x01\x01\x42\x08\n\x06_value\"c\n\x0cKVPutRequest\x12\x0b\n\x03key\x18\x01 \x01(\x0c\x12\r\n\x05value\x18\x02 \x01(\x0c\x12\x11\n\toverwrite\x18\x03 \x01(\x08\x12\x16\n\tnamespace\x18\x04 \x01(\x0cH\x00\x88\x01\x01\x42\x0c\n\n_namespace\"\'\n\rKVPutResponse\x12\x16\n\x0e\x61lready_exists\x18\x01 \x01(\x08\"X\n\x0cKVDelRequest\x12\x0b\n\x03key\x18\x01 \x01(\x0c\x12\x15\n\rdel_by_prefix\x18\x02 \x01(\x08\x12\x16\n\tnamespace\x18\x03 \x01(\x0cH\x00\x88\x01\x01\x42\x0c\n\n_namespace\"$\n\rKVDelResponse\x12\x13\n\x0b\x64\x65leted_num\x18\x01 \x01(\x05\"E\n\rKVListRequest\x12\x0e\n\x06prefix\x18\x01 \x01(\x0c\x12\x16\n\tnamespace\x18\x02 \x01(\x0cH\x00\x88\x01\x01\x42\x0c\n\n_namespace\"\x1e\n\x0eKVListResponse\x12\x0c\n\x04keys\x18\x01 \x03(\x0c\"B\n\x1d\x43lientPinRuntimeEnvURIRequest\x12\x0b\n\x03uri\x18\x01 \x01(\t\x12\x14\n\x0c\x65xpiration_s\x18\x02 \x01(\x05\" \n\x1e\x43lientPinRuntimeEnvURIResponse\"Z\n\x0bInitRequest\x12\x12\n\njob_config\x18\x01 \x01(\x0c\x12\x17\n\x0fray_init_kwargs\x18\x02 \x01(\t\x12\x1e\n\x16reconnect_grace_period\x18\x03 \x01(\x05\"\'\n\x0cInitResponse\x12\n\n\x02ok\x18\x01 \x01(\x08\x12\x0b\n\x03msg\x18\x02 \x01(\t\"\x17\n\x15PrepRuntimeEnvRequest\"\x18\n\x16PrepRuntimeEnvResponse\"6\n\x1c\x43lientListNamedActorsRequest\x12\x16\n\x0e\x61ll_namespaces\x18\x01 \x01(\x08\"4\n\x1d\x43lientListNamedActorsResponse\x12\x13\n\x0b\x61\x63tors_json\x18\x01 \x01(\t\"\x1d\n\x0eReleaseRequest\x12\x0b\n\x03ids\x18\x01 \x03(\x0c\"\x1d\n\x0fReleaseResponse\x12\n\n\x02ok\x18\x02 \x03(\x08\"\x17\n\x15\x43onnectionInfoRequest\"\x88\x01\n\x16\x43onnectionInfoResponse\x12\x13\n\x0bnum_clients\x18\x01 \x01(\x05\x12\x13\n\x0bray_version\x18\x02 \x01(\t\x12\x12\n\nray_commit\x18\x03 \x01(\t\x12\x16\n\x0epython_version\x18\x04 \x01(\t\x12\x18\n\x10protocol_version\x18\x05 \x01(\t\"\x1a\n\x18\x43onnectionCleanupRequest\"\x1b\n\x19\x43onnectionCleanupResponse\"$\n\x12\x41\x63knowledgeRequest\x12\x0e\n\x06req_id\x18\x01 \x01(\x05\"\xc4\x04\n\x0b\x44\x61taRequest\x12\x0e\n\x06req_id\x18\x01 \x01(\x05\x12\"\n\x03get\x18\x02 \x01(\x0b\x32\x13.ray.rpc.GetRequestH\x00\x12\"\n\x03put\x18\x03 \x01(\x0b\x32\x13.ray.rpc.PutRequestH\x00\x12*\n\x07release\x18\x04 \x01(\x0b\x32\x17.ray.rpc.ReleaseRequestH\x00\x12\x39\n\x0f\x63onnection_info\x18\x05 \x01(\x0b\x32\x1e.ray.rpc.ConnectionInfoRequestH\x00\x12$\n\x04init\x18\x06 \x01(\x0b\x32\x14.ray.rpc.InitRequestH\x00\x12:\n\x10prep_runtime_env\x18\x07 \x01(\x0b\x32\x1e.ray.rpc.PrepRuntimeEnvRequestH\x00\x12?\n\x12\x63onnection_cleanup\x18\x08 \x01(\x0b\x32!.ray.rpc.ConnectionCleanupRequestH\x00\x12\x32\n\x0b\x61\x63knowledge\x18\t \x01(\x0b\x32\x1b.ray.rpc.AcknowledgeRequestH\x00\x12#\n\x04task\x18\n \x01(\x0b\x32\x13.ray.rpc.ClientTaskH\x00\x12.\n\tterminate\x18\x0b \x01(\x0b\x32\x19.ray.rpc.TerminateRequestH\x00\x12\x42\n\x11list_named_actors\x18\x0c \x01(\x0b\x32%.ray.rpc.ClientListNamedActorsRequestH\x00\x42\x06\n\x04type\"\xba\x04\n\x0c\x44\x61taResponse\x12\x0e\n\x06req_id\x18\x01 \x01(\x05\x12#\n\x03get\x18\x02 \x01(\x0b\x32\x14.ray.rpc.GetResponseH\x00\x12#\n\x03put\x18\x03 \x01(\x0b\x32\x14.ray.rpc.PutResponseH\x00\x12+\n\x07release\x18\x04 \x01(\x0b\x32\x18.ray.rpc.ReleaseResponseH\x00\x12:\n\x0f\x63onnection_info\x18\x05 \x01(\x0b\x32\x1f.ray.rpc.ConnectionInfoResponseH\x00\x12%\n\x04init\x18\x06 \x01(\x0b\x32\x15.ray.rpc.InitResponseH\x00\x12;\n\x10prep_runtime_env\x18\x07 \x01(\x0b\x32\x1f.ray.rpc.PrepRuntimeEnvResponseH\x00\x12@\n\x12\x63onnection_cleanup\x18\x08 \x01(\x0b\x32\".ray.rpc.ConnectionCleanupResponseH\x00\x12\x30\n\x0btask_ticket\x18\n \x01(\x0b\x32\x19.ray.rpc.ClientTaskTicketH\x00\x12/\n\tterminate\x18\x0b \x01(\x0b\x32\x1a.ray.rpc.TerminateResponseH\x00\x12\x43\n\x11list_named_actors\x18\x0c \x01(\x0b\x32&.ray.rpc.ClientListNamedActorsResponseH\x00\x42\x06\n\x04typeJ\x04\x08\t\x10\nR\x0b\x61\x63knowledge\"7\n\x12LogSettingsRequest\x12\x0f\n\x07\x65nabled\x18\x01 \x01(\x08\x12\x10\n\x08loglevel\x18\x02 \x01(\x05\"3\n\x07LogData\x12\x0b\n\x03msg\x18\x01 \x01(\t\x12\r\n\x05level\x18\x02 \x01(\x05\x12\x0c\n\x04name\x18\x03 \x01(\t*\x13\n\x04Type\x12\x0b\n\x07\x44\x45\x46\x41ULT\x10\x00\x32\x96\x08\n\x0cRayletDriver\x12\x35\n\x04Init\x12\x14.ray.rpc.InitRequest\x1a\x15.ray.rpc.InitResponse\"\x00\x12S\n\x0ePrepRuntimeEnv\x12\x1e.ray.rpc.PrepRuntimeEnvRequest\x1a\x1f.ray.rpc.PrepRuntimeEnvResponse\"\x00\x12:\n\tGetObject\x12\x13.ray.rpc.GetRequest\x1a\x14.ray.rpc.GetResponse\"\x00\x30\x01\x12\x38\n\tPutObject\x12\x13.ray.rpc.PutRequest\x1a\x14.ray.rpc.PutResponse\"\x00\x12;\n\nWaitObject\x12\x14.ray.rpc.WaitRequest\x1a\x15.ray.rpc.WaitResponse\"\x00\x12<\n\x08Schedule\x12\x13.ray.rpc.ClientTask\x1a\x19.ray.rpc.ClientTaskTicket\"\x00\x12\x44\n\tTerminate\x12\x19.ray.rpc.TerminateRequest\x1a\x1a.ray.rpc.TerminateResponse\"\x00\x12J\n\x0b\x43lusterInfo\x12\x1b.ray.rpc.ClusterInfoRequest\x1a\x1c.ray.rpc.ClusterInfoResponse\"\x00\x12\x38\n\x05KVGet\x12\x15.ray.rpc.KVGetRequest\x1a\x16.ray.rpc.KVGetResponse\"\x00\x12\x38\n\x05KVPut\x12\x15.ray.rpc.KVPutRequest\x1a\x16.ray.rpc.KVPutResponse\"\x00\x12\x38\n\x05KVDel\x12\x15.ray.rpc.KVDelRequest\x1a\x16.ray.rpc.KVDelResponse\"\x00\x12;\n\x06KVList\x12\x16.ray.rpc.KVListRequest\x1a\x17.ray.rpc.KVListResponse\"\x00\x12\x41\n\x08KVExists\x12\x18.ray.rpc.KVExistsRequest\x1a\x19.ray.rpc.KVExistsResponse\"\x00\x12\x62\n\x0fListNamedActors\x12%.ray.rpc.ClientListNamedActorsRequest\x1a&.ray.rpc.ClientListNamedActorsResponse\"\x00\x12\x65\n\x10PinRuntimeEnvURI\x12&.ray.rpc.ClientPinRuntimeEnvURIRequest\x1a\'.ray.rpc.ClientPinRuntimeEnvURIResponse\"\x00\x32S\n\x12RayletDataStreamer\x12=\n\x08\x44\x61tapath\x12\x14.ray.rpc.DataRequest\x1a\x15.ray.rpc.DataResponse\"\x00(\x01\x30\x01\x32U\n\x11RayletLogStreamer\x12@\n\tLogstream\x12\x1b.ray.rpc.LogSettingsRequest\x1a\x10.ray.rpc.LogData\"\x00(\x01\x30\x01\x42\x03\xf8\x01\x01\x62\x06proto3') + +_TYPE = DESCRIPTOR.enum_types_by_name['Type'] +Type = enum_type_wrapper.EnumTypeWrapper(_TYPE) +DEFAULT = 0 + + +_ARG = DESCRIPTOR.message_types_by_name['Arg'] +_TASKOPTIONS = DESCRIPTOR.message_types_by_name['TaskOptions'] +_CLIENTTASK = DESCRIPTOR.message_types_by_name['ClientTask'] +_CLIENTTASK_KWARGSENTRY = _CLIENTTASK.nested_types_by_name['KwargsEntry'] +_CLIENTTASKTICKET = DESCRIPTOR.message_types_by_name['ClientTaskTicket'] +_PUTREQUEST = DESCRIPTOR.message_types_by_name['PutRequest'] +_PUTRESPONSE = DESCRIPTOR.message_types_by_name['PutResponse'] +_GETREQUEST = DESCRIPTOR.message_types_by_name['GetRequest'] +_GETRESPONSE = DESCRIPTOR.message_types_by_name['GetResponse'] +_WAITREQUEST = DESCRIPTOR.message_types_by_name['WaitRequest'] +_WAITRESPONSE = DESCRIPTOR.message_types_by_name['WaitResponse'] +_CLUSTERINFOTYPE = DESCRIPTOR.message_types_by_name['ClusterInfoType'] +_CLUSTERINFOREQUEST = DESCRIPTOR.message_types_by_name['ClusterInfoRequest'] +_CLUSTERINFORESPONSE = DESCRIPTOR.message_types_by_name['ClusterInfoResponse'] +_CLUSTERINFORESPONSE_RESOURCETABLE = _CLUSTERINFORESPONSE.nested_types_by_name['ResourceTable'] +_CLUSTERINFORESPONSE_RESOURCETABLE_TABLEENTRY = _CLUSTERINFORESPONSE_RESOURCETABLE.nested_types_by_name['TableEntry'] +_CLUSTERINFORESPONSE_RUNTIMECONTEXT = _CLUSTERINFORESPONSE.nested_types_by_name['RuntimeContext'] +_TERMINATEREQUEST = DESCRIPTOR.message_types_by_name['TerminateRequest'] +_TERMINATEREQUEST_ACTORTERMINATE = _TERMINATEREQUEST.nested_types_by_name['ActorTerminate'] +_TERMINATEREQUEST_TASKOBJECTTERMINATE = _TERMINATEREQUEST.nested_types_by_name['TaskObjectTerminate'] +_TERMINATERESPONSE = DESCRIPTOR.message_types_by_name['TerminateResponse'] +_KVEXISTSREQUEST = DESCRIPTOR.message_types_by_name['KVExistsRequest'] +_KVEXISTSRESPONSE = DESCRIPTOR.message_types_by_name['KVExistsResponse'] +_KVGETREQUEST = DESCRIPTOR.message_types_by_name['KVGetRequest'] +_KVGETRESPONSE = DESCRIPTOR.message_types_by_name['KVGetResponse'] +_KVPUTREQUEST = DESCRIPTOR.message_types_by_name['KVPutRequest'] +_KVPUTRESPONSE = DESCRIPTOR.message_types_by_name['KVPutResponse'] +_KVDELREQUEST = DESCRIPTOR.message_types_by_name['KVDelRequest'] +_KVDELRESPONSE = DESCRIPTOR.message_types_by_name['KVDelResponse'] +_KVLISTREQUEST = DESCRIPTOR.message_types_by_name['KVListRequest'] +_KVLISTRESPONSE = DESCRIPTOR.message_types_by_name['KVListResponse'] +_CLIENTPINRUNTIMEENVURIREQUEST = DESCRIPTOR.message_types_by_name['ClientPinRuntimeEnvURIRequest'] +_CLIENTPINRUNTIMEENVURIRESPONSE = DESCRIPTOR.message_types_by_name['ClientPinRuntimeEnvURIResponse'] +_INITREQUEST = DESCRIPTOR.message_types_by_name['InitRequest'] +_INITRESPONSE = DESCRIPTOR.message_types_by_name['InitResponse'] +_PREPRUNTIMEENVREQUEST = DESCRIPTOR.message_types_by_name['PrepRuntimeEnvRequest'] +_PREPRUNTIMEENVRESPONSE = DESCRIPTOR.message_types_by_name['PrepRuntimeEnvResponse'] +_CLIENTLISTNAMEDACTORSREQUEST = DESCRIPTOR.message_types_by_name['ClientListNamedActorsRequest'] +_CLIENTLISTNAMEDACTORSRESPONSE = DESCRIPTOR.message_types_by_name['ClientListNamedActorsResponse'] +_RELEASEREQUEST = DESCRIPTOR.message_types_by_name['ReleaseRequest'] +_RELEASERESPONSE = DESCRIPTOR.message_types_by_name['ReleaseResponse'] +_CONNECTIONINFOREQUEST = DESCRIPTOR.message_types_by_name['ConnectionInfoRequest'] +_CONNECTIONINFORESPONSE = DESCRIPTOR.message_types_by_name['ConnectionInfoResponse'] +_CONNECTIONCLEANUPREQUEST = DESCRIPTOR.message_types_by_name['ConnectionCleanupRequest'] +_CONNECTIONCLEANUPRESPONSE = DESCRIPTOR.message_types_by_name['ConnectionCleanupResponse'] +_ACKNOWLEDGEREQUEST = DESCRIPTOR.message_types_by_name['AcknowledgeRequest'] +_DATAREQUEST = DESCRIPTOR.message_types_by_name['DataRequest'] +_DATARESPONSE = DESCRIPTOR.message_types_by_name['DataResponse'] +_LOGSETTINGSREQUEST = DESCRIPTOR.message_types_by_name['LogSettingsRequest'] +_LOGDATA = DESCRIPTOR.message_types_by_name['LogData'] +_ARG_LOCALITY = _ARG.enum_types_by_name['Locality'] +_CLIENTTASK_REMOTEEXECTYPE = _CLIENTTASK.enum_types_by_name['RemoteExecType'] +_CLUSTERINFOTYPE_TYPEENUM = _CLUSTERINFOTYPE.enum_types_by_name['TypeEnum'] +Arg = _reflection.GeneratedProtocolMessageType('Arg', (_message.Message,), { + 'DESCRIPTOR' : _ARG, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.Arg) + }) +_sym_db.RegisterMessage(Arg) + +TaskOptions = _reflection.GeneratedProtocolMessageType('TaskOptions', (_message.Message,), { + 'DESCRIPTOR' : _TASKOPTIONS, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TaskOptions) + }) +_sym_db.RegisterMessage(TaskOptions) + +ClientTask = _reflection.GeneratedProtocolMessageType('ClientTask', (_message.Message,), { + + 'KwargsEntry' : _reflection.GeneratedProtocolMessageType('KwargsEntry', (_message.Message,), { + 'DESCRIPTOR' : _CLIENTTASK_KWARGSENTRY, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClientTask.KwargsEntry) + }) + , + 'DESCRIPTOR' : _CLIENTTASK, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClientTask) + }) +_sym_db.RegisterMessage(ClientTask) +_sym_db.RegisterMessage(ClientTask.KwargsEntry) + +ClientTaskTicket = _reflection.GeneratedProtocolMessageType('ClientTaskTicket', (_message.Message,), { + 'DESCRIPTOR' : _CLIENTTASKTICKET, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClientTaskTicket) + }) +_sym_db.RegisterMessage(ClientTaskTicket) + +PutRequest = _reflection.GeneratedProtocolMessageType('PutRequest', (_message.Message,), { + 'DESCRIPTOR' : _PUTREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PutRequest) + }) +_sym_db.RegisterMessage(PutRequest) + +PutResponse = _reflection.GeneratedProtocolMessageType('PutResponse', (_message.Message,), { + 'DESCRIPTOR' : _PUTRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PutResponse) + }) +_sym_db.RegisterMessage(PutResponse) + +GetRequest = _reflection.GeneratedProtocolMessageType('GetRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetRequest) + }) +_sym_db.RegisterMessage(GetRequest) + +GetResponse = _reflection.GeneratedProtocolMessageType('GetResponse', (_message.Message,), { + 'DESCRIPTOR' : _GETRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetResponse) + }) +_sym_db.RegisterMessage(GetResponse) + +WaitRequest = _reflection.GeneratedProtocolMessageType('WaitRequest', (_message.Message,), { + 'DESCRIPTOR' : _WAITREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WaitRequest) + }) +_sym_db.RegisterMessage(WaitRequest) + +WaitResponse = _reflection.GeneratedProtocolMessageType('WaitResponse', (_message.Message,), { + 'DESCRIPTOR' : _WAITRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.WaitResponse) + }) +_sym_db.RegisterMessage(WaitResponse) + +ClusterInfoType = _reflection.GeneratedProtocolMessageType('ClusterInfoType', (_message.Message,), { + 'DESCRIPTOR' : _CLUSTERINFOTYPE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClusterInfoType) + }) +_sym_db.RegisterMessage(ClusterInfoType) + +ClusterInfoRequest = _reflection.GeneratedProtocolMessageType('ClusterInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _CLUSTERINFOREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClusterInfoRequest) + }) +_sym_db.RegisterMessage(ClusterInfoRequest) + +ClusterInfoResponse = _reflection.GeneratedProtocolMessageType('ClusterInfoResponse', (_message.Message,), { + + 'ResourceTable' : _reflection.GeneratedProtocolMessageType('ResourceTable', (_message.Message,), { + + 'TableEntry' : _reflection.GeneratedProtocolMessageType('TableEntry', (_message.Message,), { + 'DESCRIPTOR' : _CLUSTERINFORESPONSE_RESOURCETABLE_TABLEENTRY, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClusterInfoResponse.ResourceTable.TableEntry) + }) + , + 'DESCRIPTOR' : _CLUSTERINFORESPONSE_RESOURCETABLE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClusterInfoResponse.ResourceTable) + }) + , + + 'RuntimeContext' : _reflection.GeneratedProtocolMessageType('RuntimeContext', (_message.Message,), { + 'DESCRIPTOR' : _CLUSTERINFORESPONSE_RUNTIMECONTEXT, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClusterInfoResponse.RuntimeContext) + }) + , + 'DESCRIPTOR' : _CLUSTERINFORESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClusterInfoResponse) + }) +_sym_db.RegisterMessage(ClusterInfoResponse) +_sym_db.RegisterMessage(ClusterInfoResponse.ResourceTable) +_sym_db.RegisterMessage(ClusterInfoResponse.ResourceTable.TableEntry) +_sym_db.RegisterMessage(ClusterInfoResponse.RuntimeContext) + +TerminateRequest = _reflection.GeneratedProtocolMessageType('TerminateRequest', (_message.Message,), { + + 'ActorTerminate' : _reflection.GeneratedProtocolMessageType('ActorTerminate', (_message.Message,), { + 'DESCRIPTOR' : _TERMINATEREQUEST_ACTORTERMINATE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TerminateRequest.ActorTerminate) + }) + , + + 'TaskObjectTerminate' : _reflection.GeneratedProtocolMessageType('TaskObjectTerminate', (_message.Message,), { + 'DESCRIPTOR' : _TERMINATEREQUEST_TASKOBJECTTERMINATE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TerminateRequest.TaskObjectTerminate) + }) + , + 'DESCRIPTOR' : _TERMINATEREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TerminateRequest) + }) +_sym_db.RegisterMessage(TerminateRequest) +_sym_db.RegisterMessage(TerminateRequest.ActorTerminate) +_sym_db.RegisterMessage(TerminateRequest.TaskObjectTerminate) + +TerminateResponse = _reflection.GeneratedProtocolMessageType('TerminateResponse', (_message.Message,), { + 'DESCRIPTOR' : _TERMINATERESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.TerminateResponse) + }) +_sym_db.RegisterMessage(TerminateResponse) + +KVExistsRequest = _reflection.GeneratedProtocolMessageType('KVExistsRequest', (_message.Message,), { + 'DESCRIPTOR' : _KVEXISTSREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KVExistsRequest) + }) +_sym_db.RegisterMessage(KVExistsRequest) + +KVExistsResponse = _reflection.GeneratedProtocolMessageType('KVExistsResponse', (_message.Message,), { + 'DESCRIPTOR' : _KVEXISTSRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KVExistsResponse) + }) +_sym_db.RegisterMessage(KVExistsResponse) + +KVGetRequest = _reflection.GeneratedProtocolMessageType('KVGetRequest', (_message.Message,), { + 'DESCRIPTOR' : _KVGETREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KVGetRequest) + }) +_sym_db.RegisterMessage(KVGetRequest) + +KVGetResponse = _reflection.GeneratedProtocolMessageType('KVGetResponse', (_message.Message,), { + 'DESCRIPTOR' : _KVGETRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KVGetResponse) + }) +_sym_db.RegisterMessage(KVGetResponse) + +KVPutRequest = _reflection.GeneratedProtocolMessageType('KVPutRequest', (_message.Message,), { + 'DESCRIPTOR' : _KVPUTREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KVPutRequest) + }) +_sym_db.RegisterMessage(KVPutRequest) + +KVPutResponse = _reflection.GeneratedProtocolMessageType('KVPutResponse', (_message.Message,), { + 'DESCRIPTOR' : _KVPUTRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KVPutResponse) + }) +_sym_db.RegisterMessage(KVPutResponse) + +KVDelRequest = _reflection.GeneratedProtocolMessageType('KVDelRequest', (_message.Message,), { + 'DESCRIPTOR' : _KVDELREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KVDelRequest) + }) +_sym_db.RegisterMessage(KVDelRequest) + +KVDelResponse = _reflection.GeneratedProtocolMessageType('KVDelResponse', (_message.Message,), { + 'DESCRIPTOR' : _KVDELRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KVDelResponse) + }) +_sym_db.RegisterMessage(KVDelResponse) + +KVListRequest = _reflection.GeneratedProtocolMessageType('KVListRequest', (_message.Message,), { + 'DESCRIPTOR' : _KVLISTREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KVListRequest) + }) +_sym_db.RegisterMessage(KVListRequest) + +KVListResponse = _reflection.GeneratedProtocolMessageType('KVListResponse', (_message.Message,), { + 'DESCRIPTOR' : _KVLISTRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.KVListResponse) + }) +_sym_db.RegisterMessage(KVListResponse) + +ClientPinRuntimeEnvURIRequest = _reflection.GeneratedProtocolMessageType('ClientPinRuntimeEnvURIRequest', (_message.Message,), { + 'DESCRIPTOR' : _CLIENTPINRUNTIMEENVURIREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClientPinRuntimeEnvURIRequest) + }) +_sym_db.RegisterMessage(ClientPinRuntimeEnvURIRequest) + +ClientPinRuntimeEnvURIResponse = _reflection.GeneratedProtocolMessageType('ClientPinRuntimeEnvURIResponse', (_message.Message,), { + 'DESCRIPTOR' : _CLIENTPINRUNTIMEENVURIRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClientPinRuntimeEnvURIResponse) + }) +_sym_db.RegisterMessage(ClientPinRuntimeEnvURIResponse) + +InitRequest = _reflection.GeneratedProtocolMessageType('InitRequest', (_message.Message,), { + 'DESCRIPTOR' : _INITREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InitRequest) + }) +_sym_db.RegisterMessage(InitRequest) + +InitResponse = _reflection.GeneratedProtocolMessageType('InitResponse', (_message.Message,), { + 'DESCRIPTOR' : _INITRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.InitResponse) + }) +_sym_db.RegisterMessage(InitResponse) + +PrepRuntimeEnvRequest = _reflection.GeneratedProtocolMessageType('PrepRuntimeEnvRequest', (_message.Message,), { + 'DESCRIPTOR' : _PREPRUNTIMEENVREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PrepRuntimeEnvRequest) + }) +_sym_db.RegisterMessage(PrepRuntimeEnvRequest) + +PrepRuntimeEnvResponse = _reflection.GeneratedProtocolMessageType('PrepRuntimeEnvResponse', (_message.Message,), { + 'DESCRIPTOR' : _PREPRUNTIMEENVRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PrepRuntimeEnvResponse) + }) +_sym_db.RegisterMessage(PrepRuntimeEnvResponse) + +ClientListNamedActorsRequest = _reflection.GeneratedProtocolMessageType('ClientListNamedActorsRequest', (_message.Message,), { + 'DESCRIPTOR' : _CLIENTLISTNAMEDACTORSREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClientListNamedActorsRequest) + }) +_sym_db.RegisterMessage(ClientListNamedActorsRequest) + +ClientListNamedActorsResponse = _reflection.GeneratedProtocolMessageType('ClientListNamedActorsResponse', (_message.Message,), { + 'DESCRIPTOR' : _CLIENTLISTNAMEDACTORSRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ClientListNamedActorsResponse) + }) +_sym_db.RegisterMessage(ClientListNamedActorsResponse) + +ReleaseRequest = _reflection.GeneratedProtocolMessageType('ReleaseRequest', (_message.Message,), { + 'DESCRIPTOR' : _RELEASEREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReleaseRequest) + }) +_sym_db.RegisterMessage(ReleaseRequest) + +ReleaseResponse = _reflection.GeneratedProtocolMessageType('ReleaseResponse', (_message.Message,), { + 'DESCRIPTOR' : _RELEASERESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReleaseResponse) + }) +_sym_db.RegisterMessage(ReleaseResponse) + +ConnectionInfoRequest = _reflection.GeneratedProtocolMessageType('ConnectionInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _CONNECTIONINFOREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ConnectionInfoRequest) + }) +_sym_db.RegisterMessage(ConnectionInfoRequest) + +ConnectionInfoResponse = _reflection.GeneratedProtocolMessageType('ConnectionInfoResponse', (_message.Message,), { + 'DESCRIPTOR' : _CONNECTIONINFORESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ConnectionInfoResponse) + }) +_sym_db.RegisterMessage(ConnectionInfoResponse) + +ConnectionCleanupRequest = _reflection.GeneratedProtocolMessageType('ConnectionCleanupRequest', (_message.Message,), { + 'DESCRIPTOR' : _CONNECTIONCLEANUPREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ConnectionCleanupRequest) + }) +_sym_db.RegisterMessage(ConnectionCleanupRequest) + +ConnectionCleanupResponse = _reflection.GeneratedProtocolMessageType('ConnectionCleanupResponse', (_message.Message,), { + 'DESCRIPTOR' : _CONNECTIONCLEANUPRESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ConnectionCleanupResponse) + }) +_sym_db.RegisterMessage(ConnectionCleanupResponse) + +AcknowledgeRequest = _reflection.GeneratedProtocolMessageType('AcknowledgeRequest', (_message.Message,), { + 'DESCRIPTOR' : _ACKNOWLEDGEREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.AcknowledgeRequest) + }) +_sym_db.RegisterMessage(AcknowledgeRequest) + +DataRequest = _reflection.GeneratedProtocolMessageType('DataRequest', (_message.Message,), { + 'DESCRIPTOR' : _DATAREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DataRequest) + }) +_sym_db.RegisterMessage(DataRequest) + +DataResponse = _reflection.GeneratedProtocolMessageType('DataResponse', (_message.Message,), { + 'DESCRIPTOR' : _DATARESPONSE, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DataResponse) + }) +_sym_db.RegisterMessage(DataResponse) + +LogSettingsRequest = _reflection.GeneratedProtocolMessageType('LogSettingsRequest', (_message.Message,), { + 'DESCRIPTOR' : _LOGSETTINGSREQUEST, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.LogSettingsRequest) + }) +_sym_db.RegisterMessage(LogSettingsRequest) + +LogData = _reflection.GeneratedProtocolMessageType('LogData', (_message.Message,), { + 'DESCRIPTOR' : _LOGDATA, + '__module__' : 'src.ray.protobuf.ray_client_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.LogData) + }) +_sym_db.RegisterMessage(LogData) + +_RAYLETDRIVER = DESCRIPTOR.services_by_name['RayletDriver'] +_RAYLETDATASTREAMER = DESCRIPTOR.services_by_name['RayletDataStreamer'] +_RAYLETLOGSTREAMER = DESCRIPTOR.services_by_name['RayletLogStreamer'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _CLIENTTASK_KWARGSENTRY._options = None + _CLIENTTASK_KWARGSENTRY._serialized_options = b'8\001' + _GETREQUEST.fields_by_name['id']._options = None + _GETREQUEST.fields_by_name['id']._serialized_options = b'\030\001' + _CLUSTERINFORESPONSE_RESOURCETABLE_TABLEENTRY._options = None + _CLUSTERINFORESPONSE_RESOURCETABLE_TABLEENTRY._serialized_options = b'8\001' + _TYPE._serialized_start=5092 + _TYPE._serialized_end=5111 + _ARG._serialized_start=47 + _ARG._serialized_end=196 + _ARG_LOCALITY._serialized_start=157 + _ARG_LOCALITY._serialized_end=196 + _TASKOPTIONS._serialized_start=198 + _TASKOPTIONS._serialized_end=236 + _CLIENTTASK._serialized_start=239 + _CLIENTTASK._serialized_end=743 + _CLIENTTASK_KWARGSENTRY._serialized_start=593 + _CLIENTTASK_KWARGSENTRY._serialized_end=652 + _CLIENTTASK_REMOTEEXECTYPE._serialized_start=654 + _CLIENTTASK_REMOTEEXECTYPE._serialized_end=743 + _CLIENTTASKTICKET._serialized_start=745 + _CLIENTTASKTICKET._serialized_end=813 + _PUTREQUEST._serialized_start=815 + _PUTREQUEST._serialized_end=942 + _PUTRESPONSE._serialized_start=944 + _PUTRESPONSE._serialized_end=999 + _GETREQUEST._serialized_start=1001 + _GETREQUEST._serialized_end=1105 + _GETRESPONSE._serialized_start=1107 + _GETRESPONSE._serialized_end=1224 + _WAITREQUEST._serialized_start=1226 + _WAITREQUEST._serialized_end=1316 + _WAITRESPONSE._serialized_start=1318 + _WAITRESPONSE._serialized_end=1403 + _CLUSTERINFOTYPE._serialized_start=1406 + _CLUSTERINFOTYPE._serialized_end=1579 + _CLUSTERINFOTYPE_TYPEENUM._serialized_start=1426 + _CLUSTERINFOTYPE_TYPEENUM._serialized_end=1579 + _CLUSTERINFOREQUEST._serialized_start=1581 + _CLUSTERINFOREQUEST._serialized_end=1650 + _CLUSTERINFORESPONSE._serialized_start=1653 + _CLUSTERINFORESPONSE._serialized_end=2175 + _CLUSTERINFORESPONSE_RESOURCETABLE._serialized_start=1884 + _CLUSTERINFORESPONSE_RESOURCETABLE._serialized_end=2015 + _CLUSTERINFORESPONSE_RESOURCETABLE_TABLEENTRY._serialized_start=1971 + _CLUSTERINFORESPONSE_RESOURCETABLE_TABLEENTRY._serialized_end=2015 + _CLUSTERINFORESPONSE_RUNTIMECONTEXT._serialized_start=2018 + _CLUSTERINFORESPONSE_RUNTIMECONTEXT._serialized_end=2158 + _TERMINATEREQUEST._serialized_start=2178 + _TERMINATEREQUEST._serialized_end=2481 + _TERMINATEREQUEST_ACTORTERMINATE._serialized_start=2346 + _TERMINATEREQUEST_ACTORTERMINATE._serialized_end=2394 + _TERMINATEREQUEST_TASKOBJECTTERMINATE._serialized_start=2396 + _TERMINATEREQUEST_TASKOBJECTTERMINATE._serialized_end=2463 + _TERMINATERESPONSE._serialized_start=2483 + _TERMINATERESPONSE._serialized_end=2514 + _KVEXISTSREQUEST._serialized_start=2516 + _KVEXISTSREQUEST._serialized_end=2584 + _KVEXISTSRESPONSE._serialized_start=2586 + _KVEXISTSRESPONSE._serialized_end=2620 + _KVGETREQUEST._serialized_start=2622 + _KVGETREQUEST._serialized_end=2687 + _KVGETRESPONSE._serialized_start=2689 + _KVGETRESPONSE._serialized_end=2734 + _KVPUTREQUEST._serialized_start=2736 + _KVPUTREQUEST._serialized_end=2835 + _KVPUTRESPONSE._serialized_start=2837 + _KVPUTRESPONSE._serialized_end=2876 + _KVDELREQUEST._serialized_start=2878 + _KVDELREQUEST._serialized_end=2966 + _KVDELRESPONSE._serialized_start=2968 + _KVDELRESPONSE._serialized_end=3004 + _KVLISTREQUEST._serialized_start=3006 + _KVLISTREQUEST._serialized_end=3075 + _KVLISTRESPONSE._serialized_start=3077 + _KVLISTRESPONSE._serialized_end=3107 + _CLIENTPINRUNTIMEENVURIREQUEST._serialized_start=3109 + _CLIENTPINRUNTIMEENVURIREQUEST._serialized_end=3175 + _CLIENTPINRUNTIMEENVURIRESPONSE._serialized_start=3177 + _CLIENTPINRUNTIMEENVURIRESPONSE._serialized_end=3209 + _INITREQUEST._serialized_start=3211 + _INITREQUEST._serialized_end=3301 + _INITRESPONSE._serialized_start=3303 + _INITRESPONSE._serialized_end=3342 + _PREPRUNTIMEENVREQUEST._serialized_start=3344 + _PREPRUNTIMEENVREQUEST._serialized_end=3367 + _PREPRUNTIMEENVRESPONSE._serialized_start=3369 + _PREPRUNTIMEENVRESPONSE._serialized_end=3393 + _CLIENTLISTNAMEDACTORSREQUEST._serialized_start=3395 + _CLIENTLISTNAMEDACTORSREQUEST._serialized_end=3449 + _CLIENTLISTNAMEDACTORSRESPONSE._serialized_start=3451 + _CLIENTLISTNAMEDACTORSRESPONSE._serialized_end=3503 + _RELEASEREQUEST._serialized_start=3505 + _RELEASEREQUEST._serialized_end=3534 + _RELEASERESPONSE._serialized_start=3536 + _RELEASERESPONSE._serialized_end=3565 + _CONNECTIONINFOREQUEST._serialized_start=3567 + _CONNECTIONINFOREQUEST._serialized_end=3590 + _CONNECTIONINFORESPONSE._serialized_start=3593 + _CONNECTIONINFORESPONSE._serialized_end=3729 + _CONNECTIONCLEANUPREQUEST._serialized_start=3731 + _CONNECTIONCLEANUPREQUEST._serialized_end=3757 + _CONNECTIONCLEANUPRESPONSE._serialized_start=3759 + _CONNECTIONCLEANUPRESPONSE._serialized_end=3786 + _ACKNOWLEDGEREQUEST._serialized_start=3788 + _ACKNOWLEDGEREQUEST._serialized_end=3824 + _DATAREQUEST._serialized_start=3827 + _DATAREQUEST._serialized_end=4407 + _DATARESPONSE._serialized_start=4410 + _DATARESPONSE._serialized_end=4980 + _LOGSETTINGSREQUEST._serialized_start=4982 + _LOGSETTINGSREQUEST._serialized_end=5037 + _LOGDATA._serialized_start=5039 + _LOGDATA._serialized_end=5090 + _RAYLETDRIVER._serialized_start=5114 + _RAYLETDRIVER._serialized_end=6160 + _RAYLETDATASTREAMER._serialized_start=6162 + _RAYLETDATASTREAMER._serialized_end=6245 + _RAYLETLOGSTREAMER._serialized_start=6247 + _RAYLETLOGSTREAMER._serialized_end=6332 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/ray_client_pb2_grpc.py b/python/ray/core/generated/ray_client_pb2_grpc.py new file mode 100644 index 0000000000000..f5b943012c5e9 --- /dev/null +++ b/python/ray/core/generated/ray_client_pb2_grpc.py @@ -0,0 +1,650 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import ray_client_pb2 as src_dot_ray_dot_protobuf_dot_ray__client__pb2 + + +class RayletDriverStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.Init = channel.unary_unary( + '/ray.rpc.RayletDriver/Init', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.InitRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.InitResponse.FromString, + ) + self.PrepRuntimeEnv = channel.unary_unary( + '/ray.rpc.RayletDriver/PrepRuntimeEnv', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.PrepRuntimeEnvRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.PrepRuntimeEnvResponse.FromString, + ) + self.GetObject = channel.unary_stream( + '/ray.rpc.RayletDriver/GetObject', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.GetRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.GetResponse.FromString, + ) + self.PutObject = channel.unary_unary( + '/ray.rpc.RayletDriver/PutObject', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.PutRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.PutResponse.FromString, + ) + self.WaitObject = channel.unary_unary( + '/ray.rpc.RayletDriver/WaitObject', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.WaitRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.WaitResponse.FromString, + ) + self.Schedule = channel.unary_unary( + '/ray.rpc.RayletDriver/Schedule', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientTask.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientTaskTicket.FromString, + ) + self.Terminate = channel.unary_unary( + '/ray.rpc.RayletDriver/Terminate', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.TerminateRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.TerminateResponse.FromString, + ) + self.ClusterInfo = channel.unary_unary( + '/ray.rpc.RayletDriver/ClusterInfo', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClusterInfoRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClusterInfoResponse.FromString, + ) + self.KVGet = channel.unary_unary( + '/ray.rpc.RayletDriver/KVGet', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVGetRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVGetResponse.FromString, + ) + self.KVPut = channel.unary_unary( + '/ray.rpc.RayletDriver/KVPut', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVPutRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVPutResponse.FromString, + ) + self.KVDel = channel.unary_unary( + '/ray.rpc.RayletDriver/KVDel', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVDelRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVDelResponse.FromString, + ) + self.KVList = channel.unary_unary( + '/ray.rpc.RayletDriver/KVList', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVListRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVListResponse.FromString, + ) + self.KVExists = channel.unary_unary( + '/ray.rpc.RayletDriver/KVExists', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVExistsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVExistsResponse.FromString, + ) + self.ListNamedActors = channel.unary_unary( + '/ray.rpc.RayletDriver/ListNamedActors', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientListNamedActorsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientListNamedActorsResponse.FromString, + ) + self.PinRuntimeEnvURI = channel.unary_unary( + '/ray.rpc.RayletDriver/PinRuntimeEnvURI', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientPinRuntimeEnvURIRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientPinRuntimeEnvURIResponse.FromString, + ) + + +class RayletDriverServicer(object): + """Missing associated documentation comment in .proto file.""" + + def Init(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def PrepRuntimeEnv(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetObject(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def PutObject(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def WaitObject(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Schedule(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Terminate(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ClusterInfo(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def KVGet(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def KVPut(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def KVDel(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def KVList(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def KVExists(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ListNamedActors(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def PinRuntimeEnvURI(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_RayletDriverServicer_to_server(servicer, server): + rpc_method_handlers = { + 'Init': grpc.unary_unary_rpc_method_handler( + servicer.Init, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.InitRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.InitResponse.SerializeToString, + ), + 'PrepRuntimeEnv': grpc.unary_unary_rpc_method_handler( + servicer.PrepRuntimeEnv, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.PrepRuntimeEnvRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.PrepRuntimeEnvResponse.SerializeToString, + ), + 'GetObject': grpc.unary_stream_rpc_method_handler( + servicer.GetObject, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.GetRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.GetResponse.SerializeToString, + ), + 'PutObject': grpc.unary_unary_rpc_method_handler( + servicer.PutObject, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.PutRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.PutResponse.SerializeToString, + ), + 'WaitObject': grpc.unary_unary_rpc_method_handler( + servicer.WaitObject, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.WaitRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.WaitResponse.SerializeToString, + ), + 'Schedule': grpc.unary_unary_rpc_method_handler( + servicer.Schedule, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientTask.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientTaskTicket.SerializeToString, + ), + 'Terminate': grpc.unary_unary_rpc_method_handler( + servicer.Terminate, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.TerminateRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.TerminateResponse.SerializeToString, + ), + 'ClusterInfo': grpc.unary_unary_rpc_method_handler( + servicer.ClusterInfo, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClusterInfoRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClusterInfoResponse.SerializeToString, + ), + 'KVGet': grpc.unary_unary_rpc_method_handler( + servicer.KVGet, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVGetRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVGetResponse.SerializeToString, + ), + 'KVPut': grpc.unary_unary_rpc_method_handler( + servicer.KVPut, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVPutRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVPutResponse.SerializeToString, + ), + 'KVDel': grpc.unary_unary_rpc_method_handler( + servicer.KVDel, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVDelRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVDelResponse.SerializeToString, + ), + 'KVList': grpc.unary_unary_rpc_method_handler( + servicer.KVList, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVListRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVListResponse.SerializeToString, + ), + 'KVExists': grpc.unary_unary_rpc_method_handler( + servicer.KVExists, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVExistsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVExistsResponse.SerializeToString, + ), + 'ListNamedActors': grpc.unary_unary_rpc_method_handler( + servicer.ListNamedActors, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientListNamedActorsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientListNamedActorsResponse.SerializeToString, + ), + 'PinRuntimeEnvURI': grpc.unary_unary_rpc_method_handler( + servicer.PinRuntimeEnvURI, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientPinRuntimeEnvURIRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientPinRuntimeEnvURIResponse.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.RayletDriver', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class RayletDriver(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def Init(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/Init', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.InitRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.InitResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def PrepRuntimeEnv(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/PrepRuntimeEnv', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.PrepRuntimeEnvRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.PrepRuntimeEnvResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetObject(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_stream(request, target, '/ray.rpc.RayletDriver/GetObject', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.GetRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.GetResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def PutObject(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/PutObject', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.PutRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.PutResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def WaitObject(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/WaitObject', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.WaitRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.WaitResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def Schedule(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/Schedule', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientTask.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientTaskTicket.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def Terminate(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/Terminate', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.TerminateRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.TerminateResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ClusterInfo(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/ClusterInfo', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClusterInfoRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClusterInfoResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def KVGet(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/KVGet', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVGetRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVGetResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def KVPut(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/KVPut', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVPutRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVPutResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def KVDel(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/KVDel', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVDelRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVDelResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def KVList(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/KVList', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVListRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVListResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def KVExists(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/KVExists', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVExistsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.KVExistsResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ListNamedActors(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/ListNamedActors', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientListNamedActorsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientListNamedActorsResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def PinRuntimeEnvURI(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.RayletDriver/PinRuntimeEnvURI', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientPinRuntimeEnvURIRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.ClientPinRuntimeEnvURIResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class RayletDataStreamerStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.Datapath = channel.stream_stream( + '/ray.rpc.RayletDataStreamer/Datapath', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.DataRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.DataResponse.FromString, + ) + + +class RayletDataStreamerServicer(object): + """Missing associated documentation comment in .proto file.""" + + def Datapath(self, request_iterator, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_RayletDataStreamerServicer_to_server(servicer, server): + rpc_method_handlers = { + 'Datapath': grpc.stream_stream_rpc_method_handler( + servicer.Datapath, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.DataRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.DataResponse.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.RayletDataStreamer', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class RayletDataStreamer(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def Datapath(request_iterator, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.stream_stream(request_iterator, target, '/ray.rpc.RayletDataStreamer/Datapath', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.DataRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.DataResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class RayletLogStreamerStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.Logstream = channel.stream_stream( + '/ray.rpc.RayletLogStreamer/Logstream', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.LogSettingsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.LogData.FromString, + ) + + +class RayletLogStreamerServicer(object): + """Missing associated documentation comment in .proto file.""" + + def Logstream(self, request_iterator, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_RayletLogStreamerServicer_to_server(servicer, server): + rpc_method_handlers = { + 'Logstream': grpc.stream_stream_rpc_method_handler( + servicer.Logstream, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.LogSettingsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__client__pb2.LogData.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.RayletLogStreamer', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class RayletLogStreamer(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def Logstream(request_iterator, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.stream_stream(request_iterator, target, '/ray.rpc.RayletLogStreamer/Logstream', + src_dot_ray_dot_protobuf_dot_ray__client__pb2.LogSettingsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__client__pb2.LogData.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/ray_syncer_pb2.py b/python/ray/core/generated/ray_syncer_pb2.py new file mode 100644 index 0000000000000..9ae83eb6d7945 --- /dev/null +++ b/python/ray/core/generated/ray_syncer_pb2.py @@ -0,0 +1,45 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/ray_syncer.proto +"""Generated protocol buffer code.""" +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!src/ray/protobuf/ray_syncer.proto\x12\x0eray.rpc.syncer\"{\n\x0eRaySyncMessage\x12\x0f\n\x07version\x18\x01 \x01(\x03\x12\x31\n\x0cmessage_type\x18\x02 \x01(\x0e\x32\x1b.ray.rpc.syncer.MessageType\x12\x14\n\x0csync_message\x18\x03 \x01(\x0c\x12\x0f\n\x07node_id\x18\x04 \x01(\x0c*.\n\x0bMessageType\x12\x11\n\rRESOURCE_VIEW\x10\x00\x12\x0c\n\x08\x43OMMANDS\x10\x01\x32\\\n\tRaySyncer\x12O\n\tStartSync\x12\x1e.ray.rpc.syncer.RaySyncMessage\x1a\x1e.ray.rpc.syncer.RaySyncMessage(\x01\x30\x01\x42\x03\xf8\x01\x01\x62\x06proto3') + +_MESSAGETYPE = DESCRIPTOR.enum_types_by_name['MessageType'] +MessageType = enum_type_wrapper.EnumTypeWrapper(_MESSAGETYPE) +RESOURCE_VIEW = 0 +COMMANDS = 1 + + +_RAYSYNCMESSAGE = DESCRIPTOR.message_types_by_name['RaySyncMessage'] +RaySyncMessage = _reflection.GeneratedProtocolMessageType('RaySyncMessage', (_message.Message,), { + 'DESCRIPTOR' : _RAYSYNCMESSAGE, + '__module__' : 'src.ray.protobuf.ray_syncer_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.syncer.RaySyncMessage) + }) +_sym_db.RegisterMessage(RaySyncMessage) + +_RAYSYNCER = DESCRIPTOR.services_by_name['RaySyncer'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _MESSAGETYPE._serialized_start=178 + _MESSAGETYPE._serialized_end=224 + _RAYSYNCMESSAGE._serialized_start=53 + _RAYSYNCMESSAGE._serialized_end=176 + _RAYSYNCER._serialized_start=226 + _RAYSYNCER._serialized_end=318 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/ray_syncer_pb2_grpc.py b/python/ray/core/generated/ray_syncer_pb2_grpc.py new file mode 100644 index 0000000000000..aadfd93def26c --- /dev/null +++ b/python/ray/core/generated/ray_syncer_pb2_grpc.py @@ -0,0 +1,66 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import ray_syncer_pb2 as src_dot_ray_dot_protobuf_dot_ray__syncer__pb2 + + +class RaySyncerStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.StartSync = channel.stream_stream( + '/ray.rpc.syncer.RaySyncer/StartSync', + request_serializer=src_dot_ray_dot_protobuf_dot_ray__syncer__pb2.RaySyncMessage.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_ray__syncer__pb2.RaySyncMessage.FromString, + ) + + +class RaySyncerServicer(object): + """Missing associated documentation comment in .proto file.""" + + def StartSync(self, request_iterator, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_RaySyncerServicer_to_server(servicer, server): + rpc_method_handlers = { + 'StartSync': grpc.stream_stream_rpc_method_handler( + servicer.StartSync, + request_deserializer=src_dot_ray_dot_protobuf_dot_ray__syncer__pb2.RaySyncMessage.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_ray__syncer__pb2.RaySyncMessage.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.syncer.RaySyncer', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class RaySyncer(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def StartSync(request_iterator, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.stream_stream(request_iterator, target, '/ray.rpc.syncer.RaySyncer/StartSync', + src_dot_ray_dot_protobuf_dot_ray__syncer__pb2.RaySyncMessage.SerializeToString, + src_dot_ray_dot_protobuf_dot_ray__syncer__pb2.RaySyncMessage.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/reporter_pb2.py b/python/ray/core/generated/reporter_pb2.py new file mode 100644 index 0000000000000..08891010e1655 --- /dev/null +++ b/python/ray/core/generated/reporter_pb2.py @@ -0,0 +1,173 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/reporter.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from . import metrics_pb2 as opencensus_dot_proto_dot_metrics_dot_v1_dot_metrics__pb2 +from . import common_pb2 as src_dot_ray_dot_protobuf_dot_common__pb2 + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1fsrc/ray/protobuf/reporter.proto\x12\x07ray.rpc\x1a)opencensus/proto/metrics/v1/metrics.proto\x1a\x1dsrc/ray/protobuf/common.proto\"9\n\x18GetProfilingStatsRequest\x12\x0b\n\x03pid\x18\x01 \x01(\r\x12\x10\n\x08\x64uration\x18\x02 \x01(\x05\"S\n\x16GetProfilingStatsReply\x12\x17\n\x0fprofiling_stats\x18\x01 \x01(\t\x12\x0f\n\x07std_out\x18\x02 \x01(\t\x12\x0f\n\x07std_err\x18\x03 \x01(\t\"B\n\x13GetTracebackRequest\x12\x0b\n\x03pid\x18\x01 \x01(\r\x12\x13\n\x06native\x18\x02 \x01(\x08H\x00\x88\x01\x01\x42\t\n\x07_native\"4\n\x11GetTracebackReply\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x0e\n\x06output\x18\x02 \x01(\t\"\x86\x01\n\x13\x43puProfilingRequest\x12\x0b\n\x03pid\x18\x01 \x01(\r\x12\x13\n\x06\x66ormat\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x15\n\x08\x64uration\x18\x03 \x01(\rH\x01\x88\x01\x01\x12\x13\n\x06native\x18\x04 \x01(\x08H\x02\x88\x01\x01\x42\t\n\x07_formatB\x0b\n\t_durationB\t\n\x07_native\"4\n\x11\x43puProfilingReply\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x0e\n\x06output\x18\x02 \x01(\t\"D\n\x14ReportMetricsRequest\x12,\n\x0emetrics_points\x18\x01 \x03(\x0b\x32\x14.ray.rpc.MetricPoint\"9\n\x12ReportMetricsReply\x12#\n\x1bmetrcs_description_required\x18\x01 \x01(\x08\"a\n\x16ReportOCMetricsRequest\x12\x34\n\x07metrics\x18\x01 \x03(\x0b\x32#.opencensus.proto.metrics.v1.Metric\x12\x11\n\tworker_id\x18\x02 \x01(\x0c\"\x16\n\x14ReportOCMetricsReply\"\xd3\x01\n\x10StreamLogRequest\x12\x15\n\rlog_file_name\x18\x01 \x01(\t\x12\x12\n\nkeep_alive\x18\x02 \x01(\x08\x12\x12\n\x05lines\x18\x03 \x01(\x05H\x00\x88\x01\x01\x12\x15\n\x08interval\x18\x04 \x01(\x02H\x01\x88\x01\x01\x12\x19\n\x0cstart_offset\x18\x05 \x01(\x05H\x02\x88\x01\x01\x12\x17\n\nend_offset\x18\x06 \x01(\x05H\x03\x88\x01\x01\x42\x08\n\x06_linesB\x0b\n\t_intervalB\x0f\n\r_start_offsetB\r\n\x0b_end_offset\"\x1e\n\x0eStreamLogReply\x12\x0c\n\x04\x64\x61ta\x18\x01 \x01(\x0c\"&\n\x0fListLogsRequest\x12\x13\n\x0bglob_filter\x18\x01 \x01(\t\"\"\n\rListLogsReply\x12\x11\n\tlog_files\x18\x01 \x03(\t2\x9e\x03\n\x0fReporterService\x12W\n\x11GetProfilingStats\x12!.ray.rpc.GetProfilingStatsRequest\x1a\x1f.ray.rpc.GetProfilingStatsReply\x12K\n\rReportMetrics\x12\x1d.ray.rpc.ReportMetricsRequest\x1a\x1b.ray.rpc.ReportMetricsReply\x12Q\n\x0fReportOCMetrics\x12\x1f.ray.rpc.ReportOCMetricsRequest\x1a\x1d.ray.rpc.ReportOCMetricsReply\x12H\n\x0cGetTraceback\x12\x1c.ray.rpc.GetTracebackRequest\x1a\x1a.ray.rpc.GetTracebackReply\x12H\n\x0c\x43puProfiling\x12\x1c.ray.rpc.CpuProfilingRequest\x1a\x1a.ray.rpc.CpuProfilingReply2\x8d\x01\n\nLogService\x12<\n\x08ListLogs\x12\x18.ray.rpc.ListLogsRequest\x1a\x16.ray.rpc.ListLogsReply\x12\x41\n\tStreamLog\x12\x19.ray.rpc.StreamLogRequest\x1a\x17.ray.rpc.StreamLogReply0\x01\x42\x03\xf8\x01\x01\x62\x06proto3') + + + +_GETPROFILINGSTATSREQUEST = DESCRIPTOR.message_types_by_name['GetProfilingStatsRequest'] +_GETPROFILINGSTATSREPLY = DESCRIPTOR.message_types_by_name['GetProfilingStatsReply'] +_GETTRACEBACKREQUEST = DESCRIPTOR.message_types_by_name['GetTracebackRequest'] +_GETTRACEBACKREPLY = DESCRIPTOR.message_types_by_name['GetTracebackReply'] +_CPUPROFILINGREQUEST = DESCRIPTOR.message_types_by_name['CpuProfilingRequest'] +_CPUPROFILINGREPLY = DESCRIPTOR.message_types_by_name['CpuProfilingReply'] +_REPORTMETRICSREQUEST = DESCRIPTOR.message_types_by_name['ReportMetricsRequest'] +_REPORTMETRICSREPLY = DESCRIPTOR.message_types_by_name['ReportMetricsReply'] +_REPORTOCMETRICSREQUEST = DESCRIPTOR.message_types_by_name['ReportOCMetricsRequest'] +_REPORTOCMETRICSREPLY = DESCRIPTOR.message_types_by_name['ReportOCMetricsReply'] +_STREAMLOGREQUEST = DESCRIPTOR.message_types_by_name['StreamLogRequest'] +_STREAMLOGREPLY = DESCRIPTOR.message_types_by_name['StreamLogReply'] +_LISTLOGSREQUEST = DESCRIPTOR.message_types_by_name['ListLogsRequest'] +_LISTLOGSREPLY = DESCRIPTOR.message_types_by_name['ListLogsReply'] +GetProfilingStatsRequest = _reflection.GeneratedProtocolMessageType('GetProfilingStatsRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETPROFILINGSTATSREQUEST, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetProfilingStatsRequest) + }) +_sym_db.RegisterMessage(GetProfilingStatsRequest) + +GetProfilingStatsReply = _reflection.GeneratedProtocolMessageType('GetProfilingStatsReply', (_message.Message,), { + 'DESCRIPTOR' : _GETPROFILINGSTATSREPLY, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetProfilingStatsReply) + }) +_sym_db.RegisterMessage(GetProfilingStatsReply) + +GetTracebackRequest = _reflection.GeneratedProtocolMessageType('GetTracebackRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETTRACEBACKREQUEST, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetTracebackRequest) + }) +_sym_db.RegisterMessage(GetTracebackRequest) + +GetTracebackReply = _reflection.GeneratedProtocolMessageType('GetTracebackReply', (_message.Message,), { + 'DESCRIPTOR' : _GETTRACEBACKREPLY, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetTracebackReply) + }) +_sym_db.RegisterMessage(GetTracebackReply) + +CpuProfilingRequest = _reflection.GeneratedProtocolMessageType('CpuProfilingRequest', (_message.Message,), { + 'DESCRIPTOR' : _CPUPROFILINGREQUEST, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CpuProfilingRequest) + }) +_sym_db.RegisterMessage(CpuProfilingRequest) + +CpuProfilingReply = _reflection.GeneratedProtocolMessageType('CpuProfilingReply', (_message.Message,), { + 'DESCRIPTOR' : _CPUPROFILINGREPLY, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.CpuProfilingReply) + }) +_sym_db.RegisterMessage(CpuProfilingReply) + +ReportMetricsRequest = _reflection.GeneratedProtocolMessageType('ReportMetricsRequest', (_message.Message,), { + 'DESCRIPTOR' : _REPORTMETRICSREQUEST, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportMetricsRequest) + }) +_sym_db.RegisterMessage(ReportMetricsRequest) + +ReportMetricsReply = _reflection.GeneratedProtocolMessageType('ReportMetricsReply', (_message.Message,), { + 'DESCRIPTOR' : _REPORTMETRICSREPLY, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportMetricsReply) + }) +_sym_db.RegisterMessage(ReportMetricsReply) + +ReportOCMetricsRequest = _reflection.GeneratedProtocolMessageType('ReportOCMetricsRequest', (_message.Message,), { + 'DESCRIPTOR' : _REPORTOCMETRICSREQUEST, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportOCMetricsRequest) + }) +_sym_db.RegisterMessage(ReportOCMetricsRequest) + +ReportOCMetricsReply = _reflection.GeneratedProtocolMessageType('ReportOCMetricsReply', (_message.Message,), { + 'DESCRIPTOR' : _REPORTOCMETRICSREPLY, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ReportOCMetricsReply) + }) +_sym_db.RegisterMessage(ReportOCMetricsReply) + +StreamLogRequest = _reflection.GeneratedProtocolMessageType('StreamLogRequest', (_message.Message,), { + 'DESCRIPTOR' : _STREAMLOGREQUEST, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.StreamLogRequest) + }) +_sym_db.RegisterMessage(StreamLogRequest) + +StreamLogReply = _reflection.GeneratedProtocolMessageType('StreamLogReply', (_message.Message,), { + 'DESCRIPTOR' : _STREAMLOGREPLY, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.StreamLogReply) + }) +_sym_db.RegisterMessage(StreamLogReply) + +ListLogsRequest = _reflection.GeneratedProtocolMessageType('ListLogsRequest', (_message.Message,), { + 'DESCRIPTOR' : _LISTLOGSREQUEST, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ListLogsRequest) + }) +_sym_db.RegisterMessage(ListLogsRequest) + +ListLogsReply = _reflection.GeneratedProtocolMessageType('ListLogsReply', (_message.Message,), { + 'DESCRIPTOR' : _LISTLOGSREPLY, + '__module__' : 'src.ray.protobuf.reporter_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.ListLogsReply) + }) +_sym_db.RegisterMessage(ListLogsReply) + +_REPORTERSERVICE = DESCRIPTOR.services_by_name['ReporterService'] +_LOGSERVICE = DESCRIPTOR.services_by_name['LogService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _GETPROFILINGSTATSREQUEST._serialized_start=118 + _GETPROFILINGSTATSREQUEST._serialized_end=175 + _GETPROFILINGSTATSREPLY._serialized_start=177 + _GETPROFILINGSTATSREPLY._serialized_end=260 + _GETTRACEBACKREQUEST._serialized_start=262 + _GETTRACEBACKREQUEST._serialized_end=328 + _GETTRACEBACKREPLY._serialized_start=330 + _GETTRACEBACKREPLY._serialized_end=382 + _CPUPROFILINGREQUEST._serialized_start=385 + _CPUPROFILINGREQUEST._serialized_end=519 + _CPUPROFILINGREPLY._serialized_start=521 + _CPUPROFILINGREPLY._serialized_end=573 + _REPORTMETRICSREQUEST._serialized_start=575 + _REPORTMETRICSREQUEST._serialized_end=643 + _REPORTMETRICSREPLY._serialized_start=645 + _REPORTMETRICSREPLY._serialized_end=702 + _REPORTOCMETRICSREQUEST._serialized_start=704 + _REPORTOCMETRICSREQUEST._serialized_end=801 + _REPORTOCMETRICSREPLY._serialized_start=803 + _REPORTOCMETRICSREPLY._serialized_end=825 + _STREAMLOGREQUEST._serialized_start=828 + _STREAMLOGREQUEST._serialized_end=1039 + _STREAMLOGREPLY._serialized_start=1041 + _STREAMLOGREPLY._serialized_end=1071 + _LISTLOGSREQUEST._serialized_start=1073 + _LISTLOGSREQUEST._serialized_end=1111 + _LISTLOGSREPLY._serialized_start=1113 + _LISTLOGSREPLY._serialized_end=1147 + _REPORTERSERVICE._serialized_start=1150 + _REPORTERSERVICE._serialized_end=1564 + _LOGSERVICE._serialized_start=1567 + _LOGSERVICE._serialized_end=1708 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/reporter_pb2_grpc.py b/python/ray/core/generated/reporter_pb2_grpc.py new file mode 100644 index 0000000000000..b1622f1c8ac6c --- /dev/null +++ b/python/ray/core/generated/reporter_pb2_grpc.py @@ -0,0 +1,303 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import reporter_pb2 as src_dot_ray_dot_protobuf_dot_reporter__pb2 + + +class ReporterServiceStub(object): + """Service for communicating with the reporter agent module on a remote node. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.GetProfilingStats = channel.unary_unary( + '/ray.rpc.ReporterService/GetProfilingStats', + request_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.GetProfilingStatsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.GetProfilingStatsReply.FromString, + ) + self.ReportMetrics = channel.unary_unary( + '/ray.rpc.ReporterService/ReportMetrics', + request_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.ReportMetricsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.ReportMetricsReply.FromString, + ) + self.ReportOCMetrics = channel.unary_unary( + '/ray.rpc.ReporterService/ReportOCMetrics', + request_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.ReportOCMetricsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.ReportOCMetricsReply.FromString, + ) + self.GetTraceback = channel.unary_unary( + '/ray.rpc.ReporterService/GetTraceback', + request_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.GetTracebackRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.GetTracebackReply.FromString, + ) + self.CpuProfiling = channel.unary_unary( + '/ray.rpc.ReporterService/CpuProfiling', + request_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.CpuProfilingRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.CpuProfilingReply.FromString, + ) + + +class ReporterServiceServicer(object): + """Service for communicating with the reporter agent module on a remote node. + """ + + def GetProfilingStats(self, request, context): + """Get the profiling stats. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ReportMetrics(self, request, context): + """Report metrics to the local metrics agents. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def ReportOCMetrics(self, request, context): + """Report OpenCensus metrics to the local metrics agent. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetTraceback(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def CpuProfiling(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_ReporterServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'GetProfilingStats': grpc.unary_unary_rpc_method_handler( + servicer.GetProfilingStats, + request_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.GetProfilingStatsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.GetProfilingStatsReply.SerializeToString, + ), + 'ReportMetrics': grpc.unary_unary_rpc_method_handler( + servicer.ReportMetrics, + request_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.ReportMetricsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.ReportMetricsReply.SerializeToString, + ), + 'ReportOCMetrics': grpc.unary_unary_rpc_method_handler( + servicer.ReportOCMetrics, + request_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.ReportOCMetricsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.ReportOCMetricsReply.SerializeToString, + ), + 'GetTraceback': grpc.unary_unary_rpc_method_handler( + servicer.GetTraceback, + request_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.GetTracebackRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.GetTracebackReply.SerializeToString, + ), + 'CpuProfiling': grpc.unary_unary_rpc_method_handler( + servicer.CpuProfiling, + request_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.CpuProfilingRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.CpuProfilingReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.ReporterService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class ReporterService(object): + """Service for communicating with the reporter agent module on a remote node. + """ + + @staticmethod + def GetProfilingStats(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ReporterService/GetProfilingStats', + src_dot_ray_dot_protobuf_dot_reporter__pb2.GetProfilingStatsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_reporter__pb2.GetProfilingStatsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ReportMetrics(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ReporterService/ReportMetrics', + src_dot_ray_dot_protobuf_dot_reporter__pb2.ReportMetricsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_reporter__pb2.ReportMetricsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def ReportOCMetrics(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ReporterService/ReportOCMetrics', + src_dot_ray_dot_protobuf_dot_reporter__pb2.ReportOCMetricsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_reporter__pb2.ReportOCMetricsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def GetTraceback(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ReporterService/GetTraceback', + src_dot_ray_dot_protobuf_dot_reporter__pb2.GetTracebackRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_reporter__pb2.GetTracebackReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def CpuProfiling(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.ReporterService/CpuProfiling', + src_dot_ray_dot_protobuf_dot_reporter__pb2.CpuProfilingRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_reporter__pb2.CpuProfilingReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class LogServiceStub(object): + """Service for communicating with the log agent module on a remote node. + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.ListLogs = channel.unary_unary( + '/ray.rpc.LogService/ListLogs', + request_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.ListLogsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.ListLogsReply.FromString, + ) + self.StreamLog = channel.unary_stream( + '/ray.rpc.LogService/StreamLog', + request_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.StreamLogRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.StreamLogReply.FromString, + ) + + +class LogServiceServicer(object): + """Service for communicating with the log agent module on a remote node. + """ + + def ListLogs(self, request, context): + """Get the list of logs from the agent. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def StreamLog(self, request, context): + """Streams a log file from the agent. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_LogServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'ListLogs': grpc.unary_unary_rpc_method_handler( + servicer.ListLogs, + request_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.ListLogsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.ListLogsReply.SerializeToString, + ), + 'StreamLog': grpc.unary_stream_rpc_method_handler( + servicer.StreamLog, + request_deserializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.StreamLogRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_reporter__pb2.StreamLogReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.LogService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class LogService(object): + """Service for communicating with the log agent module on a remote node. + """ + + @staticmethod + def ListLogs(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.LogService/ListLogs', + src_dot_ray_dot_protobuf_dot_reporter__pb2.ListLogsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_reporter__pb2.ListLogsReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def StreamLog(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_stream(request, target, '/ray.rpc.LogService/StreamLog', + src_dot_ray_dot_protobuf_dot_reporter__pb2.StreamLogRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_reporter__pb2.StreamLogReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/resource_pb2.py b/python/ray/core/generated/resource_pb2.py new file mode 100644 index 0000000000000..72fa9e1479758 --- /dev/null +++ b/python/ray/core/generated/resource_pb2.py @@ -0,0 +1,48 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: opencensus/proto/resource/v1/resource.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+opencensus/proto/resource/v1/resource.proto\x12\x1copencensus.proto.resource.v1\"\x8b\x01\n\x08Resource\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\x42\n\x06labels\x18\x02 \x03(\x0b\x32\x32.opencensus.proto.resource.v1.Resource.LabelsEntry\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x42\x98\x01\n\x1fio.opencensus.proto.resource.v1B\rResourceProtoP\x01ZEgithub.com/census-instrumentation/opencensus-proto/gen-go/resource/v1\xea\x02\x1cOpenCensus.Proto.Resource.V1b\x06proto3') + + + +_RESOURCE = DESCRIPTOR.message_types_by_name['Resource'] +_RESOURCE_LABELSENTRY = _RESOURCE.nested_types_by_name['LabelsEntry'] +Resource = _reflection.GeneratedProtocolMessageType('Resource', (_message.Message,), { + + 'LabelsEntry' : _reflection.GeneratedProtocolMessageType('LabelsEntry', (_message.Message,), { + 'DESCRIPTOR' : _RESOURCE_LABELSENTRY, + '__module__' : 'opencensus.proto.resource.v1.resource_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.resource.v1.Resource.LabelsEntry) + }) + , + 'DESCRIPTOR' : _RESOURCE, + '__module__' : 'opencensus.proto.resource.v1.resource_pb2' + # @@protoc_insertion_point(class_scope:opencensus.proto.resource.v1.Resource) + }) +_sym_db.RegisterMessage(Resource) +_sym_db.RegisterMessage(Resource.LabelsEntry) + +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\n\037io.opencensus.proto.resource.v1B\rResourceProtoP\001ZEgithub.com/census-instrumentation/opencensus-proto/gen-go/resource/v1\352\002\034OpenCensus.Proto.Resource.V1' + _RESOURCE_LABELSENTRY._options = None + _RESOURCE_LABELSENTRY._serialized_options = b'8\001' + _RESOURCE._serialized_start=78 + _RESOURCE._serialized_end=217 + _RESOURCE_LABELSENTRY._serialized_start=172 + _RESOURCE_LABELSENTRY._serialized_end=217 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/resource_pb2_grpc.py b/python/ray/core/generated/resource_pb2_grpc.py new file mode 100644 index 0000000000000..2daafffebfc81 --- /dev/null +++ b/python/ray/core/generated/resource_pb2_grpc.py @@ -0,0 +1,4 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + diff --git a/python/ray/core/generated/runtime_env_agent_pb2.py b/python/ray/core/generated/runtime_env_agent_pb2.py new file mode 100644 index 0000000000000..c40319502b072 --- /dev/null +++ b/python/ray/core/generated/runtime_env_agent_pb2.py @@ -0,0 +1,87 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/runtime_env_agent.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + +from . import runtime_env_common_pb2 as src_dot_ray_dot_protobuf_dot_runtime__env__common__pb2 +from . import agent_manager_pb2 as src_dot_ray_dot_protobuf_dot_agent__manager__pb2 + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n(src/ray/protobuf/runtime_env_agent.proto\x12\x07ray.rpc\x1a)src/ray/protobuf/runtime_env_common.proto\x1a$src/ray/protobuf/agent_manager.proto\"\xce\x01\n\x1cGetOrCreateRuntimeEnvRequest\x12\x1e\n\x16serialized_runtime_env\x18\x01 \x01(\t\x12\x35\n\x12runtime_env_config\x18\x02 \x01(\x0b\x32\x19.ray.rpc.RuntimeEnvConfig\x12\x0e\n\x06job_id\x18\x03 \x01(\x0c\x12/\n\'serialized_allocated_resource_instances\x18\x04 \x01(\t\x12\x16\n\x0esource_process\x18\x05 \x01(\t\"\x84\x01\n\x1aGetOrCreateRuntimeEnvReply\x12\'\n\x06status\x18\x01 \x01(\x0e\x32\x17.ray.rpc.AgentRpcStatus\x12\x15\n\rerror_message\x18\x02 \x01(\t\x12&\n\x1eserialized_runtime_env_context\x18\x03 \x01(\t\"[\n!DeleteRuntimeEnvIfPossibleRequest\x12\x1e\n\x16serialized_runtime_env\x18\x01 \x01(\t\x12\x16\n\x0esource_process\x18\x02 \x01(\t\"a\n\x1f\x44\x65leteRuntimeEnvIfPossibleReply\x12\'\n\x06status\x18\x01 \x01(\x0e\x32\x17.ray.rpc.AgentRpcStatus\x12\x15\n\rerror_message\x18\x02 \x01(\t\"9\n\x19GetRuntimeEnvsInfoRequest\x12\x12\n\x05limit\x18\x01 \x01(\x03H\x00\x88\x01\x01\x42\x08\n\x06_limit\"^\n\x17GetRuntimeEnvsInfoReply\x12\x34\n\x12runtime_env_states\x18\x01 \x03(\x0b\x32\x18.ray.rpc.RuntimeEnvState\x12\r\n\x05total\x18\x02 \x01(\x03\x42\x03\xf8\x01\x01\x62\x06proto3') + + + +_GETORCREATERUNTIMEENVREQUEST = DESCRIPTOR.message_types_by_name['GetOrCreateRuntimeEnvRequest'] +_GETORCREATERUNTIMEENVREPLY = DESCRIPTOR.message_types_by_name['GetOrCreateRuntimeEnvReply'] +_DELETERUNTIMEENVIFPOSSIBLEREQUEST = DESCRIPTOR.message_types_by_name['DeleteRuntimeEnvIfPossibleRequest'] +_DELETERUNTIMEENVIFPOSSIBLEREPLY = DESCRIPTOR.message_types_by_name['DeleteRuntimeEnvIfPossibleReply'] +_GETRUNTIMEENVSINFOREQUEST = DESCRIPTOR.message_types_by_name['GetRuntimeEnvsInfoRequest'] +_GETRUNTIMEENVSINFOREPLY = DESCRIPTOR.message_types_by_name['GetRuntimeEnvsInfoReply'] +GetOrCreateRuntimeEnvRequest = _reflection.GeneratedProtocolMessageType('GetOrCreateRuntimeEnvRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETORCREATERUNTIMEENVREQUEST, + '__module__' : 'src.ray.protobuf.runtime_env_agent_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetOrCreateRuntimeEnvRequest) + }) +_sym_db.RegisterMessage(GetOrCreateRuntimeEnvRequest) + +GetOrCreateRuntimeEnvReply = _reflection.GeneratedProtocolMessageType('GetOrCreateRuntimeEnvReply', (_message.Message,), { + 'DESCRIPTOR' : _GETORCREATERUNTIMEENVREPLY, + '__module__' : 'src.ray.protobuf.runtime_env_agent_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetOrCreateRuntimeEnvReply) + }) +_sym_db.RegisterMessage(GetOrCreateRuntimeEnvReply) + +DeleteRuntimeEnvIfPossibleRequest = _reflection.GeneratedProtocolMessageType('DeleteRuntimeEnvIfPossibleRequest', (_message.Message,), { + 'DESCRIPTOR' : _DELETERUNTIMEENVIFPOSSIBLEREQUEST, + '__module__' : 'src.ray.protobuf.runtime_env_agent_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DeleteRuntimeEnvIfPossibleRequest) + }) +_sym_db.RegisterMessage(DeleteRuntimeEnvIfPossibleRequest) + +DeleteRuntimeEnvIfPossibleReply = _reflection.GeneratedProtocolMessageType('DeleteRuntimeEnvIfPossibleReply', (_message.Message,), { + 'DESCRIPTOR' : _DELETERUNTIMEENVIFPOSSIBLEREPLY, + '__module__' : 'src.ray.protobuf.runtime_env_agent_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.DeleteRuntimeEnvIfPossibleReply) + }) +_sym_db.RegisterMessage(DeleteRuntimeEnvIfPossibleReply) + +GetRuntimeEnvsInfoRequest = _reflection.GeneratedProtocolMessageType('GetRuntimeEnvsInfoRequest', (_message.Message,), { + 'DESCRIPTOR' : _GETRUNTIMEENVSINFOREQUEST, + '__module__' : 'src.ray.protobuf.runtime_env_agent_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetRuntimeEnvsInfoRequest) + }) +_sym_db.RegisterMessage(GetRuntimeEnvsInfoRequest) + +GetRuntimeEnvsInfoReply = _reflection.GeneratedProtocolMessageType('GetRuntimeEnvsInfoReply', (_message.Message,), { + 'DESCRIPTOR' : _GETRUNTIMEENVSINFOREPLY, + '__module__' : 'src.ray.protobuf.runtime_env_agent_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.GetRuntimeEnvsInfoReply) + }) +_sym_db.RegisterMessage(GetRuntimeEnvsInfoReply) + +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _GETORCREATERUNTIMEENVREQUEST._serialized_start=135 + _GETORCREATERUNTIMEENVREQUEST._serialized_end=341 + _GETORCREATERUNTIMEENVREPLY._serialized_start=344 + _GETORCREATERUNTIMEENVREPLY._serialized_end=476 + _DELETERUNTIMEENVIFPOSSIBLEREQUEST._serialized_start=478 + _DELETERUNTIMEENVIFPOSSIBLEREQUEST._serialized_end=569 + _DELETERUNTIMEENVIFPOSSIBLEREPLY._serialized_start=571 + _DELETERUNTIMEENVIFPOSSIBLEREPLY._serialized_end=668 + _GETRUNTIMEENVSINFOREQUEST._serialized_start=670 + _GETRUNTIMEENVSINFOREQUEST._serialized_end=727 + _GETRUNTIMEENVSINFOREPLY._serialized_start=729 + _GETRUNTIMEENVSINFOREPLY._serialized_end=823 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/runtime_env_agent_pb2_grpc.py b/python/ray/core/generated/runtime_env_agent_pb2_grpc.py new file mode 100644 index 0000000000000..2daafffebfc81 --- /dev/null +++ b/python/ray/core/generated/runtime_env_agent_pb2_grpc.py @@ -0,0 +1,4 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + diff --git a/python/ray/core/generated/runtime_env_common_pb2.py b/python/ray/core/generated/runtime_env_common_pb2.py new file mode 100644 index 0000000000000..cf7e46deec2fd --- /dev/null +++ b/python/ray/core/generated/runtime_env_common_pb2.py @@ -0,0 +1,65 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/runtime_env_common.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n)src/ray/protobuf/runtime_env_common.proto\x12\x07ray.rpc\"B\n\x0eRuntimeEnvUris\x12\x17\n\x0fworking_dir_uri\x18\x01 \x01(\t\x12\x17\n\x0fpy_modules_uris\x18\x02 \x03(\t\"H\n\x10RuntimeEnvConfig\x12\x1d\n\x15setup_timeout_seconds\x18\x01 \x01(\x05\x12\x15\n\reager_install\x18\x02 \x01(\x08\"\x8e\x01\n\x0eRuntimeEnvInfo\x12\x1e\n\x16serialized_runtime_env\x18\x01 \x01(\t\x12%\n\x04uris\x18\x02 \x01(\x0b\x32\x17.ray.rpc.RuntimeEnvUris\x12\x35\n\x12runtime_env_config\x18\x03 \x01(\x0b\x32\x19.ray.rpc.RuntimeEnvConfig\"\xab\x01\n\x0fRuntimeEnvState\x12\x13\n\x0bruntime_env\x18\x01 \x01(\t\x12\x0f\n\x07ref_cnt\x18\x02 \x01(\x03\x12\x14\n\x07success\x18\x03 \x01(\x08H\x00\x88\x01\x01\x12\x12\n\x05\x65rror\x18\x04 \x01(\tH\x01\x88\x01\x01\x12\x1d\n\x10\x63reation_time_ms\x18\x05 \x01(\x03H\x02\x88\x01\x01\x42\n\n\x08_successB\x08\n\x06_errorB\x13\n\x11_creation_time_msB\x1d\n\x18io.ray.runtime.generated\xf8\x01\x01\x62\x06proto3') + + + +_RUNTIMEENVURIS = DESCRIPTOR.message_types_by_name['RuntimeEnvUris'] +_RUNTIMEENVCONFIG = DESCRIPTOR.message_types_by_name['RuntimeEnvConfig'] +_RUNTIMEENVINFO = DESCRIPTOR.message_types_by_name['RuntimeEnvInfo'] +_RUNTIMEENVSTATE = DESCRIPTOR.message_types_by_name['RuntimeEnvState'] +RuntimeEnvUris = _reflection.GeneratedProtocolMessageType('RuntimeEnvUris', (_message.Message,), { + 'DESCRIPTOR' : _RUNTIMEENVURIS, + '__module__' : 'src.ray.protobuf.runtime_env_common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RuntimeEnvUris) + }) +_sym_db.RegisterMessage(RuntimeEnvUris) + +RuntimeEnvConfig = _reflection.GeneratedProtocolMessageType('RuntimeEnvConfig', (_message.Message,), { + 'DESCRIPTOR' : _RUNTIMEENVCONFIG, + '__module__' : 'src.ray.protobuf.runtime_env_common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RuntimeEnvConfig) + }) +_sym_db.RegisterMessage(RuntimeEnvConfig) + +RuntimeEnvInfo = _reflection.GeneratedProtocolMessageType('RuntimeEnvInfo', (_message.Message,), { + 'DESCRIPTOR' : _RUNTIMEENVINFO, + '__module__' : 'src.ray.protobuf.runtime_env_common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RuntimeEnvInfo) + }) +_sym_db.RegisterMessage(RuntimeEnvInfo) + +RuntimeEnvState = _reflection.GeneratedProtocolMessageType('RuntimeEnvState', (_message.Message,), { + 'DESCRIPTOR' : _RUNTIMEENVSTATE, + '__module__' : 'src.ray.protobuf.runtime_env_common_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.RuntimeEnvState) + }) +_sym_db.RegisterMessage(RuntimeEnvState) + +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\n\030io.ray.runtime.generated\370\001\001' + _RUNTIMEENVURIS._serialized_start=54 + _RUNTIMEENVURIS._serialized_end=120 + _RUNTIMEENVCONFIG._serialized_start=122 + _RUNTIMEENVCONFIG._serialized_end=194 + _RUNTIMEENVINFO._serialized_start=197 + _RUNTIMEENVINFO._serialized_end=339 + _RUNTIMEENVSTATE._serialized_start=342 + _RUNTIMEENVSTATE._serialized_end=513 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/runtime_env_common_pb2_grpc.py b/python/ray/core/generated/runtime_env_common_pb2_grpc.py new file mode 100644 index 0000000000000..2daafffebfc81 --- /dev/null +++ b/python/ray/core/generated/runtime_env_common_pb2_grpc.py @@ -0,0 +1,4 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + diff --git a/python/ray/core/generated/serialization_pb2.py b/python/ray/core/generated/serialization_pb2.py new file mode 100644 index 0000000000000..ccb0bb86d1280 --- /dev/null +++ b/python/ray/core/generated/serialization_pb2.py @@ -0,0 +1,45 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/serialization.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$src/ray/protobuf/serialization.proto\x12\x11ray.serialization\"\x91\x01\n\x0cPythonBuffer\x12\x0f\n\x07\x61\x64\x64ress\x18\x01 \x01(\x04\x12\x0e\n\x06length\x18\x02 \x01(\x03\x12\x10\n\x08itemsize\x18\x03 \x01(\x03\x12\x0c\n\x04ndim\x18\x04 \x01(\x05\x12\x10\n\x08readonly\x18\x05 \x01(\x08\x12\x0e\n\x06\x66ormat\x18\x06 \x01(\t\x12\r\n\x05shape\x18\x07 \x03(\x03\x12\x0f\n\x07strides\x18\x08 \x03(\x03\"s\n\x0cPythonObject\x12\x18\n\x10inband_data_size\x18\x01 \x01(\x04\x12\x18\n\x10raw_buffers_size\x18\x02 \x01(\x04\x12/\n\x06\x62uffer\x18\x03 \x03(\x0b\x32\x1f.ray.serialization.PythonBufferB\x03\xf8\x01\x01\x62\x06proto3') + + + +_PYTHONBUFFER = DESCRIPTOR.message_types_by_name['PythonBuffer'] +_PYTHONOBJECT = DESCRIPTOR.message_types_by_name['PythonObject'] +PythonBuffer = _reflection.GeneratedProtocolMessageType('PythonBuffer', (_message.Message,), { + 'DESCRIPTOR' : _PYTHONBUFFER, + '__module__' : 'src.ray.protobuf.serialization_pb2' + # @@protoc_insertion_point(class_scope:ray.serialization.PythonBuffer) + }) +_sym_db.RegisterMessage(PythonBuffer) + +PythonObject = _reflection.GeneratedProtocolMessageType('PythonObject', (_message.Message,), { + 'DESCRIPTOR' : _PYTHONOBJECT, + '__module__' : 'src.ray.protobuf.serialization_pb2' + # @@protoc_insertion_point(class_scope:ray.serialization.PythonObject) + }) +_sym_db.RegisterMessage(PythonObject) + +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _PYTHONBUFFER._serialized_start=60 + _PYTHONBUFFER._serialized_end=205 + _PYTHONOBJECT._serialized_start=207 + _PYTHONOBJECT._serialized_end=322 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/serialization_pb2_grpc.py b/python/ray/core/generated/serialization_pb2_grpc.py new file mode 100644 index 0000000000000..2daafffebfc81 --- /dev/null +++ b/python/ray/core/generated/serialization_pb2_grpc.py @@ -0,0 +1,4 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + diff --git a/python/ray/core/generated/test_service_pb2.py b/python/ray/core/generated/test_service_pb2.py new file mode 100644 index 0000000000000..c09f07acd62a9 --- /dev/null +++ b/python/ray/core/generated/test_service_pb2.py @@ -0,0 +1,67 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/test_service.proto +"""Generated protocol buffer code.""" +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n#src/ray/protobuf/test_service.proto\x12\x07ray.rpc\"\x1f\n\x0bPingRequest\x12\x10\n\x08no_reply\x18\x01 \x01(\x08\"\x0b\n\tPingReply\"\x14\n\x12PingTimeoutRequest\"\x12\n\x10PingTimeoutReply2\x86\x01\n\x0bTestService\x12\x30\n\x04Ping\x12\x14.ray.rpc.PingRequest\x1a\x12.ray.rpc.PingReply\x12\x45\n\x0bPingTimeout\x12\x1b.ray.rpc.PingTimeoutRequest\x1a\x19.ray.rpc.PingTimeoutReplyb\x06proto3') + + + +_PINGREQUEST = DESCRIPTOR.message_types_by_name['PingRequest'] +_PINGREPLY = DESCRIPTOR.message_types_by_name['PingReply'] +_PINGTIMEOUTREQUEST = DESCRIPTOR.message_types_by_name['PingTimeoutRequest'] +_PINGTIMEOUTREPLY = DESCRIPTOR.message_types_by_name['PingTimeoutReply'] +PingRequest = _reflection.GeneratedProtocolMessageType('PingRequest', (_message.Message,), { + 'DESCRIPTOR' : _PINGREQUEST, + '__module__' : 'src.ray.protobuf.test_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PingRequest) + }) +_sym_db.RegisterMessage(PingRequest) + +PingReply = _reflection.GeneratedProtocolMessageType('PingReply', (_message.Message,), { + 'DESCRIPTOR' : _PINGREPLY, + '__module__' : 'src.ray.protobuf.test_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PingReply) + }) +_sym_db.RegisterMessage(PingReply) + +PingTimeoutRequest = _reflection.GeneratedProtocolMessageType('PingTimeoutRequest', (_message.Message,), { + 'DESCRIPTOR' : _PINGTIMEOUTREQUEST, + '__module__' : 'src.ray.protobuf.test_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PingTimeoutRequest) + }) +_sym_db.RegisterMessage(PingTimeoutRequest) + +PingTimeoutReply = _reflection.GeneratedProtocolMessageType('PingTimeoutReply', (_message.Message,), { + 'DESCRIPTOR' : _PINGTIMEOUTREPLY, + '__module__' : 'src.ray.protobuf.test_service_pb2' + # @@protoc_insertion_point(class_scope:ray.rpc.PingTimeoutReply) + }) +_sym_db.RegisterMessage(PingTimeoutReply) + +_TESTSERVICE = DESCRIPTOR.services_by_name['TestService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + _PINGREQUEST._serialized_start=48 + _PINGREQUEST._serialized_end=79 + _PINGREPLY._serialized_start=81 + _PINGREPLY._serialized_end=92 + _PINGTIMEOUTREQUEST._serialized_start=94 + _PINGTIMEOUTREQUEST._serialized_end=114 + _PINGTIMEOUTREPLY._serialized_start=116 + _PINGTIMEOUTREPLY._serialized_end=134 + _TESTSERVICE._serialized_start=137 + _TESTSERVICE._serialized_end=271 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/test_service_pb2_grpc.py b/python/ray/core/generated/test_service_pb2_grpc.py new file mode 100644 index 0000000000000..e03b358168f82 --- /dev/null +++ b/python/ray/core/generated/test_service_pb2_grpc.py @@ -0,0 +1,99 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import test_service_pb2 as src_dot_ray_dot_protobuf_dot_test__service__pb2 + + +class TestServiceStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.Ping = channel.unary_unary( + '/ray.rpc.TestService/Ping', + request_serializer=src_dot_ray_dot_protobuf_dot_test__service__pb2.PingRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_test__service__pb2.PingReply.FromString, + ) + self.PingTimeout = channel.unary_unary( + '/ray.rpc.TestService/PingTimeout', + request_serializer=src_dot_ray_dot_protobuf_dot_test__service__pb2.PingTimeoutRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_test__service__pb2.PingTimeoutReply.FromString, + ) + + +class TestServiceServicer(object): + """Missing associated documentation comment in .proto file.""" + + def Ping(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def PingTimeout(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_TestServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'Ping': grpc.unary_unary_rpc_method_handler( + servicer.Ping, + request_deserializer=src_dot_ray_dot_protobuf_dot_test__service__pb2.PingRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_test__service__pb2.PingReply.SerializeToString, + ), + 'PingTimeout': grpc.unary_unary_rpc_method_handler( + servicer.PingTimeout, + request_deserializer=src_dot_ray_dot_protobuf_dot_test__service__pb2.PingTimeoutRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_test__service__pb2.PingTimeoutReply.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.rpc.TestService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class TestService(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def Ping(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.TestService/Ping', + src_dot_ray_dot_protobuf_dot_test__service__pb2.PingRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_test__service__pb2.PingReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def PingTimeout(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.rpc.TestService/PingTimeout', + src_dot_ray_dot_protobuf_dot_test__service__pb2.PingTimeoutRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_test__service__pb2.PingTimeoutReply.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/python/ray/core/generated/usage_pb2.py b/python/ray/core/generated/usage_pb2.py new file mode 100644 index 0000000000000..c6bed9cc493d8 --- /dev/null +++ b/python/ray/core/generated/usage_pb2.py @@ -0,0 +1,95 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/usage.proto +"""Generated protocol buffer code.""" +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1csrc/ray/protobuf/usage.proto\x12\tray.usage*\x95\x0f\n\x06TagKey\x12\n\n\x06_TEST1\x10\x00\x12\n\n\x06_TEST2\x10\x01\x12\x13\n\x0fRLLIB_FRAMEWORK\x10\x02\x12\x13\n\x0fRLLIB_ALGORITHM\x10\x03\x12\x15\n\x11RLLIB_NUM_WORKERS\x10\x04\x12\x15\n\x11SERVE_API_VERSION\x10\x05\x12\x19\n\x15SERVE_NUM_DEPLOYMENTS\x10\x06\x12\x0f\n\x0bGCS_STORAGE\x10\x07\x12\x1d\n\x19SERVE_NUM_GPU_DEPLOYMENTS\x10\x08\x12\x16\n\x12SERVE_FASTAPI_USED\x10\t\x12\x19\n\x15SERVE_DAG_DRIVER_USED\x10\n\x12\x1b\n\x17SERVE_HTTP_ADAPTER_USED\x10\x0b\x12\x1b\n\x17SERVE_GRPC_INGRESS_USED\x10\x0c\x12\x1a\n\x16SERVE_REST_API_VERSION\x10\r\x12\x12\n\x0eSERVE_NUM_APPS\x10\x0e\x12*\n&SERVE_NUM_REPLICAS_LIGHTWEIGHT_UPDATED\x10\x0f\x12)\n%SERVE_USER_CONFIG_LIGHTWEIGHT_UPDATED\x10\x10\x12\x30\n,SERVE_AUTOSCALING_CONFIG_LIGHTWEIGHT_UPDATED\x10\x11\x12#\n\x1fSERVE_RAY_SERVE_HANDLE_API_USED\x10\x12\x12(\n$SERVE_RAY_SERVE_SYNC_HANDLE_API_USED\x10\x13\x12$\n SERVE_DEPLOYMENT_HANDLE_API_USED\x10\x14\x12\x32\n.SERVE_DEPLOYMENT_HANDLE_TO_OBJECT_REF_API_USED\x10\x15\x12\x1e\n\x1aSERVE_MULTIPLEXED_API_USED\x10\x16\x12\x19\n\x15SERVE_HTTP_PROXY_USED\x10\x17\x12\x19\n\x15SERVE_GRPC_PROXY_USED\x10\x18\x12\x19\n\x15SERVE_STATUS_API_USED\x10\x19\x12!\n\x1dSERVE_GET_APP_HANDLE_API_USED\x10\x1a\x12(\n$SERVE_GET_DEPLOYMENT_HANDLE_API_USED\x10\x1b\x12\x1e\n\x1a\x43ORE_STATE_API_LIST_ACTORS\x10\x64\x12\x1d\n\x19\x43ORE_STATE_API_LIST_TASKS\x10\x65\x12\x1c\n\x18\x43ORE_STATE_API_LIST_JOBS\x10\x66\x12\x1d\n\x19\x43ORE_STATE_API_LIST_NODES\x10g\x12(\n$CORE_STATE_API_LIST_PLACEMENT_GROUPS\x10h\x12\x1f\n\x1b\x43ORE_STATE_API_LIST_WORKERS\x10i\x12\x1f\n\x1b\x43ORE_STATE_API_LIST_OBJECTS\x10j\x12$\n CORE_STATE_API_LIST_RUNTIME_ENVS\x10k\x12&\n\"CORE_STATE_API_LIST_CLUSTER_EVENTS\x10l\x12\x1c\n\x18\x43ORE_STATE_API_LIST_LOGS\x10m\x12\x1a\n\x16\x43ORE_STATE_API_GET_LOG\x10n\x12\"\n\x1e\x43ORE_STATE_API_SUMMARIZE_TASKS\x10o\x12#\n\x1f\x43ORE_STATE_API_SUMMARIZE_ACTORS\x10p\x12$\n CORE_STATE_API_SUMMARIZE_OBJECTS\x10q\x12\x13\n\x0e\x44\x41SHBOARD_USED\x10\xc8\x01\x12)\n$DASHBOARD_METRICS_PROMETHEUS_ENABLED\x10\xc9\x01\x12&\n!DASHBOARD_METRICS_GRAFANA_ENABLED\x10\xca\x01\x12\x13\n\x0ePG_NUM_CREATED\x10\xac\x02\x12\x16\n\x11\x41\x43TOR_NUM_CREATED\x10\xad\x02\x12\x1e\n\x19WORKER_CRASH_SYSTEM_ERROR\x10\xae\x02\x12\x15\n\x10WORKER_CRASH_OOM\x10\xaf\x02\x12\x19\n\x14RAY_GET_TIMEOUT_ZERO\x10\xb0\x02\x12\x1d\n\x18NUM_ACTOR_CREATION_TASKS\x10\xb1\x02\x12\x14\n\x0fNUM_ACTOR_TASKS\x10\xb2\x02\x12\x15\n\x10NUM_NORMAL_TASKS\x10\xb3\x02\x12\x10\n\x0bNUM_DRIVERS\x10\xb4\x02\x12\"\n\x1d\x45XPERIMENTAL_STATE_API_IMPORT\x10\xb5\x02\x12\x15\n\x10\x44\x41TA_LOGICAL_OPS\x10\x90\x03\x12\x10\n\x0b\x41IR_TRAINER\x10\xf4\x03\x12\x12\n\rTUNE_SEARCHER\x10\xf5\x03\x12\x13\n\x0eTUNE_SCHEDULER\x10\xf6\x03\x12\x11\n\x0c\x41IR_ENV_VARS\x10\xf7\x03\x12%\n AIR_SETUP_WANDB_INTEGRATION_USED\x10\xf8\x03\x12&\n!AIR_SETUP_MLFLOW_INTEGRATION_USED\x10\xf9\x03\x12\x12\n\rAIR_CALLBACKS\x10\xfa\x03\x12\x1e\n\x19\x41IR_STORAGE_CONFIGURATION\x10\xfb\x03\x12\x13\n\x0e\x41IR_ENTRYPOINT\x10\xfc\x03\x42\x03\xf8\x01\x01\x62\x06proto3') + +_TAGKEY = DESCRIPTOR.enum_types_by_name['TagKey'] +TagKey = enum_type_wrapper.EnumTypeWrapper(_TAGKEY) +_TEST1 = 0 +_TEST2 = 1 +RLLIB_FRAMEWORK = 2 +RLLIB_ALGORITHM = 3 +RLLIB_NUM_WORKERS = 4 +SERVE_API_VERSION = 5 +SERVE_NUM_DEPLOYMENTS = 6 +GCS_STORAGE = 7 +SERVE_NUM_GPU_DEPLOYMENTS = 8 +SERVE_FASTAPI_USED = 9 +SERVE_DAG_DRIVER_USED = 10 +SERVE_HTTP_ADAPTER_USED = 11 +SERVE_GRPC_INGRESS_USED = 12 +SERVE_REST_API_VERSION = 13 +SERVE_NUM_APPS = 14 +SERVE_NUM_REPLICAS_LIGHTWEIGHT_UPDATED = 15 +SERVE_USER_CONFIG_LIGHTWEIGHT_UPDATED = 16 +SERVE_AUTOSCALING_CONFIG_LIGHTWEIGHT_UPDATED = 17 +SERVE_RAY_SERVE_HANDLE_API_USED = 18 +SERVE_RAY_SERVE_SYNC_HANDLE_API_USED = 19 +SERVE_DEPLOYMENT_HANDLE_API_USED = 20 +SERVE_DEPLOYMENT_HANDLE_TO_OBJECT_REF_API_USED = 21 +SERVE_MULTIPLEXED_API_USED = 22 +SERVE_HTTP_PROXY_USED = 23 +SERVE_GRPC_PROXY_USED = 24 +SERVE_STATUS_API_USED = 25 +SERVE_GET_APP_HANDLE_API_USED = 26 +SERVE_GET_DEPLOYMENT_HANDLE_API_USED = 27 +CORE_STATE_API_LIST_ACTORS = 100 +CORE_STATE_API_LIST_TASKS = 101 +CORE_STATE_API_LIST_JOBS = 102 +CORE_STATE_API_LIST_NODES = 103 +CORE_STATE_API_LIST_PLACEMENT_GROUPS = 104 +CORE_STATE_API_LIST_WORKERS = 105 +CORE_STATE_API_LIST_OBJECTS = 106 +CORE_STATE_API_LIST_RUNTIME_ENVS = 107 +CORE_STATE_API_LIST_CLUSTER_EVENTS = 108 +CORE_STATE_API_LIST_LOGS = 109 +CORE_STATE_API_GET_LOG = 110 +CORE_STATE_API_SUMMARIZE_TASKS = 111 +CORE_STATE_API_SUMMARIZE_ACTORS = 112 +CORE_STATE_API_SUMMARIZE_OBJECTS = 113 +DASHBOARD_USED = 200 +DASHBOARD_METRICS_PROMETHEUS_ENABLED = 201 +DASHBOARD_METRICS_GRAFANA_ENABLED = 202 +PG_NUM_CREATED = 300 +ACTOR_NUM_CREATED = 301 +WORKER_CRASH_SYSTEM_ERROR = 302 +WORKER_CRASH_OOM = 303 +RAY_GET_TIMEOUT_ZERO = 304 +NUM_ACTOR_CREATION_TASKS = 305 +NUM_ACTOR_TASKS = 306 +NUM_NORMAL_TASKS = 307 +NUM_DRIVERS = 308 +EXPERIMENTAL_STATE_API_IMPORT = 309 +DATA_LOGICAL_OPS = 400 +AIR_TRAINER = 500 +TUNE_SEARCHER = 501 +TUNE_SCHEDULER = 502 +AIR_ENV_VARS = 503 +AIR_SETUP_WANDB_INTEGRATION_USED = 504 +AIR_SETUP_MLFLOW_INTEGRATION_USED = 505 +AIR_CALLBACKS = 506 +AIR_STORAGE_CONFIGURATION = 507 +AIR_ENTRYPOINT = 508 + + +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\370\001\001' + _TAGKEY._serialized_start=44 + _TAGKEY._serialized_end=1985 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/core/generated/usage_pb2_grpc.py b/python/ray/core/generated/usage_pb2_grpc.py new file mode 100644 index 0000000000000..2daafffebfc81 --- /dev/null +++ b/python/ray/core/generated/usage_pb2_grpc.py @@ -0,0 +1,4 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + diff --git a/python/ray/serve/generated/serve_pb2.py b/python/ray/serve/generated/serve_pb2.py new file mode 100644 index 0000000000000..04141696cc84b --- /dev/null +++ b/python/ray/serve/generated/serve_pb2.py @@ -0,0 +1,472 @@ +# -*- coding: utf-8 -*- +# Generated by the protocol buffer compiler. DO NOT EDIT! +# source: src/ray/protobuf/serve.proto +"""Generated protocol buffer code.""" +from google.protobuf.internal import enum_type_wrapper +from google.protobuf import descriptor as _descriptor +from google.protobuf import descriptor_pool as _descriptor_pool +from google.protobuf import message as _message +from google.protobuf import reflection as _reflection +from google.protobuf import symbol_database as _symbol_database +# @@protoc_insertion_point(imports) + +_sym_db = _symbol_database.Default() + + + + +DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1csrc/ray/protobuf/serve.proto\x12\tray.serve\"\xb6\x03\n\x11\x41utoscalingConfig\x12\x14\n\x0cmin_replicas\x18\x01 \x01(\r\x12\x14\n\x0cmax_replicas\x18\x02 \x01(\r\x12/\n\'target_num_ongoing_requests_per_replica\x18\x03 \x01(\x01\x12\x1a\n\x12metrics_interval_s\x18\x04 \x01(\x01\x12\x1a\n\x12look_back_period_s\x18\x05 \x01(\x01\x12\x18\n\x10smoothing_factor\x18\x06 \x01(\x01\x12\x19\n\x11\x64ownscale_delay_s\x18\x07 \x01(\x01\x12\x17\n\x0fupscale_delay_s\x18\x08 \x01(\x01\x12\x1d\n\x10initial_replicas\x18\t \x01(\rH\x00\x88\x01\x01\x12%\n\x18upscale_smoothing_factor\x18\n \x01(\x01H\x01\x88\x01\x01\x12\'\n\x1a\x64ownscale_smoothing_factor\x18\x0b \x01(\x01H\x02\x88\x01\x01\x42\x13\n\x11_initial_replicasB\x1b\n\x19_upscale_smoothing_factorB\x1d\n\x1b_downscale_smoothing_factor\"\xb0\x03\n\x10\x44\x65ploymentConfig\x12\x14\n\x0cnum_replicas\x18\x01 \x01(\x05\x12\x1e\n\x16max_concurrent_queries\x18\x02 \x01(\x05\x12\x13\n\x0buser_config\x18\x03 \x01(\x0c\x12%\n\x1dgraceful_shutdown_wait_loop_s\x18\x04 \x01(\x01\x12#\n\x1bgraceful_shutdown_timeout_s\x18\x05 \x01(\x01\x12\x1d\n\x15health_check_period_s\x18\x06 \x01(\x01\x12\x1e\n\x16health_check_timeout_s\x18\x07 \x01(\x01\x12\x19\n\x11is_cross_language\x18\x08 \x01(\x08\x12:\n\x13\x64\x65ployment_language\x18\t \x01(\x0e\x32\x1d.ray.serve.DeploymentLanguage\x12\x38\n\x12\x61utoscaling_config\x18\n \x01(\x0b\x32\x1c.ray.serve.AutoscalingConfig\x12\x0f\n\x07version\x18\x0b \x01(\t\x12$\n\x1cuser_configured_option_names\x18\x0c \x03(\t\"\xb6\x01\n\x0fRequestMetadata\x12\x12\n\nrequest_id\x18\x01 \x01(\t\x12\x10\n\x08\x65ndpoint\x18\x02 \x01(\t\x12\x13\n\x0b\x63\x61ll_method\x18\x03 \x01(\t\x12\x38\n\x07\x63ontext\x18\x04 \x03(\x0b\x32\'.ray.serve.RequestMetadata.ContextEntry\x1a.\n\x0c\x43ontextEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x1e\n\x0eRequestWrapper\x12\x0c\n\x04\x62ody\x18\x01 \x01(\x0c\"=\n\rUpdatedObject\x12\x17\n\x0fobject_snapshot\x18\x01 \x01(\x0c\x12\x13\n\x0bsnapshot_id\x18\x02 \x01(\x05\"\x9c\x01\n\x0fLongPollRequest\x12O\n\x14keys_to_snapshot_ids\x18\x01 \x03(\x0b\x32\x31.ray.serve.LongPollRequest.KeysToSnapshotIdsEntry\x1a\x38\n\x16KeysToSnapshotIdsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x05:\x02\x38\x01\"\xa9\x01\n\x0eLongPollResult\x12\x46\n\x0fupdated_objects\x18\x01 \x03(\x0b\x32-.ray.serve.LongPollResult.UpdatedObjectsEntry\x1aO\n\x13UpdatedObjectsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\'\n\x05value\x18\x02 \x01(\x0b\x32\x18.ray.serve.UpdatedObject:\x02\x38\x01\"\x98\x01\n\x0c\x45ndpointInfo\x12\x15\n\rendpoint_name\x18\x01 \x01(\t\x12\r\n\x05route\x18\x02 \x01(\t\x12\x33\n\x06\x63onfig\x18\x03 \x03(\x0b\x32#.ray.serve.EndpointInfo.ConfigEntry\x1a-\n\x0b\x43onfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x92\x01\n\x0b\x45ndpointSet\x12\x38\n\tendpoints\x18\x01 \x03(\x0b\x32%.ray.serve.EndpointSet.EndpointsEntry\x1aI\n\x0e\x45ndpointsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12&\n\x05value\x18\x02 \x01(\x0b\x32\x17.ray.serve.EndpointInfo:\x02\x38\x01\"\x1e\n\rActorNameList\x12\r\n\x05names\x18\x01 \x03(\t\"\xde\x01\n\x11\x44\x65ploymentVersion\x12\x14\n\x0c\x63ode_version\x18\x01 \x01(\t\x12\x36\n\x11\x64\x65ployment_config\x18\x02 \x01(\x0b\x32\x1b.ray.serve.DeploymentConfig\x12\x19\n\x11ray_actor_options\x18\x03 \x01(\t\x12\x1f\n\x17placement_group_bundles\x18\x04 \x01(\t\x12 \n\x18placement_group_strategy\x18\x05 \x01(\t\x12\x1d\n\x15max_replicas_per_node\x18\x06 \x01(\x05\"\xe9\x01\n\rReplicaConfig\x12\x1b\n\x13\x64\x65ployment_def_name\x18\x01 \x01(\t\x12\x16\n\x0e\x64\x65ployment_def\x18\x02 \x01(\x0c\x12\x11\n\tinit_args\x18\x03 \x01(\x0c\x12\x13\n\x0binit_kwargs\x18\x04 \x01(\x0c\x12\x19\n\x11ray_actor_options\x18\x05 \x01(\t\x12\x1f\n\x17placement_group_bundles\x18\x06 \x01(\t\x12 \n\x18placement_group_strategy\x18\x07 \x01(\t\x12\x1d\n\x15max_replicas_per_node\x18\x08 \x01(\x05\"\xd9\x01\n\x0e\x44\x65ploymentInfo\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x36\n\x11\x64\x65ployment_config\x18\x02 \x01(\x0b\x32\x1b.ray.serve.DeploymentConfig\x12\x30\n\x0ereplica_config\x18\x03 \x01(\x0b\x32\x18.ray.serve.ReplicaConfig\x12\x15\n\rstart_time_ms\x18\x04 \x01(\x03\x12\x12\n\nactor_name\x18\x05 \x01(\t\x12\x0f\n\x07version\x18\x06 \x01(\t\x12\x13\n\x0b\x65nd_time_ms\x18\x07 \x01(\x03\"T\n\x0f\x44\x65ploymentRoute\x12\x32\n\x0f\x64\x65ployment_info\x18\x01 \x01(\x0b\x32\x19.ray.serve.DeploymentInfo\x12\r\n\x05route\x18\x02 \x01(\t\"L\n\x13\x44\x65ploymentRouteList\x12\x35\n\x11\x64\x65ployment_routes\x18\x01 \x03(\x0b\x32\x1a.ray.serve.DeploymentRoute\"b\n\x14\x44\x65ploymentStatusInfo\x12\x0c\n\x04name\x18\x01 \x01(\t\x12+\n\x06status\x18\x02 \x01(\x0e\x32\x1b.ray.serve.DeploymentStatus\x12\x0f\n\x07message\x18\x03 \x01(\t\"\\\n\x18\x44\x65ploymentStatusInfoList\x12@\n\x17\x64\x65ployment_status_infos\x18\x01 \x03(\x0b\x32\x1f.ray.serve.DeploymentStatusInfo\"t\n\x15\x41pplicationStatusInfo\x12,\n\x06status\x18\x01 \x01(\x0e\x32\x1c.ray.serve.ApplicationStatus\x12\x0f\n\x07message\x18\x02 \x01(\t\x12\x1c\n\x14\x64\x65ployment_timestamp\x18\x03 \x01(\x01\"\x96\x01\n\x0eStatusOverview\x12\x34\n\napp_status\x18\x01 \x01(\x0b\x32 .ray.serve.ApplicationStatusInfo\x12@\n\x13\x64\x65ployment_statuses\x18\x02 \x01(\x0b\x32#.ray.serve.DeploymentStatusInfoList\x12\x0c\n\x04name\x18\x03 \x01(\t\"s\n\x0ePredictRequest\x12\x33\n\x05input\x18\x02 \x03(\x0b\x32$.ray.serve.PredictRequest.InputEntry\x1a,\n\nInputEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x0c:\x02\x38\x01\"%\n\x0fPredictResponse\x12\x12\n\nprediction\x18\x01 \x01(\x0c\"\x19\n\x17ListApplicationsRequest\"5\n\x18ListApplicationsResponse\x12\x19\n\x11\x61pplication_names\x18\x01 \x03(\t\"\x10\n\x0eHealthzRequest\"\"\n\x0fHealthzResponse\x12\x0f\n\x07message\x18\x01 \x01(\t\"<\n\x12UserDefinedMessage\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0b\n\x03\x66oo\x18\x02 \x01(\t\x12\x0b\n\x03num\x18\x03 \x01(\x03\"7\n\x13UserDefinedResponse\x12\x10\n\x08greeting\x18\x01 \x01(\t\x12\x0e\n\x06num_x2\x18\x02 \x01(\x03\"\x15\n\x13UserDefinedMessage2\"(\n\x14UserDefinedResponse2\x12\x10\n\x08greeting\x18\x01 \x01(\t\"=\n\x0c\x46ruitAmounts\x12\x0e\n\x06orange\x18\x01 \x01(\x03\x12\r\n\x05\x61pple\x18\x02 \x01(\x03\x12\x0e\n\x06\x62\x61nana\x18\x03 \x01(\x03\"\x1b\n\nFruitCosts\x12\r\n\x05\x63osts\x18\x01 \x01(\x02\"\x17\n\x07RawData\x12\x0c\n\x04nums\x18\x01 \x03(\x02\"\x1d\n\x0bModelOutput\x12\x0e\n\x06output\x18\x01 \x01(\x02**\n\x12\x44\x65ploymentLanguage\x12\n\n\x06PYTHON\x10\x00\x12\x08\n\x04JAVA\x10\x01*r\n\x10\x44\x65ploymentStatus\x12\x1e\n\x1a\x44\x45PLOYMENT_STATUS_UPDATING\x10\x00\x12\x1d\n\x19\x44\x45PLOYMENT_STATUS_HEALTHY\x10\x01\x12\x1f\n\x1b\x44\x45PLOYMENT_STATUS_UNHEALTHY\x10\x02*\xe2\x01\n\x11\x41pplicationStatus\x12 \n\x1c\x41PPLICATION_STATUS_DEPLOYING\x10\x00\x12\x1e\n\x1a\x41PPLICATION_STATUS_RUNNING\x10\x01\x12$\n APPLICATION_STATUS_DEPLOY_FAILED\x10\x02\x12\x1f\n\x1b\x41PPLICATION_STATUS_DELETING\x10\x03\x12\"\n\x1e\x41PPLICATION_STATUS_NOT_STARTED\x10\x05\x12 \n\x1c\x41PPLICATION_STATUS_UNHEALTHY\x10\x06\x32V\n\x12PredictAPIsService\x12@\n\x07Predict\x12\x19.ray.serve.PredictRequest\x1a\x1a.ray.serve.PredictResponse2\xb3\x01\n\x12RayServeAPIService\x12[\n\x10ListApplications\x12\".ray.serve.ListApplicationsRequest\x1a#.ray.serve.ListApplicationsResponse\x12@\n\x07Healthz\x12\x19.ray.serve.HealthzRequest\x1a\x1a.ray.serve.HealthzResponse2\xc3\x02\n\x12UserDefinedService\x12I\n\x08__call__\x12\x1d.ray.serve.UserDefinedMessage\x1a\x1e.ray.serve.UserDefinedResponse\x12H\n\x07Method1\x12\x1d.ray.serve.UserDefinedMessage\x1a\x1e.ray.serve.UserDefinedResponse\x12J\n\x07Method2\x12\x1e.ray.serve.UserDefinedMessage2\x1a\x1f.ray.serve.UserDefinedResponse2\x12L\n\tStreaming\x12\x1d.ray.serve.UserDefinedMessage\x1a\x1e.ray.serve.UserDefinedResponse0\x01\x32L\n\x0c\x46ruitService\x12<\n\nFruitStand\x12\x17.ray.serve.FruitAmounts\x1a\x15.ray.serve.FruitCosts2S\n\x18RayServeBenchmarkService\x12\x37\n\tgrpc_call\x12\x12.ray.serve.RawData\x1a\x16.ray.serve.ModelOutputB*\n\x16io.ray.serve.generatedB\x0bServeProtosP\x01\xf8\x01\x01\x62\x06proto3') + +_DEPLOYMENTLANGUAGE = DESCRIPTOR.enum_types_by_name['DeploymentLanguage'] +DeploymentLanguage = enum_type_wrapper.EnumTypeWrapper(_DEPLOYMENTLANGUAGE) +_DEPLOYMENTSTATUS = DESCRIPTOR.enum_types_by_name['DeploymentStatus'] +DeploymentStatus = enum_type_wrapper.EnumTypeWrapper(_DEPLOYMENTSTATUS) +_APPLICATIONSTATUS = DESCRIPTOR.enum_types_by_name['ApplicationStatus'] +ApplicationStatus = enum_type_wrapper.EnumTypeWrapper(_APPLICATIONSTATUS) +PYTHON = 0 +JAVA = 1 +DEPLOYMENT_STATUS_UPDATING = 0 +DEPLOYMENT_STATUS_HEALTHY = 1 +DEPLOYMENT_STATUS_UNHEALTHY = 2 +APPLICATION_STATUS_DEPLOYING = 0 +APPLICATION_STATUS_RUNNING = 1 +APPLICATION_STATUS_DEPLOY_FAILED = 2 +APPLICATION_STATUS_DELETING = 3 +APPLICATION_STATUS_NOT_STARTED = 5 +APPLICATION_STATUS_UNHEALTHY = 6 + + +_AUTOSCALINGCONFIG = DESCRIPTOR.message_types_by_name['AutoscalingConfig'] +_DEPLOYMENTCONFIG = DESCRIPTOR.message_types_by_name['DeploymentConfig'] +_REQUESTMETADATA = DESCRIPTOR.message_types_by_name['RequestMetadata'] +_REQUESTMETADATA_CONTEXTENTRY = _REQUESTMETADATA.nested_types_by_name['ContextEntry'] +_REQUESTWRAPPER = DESCRIPTOR.message_types_by_name['RequestWrapper'] +_UPDATEDOBJECT = DESCRIPTOR.message_types_by_name['UpdatedObject'] +_LONGPOLLREQUEST = DESCRIPTOR.message_types_by_name['LongPollRequest'] +_LONGPOLLREQUEST_KEYSTOSNAPSHOTIDSENTRY = _LONGPOLLREQUEST.nested_types_by_name['KeysToSnapshotIdsEntry'] +_LONGPOLLRESULT = DESCRIPTOR.message_types_by_name['LongPollResult'] +_LONGPOLLRESULT_UPDATEDOBJECTSENTRY = _LONGPOLLRESULT.nested_types_by_name['UpdatedObjectsEntry'] +_ENDPOINTINFO = DESCRIPTOR.message_types_by_name['EndpointInfo'] +_ENDPOINTINFO_CONFIGENTRY = _ENDPOINTINFO.nested_types_by_name['ConfigEntry'] +_ENDPOINTSET = DESCRIPTOR.message_types_by_name['EndpointSet'] +_ENDPOINTSET_ENDPOINTSENTRY = _ENDPOINTSET.nested_types_by_name['EndpointsEntry'] +_ACTORNAMELIST = DESCRIPTOR.message_types_by_name['ActorNameList'] +_DEPLOYMENTVERSION = DESCRIPTOR.message_types_by_name['DeploymentVersion'] +_REPLICACONFIG = DESCRIPTOR.message_types_by_name['ReplicaConfig'] +_DEPLOYMENTINFO = DESCRIPTOR.message_types_by_name['DeploymentInfo'] +_DEPLOYMENTROUTE = DESCRIPTOR.message_types_by_name['DeploymentRoute'] +_DEPLOYMENTROUTELIST = DESCRIPTOR.message_types_by_name['DeploymentRouteList'] +_DEPLOYMENTSTATUSINFO = DESCRIPTOR.message_types_by_name['DeploymentStatusInfo'] +_DEPLOYMENTSTATUSINFOLIST = DESCRIPTOR.message_types_by_name['DeploymentStatusInfoList'] +_APPLICATIONSTATUSINFO = DESCRIPTOR.message_types_by_name['ApplicationStatusInfo'] +_STATUSOVERVIEW = DESCRIPTOR.message_types_by_name['StatusOverview'] +_PREDICTREQUEST = DESCRIPTOR.message_types_by_name['PredictRequest'] +_PREDICTREQUEST_INPUTENTRY = _PREDICTREQUEST.nested_types_by_name['InputEntry'] +_PREDICTRESPONSE = DESCRIPTOR.message_types_by_name['PredictResponse'] +_LISTAPPLICATIONSREQUEST = DESCRIPTOR.message_types_by_name['ListApplicationsRequest'] +_LISTAPPLICATIONSRESPONSE = DESCRIPTOR.message_types_by_name['ListApplicationsResponse'] +_HEALTHZREQUEST = DESCRIPTOR.message_types_by_name['HealthzRequest'] +_HEALTHZRESPONSE = DESCRIPTOR.message_types_by_name['HealthzResponse'] +_USERDEFINEDMESSAGE = DESCRIPTOR.message_types_by_name['UserDefinedMessage'] +_USERDEFINEDRESPONSE = DESCRIPTOR.message_types_by_name['UserDefinedResponse'] +_USERDEFINEDMESSAGE2 = DESCRIPTOR.message_types_by_name['UserDefinedMessage2'] +_USERDEFINEDRESPONSE2 = DESCRIPTOR.message_types_by_name['UserDefinedResponse2'] +_FRUITAMOUNTS = DESCRIPTOR.message_types_by_name['FruitAmounts'] +_FRUITCOSTS = DESCRIPTOR.message_types_by_name['FruitCosts'] +_RAWDATA = DESCRIPTOR.message_types_by_name['RawData'] +_MODELOUTPUT = DESCRIPTOR.message_types_by_name['ModelOutput'] +AutoscalingConfig = _reflection.GeneratedProtocolMessageType('AutoscalingConfig', (_message.Message,), { + 'DESCRIPTOR' : _AUTOSCALINGCONFIG, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.AutoscalingConfig) + }) +_sym_db.RegisterMessage(AutoscalingConfig) + +DeploymentConfig = _reflection.GeneratedProtocolMessageType('DeploymentConfig', (_message.Message,), { + 'DESCRIPTOR' : _DEPLOYMENTCONFIG, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.DeploymentConfig) + }) +_sym_db.RegisterMessage(DeploymentConfig) + +RequestMetadata = _reflection.GeneratedProtocolMessageType('RequestMetadata', (_message.Message,), { + + 'ContextEntry' : _reflection.GeneratedProtocolMessageType('ContextEntry', (_message.Message,), { + 'DESCRIPTOR' : _REQUESTMETADATA_CONTEXTENTRY, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.RequestMetadata.ContextEntry) + }) + , + 'DESCRIPTOR' : _REQUESTMETADATA, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.RequestMetadata) + }) +_sym_db.RegisterMessage(RequestMetadata) +_sym_db.RegisterMessage(RequestMetadata.ContextEntry) + +RequestWrapper = _reflection.GeneratedProtocolMessageType('RequestWrapper', (_message.Message,), { + 'DESCRIPTOR' : _REQUESTWRAPPER, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.RequestWrapper) + }) +_sym_db.RegisterMessage(RequestWrapper) + +UpdatedObject = _reflection.GeneratedProtocolMessageType('UpdatedObject', (_message.Message,), { + 'DESCRIPTOR' : _UPDATEDOBJECT, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.UpdatedObject) + }) +_sym_db.RegisterMessage(UpdatedObject) + +LongPollRequest = _reflection.GeneratedProtocolMessageType('LongPollRequest', (_message.Message,), { + + 'KeysToSnapshotIdsEntry' : _reflection.GeneratedProtocolMessageType('KeysToSnapshotIdsEntry', (_message.Message,), { + 'DESCRIPTOR' : _LONGPOLLREQUEST_KEYSTOSNAPSHOTIDSENTRY, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.LongPollRequest.KeysToSnapshotIdsEntry) + }) + , + 'DESCRIPTOR' : _LONGPOLLREQUEST, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.LongPollRequest) + }) +_sym_db.RegisterMessage(LongPollRequest) +_sym_db.RegisterMessage(LongPollRequest.KeysToSnapshotIdsEntry) + +LongPollResult = _reflection.GeneratedProtocolMessageType('LongPollResult', (_message.Message,), { + + 'UpdatedObjectsEntry' : _reflection.GeneratedProtocolMessageType('UpdatedObjectsEntry', (_message.Message,), { + 'DESCRIPTOR' : _LONGPOLLRESULT_UPDATEDOBJECTSENTRY, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.LongPollResult.UpdatedObjectsEntry) + }) + , + 'DESCRIPTOR' : _LONGPOLLRESULT, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.LongPollResult) + }) +_sym_db.RegisterMessage(LongPollResult) +_sym_db.RegisterMessage(LongPollResult.UpdatedObjectsEntry) + +EndpointInfo = _reflection.GeneratedProtocolMessageType('EndpointInfo', (_message.Message,), { + + 'ConfigEntry' : _reflection.GeneratedProtocolMessageType('ConfigEntry', (_message.Message,), { + 'DESCRIPTOR' : _ENDPOINTINFO_CONFIGENTRY, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.EndpointInfo.ConfigEntry) + }) + , + 'DESCRIPTOR' : _ENDPOINTINFO, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.EndpointInfo) + }) +_sym_db.RegisterMessage(EndpointInfo) +_sym_db.RegisterMessage(EndpointInfo.ConfigEntry) + +EndpointSet = _reflection.GeneratedProtocolMessageType('EndpointSet', (_message.Message,), { + + 'EndpointsEntry' : _reflection.GeneratedProtocolMessageType('EndpointsEntry', (_message.Message,), { + 'DESCRIPTOR' : _ENDPOINTSET_ENDPOINTSENTRY, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.EndpointSet.EndpointsEntry) + }) + , + 'DESCRIPTOR' : _ENDPOINTSET, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.EndpointSet) + }) +_sym_db.RegisterMessage(EndpointSet) +_sym_db.RegisterMessage(EndpointSet.EndpointsEntry) + +ActorNameList = _reflection.GeneratedProtocolMessageType('ActorNameList', (_message.Message,), { + 'DESCRIPTOR' : _ACTORNAMELIST, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.ActorNameList) + }) +_sym_db.RegisterMessage(ActorNameList) + +DeploymentVersion = _reflection.GeneratedProtocolMessageType('DeploymentVersion', (_message.Message,), { + 'DESCRIPTOR' : _DEPLOYMENTVERSION, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.DeploymentVersion) + }) +_sym_db.RegisterMessage(DeploymentVersion) + +ReplicaConfig = _reflection.GeneratedProtocolMessageType('ReplicaConfig', (_message.Message,), { + 'DESCRIPTOR' : _REPLICACONFIG, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.ReplicaConfig) + }) +_sym_db.RegisterMessage(ReplicaConfig) + +DeploymentInfo = _reflection.GeneratedProtocolMessageType('DeploymentInfo', (_message.Message,), { + 'DESCRIPTOR' : _DEPLOYMENTINFO, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.DeploymentInfo) + }) +_sym_db.RegisterMessage(DeploymentInfo) + +DeploymentRoute = _reflection.GeneratedProtocolMessageType('DeploymentRoute', (_message.Message,), { + 'DESCRIPTOR' : _DEPLOYMENTROUTE, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.DeploymentRoute) + }) +_sym_db.RegisterMessage(DeploymentRoute) + +DeploymentRouteList = _reflection.GeneratedProtocolMessageType('DeploymentRouteList', (_message.Message,), { + 'DESCRIPTOR' : _DEPLOYMENTROUTELIST, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.DeploymentRouteList) + }) +_sym_db.RegisterMessage(DeploymentRouteList) + +DeploymentStatusInfo = _reflection.GeneratedProtocolMessageType('DeploymentStatusInfo', (_message.Message,), { + 'DESCRIPTOR' : _DEPLOYMENTSTATUSINFO, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.DeploymentStatusInfo) + }) +_sym_db.RegisterMessage(DeploymentStatusInfo) + +DeploymentStatusInfoList = _reflection.GeneratedProtocolMessageType('DeploymentStatusInfoList', (_message.Message,), { + 'DESCRIPTOR' : _DEPLOYMENTSTATUSINFOLIST, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.DeploymentStatusInfoList) + }) +_sym_db.RegisterMessage(DeploymentStatusInfoList) + +ApplicationStatusInfo = _reflection.GeneratedProtocolMessageType('ApplicationStatusInfo', (_message.Message,), { + 'DESCRIPTOR' : _APPLICATIONSTATUSINFO, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.ApplicationStatusInfo) + }) +_sym_db.RegisterMessage(ApplicationStatusInfo) + +StatusOverview = _reflection.GeneratedProtocolMessageType('StatusOverview', (_message.Message,), { + 'DESCRIPTOR' : _STATUSOVERVIEW, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.StatusOverview) + }) +_sym_db.RegisterMessage(StatusOverview) + +PredictRequest = _reflection.GeneratedProtocolMessageType('PredictRequest', (_message.Message,), { + + 'InputEntry' : _reflection.GeneratedProtocolMessageType('InputEntry', (_message.Message,), { + 'DESCRIPTOR' : _PREDICTREQUEST_INPUTENTRY, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.PredictRequest.InputEntry) + }) + , + 'DESCRIPTOR' : _PREDICTREQUEST, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.PredictRequest) + }) +_sym_db.RegisterMessage(PredictRequest) +_sym_db.RegisterMessage(PredictRequest.InputEntry) + +PredictResponse = _reflection.GeneratedProtocolMessageType('PredictResponse', (_message.Message,), { + 'DESCRIPTOR' : _PREDICTRESPONSE, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.PredictResponse) + }) +_sym_db.RegisterMessage(PredictResponse) + +ListApplicationsRequest = _reflection.GeneratedProtocolMessageType('ListApplicationsRequest', (_message.Message,), { + 'DESCRIPTOR' : _LISTAPPLICATIONSREQUEST, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.ListApplicationsRequest) + }) +_sym_db.RegisterMessage(ListApplicationsRequest) + +ListApplicationsResponse = _reflection.GeneratedProtocolMessageType('ListApplicationsResponse', (_message.Message,), { + 'DESCRIPTOR' : _LISTAPPLICATIONSRESPONSE, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.ListApplicationsResponse) + }) +_sym_db.RegisterMessage(ListApplicationsResponse) + +HealthzRequest = _reflection.GeneratedProtocolMessageType('HealthzRequest', (_message.Message,), { + 'DESCRIPTOR' : _HEALTHZREQUEST, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.HealthzRequest) + }) +_sym_db.RegisterMessage(HealthzRequest) + +HealthzResponse = _reflection.GeneratedProtocolMessageType('HealthzResponse', (_message.Message,), { + 'DESCRIPTOR' : _HEALTHZRESPONSE, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.HealthzResponse) + }) +_sym_db.RegisterMessage(HealthzResponse) + +UserDefinedMessage = _reflection.GeneratedProtocolMessageType('UserDefinedMessage', (_message.Message,), { + 'DESCRIPTOR' : _USERDEFINEDMESSAGE, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.UserDefinedMessage) + }) +_sym_db.RegisterMessage(UserDefinedMessage) + +UserDefinedResponse = _reflection.GeneratedProtocolMessageType('UserDefinedResponse', (_message.Message,), { + 'DESCRIPTOR' : _USERDEFINEDRESPONSE, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.UserDefinedResponse) + }) +_sym_db.RegisterMessage(UserDefinedResponse) + +UserDefinedMessage2 = _reflection.GeneratedProtocolMessageType('UserDefinedMessage2', (_message.Message,), { + 'DESCRIPTOR' : _USERDEFINEDMESSAGE2, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.UserDefinedMessage2) + }) +_sym_db.RegisterMessage(UserDefinedMessage2) + +UserDefinedResponse2 = _reflection.GeneratedProtocolMessageType('UserDefinedResponse2', (_message.Message,), { + 'DESCRIPTOR' : _USERDEFINEDRESPONSE2, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.UserDefinedResponse2) + }) +_sym_db.RegisterMessage(UserDefinedResponse2) + +FruitAmounts = _reflection.GeneratedProtocolMessageType('FruitAmounts', (_message.Message,), { + 'DESCRIPTOR' : _FRUITAMOUNTS, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.FruitAmounts) + }) +_sym_db.RegisterMessage(FruitAmounts) + +FruitCosts = _reflection.GeneratedProtocolMessageType('FruitCosts', (_message.Message,), { + 'DESCRIPTOR' : _FRUITCOSTS, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.FruitCosts) + }) +_sym_db.RegisterMessage(FruitCosts) + +RawData = _reflection.GeneratedProtocolMessageType('RawData', (_message.Message,), { + 'DESCRIPTOR' : _RAWDATA, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.RawData) + }) +_sym_db.RegisterMessage(RawData) + +ModelOutput = _reflection.GeneratedProtocolMessageType('ModelOutput', (_message.Message,), { + 'DESCRIPTOR' : _MODELOUTPUT, + '__module__' : 'ray.serve.generated.serve_pb2' + # @@protoc_insertion_point(class_scope:ray.serve.ModelOutput) + }) +_sym_db.RegisterMessage(ModelOutput) + +_PREDICTAPISSERVICE = DESCRIPTOR.services_by_name['PredictAPIsService'] +_RAYSERVEAPISERVICE = DESCRIPTOR.services_by_name['RayServeAPIService'] +_USERDEFINEDSERVICE = DESCRIPTOR.services_by_name['UserDefinedService'] +_FRUITSERVICE = DESCRIPTOR.services_by_name['FruitService'] +_RAYSERVEBENCHMARKSERVICE = DESCRIPTOR.services_by_name['RayServeBenchmarkService'] +if _descriptor._USE_C_DESCRIPTORS == False: + + DESCRIPTOR._options = None + DESCRIPTOR._serialized_options = b'\n\026io.ray.serve.generatedB\013ServeProtosP\001\370\001\001' + _REQUESTMETADATA_CONTEXTENTRY._options = None + _REQUESTMETADATA_CONTEXTENTRY._serialized_options = b'8\001' + _LONGPOLLREQUEST_KEYSTOSNAPSHOTIDSENTRY._options = None + _LONGPOLLREQUEST_KEYSTOSNAPSHOTIDSENTRY._serialized_options = b'8\001' + _LONGPOLLRESULT_UPDATEDOBJECTSENTRY._options = None + _LONGPOLLRESULT_UPDATEDOBJECTSENTRY._serialized_options = b'8\001' + _ENDPOINTINFO_CONFIGENTRY._options = None + _ENDPOINTINFO_CONFIGENTRY._serialized_options = b'8\001' + _ENDPOINTSET_ENDPOINTSENTRY._options = None + _ENDPOINTSET_ENDPOINTSENTRY._serialized_options = b'8\001' + _PREDICTREQUEST_INPUTENTRY._options = None + _PREDICTREQUEST_INPUTENTRY._serialized_options = b'8\001' + _DEPLOYMENTLANGUAGE._serialized_start=3800 + _DEPLOYMENTLANGUAGE._serialized_end=3842 + _DEPLOYMENTSTATUS._serialized_start=3844 + _DEPLOYMENTSTATUS._serialized_end=3958 + _APPLICATIONSTATUS._serialized_start=3961 + _APPLICATIONSTATUS._serialized_end=4187 + _AUTOSCALINGCONFIG._serialized_start=44 + _AUTOSCALINGCONFIG._serialized_end=482 + _DEPLOYMENTCONFIG._serialized_start=485 + _DEPLOYMENTCONFIG._serialized_end=917 + _REQUESTMETADATA._serialized_start=920 + _REQUESTMETADATA._serialized_end=1102 + _REQUESTMETADATA_CONTEXTENTRY._serialized_start=1056 + _REQUESTMETADATA_CONTEXTENTRY._serialized_end=1102 + _REQUESTWRAPPER._serialized_start=1104 + _REQUESTWRAPPER._serialized_end=1134 + _UPDATEDOBJECT._serialized_start=1136 + _UPDATEDOBJECT._serialized_end=1197 + _LONGPOLLREQUEST._serialized_start=1200 + _LONGPOLLREQUEST._serialized_end=1356 + _LONGPOLLREQUEST_KEYSTOSNAPSHOTIDSENTRY._serialized_start=1300 + _LONGPOLLREQUEST_KEYSTOSNAPSHOTIDSENTRY._serialized_end=1356 + _LONGPOLLRESULT._serialized_start=1359 + _LONGPOLLRESULT._serialized_end=1528 + _LONGPOLLRESULT_UPDATEDOBJECTSENTRY._serialized_start=1449 + _LONGPOLLRESULT_UPDATEDOBJECTSENTRY._serialized_end=1528 + _ENDPOINTINFO._serialized_start=1531 + _ENDPOINTINFO._serialized_end=1683 + _ENDPOINTINFO_CONFIGENTRY._serialized_start=1638 + _ENDPOINTINFO_CONFIGENTRY._serialized_end=1683 + _ENDPOINTSET._serialized_start=1686 + _ENDPOINTSET._serialized_end=1832 + _ENDPOINTSET_ENDPOINTSENTRY._serialized_start=1759 + _ENDPOINTSET_ENDPOINTSENTRY._serialized_end=1832 + _ACTORNAMELIST._serialized_start=1834 + _ACTORNAMELIST._serialized_end=1864 + _DEPLOYMENTVERSION._serialized_start=1867 + _DEPLOYMENTVERSION._serialized_end=2089 + _REPLICACONFIG._serialized_start=2092 + _REPLICACONFIG._serialized_end=2325 + _DEPLOYMENTINFO._serialized_start=2328 + _DEPLOYMENTINFO._serialized_end=2545 + _DEPLOYMENTROUTE._serialized_start=2547 + _DEPLOYMENTROUTE._serialized_end=2631 + _DEPLOYMENTROUTELIST._serialized_start=2633 + _DEPLOYMENTROUTELIST._serialized_end=2709 + _DEPLOYMENTSTATUSINFO._serialized_start=2711 + _DEPLOYMENTSTATUSINFO._serialized_end=2809 + _DEPLOYMENTSTATUSINFOLIST._serialized_start=2811 + _DEPLOYMENTSTATUSINFOLIST._serialized_end=2903 + _APPLICATIONSTATUSINFO._serialized_start=2905 + _APPLICATIONSTATUSINFO._serialized_end=3021 + _STATUSOVERVIEW._serialized_start=3024 + _STATUSOVERVIEW._serialized_end=3174 + _PREDICTREQUEST._serialized_start=3176 + _PREDICTREQUEST._serialized_end=3291 + _PREDICTREQUEST_INPUTENTRY._serialized_start=3247 + _PREDICTREQUEST_INPUTENTRY._serialized_end=3291 + _PREDICTRESPONSE._serialized_start=3293 + _PREDICTRESPONSE._serialized_end=3330 + _LISTAPPLICATIONSREQUEST._serialized_start=3332 + _LISTAPPLICATIONSREQUEST._serialized_end=3357 + _LISTAPPLICATIONSRESPONSE._serialized_start=3359 + _LISTAPPLICATIONSRESPONSE._serialized_end=3412 + _HEALTHZREQUEST._serialized_start=3414 + _HEALTHZREQUEST._serialized_end=3430 + _HEALTHZRESPONSE._serialized_start=3432 + _HEALTHZRESPONSE._serialized_end=3466 + _USERDEFINEDMESSAGE._serialized_start=3468 + _USERDEFINEDMESSAGE._serialized_end=3528 + _USERDEFINEDRESPONSE._serialized_start=3530 + _USERDEFINEDRESPONSE._serialized_end=3585 + _USERDEFINEDMESSAGE2._serialized_start=3587 + _USERDEFINEDMESSAGE2._serialized_end=3608 + _USERDEFINEDRESPONSE2._serialized_start=3610 + _USERDEFINEDRESPONSE2._serialized_end=3650 + _FRUITAMOUNTS._serialized_start=3652 + _FRUITAMOUNTS._serialized_end=3713 + _FRUITCOSTS._serialized_start=3715 + _FRUITCOSTS._serialized_end=3742 + _RAWDATA._serialized_start=3744 + _RAWDATA._serialized_end=3767 + _MODELOUTPUT._serialized_start=3769 + _MODELOUTPUT._serialized_end=3798 + _PREDICTAPISSERVICE._serialized_start=4189 + _PREDICTAPISSERVICE._serialized_end=4275 + _RAYSERVEAPISERVICE._serialized_start=4278 + _RAYSERVEAPISERVICE._serialized_end=4457 + _USERDEFINEDSERVICE._serialized_start=4460 + _USERDEFINEDSERVICE._serialized_end=4783 + _FRUITSERVICE._serialized_start=4785 + _FRUITSERVICE._serialized_end=4861 + _RAYSERVEBENCHMARKSERVICE._serialized_start=4863 + _RAYSERVEBENCHMARKSERVICE._serialized_end=4946 +# @@protoc_insertion_point(module_scope) diff --git a/python/ray/serve/generated/serve_pb2_grpc.py b/python/ray/serve/generated/serve_pb2_grpc.py new file mode 100644 index 0000000000000..99746815165b0 --- /dev/null +++ b/python/ray/serve/generated/serve_pb2_grpc.py @@ -0,0 +1,442 @@ +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +"""Client and server classes corresponding to protobuf-defined services.""" +import grpc + +from . import serve_pb2 as src_dot_ray_dot_protobuf_dot_serve__pb2 + + +class PredictAPIsServiceStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.Predict = channel.unary_unary( + '/ray.serve.PredictAPIsService/Predict', + request_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.PredictRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.PredictResponse.FromString, + ) + + +class PredictAPIsServiceServicer(object): + """Missing associated documentation comment in .proto file.""" + + def Predict(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_PredictAPIsServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'Predict': grpc.unary_unary_rpc_method_handler( + servicer.Predict, + request_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.PredictRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.PredictResponse.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.serve.PredictAPIsService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class PredictAPIsService(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def Predict(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.serve.PredictAPIsService/Predict', + src_dot_ray_dot_protobuf_dot_serve__pb2.PredictRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_serve__pb2.PredictResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class RayServeAPIServiceStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.ListApplications = channel.unary_unary( + '/ray.serve.RayServeAPIService/ListApplications', + request_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.ListApplicationsRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.ListApplicationsResponse.FromString, + ) + self.Healthz = channel.unary_unary( + '/ray.serve.RayServeAPIService/Healthz', + request_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.HealthzRequest.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.HealthzResponse.FromString, + ) + + +class RayServeAPIServiceServicer(object): + """Missing associated documentation comment in .proto file.""" + + def ListApplications(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Healthz(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_RayServeAPIServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'ListApplications': grpc.unary_unary_rpc_method_handler( + servicer.ListApplications, + request_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.ListApplicationsRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.ListApplicationsResponse.SerializeToString, + ), + 'Healthz': grpc.unary_unary_rpc_method_handler( + servicer.Healthz, + request_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.HealthzRequest.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.HealthzResponse.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.serve.RayServeAPIService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class RayServeAPIService(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def ListApplications(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.serve.RayServeAPIService/ListApplications', + src_dot_ray_dot_protobuf_dot_serve__pb2.ListApplicationsRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_serve__pb2.ListApplicationsResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def Healthz(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.serve.RayServeAPIService/Healthz', + src_dot_ray_dot_protobuf_dot_serve__pb2.HealthzRequest.SerializeToString, + src_dot_ray_dot_protobuf_dot_serve__pb2.HealthzResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class UserDefinedServiceStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.__call__ = channel.unary_unary( + '/ray.serve.UserDefinedService/__call__', + request_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedMessage.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedResponse.FromString, + ) + self.Method1 = channel.unary_unary( + '/ray.serve.UserDefinedService/Method1', + request_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedMessage.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedResponse.FromString, + ) + self.Method2 = channel.unary_unary( + '/ray.serve.UserDefinedService/Method2', + request_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedMessage2.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedResponse2.FromString, + ) + self.Streaming = channel.unary_stream( + '/ray.serve.UserDefinedService/Streaming', + request_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedMessage.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedResponse.FromString, + ) + + +class UserDefinedServiceServicer(object): + """Missing associated documentation comment in .proto file.""" + + def __call__(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Method1(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Method2(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def Streaming(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_UserDefinedServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + '__call__': grpc.unary_unary_rpc_method_handler( + servicer.__call__, + request_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedMessage.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedResponse.SerializeToString, + ), + 'Method1': grpc.unary_unary_rpc_method_handler( + servicer.Method1, + request_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedMessage.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedResponse.SerializeToString, + ), + 'Method2': grpc.unary_unary_rpc_method_handler( + servicer.Method2, + request_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedMessage2.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedResponse2.SerializeToString, + ), + 'Streaming': grpc.unary_stream_rpc_method_handler( + servicer.Streaming, + request_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedMessage.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedResponse.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.serve.UserDefinedService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class UserDefinedService(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def __call__(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.serve.UserDefinedService/__call__', + src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedMessage.SerializeToString, + src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def Method1(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.serve.UserDefinedService/Method1', + src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedMessage.SerializeToString, + src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def Method2(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.serve.UserDefinedService/Method2', + src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedMessage2.SerializeToString, + src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedResponse2.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + @staticmethod + def Streaming(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_stream(request, target, '/ray.serve.UserDefinedService/Streaming', + src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedMessage.SerializeToString, + src_dot_ray_dot_protobuf_dot_serve__pb2.UserDefinedResponse.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class FruitServiceStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.FruitStand = channel.unary_unary( + '/ray.serve.FruitService/FruitStand', + request_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.FruitAmounts.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.FruitCosts.FromString, + ) + + +class FruitServiceServicer(object): + """Missing associated documentation comment in .proto file.""" + + def FruitStand(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_FruitServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'FruitStand': grpc.unary_unary_rpc_method_handler( + servicer.FruitStand, + request_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.FruitAmounts.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.FruitCosts.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.serve.FruitService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class FruitService(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def FruitStand(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.serve.FruitService/FruitStand', + src_dot_ray_dot_protobuf_dot_serve__pb2.FruitAmounts.SerializeToString, + src_dot_ray_dot_protobuf_dot_serve__pb2.FruitCosts.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) + + +class RayServeBenchmarkServiceStub(object): + """Missing associated documentation comment in .proto file.""" + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.grpc_call = channel.unary_unary( + '/ray.serve.RayServeBenchmarkService/grpc_call', + request_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.RawData.SerializeToString, + response_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.ModelOutput.FromString, + ) + + +class RayServeBenchmarkServiceServicer(object): + """Missing associated documentation comment in .proto file.""" + + def grpc_call(self, request, context): + """Missing associated documentation comment in .proto file.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_RayServeBenchmarkServiceServicer_to_server(servicer, server): + rpc_method_handlers = { + 'grpc_call': grpc.unary_unary_rpc_method_handler( + servicer.grpc_call, + request_deserializer=src_dot_ray_dot_protobuf_dot_serve__pb2.RawData.FromString, + response_serializer=src_dot_ray_dot_protobuf_dot_serve__pb2.ModelOutput.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'ray.serve.RayServeBenchmarkService', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) + + + # This class is part of an EXPERIMENTAL API. +class RayServeBenchmarkService(object): + """Missing associated documentation comment in .proto file.""" + + @staticmethod + def grpc_call(request, + target, + options=(), + channel_credentials=None, + call_credentials=None, + insecure=False, + compression=None, + wait_for_ready=None, + timeout=None, + metadata=None): + return grpc.experimental.unary_unary(request, target, '/ray.serve.RayServeBenchmarkService/grpc_call', + src_dot_ray_dot_protobuf_dot_serve__pb2.RawData.SerializeToString, + src_dot_ray_dot_protobuf_dot_serve__pb2.ModelOutput.FromString, + options, channel_credentials, + insecure, call_credentials, compression, wait_for_ready, timeout, metadata) diff --git a/release/requirements_buildkite.in b/release/requirements_buildkite.in index a5c134efbd815..a7d80acaf4927 100644 --- a/release/requirements_buildkite.in +++ b/release/requirements_buildkite.in @@ -15,4 +15,3 @@ pybuildkite PyGithub requests retry - diff --git a/src/ray/common/asio/periodical_runner.h b/src/ray/common/asio/periodical_runner.h index 292859eb015f5..d0285ee8c90e2 100644 --- a/src/ray/common/asio/periodical_runner.h +++ b/src/ray/common/asio/periodical_runner.h @@ -38,22 +38,24 @@ class PeriodicalRunner { void RunFnPeriodically(std::function fn, uint64_t period_ms, - const std::string name = "UNKNOWN") LOCKS_EXCLUDED(mutex_); + const std::string name = "UNKNOWN") ABSL_LOCKS_EXCLUDED(mutex_); private: void DoRunFnPeriodically(const std::function &fn, boost::posix_time::milliseconds period, std::shared_ptr timer) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); void DoRunFnPeriodicallyInstrumented(const std::function &fn, boost::posix_time::milliseconds period, std::shared_ptr timer, - const std::string name) LOCKS_EXCLUDED(mutex_); + const std::string name) + ABSL_LOCKS_EXCLUDED(mutex_); instrumented_io_context &io_service_; mutable absl::Mutex mutex_; - std::vector> timers_ GUARDED_BY(mutex_); + std::vector> timers_ + ABSL_GUARDED_BY(mutex_); std::shared_ptr stopped_; }; diff --git a/src/ray/common/event_stats.h b/src/ray/common/event_stats.h index 84adbb99c3987..74fbed5c56c7b 100644 --- a/src/ray/common/event_stats.h +++ b/src/ray/common/event_stats.h @@ -43,7 +43,7 @@ struct GlobalStats { /// A mutex wrapper around a handler stats struct. struct GuardedEventStats { // Stats for some handler. - EventStats stats GUARDED_BY(mutex); + EventStats stats ABSL_GUARDED_BY(mutex); // The mutex protecting the reading and writing of these stats. // This mutex should be acquired with a reader lock before reading, and should be @@ -54,7 +54,7 @@ struct GuardedEventStats { /// A mutex wrapper around a handler stats struct. struct GuardedGlobalStats { // Stats over all handlers. - GlobalStats stats GUARDED_BY(mutex); + GlobalStats stats ABSL_GUARDED_BY(mutex); // The mutex protecting the reading and writing of these stats. // This mutex should be acquired with a reader lock before reading, and should be @@ -132,21 +132,21 @@ class EventTracker { /// \param event_name The name of the event whose stats should be returned. /// \return A snapshot view of the event's stats. absl::optional get_event_stats(const std::string &event_name) const - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Returns snapshot views of the count, queueing, and execution statistics for all /// events. /// /// \return A vector containing snapshot views of stats for all events. std::vector> get_event_stats() const - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Builds and returns a statistics summary string. Used by the DebugString() of /// objects that used this io_context wrapper, such as the raylet and the core worker. /// /// \return A stats summary string, suitable for inclusion in an object's /// DebugString(). - std::string StatsString() const LOCKS_EXCLUDED(mutex_); + std::string StatsString() const ABSL_LOCKS_EXCLUDED(mutex_); private: using EventStatsTable = @@ -163,7 +163,7 @@ class EventTracker { /// Table of per-handler post stats. /// We use a std::shared_ptr value in order to ensure pointer stability. - EventStatsTable post_handler_stats_ GUARDED_BY(mutex_); + EventStatsTable post_handler_stats_ ABSL_GUARDED_BY(mutex_); /// Protects access to the per-handler post stats table. mutable absl::Mutex mutex_; diff --git a/src/ray/common/task/task_spec.h b/src/ray/common/task/task_spec.h index 22aebdeb18a99..9cc244efc0b47 100644 --- a/src/ray/common/task/task_spec.h +++ b/src/ray/common/task/task_spec.h @@ -501,10 +501,10 @@ class TaskSpecification : public MessageWrapper { static absl::Mutex mutex_; /// Keep global static id mappings for SchedulingClass for performance. static absl::flat_hash_map sched_cls_to_id_ - GUARDED_BY(mutex_); + ABSL_GUARDED_BY(mutex_); static absl::flat_hash_map sched_id_to_cls_ - GUARDED_BY(mutex_); - static int next_sched_id_ GUARDED_BY(mutex_); + ABSL_GUARDED_BY(mutex_); + static int next_sched_id_ ABSL_GUARDED_BY(mutex_); }; /// \class WorkerCacheKey diff --git a/src/ray/common/test/client_connection_test.cc b/src/ray/common/test/client_connection_test.cc index 8c2fa89b8ddec..39a4e6ac91980 100644 --- a/src/ray/common/test/client_connection_test.cc +++ b/src/ray/common/test/client_connection_test.cc @@ -30,7 +30,7 @@ class ClientConnectionTest : public ::testing::Test { public: ClientConnectionTest() : io_service_(), in_(io_service_), out_(io_service_), error_message_type_(1) { -#if defined(BOOST_ASIO_HAS_LOCAL_SOCKETS) and !defined(_WIN64) +#if defined(BOOST_ASIO_HAS_LOCAL_SOCKETS) && !defined(_WIN32) boost::asio::local::stream_protocol::socket input(io_service_), output(io_service_); boost::asio::local::connect_pair(input, output); in_ = std::move(input); diff --git a/src/ray/common/test/ray_syncer_test.cc b/src/ray/common/test/ray_syncer_test.cc index de5e0617d8245..5d67d52c9893e 100644 --- a/src/ray/common/test/ray_syncer_test.cc +++ b/src/ray/common/test/ray_syncer_test.cc @@ -650,12 +650,12 @@ bool CompareViews(const std::vector &servers, if (!google::protobuf::util::MessageDifferencer::Equals(*v[0], *vv[0])) { RAY_LOG(ERROR) << i << ": FAIL RESOURCE: " << v[0] << ", " << vv[0]; std::string dbg_message; - google::protobuf::util::MessageToJsonString(*v[0], &dbg_message); + RAY_CHECK(google::protobuf::util::MessageToJsonString(*v[0], &dbg_message).ok()); RAY_LOG(ERROR) << "server[0] >> " << NodeID::FromBinary(servers[0]->syncer->GetLocalNodeID()) << ": " << dbg_message << " - " << NodeID::FromBinary(v[0]->node_id()); dbg_message.clear(); - google::protobuf::util::MessageToJsonString(*vv[0], &dbg_message); + RAY_CHECK(google::protobuf::util::MessageToJsonString(*vv[0], &dbg_message).ok()); RAY_LOG(ERROR) << "server[i] << " << NodeID::FromBinary(servers[i]->syncer->GetLocalNodeID()) << ": " << dbg_message << " - " << NodeID::FromBinary(vv[0]->node_id()); diff --git a/src/ray/core_worker/actor_handle.h b/src/ray/core_worker/actor_handle.h index d348077a043ff..209fc8c149280 100644 --- a/src/ray/core_worker/actor_handle.h +++ b/src/ray/core_worker/actor_handle.h @@ -103,7 +103,7 @@ class ActorHandle { // Protobuf-defined persistent state of the actor handle. const rpc::ActorHandle inner_; // Number of tasks that have been submitted on this handle. - uint64_t task_counter_ GUARDED_BY(mutex_) = 0; + uint64_t task_counter_ ABSL_GUARDED_BY(mutex_) = 0; /// Mutex to protect fields in the actor handle. mutable absl::Mutex mutex_; diff --git a/src/ray/core_worker/actor_manager.h b/src/ray/core_worker/actor_manager.h index dde6bf245d108..53dec9a3ff6df 100644 --- a/src/ray/core_worker/actor_manager.h +++ b/src/ray/core_worker/actor_manager.h @@ -193,7 +193,7 @@ class ActorManager { /// Map from actor ID to a handle to that actor. /// Actor handle is a logical abstraction that holds actor handle's states. absl::flat_hash_map> actor_handles_ - GUARDED_BY(mutex_); + ABSL_GUARDED_BY(mutex_); /// Protects access `cached_actor_name_to_ids_` and `subscribed_actors_`. mutable absl::Mutex cache_mutex_; @@ -201,11 +201,11 @@ class ActorManager { /// The map to cache name and id of the named actors in this worker locally, to avoid /// getting them from GCS frequently. absl::flat_hash_map cached_actor_name_to_ids_ - GUARDED_BY(cache_mutex_); + ABSL_GUARDED_BY(cache_mutex_); /// id -> is_killed_or_out_of_scope /// The state of actor is true When the actor is out of scope or is killed - absl::flat_hash_map subscribed_actors_ GUARDED_BY(cache_mutex_); + absl::flat_hash_map subscribed_actors_ ABSL_GUARDED_BY(cache_mutex_); FRIEND_TEST(ActorManagerTest, TestNamedActorIsKilledAfterSubscribeFinished); FRIEND_TEST(ActorManagerTest, TestNamedActorIsKilledBeforeSubscribeFinished); diff --git a/src/ray/core_worker/context.cc b/src/ray/core_worker/context.cc index 6483312da1578..826c94cb2dc8d 100644 --- a/src/ray/core_worker/context.cc +++ b/src/ray/core_worker/context.cc @@ -268,7 +268,8 @@ void WorkerContext::SetCurrentTaskId(const TaskID &task_id, uint64_t attempt_num GetThreadContext().SetCurrentTaskId(task_id, attempt_number); } -void WorkerContext::SetCurrentActorId(const ActorID &actor_id) LOCKS_EXCLUDED(mutex_) { +void WorkerContext::SetCurrentActorId(const ActorID &actor_id) + ABSL_LOCKS_EXCLUDED(mutex_) { absl::WriterMutexLock lock(&mutex_); if (!current_actor_id_.IsNil()) { RAY_CHECK(current_actor_id_ == actor_id); diff --git a/src/ray/core_worker/context.h b/src/ray/core_worker/context.h index b7d2d50e7260e..26a926080c47d 100644 --- a/src/ray/core_worker/context.h +++ b/src/ray/core_worker/context.h @@ -55,45 +55,45 @@ class WorkerContext { const WorkerID &GetWorkerID() const; - JobID GetCurrentJobID() const LOCKS_EXCLUDED(mutex_); - rpc::JobConfig GetCurrentJobConfig() const LOCKS_EXCLUDED(mutex_); + JobID GetCurrentJobID() const ABSL_LOCKS_EXCLUDED(mutex_); + rpc::JobConfig GetCurrentJobConfig() const ABSL_LOCKS_EXCLUDED(mutex_); const TaskID &GetCurrentTaskID() const; const TaskID GetMainThreadOrActorCreationTaskID() const; - const PlacementGroupID &GetCurrentPlacementGroupId() const LOCKS_EXCLUDED(mutex_); + const PlacementGroupID &GetCurrentPlacementGroupId() const ABSL_LOCKS_EXCLUDED(mutex_); - bool ShouldCaptureChildTasksInPlacementGroup() const LOCKS_EXCLUDED(mutex_); + bool ShouldCaptureChildTasksInPlacementGroup() const ABSL_LOCKS_EXCLUDED(mutex_); const std::shared_ptr GetCurrentRuntimeEnvInfo() const - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); - const std::string &GetCurrentSerializedRuntimeEnv() const LOCKS_EXCLUDED(mutex_); + const std::string &GetCurrentSerializedRuntimeEnv() const ABSL_LOCKS_EXCLUDED(mutex_); - std::shared_ptr GetCurrentRuntimeEnv() const LOCKS_EXCLUDED(mutex_); + std::shared_ptr GetCurrentRuntimeEnv() const ABSL_LOCKS_EXCLUDED(mutex_); // Initialize worker's job_id and job_config if they haven't already. // Note a worker's job config can't be changed after initialization. void MaybeInitializeJobInfo(const JobID &job_id, const rpc::JobConfig &job_config) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); // TODO(edoakes): remove this once Python core worker uses the task interfaces. void SetCurrentTaskId(const TaskID &task_id, uint64_t attempt_number); const TaskID &GetCurrentInternalTaskId() const; - void SetCurrentActorId(const ActorID &actor_id) LOCKS_EXCLUDED(mutex_); + void SetCurrentActorId(const ActorID &actor_id) ABSL_LOCKS_EXCLUDED(mutex_); - void SetTaskDepth(int64_t depth) EXCLUSIVE_LOCKS_REQUIRED(mutex_); + void SetTaskDepth(int64_t depth) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); - void SetCurrentTask(const TaskSpecification &task_spec) LOCKS_EXCLUDED(mutex_); + void SetCurrentTask(const TaskSpecification &task_spec) ABSL_LOCKS_EXCLUDED(mutex_); void ResetCurrentTask(); std::shared_ptr GetCurrentTask() const; - const ActorID &GetCurrentActorID() const LOCKS_EXCLUDED(mutex_); + const ActorID &GetCurrentActorID() const ABSL_LOCKS_EXCLUDED(mutex_); /// Returns whether the current thread is the main worker thread. bool CurrentThreadIsMain() const; @@ -103,17 +103,17 @@ class WorkerContext { bool ShouldReleaseResourcesOnBlockingCalls() const; /// Returns whether we are in a direct call actor. - bool CurrentActorIsDirectCall() const LOCKS_EXCLUDED(mutex_); + bool CurrentActorIsDirectCall() const ABSL_LOCKS_EXCLUDED(mutex_); /// Returns whether we are in a direct call task. This encompasses both direct /// actor and normal tasks. - bool CurrentTaskIsDirectCall() const LOCKS_EXCLUDED(mutex_); + bool CurrentTaskIsDirectCall() const ABSL_LOCKS_EXCLUDED(mutex_); - int CurrentActorMaxConcurrency() const LOCKS_EXCLUDED(mutex_); + int CurrentActorMaxConcurrency() const ABSL_LOCKS_EXCLUDED(mutex_); - bool CurrentActorIsAsync() const LOCKS_EXCLUDED(mutex_); + bool CurrentActorIsAsync() const ABSL_LOCKS_EXCLUDED(mutex_); - bool CurrentActorDetached() const LOCKS_EXCLUDED(mutex_); + bool CurrentActorDetached() const ABSL_LOCKS_EXCLUDED(mutex_); uint64_t GetNextTaskIndex(); @@ -134,28 +134,28 @@ class WorkerContext { const WorkerID worker_id_; // a worker's job infomation might be lazily initialized. - JobID current_job_id_ GUARDED_BY(mutex_); - std::optional job_config_ GUARDED_BY(mutex_); - - int64_t task_depth_ GUARDED_BY(mutex_) = 0; - ActorID current_actor_id_ GUARDED_BY(mutex_); - int current_actor_max_concurrency_ GUARDED_BY(mutex_) = 1; - bool current_actor_is_asyncio_ GUARDED_BY(mutex_) = false; - bool is_detached_actor_ GUARDED_BY(mutex_) = false; + JobID current_job_id_ ABSL_GUARDED_BY(mutex_); + std::optional job_config_ ABSL_GUARDED_BY(mutex_); + + int64_t task_depth_ ABSL_GUARDED_BY(mutex_) = 0; + ActorID current_actor_id_ ABSL_GUARDED_BY(mutex_); + int current_actor_max_concurrency_ ABSL_GUARDED_BY(mutex_) = 1; + bool current_actor_is_asyncio_ ABSL_GUARDED_BY(mutex_) = false; + bool is_detached_actor_ ABSL_GUARDED_BY(mutex_) = false; // The placement group id that the current actor belongs to. - PlacementGroupID current_actor_placement_group_id_ GUARDED_BY(mutex_); + PlacementGroupID current_actor_placement_group_id_ ABSL_GUARDED_BY(mutex_); // Whether or not we should implicitly capture parent's placement group. - bool placement_group_capture_child_tasks_ GUARDED_BY(mutex_); + bool placement_group_capture_child_tasks_ ABSL_GUARDED_BY(mutex_); // The runtime env for the current actor or task. - std::shared_ptr runtime_env_ GUARDED_BY(mutex_); + std::shared_ptr runtime_env_ ABSL_GUARDED_BY(mutex_); // The runtime env info. - std::shared_ptr runtime_env_info_ GUARDED_BY(mutex_); + std::shared_ptr runtime_env_info_ ABSL_GUARDED_BY(mutex_); /// The id of the (main) thread that constructed this worker context. const boost::thread::id main_thread_id_; /// The currently executing main thread's task id. It's the actor creation task id /// for concurrent actor, or the main thread's task id for other cases. /// Used merely for observability purposes to track task hierarchy. - TaskID main_thread_or_actor_creation_task_id_ GUARDED_BY(mutex_); + TaskID main_thread_or_actor_creation_task_id_ ABSL_GUARDED_BY(mutex_); // To protect access to mutable members; mutable absl::Mutex mutex_; diff --git a/src/ray/core_worker/core_worker.h b/src/ray/core_worker/core_worker.h index 5b0d10bd4829d..2556a1d1dc1d7 100644 --- a/src/ray/core_worker/core_worker.h +++ b/src/ray/core_worker/core_worker.h @@ -71,7 +71,7 @@ class TaskCounter { TaskCounter() { counter_.SetOnChangeCallback( [this](const std::tuple &key) - EXCLUSIVE_LOCKS_REQUIRED(&mu_) mutable { + ABSL_EXCLUSIVE_LOCKS_REQUIRED(&mu_) mutable { if (std::get<1>(key) != kRunning) { return; } @@ -127,7 +127,7 @@ class TaskCounter { job_id_ = job_id.Hex(); } - bool IsActor() EXCLUSIVE_LOCKS_REQUIRED(&mu_) { return actor_name_.size() > 0; } + bool IsActor() ABSL_EXCLUSIVE_LOCKS_REQUIRED(&mu_) { return actor_name_.size() > 0; } void RecordMetrics() { absl::MutexLock l(&mu_); @@ -243,17 +243,18 @@ class TaskCounter { private: mutable absl::Mutex mu_; // Tracks all tasks submitted to this worker by state, is_retry. - CounterMap> counter_ GUARDED_BY(&mu_); + CounterMap> counter_ + ABSL_GUARDED_BY(&mu_); // Additionally tracks the sub-states of RUNNING_IN_RAY_GET/WAIT. The counters here // overlap with those of counter_. - CounterMap> running_in_get_counter_ GUARDED_BY(&mu_); - CounterMap> running_in_wait_counter_ GUARDED_BY(&mu_); + CounterMap> running_in_get_counter_ ABSL_GUARDED_BY(&mu_); + CounterMap> running_in_wait_counter_ ABSL_GUARDED_BY(&mu_); - std::string job_id_ GUARDED_BY(&mu_) = ""; + std::string job_id_ ABSL_GUARDED_BY(&mu_) = ""; // Used for actor state tracking. - std::string actor_name_ GUARDED_BY(&mu_) = ""; - int64_t num_tasks_running_ GUARDED_BY(&mu_) = 0; + std::string actor_name_ ABSL_GUARDED_BY(&mu_) = ""; + int64_t num_tasks_running_ ABSL_GUARDED_BY(&mu_) = 0; }; struct TaskToRetry { @@ -803,7 +804,7 @@ class CoreWorker : public rpc::CoreWorkerServiceHandler { /// For actors, this is the current actor ID. To make sure that all caller /// IDs have the same type, we embed the actor ID in a TaskID with the rest /// of the bytes zeroed out. - TaskID GetCallerId() const LOCKS_EXCLUDED(mutex_); + TaskID GetCallerId() const ABSL_LOCKS_EXCLUDED(mutex_); /// Push an error to the relevant driver. /// @@ -1600,14 +1601,14 @@ class CoreWorker : public rpc::CoreWorkerServiceHandler { /// The ID of the current task being executed by the main thread. If there /// are multiple threads, they will have a thread-local task ID stored in the /// worker context. - TaskID main_thread_task_id_ GUARDED_BY(mutex_); + TaskID main_thread_task_id_ ABSL_GUARDED_BY(mutex_); - std::string main_thread_task_name_ GUARDED_BY(mutex_); + std::string main_thread_task_name_ ABSL_GUARDED_BY(mutex_); /// States that used for initialization. absl::Mutex initialize_mutex_; absl::CondVar intialize_cv_; - bool initialized_ GUARDED_BY(initialize_mutex_) = false; + bool initialized_ ABSL_GUARDED_BY(initialize_mutex_) = false; /// Event loop where the IO events are handled. e.g. async GCS operations. instrumented_io_context io_service_; @@ -1705,20 +1706,20 @@ class CoreWorker : public rpc::CoreWorkerServiceHandler { mutable absl::Mutex mutex_; /// Our actor ID. If this is nil, then we execute only stateless tasks. - ActorID actor_id_ GUARDED_BY(mutex_); + ActorID actor_id_ ABSL_GUARDED_BY(mutex_); /// The currently executing task spec. We have to track this separately since /// we cannot access the thread-local worker contexts from GetCoreWorkerStats() - absl::flat_hash_map current_tasks_ GUARDED_BY(mutex_); + absl::flat_hash_map current_tasks_ ABSL_GUARDED_BY(mutex_); /// Key value pairs to be displayed on Web UI. - std::unordered_map webui_display_ GUARDED_BY(mutex_); + std::unordered_map webui_display_ ABSL_GUARDED_BY(mutex_); /// Actor title that consists of class name, args, kwargs for actor construction. - std::string actor_title_ GUARDED_BY(mutex_); + std::string actor_title_ ABSL_GUARDED_BY(mutex_); /// Actor repr name if overrides by the user, empty string if not. - std::string actor_repr_name_ GUARDED_BY(mutex_) = ""; + std::string actor_repr_name_ ABSL_GUARDED_BY(mutex_) = ""; /// Number of tasks that have been pushed to the actor but not executed. std::atomic task_queue_length_; @@ -1729,7 +1730,7 @@ class CoreWorker : public rpc::CoreWorkerServiceHandler { /// A map from resource name to the resource IDs that are currently reserved /// for this worker. Each pair consists of the resource ID and the fraction /// of that resource allocated for this worker. This is set on task assignment. - std::shared_ptr resource_ids_ GUARDED_BY(mutex_); + std::shared_ptr resource_ids_ ABSL_GUARDED_BY(mutex_); /// Common rpc service for all worker modules. rpc::CoreWorkerGrpcService grpc_service_; @@ -1751,7 +1752,7 @@ class CoreWorker : public rpc::CoreWorkerServiceHandler { // Queue of tasks to resubmit when the specified time passes. std::priority_queue, TaskToRetryDescComparator> - to_resubmit_ GUARDED_BY(mutex_); + to_resubmit_ ABSL_GUARDED_BY(mutex_); /// Map of named actor registry. It doesn't need to hold a lock because /// local mode is single-threaded. @@ -1762,7 +1763,7 @@ class CoreWorker : public rpc::CoreWorkerServiceHandler { // Callbacks for when when a plasma object becomes ready. absl::flat_hash_map>> - async_plasma_callbacks_ GUARDED_BY(plasma_mutex_); + async_plasma_callbacks_ ABSL_GUARDED_BY(plasma_mutex_); // Fallback for when GetAsync cannot directly get the requested object. void PlasmaCallback(SetResultCallback success, @@ -1772,7 +1773,7 @@ class CoreWorker : public rpc::CoreWorkerServiceHandler { /// The detail reason why the core worker has exited. /// If this value is set, it means the exit process has begun. - std::optional exiting_detail_ GUARDED_BY(mutex_); + std::optional exiting_detail_ ABSL_GUARDED_BY(mutex_); std::atomic is_shutdown_ = false; diff --git a/src/ray/core_worker/core_worker_process.h b/src/ray/core_worker/core_worker_process.h index 09f3853d3ddc2..401cede54e57b 100644 --- a/src/ray/core_worker/core_worker_process.h +++ b/src/ray/core_worker/core_worker_process.h @@ -141,7 +141,7 @@ class CoreWorkerProcessImpl { const CoreWorkerOptions options_; /// The core worker instance of this worker process. - std::shared_ptr core_worker_ GUARDED_BY(mutex_); + std::shared_ptr core_worker_ ABSL_GUARDED_BY(mutex_); /// The worker ID of this worker. const WorkerID worker_id_; diff --git a/src/ray/core_worker/object_recovery_manager.h b/src/ray/core_worker/object_recovery_manager.h index b224e972d246b..15cf18503f09d 100644 --- a/src/ray/core_worker/object_recovery_manager.h +++ b/src/ray/core_worker/object_recovery_manager.h @@ -136,11 +136,11 @@ class ObjectRecoveryManager { /// Cache of gRPC clients to remote raylets for pinning objects. absl::flat_hash_map> - remote_object_pinning_clients_ GUARDED_BY(mu_); + remote_object_pinning_clients_ ABSL_GUARDED_BY(mu_); /// Objects that are currently pending recovery. Calls to RecoverObject for /// objects currently in this set are idempotent. - absl::flat_hash_set objects_pending_recovery_ GUARDED_BY(mu_); + absl::flat_hash_set objects_pending_recovery_ ABSL_GUARDED_BY(mu_); }; } // namespace core diff --git a/src/ray/core_worker/reference_count.h b/src/ray/core_worker/reference_count.h index 3243532a66cc3..6be0c6daa8012 100644 --- a/src/ray/core_worker/reference_count.h +++ b/src/ray/core_worker/reference_count.h @@ -100,14 +100,14 @@ class ReferenceCounter : public ReferenceCounterInterface, /// /// \param[in] object_id The object to to increment the count for. void AddLocalReference(const ObjectID &object_id, const std::string &call_site) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Decrease the local reference count for the ObjectID by one. /// /// \param[in] object_id The object to decrement the count for. /// \param[out] deleted List to store objects that hit zero ref count. void RemoveLocalReference(const ObjectID &object_id, std::vector *deleted) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Add references for the provided object IDs that correspond to them being /// dependencies to a submitted task. If lineage pinning is enabled, then @@ -124,7 +124,7 @@ class ReferenceCounter : public ReferenceCounterInterface, const std::vector return_ids, const std::vector &argument_ids_to_add, const std::vector &argument_ids_to_remove = std::vector(), - std::vector *deleted = nullptr) LOCKS_EXCLUDED(mutex_); + std::vector *deleted = nullptr) ABSL_LOCKS_EXCLUDED(mutex_); /// Add references for the object dependencies of a resubmitted task. This /// does not increment the arguments' lineage ref counts because we should @@ -133,7 +133,7 @@ class ReferenceCounter : public ReferenceCounterInterface, /// \param[in] argument_ids The arguments of the task to add references for. void UpdateResubmittedTaskReferences(const std::vector return_ids, const std::vector &argument_ids) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Update object references that were given to a submitted task. The task /// may still be borrowing any object IDs that were contained in its @@ -155,7 +155,7 @@ class ReferenceCounter : public ReferenceCounterInterface, const rpc::Address &worker_addr, const ReferenceTableProto &borrowed_refs, std::vector *deleted) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Add an object that we own. The object may depend on other objects. /// Dependencies for each ObjectID must be set at most once. The local @@ -188,7 +188,7 @@ class ReferenceCounter : public ReferenceCounterInterface, bool is_reconstructable, bool add_local_ref, const absl::optional &pinned_at_raylet_id = - absl::optional()) LOCKS_EXCLUDED(mutex_); + absl::optional()) ABSL_LOCKS_EXCLUDED(mutex_); /// Add an owned object that was dynamically created. These are objects that /// were created by a task that we called, but that we own. @@ -199,7 +199,7 @@ class ReferenceCounter : public ReferenceCounterInterface, /// update its ref count info to show that it contains the dynamically /// created ObjectID. void AddDynamicReturn(const ObjectID &object_id, const ObjectID &generator_id) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Own an object that the current owner (current process) dynamically created. /// @@ -221,14 +221,14 @@ class ReferenceCounter : public ReferenceCounterInterface, /// \param[in] generator_id The Object ID of the streaming generator task. void OwnDynamicStreamingTaskReturnRef(const ObjectID &object_id, const ObjectID &generator_id) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Update the size of the object. /// /// \param[in] object_id The ID of the object. /// \param[in] size The known size of the object. void UpdateObjectSize(const ObjectID &object_id, int64_t object_size) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Add an object that we are borrowing. /// @@ -243,7 +243,7 @@ class ReferenceCounter : public ReferenceCounterInterface, const ObjectID &outer_id, const rpc::Address &owner_address, bool foreign_owner_already_monitoring = false) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Get the owner address of the given object. /// @@ -256,13 +256,13 @@ class ReferenceCounter : public ReferenceCounterInterface, /// ownership information. The latter can happen when object IDs are pasesd /// out of band. bool GetOwner(const ObjectID &object_id, rpc::Address *owner_address = nullptr) const - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Check if the object has an owner. /// /// \param[in] object_id The ID of the object. /// \return if the object has an owner. - bool HasOwner(const ObjectID &object_id) const LOCKS_EXCLUDED(mutex_); + bool HasOwner(const ObjectID &object_id) const ABSL_LOCKS_EXCLUDED(mutex_); /// Get the owner addresses of the given objects. The owner address /// must be registered for these objects. @@ -290,16 +290,17 @@ class ReferenceCounter : public ReferenceCounterInterface, /// Release the underlying value from plasma (if any) for these objects. /// /// \param[in] object_ids The IDs whose values to free. - void FreePlasmaObjects(const std::vector &object_ids) LOCKS_EXCLUDED(mutex_); + void FreePlasmaObjects(const std::vector &object_ids) + ABSL_LOCKS_EXCLUDED(mutex_); /// Sets the callback that will be run when the object goes out of scope. /// Returns true if the object was in scope and the callback was added, else false. bool SetDeleteCallback(const ObjectID &object_id, const std::function callback) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); void ResetDeleteCallbacks(const std::vector &object_ids) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Set a callback for when we are no longer borrowing this object (when our /// ref count goes to 0). @@ -316,7 +317,7 @@ class ReferenceCounter : public ReferenceCounterInterface, const ObjectID &contained_in_id, const rpc::Address &owner_address, const ReferenceRemovedCallback &ref_removed_callback) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Set a callback to call whenever a Reference that we own is deleted. A /// Reference can only be deleted if: @@ -332,24 +333,24 @@ class ReferenceCounter : public ReferenceCounterInterface, /// RefCount() for the object ID goes to 0. /// /// \param[in] object_id The object that we were borrowing. - void HandleRefRemoved(const ObjectID &object_id) EXCLUSIVE_LOCKS_REQUIRED(mutex_); + void HandleRefRemoved(const ObjectID &object_id) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Returns the total number of ObjectIDs currently in scope. - size_t NumObjectIDsInScope() const LOCKS_EXCLUDED(mutex_); + size_t NumObjectIDsInScope() const ABSL_LOCKS_EXCLUDED(mutex_); /// Returns the total number of objects owned by this worker. - size_t NumObjectsOwnedByUs() const LOCKS_EXCLUDED(mutex_); + size_t NumObjectsOwnedByUs() const ABSL_LOCKS_EXCLUDED(mutex_); /// Returns the total number of actors owned by this worker. - size_t NumActorsOwnedByUs() const LOCKS_EXCLUDED(mutex_); + size_t NumActorsOwnedByUs() const ABSL_LOCKS_EXCLUDED(mutex_); /// Returns a set of all ObjectIDs currently in scope (i.e., nonzero reference count). - std::unordered_set GetAllInScopeObjectIDs() const LOCKS_EXCLUDED(mutex_); + std::unordered_set GetAllInScopeObjectIDs() const ABSL_LOCKS_EXCLUDED(mutex_); /// Returns a map of all ObjectIDs currently in scope with a pair of their /// (local, submitted_task) reference counts. For debugging purposes. std::unordered_map> GetAllReferenceCounts() const - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Populate a table with ObjectIDs that we were or are still borrowing. /// This should be called when a task returns, and the argument should be any @@ -372,7 +373,8 @@ class ReferenceCounter : public ReferenceCounterInterface, /// references. void PopAndClearLocalBorrowers(const std::vector &borrowed_ids, ReferenceTableProto *proto, - std::vector *deleted) LOCKS_EXCLUDED(mutex_); + std::vector *deleted) + ABSL_LOCKS_EXCLUDED(mutex_); /// Mark that this ObjectID contains another ObjectID(s). This should be /// called in two cases: @@ -392,14 +394,15 @@ class ReferenceCounter : public ReferenceCounterInterface, /// IDs. void AddNestedObjectIds(const ObjectID &object_id, const std::vector &inner_ids, - const rpc::WorkerAddress &owner_address) LOCKS_EXCLUDED(mutex_); + const rpc::WorkerAddress &owner_address) + ABSL_LOCKS_EXCLUDED(mutex_); /// Update the pinned location of an object stored in plasma. /// /// \param[in] object_id The object to update. /// \param[in] raylet_id The raylet that is now pinning the object ID. void UpdateObjectPinnedAtRaylet(const ObjectID &object_id, const NodeID &raylet_id) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Check whether the object is pinned at a remote plasma store node or /// spilled to external storage. In either case, a copy of the object is @@ -415,7 +418,7 @@ class ReferenceCounter : public ReferenceCounterInterface, bool IsPlasmaObjectPinnedOrSpilled(const ObjectID &object_id, bool *owned_by_us, NodeID *pinned_at, - bool *spilled) const LOCKS_EXCLUDED(mutex_); + bool *spilled) const ABSL_LOCKS_EXCLUDED(mutex_); /// Get and reset the objects that were pinned or spilled on the given node. /// This method should be called upon a node failure, to trigger @@ -434,7 +437,7 @@ class ReferenceCounter : public ReferenceCounterInterface, /// /// \param[in] object_id The object ID to check for. /// \return Whether we have a reference to the object ID. - bool HasReference(const ObjectID &object_id) const LOCKS_EXCLUDED(mutex_); + bool HasReference(const ObjectID &object_id) const ABSL_LOCKS_EXCLUDED(mutex_); /// Write the current reference table to the given proto. /// @@ -442,7 +445,7 @@ class ReferenceCounter : public ReferenceCounterInterface, void AddObjectRefStats( const absl::flat_hash_map> pinned_objects, rpc::CoreWorkerStats *stats, - const int64_t limit) const LOCKS_EXCLUDED(mutex_); + const int64_t limit) const ABSL_LOCKS_EXCLUDED(mutex_); /// Add a new location for the given object. The owner must have the object ref in /// scope. @@ -451,7 +454,7 @@ class ReferenceCounter : public ReferenceCounterInterface, /// \param[in] node_id The new object location to be added. /// \return True if the reference exists, false otherwise. bool AddObjectLocation(const ObjectID &object_id, const NodeID &node_id) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Remove a location for the given object. The owner must have the object ref in /// scope. @@ -460,7 +463,7 @@ class ReferenceCounter : public ReferenceCounterInterface, /// \param[in] node_id The object location to be removed. /// \return True if the reference exists, false otherwise. bool RemoveObjectLocation(const ObjectID &object_id, const NodeID &node_id) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Get the locations of the given object. The owner must have the object ref in /// scope. @@ -469,14 +472,15 @@ class ReferenceCounter : public ReferenceCounterInterface, /// \return The nodes that have the object if the reference exists, empty optional /// otherwise. absl::optional> GetObjectLocations( - const ObjectID &object_id) LOCKS_EXCLUDED(mutex_); + const ObjectID &object_id) ABSL_LOCKS_EXCLUDED(mutex_); /// Publish the snapshot of the object location for the given object id. /// Publish the empty locations if object is already evicted or not owned by this /// worker. /// /// \param[in] object_id The object whose locations we want. - void PublishObjectLocationSnapshot(const ObjectID &object_id) LOCKS_EXCLUDED(mutex_); + void PublishObjectLocationSnapshot(const ObjectID &object_id) + ABSL_LOCKS_EXCLUDED(mutex_); /// Fill up the object information. /// @@ -487,7 +491,7 @@ class ReferenceCounter : public ReferenceCounterInterface, /// doesn't exist. Status FillObjectInformation(const ObjectID &object_id, rpc::WorkerObjectLocationsPubMessage *object_info) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Handle an object has been spilled to external storage. /// @@ -526,7 +530,7 @@ class ReferenceCounter : public ReferenceCounterInterface, /// \param[in] object_id The ID of Object whose been borrowed. /// \param[in] borrower_address The address of borrower. void AddBorrowerAddress(const ObjectID &object_id, const rpc::Address &borrower_address) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); bool IsObjectReconstructable(const ObjectID &object_id, bool *lineage_evicted) const; @@ -794,14 +798,14 @@ class ReferenceCounter : public ReferenceCounterInterface, bool is_reconstructable, bool add_local_ref, const absl::optional &pinned_at_raylet_id) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); void SetNestedRefInUseRecursive(ReferenceTable::iterator inner_ref_it) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); bool GetOwnerInternal(const ObjectID &object_id, rpc::Address *owner_address = nullptr) const - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Release the pinned plasma object, if any. Also unsets the raylet address /// that the object was pinned at, if the address was set. @@ -809,7 +813,7 @@ class ReferenceCounter : public ReferenceCounterInterface, /// Shutdown if all references have gone out of scope and shutdown /// is scheduled. - void ShutdownIfNeeded() EXCLUSIVE_LOCKS_REQUIRED(mutex_); + void ShutdownIfNeeded() ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Deserialize a ReferenceTable. static ReferenceTable ReferenceTableFromProto(const ReferenceTableProto &proto); @@ -826,7 +830,7 @@ class ReferenceCounter : public ReferenceCounterInterface, void RemoveSubmittedTaskReferences(const std::vector &argument_ids, bool release_lineage, std::vector *deleted) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Helper method to mark that this ObjectID contains another ObjectID(s). /// @@ -840,7 +844,7 @@ class ReferenceCounter : public ReferenceCounterInterface, void AddNestedObjectIdsInternal(const ObjectID &object_id, const std::vector &inner_ids, const rpc::WorkerAddress &owner_address) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Populates the table with the ObjectID that we were or are still /// borrowing. The table also includes any IDs that we discovered were @@ -866,7 +870,7 @@ class ReferenceCounter : public ReferenceCounterInterface, bool for_ref_removed, bool deduct_local_ref, ReferenceProtoTable *borrowed_refs) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Merge remote borrowers into our local ref count. This will add any /// workers that are still borrowing the given object ID to the local ref @@ -884,7 +888,7 @@ class ReferenceCounter : public ReferenceCounterInterface, void MergeRemoteBorrowers(const ObjectID &object_id, const rpc::WorkerAddress &worker_addr, const ReferenceTable &borrowed_refs) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Wait for a borrower to stop using its reference. This should only be /// called by the owner of the ID. @@ -897,7 +901,7 @@ class ReferenceCounter : public ReferenceCounterInterface, void WaitForRefRemoved(const ReferenceTable::iterator &reference_it, const rpc::WorkerAddress &addr, const ObjectID &contained_in_id = ObjectID::Nil()) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Helper method to add an object that we are borrowing. This is used when /// deserializing IDs from a task's arguments, or when deserializing an ID @@ -910,23 +914,24 @@ class ReferenceCounter : public ReferenceCounterInterface, const ObjectID &outer_id, const rpc::Address &owner_address, bool foreign_owner_already_monitoring) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Helper method to delete an entry from the reference map and run any necessary /// callbacks. Assumes that the entry is in object_id_refs_ and invalidates the /// iterator. void DeleteReferenceInternal(ReferenceTable::iterator entry, std::vector *deleted) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Erase the Reference from the table. Assumes that the entry has no more /// references, normal or lineage. - void EraseReference(ReferenceTable::iterator entry) EXCLUSIVE_LOCKS_REQUIRED(mutex_); + void EraseReference(ReferenceTable::iterator entry) + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Helper method to garbage-collect all out-of-scope References in the /// lineage for this object. int64_t ReleaseLineageReferences(ReferenceTable::iterator entry) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Add a new location for the given object. The owner must have the object ref in /// scope, and the caller must have already acquired mutex_. @@ -934,7 +939,7 @@ class ReferenceCounter : public ReferenceCounterInterface, /// \param[in] it The reference iterator for the object. /// \param[in] node_id The new object location to be added. void AddObjectLocationInternal(ReferenceTable::iterator it, const NodeID &node_id) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Remove a location for the given object. The owner must have the object ref in /// scope, and the caller must have already acquired mutex_. @@ -942,22 +947,22 @@ class ReferenceCounter : public ReferenceCounterInterface, /// \param[in] it The reference iterator for the object. /// \param[in] node_id The object location to be removed. void RemoveObjectLocationInternal(ReferenceTable::iterator it, const NodeID &node_id) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); void UpdateObjectPendingCreationInternal(const ObjectID &object_id, bool pending_creation) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Publish object locations to all subscribers. /// /// \param[in] it The reference iterator for the object. void PushToLocationSubscribers(ReferenceTable::iterator it) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Fill up the object information for the given iterator. void FillObjectInformationInternal(ReferenceTable::iterator it, rpc::WorkerObjectLocationsPubMessage *object_info) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Clean up borrowers and references when the reference is removed from borrowers. /// It should be used as a WaitForRefRemoved callback. @@ -970,7 +975,7 @@ class ReferenceCounter : public ReferenceCounterInterface, /// calling this method. void RemoveLocalReferenceInternal(const ObjectID &object_id, std::vector *deleted) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Address of our RPC server. This is used to determine whether we own a /// given object or not, by comparing our WorkerID with the WorkerID of the @@ -995,13 +1000,13 @@ class ReferenceCounter : public ReferenceCounterInterface, mutable absl::Mutex mutex_; /// Holds all reference counts and dependency information for tracked ObjectIDs. - ReferenceTable object_id_refs_ GUARDED_BY(mutex_); + ReferenceTable object_id_refs_ ABSL_GUARDED_BY(mutex_); /// Objects whose values have been freed by the language frontend. /// The values in plasma will not be pinned. An object ID is /// removed from this set once its Reference has been deleted /// locally. - absl::flat_hash_set freed_objects_ GUARDED_BY(mutex_); + absl::flat_hash_set freed_objects_ ABSL_GUARDED_BY(mutex_); /// The callback to call once an object ID that we own is no longer in scope /// and it has no tasks that depend on it that may be retried in the future. @@ -1010,7 +1015,7 @@ class ReferenceCounter : public ReferenceCounterInterface, LineageReleasedCallback on_lineage_released_; /// Optional shutdown hook to call when all references have gone /// out of scope. - std::function shutdown_hook_ GUARDED_BY(mutex_) = nullptr; + std::function shutdown_hook_ ABSL_GUARDED_BY(mutex_) = nullptr; /// Object status publisher. It is used to publish the ref removed message for the /// reference counting protocol. It is not guarded by a lock because the class itself is @@ -1025,13 +1030,13 @@ class ReferenceCounter : public ReferenceCounterInterface, /// that may be reconstructed. These objects may have pinned lineage that /// should be evicted on memory pressure. The queue is in FIFO order, based /// on ObjectRef creation time. - std::list reconstructable_owned_objects_ GUARDED_BY(mutex_); + std::list reconstructable_owned_objects_ ABSL_GUARDED_BY(mutex_); /// We keep a FIFO queue of objects in scope so that we can choose lineage to /// evict under memory pressure. This is an index from ObjectID to the /// object's place in the queue. absl::flat_hash_map::iterator> - reconstructable_owned_objects_index_ GUARDED_BY(mutex_); + reconstructable_owned_objects_index_ ABSL_GUARDED_BY(mutex_); /// Called to check whether a raylet is still alive. This is used when adding /// the primary or spilled location of an object. If the node is dead, then @@ -1041,13 +1046,13 @@ class ReferenceCounter : public ReferenceCounterInterface, /// A buffer of the objects whose primary or spilled locations have been lost /// due to node failure. These objects are still in scope and need to be /// recovered. - std::vector objects_to_recover_ GUARDED_BY(mutex_); + std::vector objects_to_recover_ ABSL_GUARDED_BY(mutex_); /// Keep track of objects owend by this worker. - size_t num_objects_owned_by_us_ GUARDED_BY(mutex_) = 0; + size_t num_objects_owned_by_us_ ABSL_GUARDED_BY(mutex_) = 0; /// Keep track of actors owend by this worker. - size_t num_actors_owned_by_us_ GUARDED_BY(mutex_) = 0; + size_t num_actors_owned_by_us_ ABSL_GUARDED_BY(mutex_) = 0; }; } // namespace core diff --git a/src/ray/core_worker/store_provider/memory_store/memory_store.h b/src/ray/core_worker/store_provider/memory_store/memory_store.h index bd72263b52204..9b38bd0f5ef78 100644 --- a/src/ray/core_worker/store_provider/memory_store/memory_store.h +++ b/src/ray/core_worker/store_provider/memory_store/memory_store.h @@ -186,11 +186,12 @@ class CoreWorkerMemoryStore { /// Emplace the given object entry to the in-memory-store and update stats properly. void EmplaceObjectAndUpdateStats(const ObjectID &object_id, std::shared_ptr &object_entry) - EXCLUSIVE_LOCKS_REQUIRED(mu_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// Erase the object of the object id from the in memory store and update stats /// properly. - void EraseObjectAndUpdateStats(const ObjectID &object_id) EXCLUSIVE_LOCKS_REQUIRED(mu_); + void EraseObjectAndUpdateStats(const ObjectID &object_id) + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// If enabled, holds a reference to local worker ref counter. TODO(ekl) make this /// mandatory once Java is supported. @@ -205,16 +206,16 @@ class CoreWorkerMemoryStore { /// Map from object ID to `RayObject`. /// NOTE: This map should be modified by EmplaceObjectAndUpdateStats and /// EraseObjectAndUpdateStats. - absl::flat_hash_map> objects_ GUARDED_BY(mu_); + absl::flat_hash_map> objects_ ABSL_GUARDED_BY(mu_); /// Map from object ID to its get requests. absl::flat_hash_map>> - object_get_requests_ GUARDED_BY(mu_); + object_get_requests_ ABSL_GUARDED_BY(mu_); /// Map from object ID to its async get requests. absl::flat_hash_map)>>> - object_async_get_requests_ GUARDED_BY(mu_); + object_async_get_requests_ ABSL_GUARDED_BY(mu_); /// Function passed in to be called to check for signals (e.g., Ctrl-C). std::function check_signals_; @@ -226,13 +227,13 @@ class CoreWorkerMemoryStore { /// Below information is stats. /// /// Number of objects in the plasma store for this memory store. - int32_t num_in_plasma_ GUARDED_BY(mu_) = 0; + int32_t num_in_plasma_ ABSL_GUARDED_BY(mu_) = 0; /// Number of objects that don't exist in the plasma store. - int32_t num_local_objects_ GUARDED_BY(mu_) = 0; + int32_t num_local_objects_ ABSL_GUARDED_BY(mu_) = 0; /// Number of bytes used by this memory store on heap, including both /// placeholder values for objects in plasma and inlined small returned /// objects from task. - int64_t num_local_objects_bytes_ GUARDED_BY(mu_) = 0; + int64_t num_local_objects_bytes_ ABSL_GUARDED_BY(mu_) = 0; /// This lambda is used to allow language frontend to allocate the objects /// in the memory store. diff --git a/src/ray/core_worker/store_provider/plasma_store_provider.h b/src/ray/core_worker/store_provider/plasma_store_provider.h index 4bcfad197b1f1..2e08309c6cc88 100644 --- a/src/ray/core_worker/store_provider/plasma_store_provider.h +++ b/src/ray/core_worker/store_provider/plasma_store_provider.h @@ -52,7 +52,7 @@ class BufferTracker { // identifies a buffer. It should not be a shared ptr since that would keep the Buffer // alive forever (i.e., this is a weak ref map). absl::flat_hash_map, std::string> active_buffers_ - GUARDED_BY(active_buffers_mutex_); + ABSL_GUARDED_BY(active_buffers_mutex_); }; /// This can be used to hold the reference to a buffer. diff --git a/src/ray/core_worker/task_event_buffer.h b/src/ray/core_worker/task_event_buffer.h index f465ef2488e54..4156a73ca1227 100644 --- a/src/ray/core_worker/task_event_buffer.h +++ b/src/ray/core_worker/task_event_buffer.h @@ -260,13 +260,13 @@ class TaskEventBufferImpl : public TaskEventBuffer { ~TaskEventBufferImpl() override; void AddTaskEvent(std::unique_ptr task_event) - LOCKS_EXCLUDED(mutex_) override; + ABSL_LOCKS_EXCLUDED(mutex_) override; - void FlushEvents(bool forced) LOCKS_EXCLUDED(mutex_) override; + void FlushEvents(bool forced) ABSL_LOCKS_EXCLUDED(mutex_) override; - Status Start(bool auto_flush = true) LOCKS_EXCLUDED(mutex_) override; + Status Start(bool auto_flush = true) ABSL_LOCKS_EXCLUDED(mutex_) override; - void Stop() LOCKS_EXCLUDED(mutex_) override; + void Stop() ABSL_LOCKS_EXCLUDED(mutex_) override; bool Enabled() const override; @@ -322,13 +322,13 @@ class TaskEventBufferImpl : public TaskEventBuffer { PeriodicalRunner periodical_runner_; /// Client to the GCS used to push profile events to it. - std::unique_ptr gcs_client_ GUARDED_BY(mutex_); + std::unique_ptr gcs_client_ ABSL_GUARDED_BY(mutex_); /// True if the TaskEventBuffer is enabled. std::atomic enabled_ = false; /// Circular buffered task events. - boost::circular_buffer> buffer_ GUARDED_BY(mutex_); + boost::circular_buffer> buffer_ ABSL_GUARDED_BY(mutex_); /// Stats counter map. CounterMapThreadSafe stats_counter_; diff --git a/src/ray/core_worker/task_manager.h b/src/ray/core_worker/task_manager.h index 83e1cf9b0b747..932e1a3d5fdbb 100644 --- a/src/ray/core_worker/task_manager.h +++ b/src/ray/core_worker/task_manager.h @@ -188,7 +188,7 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa task_event_buffer_(task_event_buffer) { task_counter_.SetOnChangeCallback( [this](const std::tuple key) - EXCLUSIVE_LOCKS_REQUIRED(&mu_) { + ABSL_EXCLUSIVE_LOCKS_REQUIRED(&mu_) { ray::stats::STATS_tasks.Record( task_counter_.Get(key), {{"State", rpc::TaskStatus_Name(std::get<1>(key))}, @@ -283,7 +283,7 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa /// /// \return True if a task return is registered. False otherwise. bool HandleReportGeneratorItemReturns( - const rpc::ReportGeneratorItemReturnsRequest &request) LOCKS_EXCLUDED(mu_); + const rpc::ReportGeneratorItemReturnsRequest &request) ABSL_LOCKS_EXCLUDED(mu_); /// Temporarily register a given generator return reference. /// @@ -310,7 +310,7 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa /// \return True if we temporarily owned the reference. False otherwise. bool TemporarilyOwnGeneratorReturnRefIfNeeded(const ObjectID &object_id, const ObjectID &generator_id) - LOCKS_EXCLUDED(mu_); + ABSL_LOCKS_EXCLUDED(mu_); /// Delete the object ref stream. /// @@ -325,13 +325,13 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa /// /// \param[in] generator_id The object ref id of the streaming /// generator task. - void DelObjectRefStream(const ObjectID &generator_id) LOCKS_EXCLUDED(mu_); + void DelObjectRefStream(const ObjectID &generator_id) ABSL_LOCKS_EXCLUDED(mu_); /// Return true if the object ref stream exists. /// /// \param[in] generator_id The object ref id of the streaming /// generator task. - bool ObjectRefStreamExists(const ObjectID &generator_id) LOCKS_EXCLUDED(mu_); + bool ObjectRefStreamExists(const ObjectID &generator_id) ABSL_LOCKS_EXCLUDED(mu_); /// Read object reference of the next index from the /// object stream of a generator_id. @@ -349,7 +349,7 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa /// Nil ID is returned if the next index hasn't been written. /// \return ObjectRefEndOfStream if it reaches to EoF. Ok otherwise. Status TryReadObjectRefStream(const ObjectID &generator_id, ObjectID *object_id_out) - LOCKS_EXCLUDED(mu_); + ABSL_LOCKS_EXCLUDED(mu_); /// Read the next index of a ObjectRefStream of generator_id without /// consuming an index. @@ -360,7 +360,7 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa /// generator task. /// \return A object reference of the next index. /// It should not be nil. - ObjectID PeekObjectRefStream(const ObjectID &generator_id) LOCKS_EXCLUDED(mu_); + ObjectID PeekObjectRefStream(const ObjectID &generator_id) ABSL_LOCKS_EXCLUDED(mu_); /// Returns true if task can be retried. /// @@ -416,7 +416,7 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa const TaskSpecification &spec, rpc::ErrorType error_type, const rpc::RayErrorInfo *ray_error_info, - const absl::flat_hash_set &store_in_plasma_ids) LOCKS_EXCLUDED(mu_); + const absl::flat_hash_set &store_in_plasma_ids) ABSL_LOCKS_EXCLUDED(mu_); /// A task's dependencies were inlined in the task spec. This will decrement /// the ref count for the dependency IDs. If the dependencies contained other @@ -631,7 +631,7 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa bool HandleTaskReturn(const ObjectID &object_id, const rpc::ReturnObject &return_object, const NodeID &worker_raylet_id, - bool store_in_plasma) LOCKS_EXCLUDED(mu_); + bool store_in_plasma) ABSL_LOCKS_EXCLUDED(mu_); /// Remove a lineage reference to this object ID. This should be called /// whenever a task that depended on this object ID can no longer be retried. @@ -642,7 +642,7 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa /// \param[out] The amount of lineage in bytes that was removed. int64_t RemoveLineageReference(const ObjectID &object_id, std::vector *ids_to_release) - LOCKS_EXCLUDED(mu_); + ABSL_LOCKS_EXCLUDED(mu_); /// Helper function to call RemoveSubmittedTaskReferences on the remaining /// dependencies of the given task spec after the task has finished or @@ -668,10 +668,11 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa /// \param [out] Return objects that should be stored in plasma. If the /// task has been already terminated, it returns an empty set. absl::flat_hash_set GetTaskReturnObjectsToStoreInPlasma( - const TaskID &task_id, bool *first_execution = nullptr) const LOCKS_EXCLUDED(mu_); + const TaskID &task_id, bool *first_execution = nullptr) const + ABSL_LOCKS_EXCLUDED(mu_); /// Shutdown if all tasks are finished and shutdown is scheduled. - void ShutdownIfNeeded() LOCKS_EXCLUDED(mu_); + void ShutdownIfNeeded() ABSL_LOCKS_EXCLUDED(mu_); /// Set the TaskStatus /// @@ -713,12 +714,12 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa /// this should be used when a task fails (which means we know the task won't /// report any more generator return values). void MarkEndOfStream(const ObjectID &generator_id, int64_t end_of_stream_index) - LOCKS_EXCLUDED(objet_ref_stream_ops_mu_) LOCKS_EXCLUDED(mu_); + ABSL_LOCKS_EXCLUDED(objet_ref_stream_ops_mu_) ABSL_LOCKS_EXCLUDED(mu_); /// See TemporarilyOwnGeneratorReturnRefIfNeeded for a docstring. bool TemporarilyOwnGeneratorReturnRefIfNeededInternal(const ObjectID &object_id, const ObjectID &generator_id) - EXCLUSIVE_LOCKS_REQUIRED(objet_ref_stream_ops_mu_) LOCKS_EXCLUDED(mu_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(objet_ref_stream_ops_mu_) ABSL_LOCKS_EXCLUDED(mu_); /// Used to store task results. std::shared_ptr in_memory_store_; @@ -730,7 +731,7 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa /// Mapping from a streaming generator task id -> object ref stream. absl::flat_hash_map object_ref_streams_ - GUARDED_BY(objet_ref_stream_ops_mu_); + ABSL_GUARDED_BY(objet_ref_stream_ops_mu_); /// Callback to store objects in plasma. This is used for objects that were /// originally stored in plasma. During reconstruction, we ensure that these @@ -747,10 +748,10 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa const int64_t max_lineage_bytes_; // The number of task failures we have logged total. - int64_t num_failure_logs_ GUARDED_BY(mu_) = 0; + int64_t num_failure_logs_ ABSL_GUARDED_BY(mu_) = 0; // The last time we logged a task failure. - int64_t last_log_time_ms_ GUARDED_BY(mu_) = 0; + int64_t last_log_time_ms_ ABSL_GUARDED_BY(mu_) = 0; /// Protects below fields. mutable absl::Mutex mu_; @@ -760,23 +761,23 @@ class TaskManager : public TaskFinisherInterface, public TaskResubmissionInterfa mutable absl::Mutex objet_ref_stream_ops_mu_; /// Tracks per-task-state counters for metric purposes. - TaskStatusCounter task_counter_ GUARDED_BY(mu_); + TaskStatusCounter task_counter_ ABSL_GUARDED_BY(mu_); /// This map contains one entry per task that may be submitted for /// execution. This includes both tasks that are currently pending execution /// and tasks that finished execution but that may be retried again in the /// future. - absl::flat_hash_map submissible_tasks_ GUARDED_BY(mu_); + absl::flat_hash_map submissible_tasks_ ABSL_GUARDED_BY(mu_); /// Number of tasks that are pending. This is a count of all tasks in /// submissible_tasks_ that have been submitted and are currently pending /// execution. size_t num_pending_tasks_ = 0; - int64_t total_lineage_footprint_bytes_ GUARDED_BY(mu_) = 0; + int64_t total_lineage_footprint_bytes_ ABSL_GUARDED_BY(mu_) = 0; /// Optional shutdown hook to call when pending tasks all finish. - std::function shutdown_hook_ GUARDED_BY(mu_) = nullptr; + std::function shutdown_hook_ ABSL_GUARDED_BY(mu_) = nullptr; /// A task state events buffer initialized managed by the CoreWorker. /// task_event_buffer_.Enabled() will return false if disabled (due to config or set-up diff --git a/src/ray/core_worker/transport/actor_scheduling_queue.h b/src/ray/core_worker/transport/actor_scheduling_queue.h index f94266bed4dcb..f73d80fbae554 100644 --- a/src/ray/core_worker/transport/actor_scheduling_queue.h +++ b/src/ray/core_worker/transport/actor_scheduling_queue.h @@ -107,7 +107,7 @@ class ActorSchedulingQueue : public SchedulingQueue { absl::Mutex mu_; /// A map of actor task IDs -> is_canceled /// Pending means tasks are queued or running. - absl::flat_hash_map pending_task_id_to_is_canceled GUARDED_BY(mu_); + absl::flat_hash_map pending_task_id_to_is_canceled ABSL_GUARDED_BY(mu_); friend class SchedulingQueueTest; }; diff --git a/src/ray/core_worker/transport/dependency_resolver.h b/src/ray/core_worker/transport/dependency_resolver.h index 9593247ef11c5..14a651130291c 100644 --- a/src/ray/core_worker/transport/dependency_resolver.h +++ b/src/ray/core_worker/transport/dependency_resolver.h @@ -102,7 +102,8 @@ class LocalDependencyResolver { /// Number of tasks pending dependency resolution. std::atomic num_pending_; - absl::flat_hash_map> pending_tasks_ GUARDED_BY(mu_); + absl::flat_hash_map> pending_tasks_ + ABSL_GUARDED_BY(mu_); /// Protects against concurrent access to internal state. mutable absl::Mutex mu_; diff --git a/src/ray/core_worker/transport/direct_actor_task_submitter.h b/src/ray/core_worker/transport/direct_actor_task_submitter.h index 4da3cc761e21d..602ea66a7c3b0 100644 --- a/src/ray/core_worker/transport/direct_actor_task_submitter.h +++ b/src/ray/core_worker/transport/direct_actor_task_submitter.h @@ -324,33 +324,33 @@ class CoreWorkerDirectActorTaskSubmitter /// \return Void. void PushActorTask(ClientQueue &queue, const TaskSpecification &task_spec, - bool skip_queue) EXCLUSIVE_LOCKS_REQUIRED(mu_); + bool skip_queue) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); void HandlePushTaskReply(const Status &status, const rpc::PushTaskReply &reply, const rpc::Address &addr, - const TaskSpecification &task_spec) LOCKS_EXCLUDED(mu_); + const TaskSpecification &task_spec) ABSL_LOCKS_EXCLUDED(mu_); /// Send all pending tasks for an actor. /// /// \param[in] actor_id Actor ID. /// \return Void. - void SendPendingTasks(const ActorID &actor_id) EXCLUSIVE_LOCKS_REQUIRED(mu_); + void SendPendingTasks(const ActorID &actor_id) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// Resend all previously-received, out-of-order, received tasks for an actor. /// When sending these tasks, the tasks will have the flag skip_execution=true. /// /// \param[in] actor_id Actor ID. /// \return Void. - void ResendOutOfOrderTasks(const ActorID &actor_id) EXCLUSIVE_LOCKS_REQUIRED(mu_); + void ResendOutOfOrderTasks(const ActorID &actor_id) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// Disconnect the RPC client for an actor. - void DisconnectRpcClient(ClientQueue &queue) EXCLUSIVE_LOCKS_REQUIRED(mu_); + void DisconnectRpcClient(ClientQueue &queue) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// Fail all in-flight tasks. void FailInflightTasks( const absl::flat_hash_map> - &inflight_task_callbacks) LOCKS_EXCLUDED(mu_); + &inflight_task_callbacks) ABSL_LOCKS_EXCLUDED(mu_); /// Pool for producing new core worker clients. rpc::CoreWorkerClientPool &core_worker_client_pool_; @@ -358,7 +358,7 @@ class CoreWorkerDirectActorTaskSubmitter /// Mutex to protect the various maps below. mutable absl::Mutex mu_; - absl::flat_hash_map client_queues_ GUARDED_BY(mu_); + absl::flat_hash_map client_queues_ ABSL_GUARDED_BY(mu_); /// Resolve direct call object dependencies. LocalDependencyResolver resolver_; diff --git a/src/ray/core_worker/transport/direct_task_transport.h b/src/ray/core_worker/transport/direct_task_transport.h index a095c737608af..253c2f0b3800e 100644 --- a/src/ray/core_worker/transport/direct_task_transport.h +++ b/src/ray/core_worker/transport/direct_task_transport.h @@ -158,21 +158,21 @@ class CoreWorkerDirectTaskSubmitter { bool was_error, bool worker_exiting, const google::protobuf::RepeatedPtrField &assigned_resources) - EXCLUSIVE_LOCKS_REQUIRED(mu_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// Get an existing lease client or connect a new one. If a raylet_address is /// provided, this connects to a remote raylet. Else, this connects to the /// local raylet. std::shared_ptr GetOrConnectLeaseClient( - const rpc::Address *raylet_address) EXCLUSIVE_LOCKS_REQUIRED(mu_); + const rpc::Address *raylet_address) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// Report worker backlog information to the local raylet - void ReportWorkerBacklogInternal() EXCLUSIVE_LOCKS_REQUIRED(mu_); + void ReportWorkerBacklogInternal() ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// Report backlog if the backlog size is changed for this scheduling key /// since last report void ReportWorkerBacklogIfNeeded(const SchedulingKey &scheduling_key) - EXCLUSIVE_LOCKS_REQUIRED(mu_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// Request a new worker from the raylet if no such requests are currently in /// flight and there are tasks queued. If a raylet address is provided, then @@ -180,14 +180,14 @@ class CoreWorkerDirectTaskSubmitter { /// worker should be requested from the local raylet. void RequestNewWorkerIfNeeded(const SchedulingKey &task_queue_key, const rpc::Address *raylet_address = nullptr) - EXCLUSIVE_LOCKS_REQUIRED(mu_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// Cancel a pending worker lease and retry until the cancellation succeeds /// (i.e., the raylet drops the request). This should be called when there /// are no more tasks queued with the given scheduling key and there is an /// in-flight lease request for that key. void CancelWorkerLeaseIfNeeded(const SchedulingKey &scheduling_key) - EXCLUSIVE_LOCKS_REQUIRED(mu_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// Set up client state for newly granted worker lease. void AddWorkerLeaseClient( @@ -195,7 +195,7 @@ class CoreWorkerDirectTaskSubmitter { std::shared_ptr lease_client, const google::protobuf::RepeatedPtrField &assigned_resources, const SchedulingKey &scheduling_key, - const TaskID &task_id) EXCLUSIVE_LOCKS_REQUIRED(mu_); + const TaskID &task_id) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// This function takes care of returning a worker to the Raylet. /// \param[in] addr The address of the worker. @@ -204,10 +204,11 @@ class CoreWorkerDirectTaskSubmitter { void ReturnWorker(const rpc::WorkerAddress addr, bool was_error, bool worker_exiting, - const SchedulingKey &scheduling_key) EXCLUSIVE_LOCKS_REQUIRED(mu_); + const SchedulingKey &scheduling_key) + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); /// Check that the scheduling_key_entries_ hashmap is empty. - inline bool CheckNoSchedulingKeyEntries() const EXCLUSIVE_LOCKS_REQUIRED(mu_) { + inline bool CheckNoSchedulingKeyEntries() const ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_) { return scheduling_key_entries_.empty(); } @@ -236,7 +237,7 @@ class CoreWorkerDirectTaskSubmitter { /// Cache of gRPC clients to remote raylets. absl::flat_hash_map> remote_lease_clients_ - GUARDED_BY(mu_); + ABSL_GUARDED_BY(mu_); /// Factory for producing new clients to request leases from remote nodes. LeaseClientFactoryFn lease_client_factory_; @@ -306,7 +307,7 @@ class CoreWorkerDirectTaskSubmitter { // Map from worker address to a LeaseEntry struct containing the lease's metadata. absl::flat_hash_map worker_to_lease_entry_ - GUARDED_BY(mu_); + ABSL_GUARDED_BY(mu_); struct SchedulingKeyEntry { // Keep track of pending worker lease requests to the raylet. @@ -356,13 +357,13 @@ class CoreWorkerDirectTaskSubmitter { // with the queue of tasks belonging to that SchedulingKey, together with the other // fields that are needed to orchestrate the execution of those tasks by the workers. absl::flat_hash_map scheduling_key_entries_ - GUARDED_BY(mu_); + ABSL_GUARDED_BY(mu_); // Tasks that were cancelled while being resolved. - absl::flat_hash_set cancelled_tasks_ GUARDED_BY(mu_); + absl::flat_hash_set cancelled_tasks_ ABSL_GUARDED_BY(mu_); // Keeps track of where currently executing tasks are being run. - absl::flat_hash_map executing_tasks_ GUARDED_BY(mu_); + absl::flat_hash_map executing_tasks_ ABSL_GUARDED_BY(mu_); // Ratelimiter controls the num of pending lease requests. std::shared_ptr lease_request_rate_limiter_; @@ -371,7 +372,7 @@ class CoreWorkerDirectTaskSubmitter { absl::optional cancel_retry_timer_; int64_t num_tasks_submitted_ = 0; - int64_t num_leases_requested_ GUARDED_BY(mu_) = 0; + int64_t num_leases_requested_ ABSL_GUARDED_BY(mu_) = 0; }; } // namespace core diff --git a/src/ray/core_worker/transport/normal_scheduling_queue.h b/src/ray/core_worker/transport/normal_scheduling_queue.h index a2c451fcc0cab..7d5c7d2464b01 100644 --- a/src/ray/core_worker/transport/normal_scheduling_queue.h +++ b/src/ray/core_worker/transport/normal_scheduling_queue.h @@ -60,7 +60,7 @@ class NormalSchedulingQueue : public SchedulingQueue { /// Protects access to the dequeue below. mutable absl::Mutex mu_; /// Queue with (accept, rej) callbacks for non-actor tasks - std::deque pending_normal_tasks_ GUARDED_BY(mu_); + std::deque pending_normal_tasks_ ABSL_GUARDED_BY(mu_); friend class SchedulingQueueTest; }; diff --git a/src/ray/core_worker/transport/out_of_order_actor_scheduling_queue.h b/src/ray/core_worker/transport/out_of_order_actor_scheduling_queue.h index 667a3b93093a2..acbf17e2af905 100644 --- a/src/ray/core_worker/transport/out_of_order_actor_scheduling_queue.h +++ b/src/ray/core_worker/transport/out_of_order_actor_scheduling_queue.h @@ -94,7 +94,7 @@ class OutOfOrderActorSchedulingQueue : public SchedulingQueue { absl::Mutex mu_; /// A map of actor task IDs -> is_canceled. // Pending means tasks are queued or running. - absl::flat_hash_map pending_task_id_to_is_canceled GUARDED_BY(mu_); + absl::flat_hash_map pending_task_id_to_is_canceled ABSL_GUARDED_BY(mu_); friend class SchedulingQueueTest; }; diff --git a/src/ray/gcs/gcs_client/accessor.h b/src/ray/gcs/gcs_client/accessor.h index acbac311e1232..b9942a558ef1f 100644 --- a/src/ray/gcs/gcs_client/accessor.h +++ b/src/ray/gcs/gcs_client/accessor.h @@ -188,11 +188,11 @@ class ActorInfoAccessor { /// Resubscribe operations for actors. absl::flat_hash_map resubscribe_operations_ - GUARDED_BY(mutex_); + ABSL_GUARDED_BY(mutex_); /// Save the fetch data operation of actors. absl::flat_hash_map fetch_data_operations_ - GUARDED_BY(mutex_); + ABSL_GUARDED_BY(mutex_); GcsClient *client_impl_; }; @@ -488,7 +488,7 @@ class NodeResourceInfoAccessor { /// Save the resource usage data, so we can resend it again when GCS server restarts /// from a failure. - rpc::ReportResourceUsageRequest cached_resource_usage_ GUARDED_BY(mutex_); + rpc::ReportResourceUsageRequest cached_resource_usage_ ABSL_GUARDED_BY(mutex_); /// Save the subscribe operation in this function, so we can call it again when PubSub /// server restarts from a failure. diff --git a/src/ray/gcs/gcs_client/global_state_accessor.h b/src/ray/gcs/gcs_client/global_state_accessor.h index 384690e7cdd12..1ca2e6c850bf2 100644 --- a/src/ray/gcs/gcs_client/global_state_accessor.h +++ b/src/ray/gcs/gcs_client/global_state_accessor.h @@ -34,37 +34,37 @@ class GlobalStateAccessor { /// \param gcs_client_options The client options to connect to gcs explicit GlobalStateAccessor(const GcsClientOptions &gcs_client_options); - ~GlobalStateAccessor() LOCKS_EXCLUDED(mutex_); + ~GlobalStateAccessor() ABSL_LOCKS_EXCLUDED(mutex_); /// Connect gcs server. /// /// \return Whether the connection is successful. - bool Connect() LOCKS_EXCLUDED(mutex_); + bool Connect() ABSL_LOCKS_EXCLUDED(mutex_); /// Disconnect from gcs server. - void Disconnect() LOCKS_EXCLUDED(mutex_); + void Disconnect() ABSL_LOCKS_EXCLUDED(mutex_); /// Get information of all jobs from GCS Service. /// /// \return All job info. To support multi-language, we serialize each JobTableData and /// return the serialized string. Where used, it needs to be deserialized with /// protobuf function. - std::vector GetAllJobInfo() LOCKS_EXCLUDED(mutex_); + std::vector GetAllJobInfo() ABSL_LOCKS_EXCLUDED(mutex_); /// Get next job id from GCS Service. /// /// \return Next job id. - JobID GetNextJobID() LOCKS_EXCLUDED(mutex_); + JobID GetNextJobID() ABSL_LOCKS_EXCLUDED(mutex_); /// Get all node information from GCS. /// /// \return A list of `GcsNodeInfo` objects serialized in protobuf format. - std::vector GetAllNodeInfo() LOCKS_EXCLUDED(mutex_); + std::vector GetAllNodeInfo() ABSL_LOCKS_EXCLUDED(mutex_); /// Get information of all task events from GCS Service. /// /// \return All task events info. - std::vector GetAllTaskEvents() LOCKS_EXCLUDED(mutex_); + std::vector GetAllTaskEvents() ABSL_LOCKS_EXCLUDED(mutex_); /// Get information of a node resource from GCS Service. /// @@ -72,19 +72,19 @@ class GlobalStateAccessor { /// \return node resource map info. To support multi-language, we serialize each /// ResourceTableData and return the serialized string. Where used, it needs to be /// deserialized with protobuf function. - std::string GetNodeResourceInfo(const NodeID &node_id) LOCKS_EXCLUDED(mutex_); + std::string GetNodeResourceInfo(const NodeID &node_id) ABSL_LOCKS_EXCLUDED(mutex_); /// Get available resources of all nodes. /// /// \return available resources of all nodes. To support multi-language, we serialize /// each AvailableResources and return the serialized string. Where used, it needs to be /// deserialized with protobuf function. - std::vector GetAllAvailableResources() LOCKS_EXCLUDED(mutex_); + std::vector GetAllAvailableResources() ABSL_LOCKS_EXCLUDED(mutex_); /// Get ids of draining nodes. /// /// \return ids of draining nodes. - std::vector GetDrainingNodes() LOCKS_EXCLUDED(mutex_); + std::vector GetDrainingNodes() ABSL_LOCKS_EXCLUDED(mutex_); /// Get newest resource usage of all nodes from GCS Service. Only used when light /// rerouce usage report enabled. @@ -92,14 +92,14 @@ class GlobalStateAccessor { /// \return resource usage info. To support multi-language, we serialize each /// data and return the serialized string. Where used, it needs to be /// deserialized with protobuf function. - std::unique_ptr GetAllResourceUsage() LOCKS_EXCLUDED(mutex_); + std::unique_ptr GetAllResourceUsage() ABSL_LOCKS_EXCLUDED(mutex_); /// Get information of all actors from GCS Service. /// /// \return All actor info. To support multi-language, we serialize each ActorTableData /// and return the serialized string. Where used, it needs to be deserialized with /// protobuf function. - std::vector GetAllActorInfo() LOCKS_EXCLUDED(mutex_); + std::vector GetAllActorInfo() ABSL_LOCKS_EXCLUDED(mutex_); /// Get information of an actor from GCS Service. /// @@ -108,7 +108,7 @@ class GlobalStateAccessor { /// return the serialized string. Where used, it needs to be deserialized with /// protobuf function. std::unique_ptr GetActorInfo(const ActorID &actor_id) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Get information of a worker from GCS Service. /// @@ -117,28 +117,28 @@ class GlobalStateAccessor { /// and return the serialized string. Where used, it needs to be deserialized with /// protobuf function. std::unique_ptr GetWorkerInfo(const WorkerID &worker_id) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Get information of all workers from GCS Service. /// /// \return All worker info. To support multi-language, we serialize each /// WorkerTableData and return the serialized string. Where used, it needs to be /// deserialized with protobuf function. - std::vector GetAllWorkerInfo() LOCKS_EXCLUDED(mutex_); + std::vector GetAllWorkerInfo() ABSL_LOCKS_EXCLUDED(mutex_); /// Add information of a worker to GCS Service. /// /// \param serialized_string The serialized data of worker to be added in the GCS /// Service, use string is convenient for python to use. /// \return Is operation success. - bool AddWorkerInfo(const std::string &serialized_string) LOCKS_EXCLUDED(mutex_); + bool AddWorkerInfo(const std::string &serialized_string) ABSL_LOCKS_EXCLUDED(mutex_); /// Get information of all placement group from GCS Service. /// /// \return All placement group info. To support multi-language, we serialize each /// PlacementGroupTableData and return the serialized string. Where used, it needs to be /// deserialized with protobuf function. - std::vector GetAllPlacementGroupInfo() LOCKS_EXCLUDED(mutex_); + std::vector GetAllPlacementGroupInfo() ABSL_LOCKS_EXCLUDED(mutex_); /// Get information of a placement group from GCS Service by ID. /// @@ -147,7 +147,7 @@ class GlobalStateAccessor { /// PlacementGroupTableData and return the serialized string. Where used, it needs to be /// deserialized with protobuf function. std::unique_ptr GetPlacementGroupInfo( - const PlacementGroupID &placement_group_id) LOCKS_EXCLUDED(mutex_); + const PlacementGroupID &placement_group_id) ABSL_LOCKS_EXCLUDED(mutex_); /// Get information of a placement group from GCS Service by name. /// @@ -158,7 +158,7 @@ class GlobalStateAccessor { /// deserialized with protobuf function. std::unique_ptr GetPlacementGroupByName( const std::string &placement_group_name, const std::string &ray_namespace) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Get value of the key from GCS Service. /// @@ -167,12 +167,12 @@ class GlobalStateAccessor { /// \return Value of the key. std::unique_ptr GetInternalKV(const std::string &ns, const std::string &key) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); /// Get the serialized system config from GCS. /// /// \return The serialized system config. - std::string GetSystemConfig() LOCKS_EXCLUDED(mutex_); + std::string GetSystemConfig() ABSL_LOCKS_EXCLUDED(mutex_); /// Get the node to connect for a Ray driver. /// @@ -182,7 +182,7 @@ class GlobalStateAccessor { /// Where used, it needs to be deserialized with protobuf function. ray::Status GetNodeToConnectForDriver(const std::string &node_ip_address, std::string *node_to_connect) - LOCKS_EXCLUDED(mutex_); + ABSL_LOCKS_EXCLUDED(mutex_); private: /// MultiItem transformation helper in template style. @@ -235,11 +235,11 @@ class GlobalStateAccessor { mutable absl::Mutex mutex_; /// Whether this client is connected to gcs server. - bool is_connected_ GUARDED_BY(mutex_) = false; + bool is_connected_ ABSL_GUARDED_BY(mutex_) = false; std::unique_ptr thread_io_service_; std::unique_ptr io_service_; - std::unique_ptr gcs_client_ GUARDED_BY(mutex_); + std::unique_ptr gcs_client_ ABSL_GUARDED_BY(mutex_); }; } // namespace gcs diff --git a/src/ray/gcs/gcs_server/gcs_table_storage.h b/src/ray/gcs/gcs_server/gcs_table_storage.h index caabd5b293b76..03ef5b5f86b30 100644 --- a/src/ray/gcs/gcs_server/gcs_table_storage.h +++ b/src/ray/gcs/gcs_server/gcs_table_storage.h @@ -155,7 +155,7 @@ class GcsTableWithJobId : public GcsTable { virtual JobID GetJobIdFromKey(const Key &key) = 0; absl::Mutex mutex_; - absl::flat_hash_map> index_ GUARDED_BY(mutex_); + absl::flat_hash_map> index_ ABSL_GUARDED_BY(mutex_); }; class GcsJobTable : public GcsTable { diff --git a/src/ray/gcs/gcs_server/gcs_task_manager.h b/src/ray/gcs/gcs_server/gcs_task_manager.h index c1d1b204a2d13..7cd984e17048f 100644 --- a/src/ray/gcs/gcs_server/gcs_task_manager.h +++ b/src/ray/gcs/gcs_server/gcs_task_manager.h @@ -122,10 +122,11 @@ class GcsTaskManager : public rpc::TaskInfoHandler { std::string DebugString(); /// Record metrics. - void RecordMetrics() LOCKS_EXCLUDED(mutex_); + void RecordMetrics() ABSL_LOCKS_EXCLUDED(mutex_); /// Set telemetry client. - void SetUsageStatsClient(UsageStatsClient *usage_stats_client) LOCKS_EXCLUDED(mutex_); + void SetUsageStatsClient(UsageStatsClient *usage_stats_client) + ABSL_LOCKS_EXCLUDED(mutex_); /// A storage component that stores the task events. /// @@ -298,7 +299,7 @@ class GcsTaskManager : public rpc::TaskInfoHandler { // Mutex guarding the usage stats client absl::Mutex mutex_; - UsageStatsClient *usage_stats_client_ GUARDED_BY(mutex_) = nullptr; + UsageStatsClient *usage_stats_client_ ABSL_GUARDED_BY(mutex_) = nullptr; /// Counter map for GcsTaskManager stats. CounterMapThreadSafe stats_counter_; diff --git a/src/ray/gcs/gcs_server/gcs_worker_manager.cc b/src/ray/gcs/gcs_server/gcs_worker_manager.cc index 6e1a5846c2b0d..1e4ab08154e5f 100644 --- a/src/ray/gcs/gcs_server/gcs_worker_manager.cc +++ b/src/ray/gcs/gcs_server/gcs_worker_manager.cc @@ -114,8 +114,9 @@ void GcsWorkerManager::HandleReportWorkerFailure( worker_crash_system_error_count_ += 1; key = usage::TagKey::WORKER_CRASH_SYSTEM_ERROR; count = worker_crash_system_error_count_; - } else if (request.worker_failure().exit_type() == - rpc::WorkerExitType::NODE_OUT_OF_MEMORY) { + } else { + RAY_CHECK_EQ(request.worker_failure().exit_type(), + rpc::WorkerExitType::NODE_OUT_OF_MEMORY); worker_crash_oom_count_ += 1; key = usage::TagKey::WORKER_CRASH_OOM; count = worker_crash_oom_count_; diff --git a/src/ray/gcs/gcs_server/grpc_based_resource_broadcaster.h b/src/ray/gcs/gcs_server/grpc_based_resource_broadcaster.h index 4b52c0fb69ec9..83e9ef5a2b188 100644 --- a/src/ray/gcs/gcs_server/grpc_based_resource_broadcaster.h +++ b/src/ray/gcs/gcs_server/grpc_based_resource_broadcaster.h @@ -46,10 +46,10 @@ class GrpcBasedResourceBroadcaster { void Initialize(const GcsInitData &gcs_init_data); /// Event handler when a new node joins the cluster. - void HandleNodeAdded(const rpc::GcsNodeInfo &node_info) LOCKS_EXCLUDED(mutex_); + void HandleNodeAdded(const rpc::GcsNodeInfo &node_info) ABSL_LOCKS_EXCLUDED(mutex_); /// Event handler when a node leaves the cluster. - void HandleNodeRemoved(const rpc::GcsNodeInfo &node_info) LOCKS_EXCLUDED(mutex_); + void HandleNodeRemoved(const rpc::GcsNodeInfo &node_info) ABSL_LOCKS_EXCLUDED(mutex_); std::string DebugString(); void SendBroadcast(rpc::ResourceUsageBroadcastData batch); @@ -70,7 +70,7 @@ class GrpcBasedResourceBroadcaster { /// A lock to protect the data structures. absl::Mutex mutex_; /// The set of nodes and their addresses which are subscribed to resource usage changes. - absl::flat_hash_map nodes_ GUARDED_BY(mutex_); + absl::flat_hash_map nodes_ ABSL_GUARDED_BY(mutex_); friend class GrpcBasedResourceBroadcasterTest; }; diff --git a/src/ray/gcs/gcs_server/test/gcs_placement_group_scheduler_test.cc b/src/ray/gcs/gcs_server/test/gcs_placement_group_scheduler_test.cc index ebd8d756eeebd..8185e35160fe0 100644 --- a/src/ray/gcs/gcs_server/test/gcs_placement_group_scheduler_test.cc +++ b/src/ray/gcs/gcs_server/test/gcs_placement_group_scheduler_test.cc @@ -292,9 +292,9 @@ class GcsPlacementGroupSchedulerTest : public ::testing::Test { std::shared_ptr gcs_node_manager_; std::shared_ptr scheduler_; std::vector> success_placement_groups_ - GUARDED_BY(placement_group_requests_mutex_); + ABSL_GUARDED_BY(placement_group_requests_mutex_); std::vector> failure_placement_groups_ - GUARDED_BY(placement_group_requests_mutex_); + ABSL_GUARDED_BY(placement_group_requests_mutex_); std::shared_ptr gcs_publisher_; std::shared_ptr gcs_table_storage_; std::shared_ptr raylet_client_pool_; diff --git a/src/ray/gcs/pubsub/gcs_pub_sub.h b/src/ray/gcs/pubsub/gcs_pub_sub.h index 12479b7bec6bb..4b9cadda0123b 100644 --- a/src/ray/gcs/pubsub/gcs_pub_sub.h +++ b/src/ray/gcs/pubsub/gcs_pub_sub.h @@ -206,11 +206,11 @@ class RAY_EXPORT PythonGcsSubscriber { const std::string subscriber_id_; std::string publisher_id_; const std::string worker_id_; - int64_t max_processed_sequence_id_ GUARDED_BY(mu_); - int64_t last_batch_size_ GUARDED_BY(mu_); - std::deque queue_ GUARDED_BY(mu_); - bool closed_ GUARDED_BY(mu_); - std::shared_ptr current_polling_context_ GUARDED_BY(mu_); + int64_t max_processed_sequence_id_ ABSL_GUARDED_BY(mu_); + int64_t last_batch_size_ ABSL_GUARDED_BY(mu_); + std::deque queue_ ABSL_GUARDED_BY(mu_); + bool closed_ ABSL_GUARDED_BY(mu_); + std::shared_ptr current_polling_context_ ABSL_GUARDED_BY(mu_); }; /// Get the .lines() attribute of a LogBatch as a std::vector diff --git a/src/ray/gcs/store_client/in_memory_store_client.h b/src/ray/gcs/store_client/in_memory_store_client.h index 730e876cee85d..a4ea7bc47ac63 100644 --- a/src/ray/gcs/store_client/in_memory_store_client.h +++ b/src/ray/gcs/store_client/in_memory_store_client.h @@ -73,7 +73,7 @@ class InMemoryStoreClient : public StoreClient { /// Mutex to protect the records_ field and the index_keys_ field. absl::Mutex mutex_; // Mapping from key to data. - absl::flat_hash_map records_ GUARDED_BY(mutex_); + absl::flat_hash_map records_ ABSL_GUARDED_BY(mutex_); }; std::shared_ptr GetOrCreateTable( @@ -82,7 +82,7 @@ class InMemoryStoreClient : public StoreClient { /// Mutex to protect the tables_ field. absl::Mutex mutex_; absl::flat_hash_map> tables_ - GUARDED_BY(mutex_); + ABSL_GUARDED_BY(mutex_); /// Async API Callback needs to post to main_io_service_ to ensure the orderly execution /// of the callback. diff --git a/src/ray/gcs/store_client/redis_store_client.h b/src/ray/gcs/store_client/redis_store_client.h index 76ef844e1a894..7307a8b1a9c2b 100644 --- a/src/ray/gcs/store_client/redis_store_client.h +++ b/src/ray/gcs/store_client/redis_store_client.h @@ -122,7 +122,7 @@ class RedisStoreClient : public StoreClient { // only when there is no in-flight request for the key. size_t PushToSendingQueue(const std::vector &keys, std::function send_request) - EXCLUSIVE_LOCKS_REQUIRED(mu_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); // Take requests from the sending queue and erase the queue if it's // empty. @@ -131,7 +131,7 @@ class RedisStoreClient : public StoreClient { // // \return The requests to send. std::vector> TakeRequestsFromSendingQueue( - const std::vector &keys) EXCLUSIVE_LOCKS_REQUIRED(mu_); + const std::vector &keys) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_); Status DoPut(const std::string &key, const std::string &data, @@ -163,7 +163,7 @@ class RedisStoreClient : public StoreClient { // The pending redis requests queue for each key. // The queue will be poped when the request is processed. absl::flat_hash_map>> - pending_redis_request_by_key_ GUARDED_BY(mu_); + pending_redis_request_by_key_ ABSL_GUARDED_BY(mu_); FRIEND_TEST(RedisStoreClientTest, Random); }; diff --git a/src/ray/object_manager/object_buffer_pool.h b/src/ray/object_manager/object_buffer_pool.h index 321a9ba1b0e7d..b87cb0c705c15 100644 --- a/src/ray/object_manager/object_buffer_pool.h +++ b/src/ray/object_manager/object_buffer_pool.h @@ -89,7 +89,8 @@ class ObjectBufferPool { /// this method. An IOError status is returned if the Get call on the plasma store /// fails, and the MemoryObjectReader will be empty. std::pair, ray::Status> CreateObjectReader( - const ObjectID &object_id, rpc::Address owner_address) LOCKS_EXCLUDED(pool_mutex_); + const ObjectID &object_id, rpc::Address owner_address) + ABSL_LOCKS_EXCLUDED(pool_mutex_); /// Returns a chunk of an empty object at the given chunk_index. The object chunk /// serves as the buffer that is to be written to by a connection receiving an @@ -112,7 +113,7 @@ class ObjectBufferPool { const rpc::Address &owner_address, uint64_t data_size, uint64_t metadata_size, - uint64_t chunk_index) LOCKS_EXCLUDED(pool_mutex_); + uint64_t chunk_index) ABSL_LOCKS_EXCLUDED(pool_mutex_); /// Write to a Chunk of an object. If all chunks of an object is written, /// it seals the object. @@ -128,22 +129,23 @@ class ObjectBufferPool { uint64_t data_size, uint64_t metadata_size, uint64_t chunk_index, - const std::string &data) LOCKS_EXCLUDED(pool_mutex_); + const std::string &data) ABSL_LOCKS_EXCLUDED(pool_mutex_); /// Free a list of objects from object store. /// /// \param object_ids the The list of ObjectIDs to be deleted. /// \return Void. - void FreeObjects(const std::vector &object_ids) LOCKS_EXCLUDED(pool_mutex_); + void FreeObjects(const std::vector &object_ids) + ABSL_LOCKS_EXCLUDED(pool_mutex_); /// Abort the create operation associated with an object. This destroys the buffer /// state, including create operations in progress for all chunks of the object. - void AbortCreate(const ObjectID &object_id) LOCKS_EXCLUDED(pool_mutex_); + void AbortCreate(const ObjectID &object_id) ABSL_LOCKS_EXCLUDED(pool_mutex_); /// Returns debug string for class. /// /// \return string. - std::string DebugString() const LOCKS_EXCLUDED(pool_mutex_); + std::string DebugString() const ABSL_LOCKS_EXCLUDED(pool_mutex_); private: /// Splits an object into ceil(data_size/chunk_size) chunks, which will @@ -152,7 +154,7 @@ class ObjectBufferPool { uint8_t *data, uint64_t data_size, std::shared_ptr buffer_ref) - EXCLUSIVE_LOCKS_REQUIRED(pool_mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(pool_mutex_); /// Ensures buffer for the object exists, and creates the buffer if needed. /// Returns OK if buffer exists. @@ -163,10 +165,10 @@ class ObjectBufferPool { uint64_t data_size, uint64_t metadata_size, uint64_t chunk_index) - EXCLUSIVE_LOCKS_REQUIRED(pool_mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(pool_mutex_); void AbortCreateInternal(const ObjectID &object_id) - EXCLUSIVE_LOCKS_REQUIRED(pool_mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(pool_mutex_); /// The state of a chunk associated with a create operation. enum class CreateChunkState : unsigned int { AVAILABLE = 0, REFERENCED, SEALED }; @@ -208,10 +210,10 @@ class ObjectBufferPool { /// to complete. If successful, the corresponding entry in create_buffer_state_ /// will be created. absl::flat_hash_map> create_buffer_ops_ - GUARDED_BY(pool_mutex_); + ABSL_GUARDED_BY(pool_mutex_); /// The state of a buffer that's currently being used. absl::flat_hash_map create_buffer_state_ - GUARDED_BY(pool_mutex_); + ABSL_GUARDED_BY(pool_mutex_); /// Plasma client pool. std::shared_ptr store_client_; diff --git a/src/ray/object_manager/plasma/store.h b/src/ray/object_manager/plasma/store.h index 61f677faa9d5f..122e04b784e02 100644 --- a/src/ray/object_manager/plasma/store.h +++ b/src/ray/object_manager/plasma/store.h @@ -78,14 +78,14 @@ class PlasmaStore { /// NOTE: Avoid using this method outside object spilling context (e.g., unless you /// absolutely know what's going on). This method won't work correctly if it is used /// before the object is pinned by raylet for the first time. - bool IsObjectSpillable(const ObjectID &object_id) LOCKS_EXCLUDED(mutex_); + bool IsObjectSpillable(const ObjectID &object_id) ABSL_LOCKS_EXCLUDED(mutex_); /// Return the plasma object bytes that are consumed by core workers. int64_t GetConsumedBytes(); /// Get the available memory for new objects to be created. This includes /// memory that is currently being used for created but unsealed objects. - size_t GetAvailableMemory() const LOCKS_EXCLUDED(mutex_) { + size_t GetAvailableMemory() const ABSL_LOCKS_EXCLUDED(mutex_) { absl::MutexLock lock(&mutex_); RAY_CHECK((object_lifecycle_mgr_.GetNumBytesUnsealed() > 0 && object_lifecycle_mgr_.GetNumObjectsUnsealed() > 0) || @@ -126,7 +126,7 @@ class PlasmaStore { plasma::flatbuf::ObjectSource source, const std::shared_ptr &client, bool fallback_allocator, - PlasmaObject *result) EXCLUSIVE_LOCKS_REQUIRED(mutex_); + PlasmaObject *result) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Abort a created but unsealed object. If the client is not the /// creator, then the abort will fail. @@ -136,7 +136,7 @@ class PlasmaStore { /// match the creator of the object, then the abort will fail. /// \return 1 if the abort succeeds, else 0. int AbortObject(const ObjectID &object_id, const std::shared_ptr &client) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Delete a specific object by object_id that have been created in the hash table. /// @@ -145,7 +145,7 @@ class PlasmaStore { /// - PlasmaError::OK, if the object was delete successfully. /// - PlasmaError::ObjectNonexistent, if ths object isn't existed. /// - PlasmaError::ObjectInUse, if the object is in use. - PlasmaError DeleteObject(ObjectID &object_id) EXCLUSIVE_LOCKS_REQUIRED(mutex_); + PlasmaError DeleteObject(ObjectID &object_id) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Process a get request from a client. This method assumes that we will /// eventually have these objects sealed. If one of the objects has not yet @@ -161,72 +161,72 @@ class PlasmaStore { void ProcessGetRequest(const std::shared_ptr &client, const std::vector &object_ids, int64_t timeout_ms, - bool is_from_worker) EXCLUSIVE_LOCKS_REQUIRED(mutex_); + bool is_from_worker) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Process queued requests to create an object. - void ProcessCreateRequests() EXCLUSIVE_LOCKS_REQUIRED(mutex_); + void ProcessCreateRequests() ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Seal a vector of objects. The objects are now immutable and can be accessed with /// get. /// /// \param object_ids The vector of Object IDs of the objects to be sealed. void SealObjects(const std::vector &object_ids) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Record the fact that a particular client is no longer using an object. /// /// \param object_id The object ID of the object that is being released. /// \param client The client making this request. void ReleaseObject(const ObjectID &object_id, const std::shared_ptr &client) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Connect a new client to the PlasmaStore. /// /// \param error The error code from the acceptor. void ConnectClient(const boost::system::error_code &error) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Disconnect a client from the PlasmaStore. /// /// \param client The client that is disconnected. void DisconnectClient(const std::shared_ptr &client) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); Status ProcessMessage(const std::shared_ptr &client, plasma::flatbuf::MessageType type, - const std::vector &message) LOCKS_EXCLUDED(mutex_); + const std::vector &message) ABSL_LOCKS_EXCLUDED(mutex_); PlasmaError HandleCreateObjectRequest(const std::shared_ptr &client, const std::vector &message, bool fallback_allocator, PlasmaObject *object) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); void ReplyToCreateClient(const std::shared_ptr &client, const ObjectID &object_id, - uint64_t req_id) EXCLUSIVE_LOCKS_REQUIRED(mutex_); + uint64_t req_id) ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); void AddToClientObjectIds(const ObjectID &object_id, const std::shared_ptr &client) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); void ReturnFromGet(const std::shared_ptr &get_request); int RemoveFromClientObjectIds(const ObjectID &object_id, const std::shared_ptr &client) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); // Start listening for clients. void DoAccept(); - void PrintAndRecordDebugDump() const LOCKS_EXCLUDED(mutex_); + void PrintAndRecordDebugDump() const ABSL_LOCKS_EXCLUDED(mutex_); - std::string GetDebugDump() const EXCLUSIVE_LOCKS_REQUIRED(mutex_); + std::string GetDebugDump() const ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); private: friend class GetRequestQueue; - void ScheduleRecordMetrics() const LOCKS_EXCLUDED(mutex_); + void ScheduleRecordMetrics() const ABSL_LOCKS_EXCLUDED(mutex_); // A reference to the asio io context. instrumented_io_context &io_context_; @@ -246,7 +246,7 @@ class PlasmaStore { mutable absl::Mutex mutex_; /// The allocator that allocates mmaped memory. - IAllocator &allocator_ GUARDED_BY(mutex_); + IAllocator &allocator_ ABSL_GUARDED_BY(mutex_); /// Monitor the disk utilization. ray::FileSystemMonitor &fs_monitor_; @@ -261,7 +261,7 @@ class PlasmaStore { /// shared with the main raylet thread. const ray::DeleteObjectCallback delete_object_callback_; - ObjectLifecycleManager object_lifecycle_mgr_ GUARDED_BY(mutex_); + ObjectLifecycleManager object_lifecycle_mgr_ ABSL_GUARDED_BY(mutex_); /// The amount of time to wait before retrying a creation request after an /// OOM error. @@ -270,25 +270,27 @@ class PlasmaStore { /// A timer that is set when the first request in the queue is not /// serviceable because there is not enough memory. The request will be /// retried when this timer expires. - std::shared_ptr create_timer_ GUARDED_BY(mutex_); + std::shared_ptr create_timer_ ABSL_GUARDED_BY(mutex_); /// Timer for printing debug information. - mutable std::shared_ptr stats_timer_ GUARDED_BY(mutex_); + mutable std::shared_ptr stats_timer_ + ABSL_GUARDED_BY(mutex_); /// Timer for recording object store metrics. - mutable std::shared_ptr metric_timer_ GUARDED_BY(mutex_); + mutable std::shared_ptr metric_timer_ + ABSL_GUARDED_BY(mutex_); /// Queue of object creation requests. - CreateRequestQueue create_request_queue_ GUARDED_BY(mutex_); + CreateRequestQueue create_request_queue_ ABSL_GUARDED_BY(mutex_); /// Total plasma object bytes that are consumed by core workers. std::atomic total_consumed_bytes_; /// Whether we have dumped debug information on OOM yet. This limits dump /// (which can be expensive) to once per OOM event. - bool dumped_on_oom_ GUARDED_BY(mutex_) = false; + bool dumped_on_oom_ ABSL_GUARDED_BY(mutex_) = false; - GetRequestQueue get_request_queue_ GUARDED_BY(mutex_); + GetRequestQueue get_request_queue_ ABSL_GUARDED_BY(mutex_); }; } // namespace plasma diff --git a/src/ray/object_manager/pull_manager.h b/src/ray/object_manager/pull_manager.h index dd9213422d234..101b490bbf64f 100644 --- a/src/ray/object_manager/pull_manager.h +++ b/src/ray/object_manager/pull_manager.h @@ -364,7 +364,7 @@ class PullManager { /// request or if it is already local. This also sets a timeout for when to /// make the next attempt to make the object local. void TryToMakeObjectLocal(const ObjectID &object_id) - EXCLUSIVE_LOCKS_REQUIRED(active_objects_mu_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(active_objects_mu_); /// Returns whether the set of active pull requests exceeds the memory allowance /// for pulls. Note that exceeding the quota is allowed in certain situations, @@ -494,7 +494,7 @@ class PullManager { /// is the number of bytes that we are currently pulling, and it must be less /// than the bytes available. absl::flat_hash_map> - active_object_pull_requests_ GUARDED_BY(active_objects_mu_); + active_object_pull_requests_ ABSL_GUARDED_BY(active_objects_mu_); /// Tracks the objects we have pinned. Keys are subset of active_object_pull_requests_. /// We need to pin these objects so that parts of in-progress bundles aren't evicted diff --git a/src/ray/protobuf/BUILD b/src/ray/protobuf/BUILD index a0ec9a67bf4ce..263113f1b4705 100644 --- a/src/ray/protobuf/BUILD +++ b/src/ray/protobuf/BUILD @@ -380,3 +380,18 @@ cc_proto_library( name = "autoscaler_cc_proto", deps = [":autoscaler_proto"], ) + +serve_protos = ["serve.proto"] +filegroup( + name = "serve_protos", + srcs = serve_protos, + visibility = ["//visibility:public"], +) + +filegroup( + name = "core_protos", + srcs = glob([ + "**/*.proto", + ], exclude=serve_protos), + visibility = ["//visibility:public"], +) diff --git a/src/ray/pubsub/publisher.h b/src/ray/pubsub/publisher.h index f14b0c4e37757..dbb0f3376a412 100644 --- a/src/ray/pubsub/publisher.h +++ b/src/ray/pubsub/publisher.h @@ -441,7 +441,7 @@ class Publisher : public PublisherInterface { /// int UnregisterSubscriberInternal(const SubscriberID &subscriber_id) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); // Periodic runner to invoke CheckDeadSubscribers. PeriodicalRunner *periodical_runner_; @@ -458,16 +458,17 @@ class Publisher : public PublisherInterface { /// Mapping of node id -> subscribers. absl::flat_hash_map> - subscribers_ GUARDED_BY(mutex_); + subscribers_ ABSL_GUARDED_BY(mutex_); /// Index that stores the mapping of messages <-> subscribers. absl::flat_hash_map - subscription_index_map_ GUARDED_BY(mutex_); + subscription_index_map_ ABSL_GUARDED_BY(mutex_); /// The maximum number of objects to publish for each publish calls. int publish_batch_size_; - absl::flat_hash_map cum_pub_message_cnt_ GUARDED_BY(mutex_); + absl::flat_hash_map cum_pub_message_cnt_ + ABSL_GUARDED_BY(mutex_); /// The monotonically increasing sequence_id for this publisher. /// The publisher will add this sequence_id to every message to be published. @@ -480,7 +481,7 @@ class Publisher : public PublisherInterface { /// - the subscriber doesn't expect the sequences it receives are contiguous. /// this is due the fact a subscriber can only subscribe a subset /// of a channel. - int64_t next_sequence_id_ GUARDED_BY(mutex_) = 0; + int64_t next_sequence_id_ ABSL_GUARDED_BY(mutex_) = 0; /// A unique identifier identifies the publisher_id. /// TODO(scv119) add docs about the semantics. diff --git a/src/ray/pubsub/subscriber.h b/src/ray/pubsub/subscriber.h index 8c73716d88449..97b32af748cdc 100644 --- a/src/ray/pubsub/subscriber.h +++ b/src/ray/pubsub/subscriber.h @@ -375,7 +375,7 @@ class Subscriber : public SubscriberInterface { /// Return the Channel of the given channel type. Subscriber keeps ownership. SubscriberChannel *Channel(const rpc::ChannelType channel_type) const - EXCLUSIVE_LOCKS_REQUIRED(mutex_) { + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_) { const auto it = channels_.find(channel_type); if (it == channels_.end()) { return nullptr; @@ -402,7 +402,7 @@ class Subscriber : public SubscriberInterface { FRIEND_TEST(SubscriberTest, TestUnsubscribeInSubscriptionCallback); FRIEND_TEST(SubscriberTest, TestCommandsCleanedUponPublishFailure); // Testing only. Check if there are leaks. - bool CheckNoLeaks() const LOCKS_EXCLUDED(mutex_); + bool CheckNoLeaks() const ABSL_LOCKS_EXCLUDED(mutex_); /// /// Private fields @@ -426,19 +426,19 @@ class Subscriber : public SubscriberInterface { /// objects. /// \param subscriber_address The address of the subscriber. void MakeLongPollingPubsubConnection(const rpc::Address &publisher_address) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Private method to handle long polling responses. Long polling responses contain the /// published messages. void HandleLongPollingResponse(const rpc::Address &publisher_address, const Status &status, const rpc::PubsubLongPollingReply &reply) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Make a long polling connection if it never made the one with this publisher for /// pubsub operations. void MakeLongPollingConnectionIfNotConnected(const rpc::Address &publisher_address) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Send a command batch to the publisher. To ensure the FIFO order with unary GRPC /// requests (which don't guarantee ordering), the subscriber module only allows to have @@ -448,11 +448,11 @@ class Subscriber : public SubscriberInterface { /// This RPC should be independent from the long polling RPC to receive published /// messages. void SendCommandBatchIfPossible(const rpc::Address &publisher_address) - EXCLUSIVE_LOCKS_REQUIRED(mutex_); + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_); /// Return true if the given publisher id has subscription to any of channel. bool SubscriptionExists(const PublisherID &publisher_id) - EXCLUSIVE_LOCKS_REQUIRED(mutex_) { + ABSL_EXCLUSIVE_LOCKS_REQUIRED(mutex_) { return std::any_of(channels_.begin(), channels_.end(), [publisher_id](const auto &p) { return p.second->SubscriptionExists(publisher_id); }); @@ -480,23 +480,23 @@ class Subscriber : public SubscriberInterface { SubscribeDoneCallback done_cb; }; using CommandQueue = std::queue>; - absl::flat_hash_map commands_ GUARDED_BY(mutex_); + absl::flat_hash_map commands_ ABSL_GUARDED_BY(mutex_); /// A set to cache the connected publisher ids. "Connected" means the long polling /// request is in flight. - absl::flat_hash_set publishers_connected_ GUARDED_BY(mutex_); + absl::flat_hash_set publishers_connected_ ABSL_GUARDED_BY(mutex_); /// A set to keep track of in-flight command batch requests - absl::flat_hash_set command_batch_sent_ GUARDED_BY(mutex_); + absl::flat_hash_set command_batch_sent_ ABSL_GUARDED_BY(mutex_); /// Mapping of channel type to channels. absl::flat_hash_map> channels_ - GUARDED_BY(mutex_); + ABSL_GUARDED_BY(mutex_); /// Keeps track of last processed by publisher. /// Note the publisher_id only change if gcs failover. absl::flat_hash_map> processed_sequences_ - GUARDED_BY(mutex_); + ABSL_GUARDED_BY(mutex_); }; } // namespace pubsub diff --git a/src/ray/raylet/local_object_manager.h b/src/ray/raylet/local_object_manager.h index 116776bd1d16e..c7619d0e31c45 100644 --- a/src/ray/raylet/local_object_manager.h +++ b/src/ray/raylet/local_object_manager.h @@ -318,7 +318,7 @@ class LocalObjectManager { mutable absl::Mutex mutex_; /// The current number of active spill workers. - int64_t num_active_workers_ GUARDED_BY(mutex_); + int64_t num_active_workers_ ABSL_GUARDED_BY(mutex_); /// The max number of active spill workers. const int64_t max_active_workers_; diff --git a/src/ray/raylet/node_manager.h b/src/ray/raylet/node_manager.h index fd031c10369c3..d364996a31648 100644 --- a/src/ray/raylet/node_manager.h +++ b/src/ray/raylet/node_manager.h @@ -811,7 +811,8 @@ class NodeManager : public rpc::NodeManagerServiceHandler, /// Keeps track of workers waiting for objects absl::flat_hash_map>> - async_plasma_objects_notification_ GUARDED_BY(plasma_object_notification_lock_); + async_plasma_objects_notification_ + ABSL_GUARDED_BY(plasma_object_notification_lock_); /// Fields that are used to report metrics. /// The period between debug state dumps. diff --git a/src/ray/raylet/placement_group_resource_manager_test.cc b/src/ray/raylet/placement_group_resource_manager_test.cc index 8ee5549518364..e50125290f0bc 100644 --- a/src/ray/raylet/placement_group_resource_manager_test.cc +++ b/src/ray/raylet/placement_group_resource_manager_test.cc @@ -29,6 +29,7 @@ namespace ray { class NewPlacementGroupResourceManagerTest : public ::testing::Test { public: + instrumented_io_context io_context; std::unique_ptr new_placement_group_resource_manager_; std::shared_ptr cluster_resource_scheduler_; @@ -73,7 +74,6 @@ class NewPlacementGroupResourceManagerTest : public ::testing::Test { std::make_shared(std::move(bundle_spec))); return bundle_specs; } - instrumented_io_context io_context; }; TEST_F(NewPlacementGroupResourceManagerTest, diff --git a/src/ray/raylet/worker_pool.cc b/src/ray/raylet/worker_pool.cc index f77c4cfc66df8..8eb653abb6943 100644 --- a/src/ray/raylet/worker_pool.cc +++ b/src/ray/raylet/worker_pool.cc @@ -1367,16 +1367,21 @@ void WorkerPool::DisconnectWorker(const std::shared_ptr &worker auto &state = GetStateForLanguage(worker->GetLanguage()); auto it = state.worker_processes.find(worker->GetStartupToken()); if (it != state.worker_processes.end()) { + const auto serialized_runtime_env = + it->second.runtime_env_info.serialized_runtime_env(); if (it->second.is_pending_registration) { // Worker is either starting or started, // if it's not started, we should remove it from starting. it->second.is_pending_registration = false; if (worker->GetWorkerType() == rpc::WorkerType::WORKER) { + // This may add new workers to state.worker_processes + // and invalidate the iterator, do not use `it` + // after this call. TryPendingPopWorkerRequests(worker->GetLanguage()); } } - DeleteRuntimeEnvIfPossible(it->second.runtime_env_info.serialized_runtime_env()); + DeleteRuntimeEnvIfPossible(serialized_runtime_env); RemoveWorkerProcess(state, worker->GetStartupToken()); } RAY_CHECK(RemoveWorker(state.registered_workers, worker)); diff --git a/src/ray/rpc/client_call.h b/src/ray/rpc/client_call.h index 15c5f6b2f8cdc..470347f317196 100644 --- a/src/ray/rpc/client_call.h +++ b/src/ray/rpc/client_call.h @@ -131,7 +131,7 @@ class ClientCallImpl : public ClientCall { /// return_status_ = GrpcStatusToRayStatus(status_) but need /// a separate variable because status_ is set internally by /// GRPC and we cannot control it holding the lock. - ray::Status return_status_ GUARDED_BY(mutex_); + ray::Status return_status_ ABSL_GUARDED_BY(mutex_); /// Context for the client. It could be used to convey extra information to /// the server and/or tweak certain RPC behaviors. diff --git a/src/ray/rpc/node_manager/node_manager_client_pool.h b/src/ray/rpc/node_manager/node_manager_client_pool.h index 6e28981e57fa8..0e575a460cfb4 100644 --- a/src/ray/rpc/node_manager/node_manager_client_pool.h +++ b/src/ray/rpc/node_manager/node_manager_client_pool.h @@ -78,7 +78,7 @@ class NodeManagerClientPool { /// A pool of open connections by host:port. Clients can reuse the connection /// objects in this pool by requesting them absl::flat_hash_map> client_map_ - GUARDED_BY(mu_); + ABSL_GUARDED_BY(mu_); }; } // namespace rpc diff --git a/src/ray/rpc/worker/core_worker_client.h b/src/ray/rpc/worker/core_worker_client.h index b2f3f73c335d9..449b8f3f8ba3d 100644 --- a/src/ray/rpc/worker/core_worker_client.h +++ b/src/ray/rpc/worker/core_worker_client.h @@ -470,13 +470,13 @@ class CoreWorkerClient : public std::enable_shared_from_this, /// Queue of requests to send. std::deque, ClientCallback>> - send_queue_ GUARDED_BY(mutex_); + send_queue_ ABSL_GUARDED_BY(mutex_); /// The number of bytes currently in flight. - int64_t rpc_bytes_in_flight_ GUARDED_BY(mutex_) = 0; + int64_t rpc_bytes_in_flight_ ABSL_GUARDED_BY(mutex_) = 0; /// The max sequence number we have processed responses for. - int64_t max_finished_seq_no_ GUARDED_BY(mutex_) = -1; + int64_t max_finished_seq_no_ ABSL_GUARDED_BY(mutex_) = -1; }; typedef std::function(const rpc::Address &)> diff --git a/src/ray/rpc/worker/core_worker_client_pool.h b/src/ray/rpc/worker/core_worker_client_pool.h index b3e581e529cb2..8d902ce6a666e 100644 --- a/src/ray/rpc/worker/core_worker_client_pool.h +++ b/src/ray/rpc/worker/core_worker_client_pool.h @@ -73,7 +73,7 @@ class CoreWorkerClientPool { /// A pool of open connections by WorkerID. Clients can reuse the connection /// objects in this pool by requesting them. absl::flat_hash_map> client_map_ - GUARDED_BY(mu_); + ABSL_GUARDED_BY(mu_); }; } // namespace rpc diff --git a/src/ray/stats/metric_exporter.h b/src/ray/stats/metric_exporter.h index 4212736da3a9d..4340fba4a42a3 100644 --- a/src/ray/stats/metric_exporter.h +++ b/src/ray/stats/metric_exporter.h @@ -120,7 +120,7 @@ class OpenCensusProtoExporter final : public opencensus::stats::StatsExporter::H /// Lock to protect the client mutable absl::Mutex mu_; /// Client to call a metrics agent gRPC server. - std::unique_ptr client_ GUARDED_BY(&mu_); + std::unique_ptr client_ ABSL_GUARDED_BY(&mu_); /// The worker ID of the current component. WorkerID worker_id_; }; diff --git a/src/ray/util/counter_map.h b/src/ray/util/counter_map.h index 8f61bcb7ab32e..6c7acbe046b8f 100644 --- a/src/ray/util/counter_map.h +++ b/src/ray/util/counter_map.h @@ -151,22 +151,22 @@ class CounterMapThreadSafe { CounterMapThreadSafe() = default; void SetOnChangeCallback(std::function on_change) - LOCKS_EXCLUDED(mutex_) { + ABSL_LOCKS_EXCLUDED(mutex_) { absl::WriterMutexLock lock(&mutex_); counter_map_.SetOnChangeCallback(std::move(on_change)); } - void FlushOnChangeCallbacks() LOCKS_EXCLUDED(mutex_) { + void FlushOnChangeCallbacks() ABSL_LOCKS_EXCLUDED(mutex_) { absl::WriterMutexLock lock(&mutex_); counter_map_.FlushOnChangeCallbacks(); } - void Increment(const K &key, int64_t val = 1) LOCKS_EXCLUDED(mutex_) { + void Increment(const K &key, int64_t val = 1) ABSL_LOCKS_EXCLUDED(mutex_) { absl::WriterMutexLock lock(&mutex_); counter_map_.Increment(key, val); } - void Decrement(const K &key, int64_t val = 1) LOCKS_EXCLUDED(mutex_) { + void Decrement(const K &key, int64_t val = 1) ABSL_LOCKS_EXCLUDED(mutex_) { absl::WriterMutexLock lock(&mutex_); counter_map_.Decrement(key, val); } @@ -176,7 +176,8 @@ class CounterMapThreadSafe { return counter_map_.Get(key); } - void Swap(const K &old_key, const K &new_key, int64_t val = 1) LOCKS_EXCLUDED(mutex_) { + void Swap(const K &old_key, const K &new_key, int64_t val = 1) + ABSL_LOCKS_EXCLUDED(mutex_) { absl::WriterMutexLock lock(&mutex_); counter_map_.Swap(old_key, new_key, val); } diff --git a/src/ray/util/sequencer.h b/src/ray/util/sequencer.h index ffa52a1b92ac0..3a49890405840 100644 --- a/src/ray/util/sequencer.h +++ b/src/ray/util/sequencer.h @@ -74,7 +74,7 @@ class Sequencer { absl::flat_hash_map< KEY, std::deque>> - pending_operations_ GUARDED_BY(mutex_); + pending_operations_ ABSL_GUARDED_BY(mutex_); }; } // namespace ray diff --git a/src/ray/util/util.cc b/src/ray/util/util.cc index 6299b3922774c..43cae6e91d264 100644 --- a/src/ray/util/util.cc +++ b/src/ray/util/util.cc @@ -77,7 +77,7 @@ std::string EndpointToUrl( result = ss.str(); break; } -#if defined(BOOST_ASIO_HAS_LOCAL_SOCKETS) and !defined(_WIN64) +#if defined(BOOST_ASIO_HAS_LOCAL_SOCKETS) && !defined(_WIN32) case AF_UNIX: scheme = "unix://"; result.append(reinterpret_cast(ep.data())->sun_path, @@ -112,7 +112,7 @@ ParseUrlEndpoint(const std::string &endpoint, int default_port) { scheme = "tcp://"; } if (scheme == "unix://") { -#if defined(BOOST_ASIO_HAS_LOCAL_SOCKETS) and !defined(_WIN64) +#if defined(BOOST_ASIO_HAS_LOCAL_SOCKETS) && !defined(_WIN32) size_t maxlen = sizeof(sockaddr_un().sun_path) / sizeof(*sockaddr_un().sun_path) - 1; RAY_CHECK(address.size() <= maxlen) << "AF_UNIX path length cannot exceed " << maxlen << " bytes: " << address; diff --git a/thirdparty/patches/grpc-cython-copts.patch b/thirdparty/patches/grpc-cython-copts.patch index db3a3911f7619..c893cab1bd78e 100644 --- a/thirdparty/patches/grpc-cython-copts.patch +++ b/thirdparty/patches/grpc-cython-copts.patch @@ -30,6 +30,8 @@ diff --git bazel/cython_library.bzl bazel/cython_library.bzl + srcs = [stem + ".cpp"] + cc_kwargs.pop("srcs", []), - deps = deps + ["@local_config_python//:python_headers"], + deps = deps + ["@local_config_python//:python_headers"] + cc_kwargs.pop("deps", []), +- defines = defines, ++ defines = defines, - linkshared = 1, + linkshared = cc_kwargs.pop("linkshared", 1), + **cc_kwargs diff --git a/thirdparty/patches/grpc-python.patch b/thirdparty/patches/grpc-python.patch deleted file mode 100644 index 263d47cc14b2d..0000000000000 --- a/thirdparty/patches/grpc-python.patch +++ /dev/null @@ -1,12 +0,0 @@ -diff --git third_party/py/python_configure.bzl third_party/py/python_configure.bzl ---- third_party/py/python_configure.bzl -+++ third_party/py/python_configure.bzl -@@ -345,7 +345,7 @@ def _python_autoconf_impl(repository_ctx): - repository_ctx, - "_python3", - _PYTHON3_BIN_PATH, -- "python3", -+ "python3" if not _is_windows(repository_ctx) else "python.exe", - _PYTHON3_LIB_PATH, - False - ) diff --git a/thirdparty/patches/opencensus-cpp-shutdown-api.patch b/thirdparty/patches/opencensus-cpp-shutdown-api.patch index 4cd0296c0e7fe..0ec278b1f7aa2 100644 --- a/thirdparty/patches/opencensus-cpp-shutdown-api.patch +++ b/thirdparty/patches/opencensus-cpp-shutdown-api.patch @@ -5,7 +5,7 @@ index 3bb5962..706b9b4 100644 @@ -25,7 +25,7 @@ namespace opencensus { namespace exporters { namespace stats { - + -std::vector PrometheusExporter::Collect() const { +std::vector PrometheusExporter::Collect() { const auto data = opencensus::stats::StatsExporter::GetViewData(); @@ -22,7 +22,7 @@ index bbb285d..ab6471f 100644 - std::vector Collect() const override; + std::vector Collect() override; }; - + } // namespace stats diff --git a/opencensus/stats/internal/delta_producer.cc b/opencensus/stats/internal/delta_producer.cc index 1d00504..7eb0d8a 100644 @@ -31,7 +31,7 @@ index 1d00504..7eb0d8a 100644 @@ -75,6 +75,20 @@ DeltaProducer* DeltaProducer::Get() { return global_delta_producer; } - + +void DeltaProducer::Shutdown() { + { + absl::MutexLock l(&mu_); @@ -51,14 +51,14 @@ index 1d00504..7eb0d8a 100644 absl::MutexLock harvester_lock(&harvester_mu_); @@ -115,7 +129,10 @@ void DeltaProducer::Flush() { } - + DeltaProducer::DeltaProducer() - : harvester_thread_(&DeltaProducer::RunHarvesterLoop, this) {} + : harvester_thread_(&DeltaProducer::RunHarvesterLoop, this) { + absl::MutexLock l(&mu_); + thread_started_ = true; +} - + void DeltaProducer::SwapDeltas() { ABSL_ASSERT(last_delta_.delta().empty() && "Last delta was not consumed."); @@ -131,11 +148,19 @@ void DeltaProducer::RunHarvesterLoop() { @@ -83,7 +83,7 @@ index 1d00504..7eb0d8a 100644 + } } } - + diff --git a/opencensus/stats/internal/delta_producer.h b/opencensus/stats/internal/delta_producer.h index e565f6a..453b4ef 100644 --- a/opencensus/stats/internal/delta_producer.h @@ -91,21 +91,21 @@ index e565f6a..453b4ef 100644 @@ -71,6 +71,8 @@ class DeltaProducer final { // Returns a pointer to the singleton DeltaProducer. static DeltaProducer* Get(); - + + void Shutdown(); + // Adds a new Measure. void AddMeasure(); - + @@ -124,6 +126,9 @@ class DeltaProducer final { // thread when calling a flush during harvesting. Delta last_delta_ ABSL_GUARDED_BY(harvester_mu_); std::thread harvester_thread_ ABSL_GUARDED_BY(harvester_mu_); + + mutable absl::Mutex mu_; -+ bool thread_started_ GUARDED_BY(mu_) = false; ++ bool thread_started_ ABSL_GUARDED_BY(mu_) = false; }; - + } // namespace stats diff --git a/opencensus/stats/internal/stats_exporter.cc b/opencensus/stats/internal/stats_exporter.cc index 7de96d6..f9cac57 100644 @@ -113,7 +113,7 @@ index 7de96d6..f9cac57 100644 +++ b/opencensus/stats/internal/stats_exporter.cc @@ -95,25 +95,57 @@ void StatsExporterImpl::ClearHandlersForTesting() { } - + void StatsExporterImpl::StartExportThread() ABSL_EXCLUSIVE_LOCKS_REQUIRED(mu_) { - t_ = std::thread(&StatsExporterImpl::RunWorkerLoop, this); thread_started_ = true; @@ -133,7 +133,7 @@ index 7de96d6..f9cac57 100644 + t_.join(); + } } - + void StatsExporterImpl::RunWorkerLoop() { absl::Time next_export_time = GetNextExportTime(); while (true) { @@ -156,10 +156,10 @@ index 7de96d6..f9cac57 100644 + } } } - + // StatsExporter // ------------- - + +void StatsExporter::Shutdown() { + StatsExporterImpl::Get()->Shutdown(); + StatsExporterImpl::Get()->ClearHandlersForTesting(); @@ -190,11 +190,11 @@ index 6756858..228069b 100644 --- a/opencensus/stats/stats_exporter.h +++ b/opencensus/stats/stats_exporter.h @@ -44,6 +44,8 @@ class StatsExporter final { - + // Removes the view with 'name' from the registry, if one is registered. static void RemoveView(absl::string_view name); + static void Shutdown(); + static void ExportNow(); - + // StatsExporter::Handler is the interface for push exporters that export // recorded data for registered views. The exporter should provide a static