forked from ryihan/Subrics-Game
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
37 changed files
with
7,166 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,118 @@ | ||
# Copyright 2021 The gRPC Authors | ||
# | ||
# Licensed under the Apache License, Version 2.0 (the "License"); | ||
# you may not use this file except in compliance with the License. | ||
# You may obtain a copy of the License at | ||
# | ||
# http://www.apache.org/licenses/LICENSE-2.0 | ||
# | ||
# Unless required by applicable law or agreed to in writing, software | ||
# distributed under the License is distributed on an "AS IS" BASIS, | ||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
# See the License for the specific language governing permissions and | ||
# limitations under the License. | ||
"""Generates and compiles C++ grpc stubs from proto_library rules.""" | ||
|
||
load("@rules_proto//proto:defs.bzl", "proto_library") | ||
load("//bazel:generate_cc.bzl", "generate_cc") | ||
load("//bazel:protobuf.bzl", "well_known_proto_libs") | ||
|
||
def cc_grpc_library( | ||
name, | ||
srcs, | ||
deps, | ||
proto_only = False, | ||
well_known_protos = False, | ||
generate_mocks = False, | ||
use_external = False, | ||
grpc_only = False, | ||
**kwargs): | ||
"""Generates C++ grpc classes for services defined in a proto file. | ||
If grpc_only is True, this rule is compatible with proto_library and | ||
cc_proto_library native rules such that it expects proto_library target | ||
as srcs argument and generates only grpc library classes, expecting | ||
protobuf messages classes library (cc_proto_library target) to be passed in | ||
deps argument. By default grpc_only is False which makes this rule to behave | ||
in a backwards-compatible mode (trying to generate both proto and grpc | ||
classes). | ||
Assumes the generated classes will be used in cc_api_version = 2. | ||
Args: | ||
name (str): Name of rule. | ||
srcs (list): A single .proto file which contains services definitions, | ||
or if grpc_only parameter is True, a single proto_library which | ||
contains services descriptors. | ||
deps (list): A list of C++ proto_library (or cc_proto_library) which | ||
provides the compiled code of any message that the services depend on. | ||
proto_only (bool): If True, create only C++ proto classes library, | ||
avoid creating C++ grpc classes library (expect it in deps). | ||
Deprecated, use native cc_proto_library instead. False by default. | ||
well_known_protos (bool): Should this library additionally depend on | ||
well known protos. Deprecated, the well known protos should be | ||
specified as explicit dependencies of the proto_library target | ||
(passed in srcs parameter) instead. False by default. | ||
generate_mocks (bool): when True, Google Mock code for client stub is | ||
generated. False by default. | ||
use_external (bool): Not used. | ||
grpc_only (bool): if True, generate only grpc library, expecting | ||
protobuf messages library (cc_proto_library target) to be passed as | ||
deps. False by default (will become True by default eventually). | ||
**kwargs: rest of arguments, e.g., compatible_with and visibility | ||
""" | ||
if len(srcs) > 1: | ||
fail("Only one srcs value supported", "srcs") | ||
if grpc_only and proto_only: | ||
fail("A mutualy exclusive configuration is specified: grpc_only = True and proto_only = True") | ||
|
||
extra_deps = [] | ||
proto_targets = [] | ||
|
||
if not grpc_only: | ||
proto_target = "_" + name + "_only" | ||
cc_proto_target = name if proto_only else "_" + name + "_cc_proto" | ||
|
||
proto_deps = ["_" + dep + "_only" for dep in deps if dep.find(":") == -1] | ||
proto_deps += [dep.split(":")[0] + ":" + "_" + dep.split(":")[1] + "_only" for dep in deps if dep.find(":") != -1] | ||
if well_known_protos: | ||
proto_deps += well_known_proto_libs() | ||
proto_library( | ||
name = proto_target, | ||
srcs = srcs, | ||
deps = proto_deps, | ||
**kwargs | ||
) | ||
|
||
native.cc_proto_library( | ||
name = cc_proto_target, | ||
deps = [":" + proto_target], | ||
**kwargs | ||
) | ||
extra_deps.append(":" + cc_proto_target) | ||
proto_targets.append(proto_target) | ||
else: | ||
if not srcs: | ||
fail("srcs cannot be empty", "srcs") | ||
proto_targets += srcs | ||
|
||
if not proto_only: | ||
codegen_grpc_target = "_" + name + "_grpc_codegen" | ||
generate_cc( | ||
name = codegen_grpc_target, | ||
srcs = proto_targets, | ||
plugin = "@com_github_grpc_grpc//src/compiler:grpc_cpp_plugin", | ||
well_known_protos = well_known_protos, | ||
generate_mocks = generate_mocks, | ||
**kwargs | ||
) | ||
|
||
native.cc_library( | ||
name = name, | ||
srcs = [":" + codegen_grpc_target], | ||
hdrs = [":" + codegen_grpc_target], | ||
deps = deps + | ||
extra_deps + | ||
["@com_github_grpc_grpc//:grpc++_codegen_proto"], | ||
**kwargs | ||
) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,56 @@ | ||
# Copyright 2021 the gRPC authors. | ||
# | ||
# Licensed under the Apache License, Version 2.0 (the "License"); | ||
# you may not use this file except in compliance with the License. | ||
# You may obtain a copy of the License at | ||
# | ||
# http://www.apache.org/licenses/LICENSE-2.0 | ||
# | ||
# Unless required by applicable law or agreed to in writing, software | ||
# distributed under the License is distributed on an "AS IS" BASIS, | ||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
# See the License for the specific language governing permissions and | ||
# limitations under the License. | ||
|
||
# This is a list of llvm flags to be used when being built with use_strict_warning=1 | ||
GRPC_LLVM_WARNING_FLAGS = [ | ||
# Enable all & extra warnings | ||
"-Wall", | ||
"-Wextra", | ||
# Avoid some known traps | ||
"-Wimplicit-fallthrough", | ||
# Consider warnings as errors | ||
"-Werror", | ||
# Ignore unknown warning flags | ||
"-Wno-unknown-warning-option", | ||
# A list of enabled flags coming from internal build system | ||
"-Wc++20-extensions", | ||
"-Wctad-maybe-unsupported", | ||
"-Wdeprecated-increment-bool", | ||
"-Wfloat-overflow-conversion", | ||
"-Wfloat-zero-conversion", | ||
"-Wfor-loop-analysis", | ||
"-Wformat-security", | ||
"-Wgnu-redeclared-enum", | ||
"-Winfinite-recursion", | ||
"-Wliteral-conversion", | ||
"-Wnon-virtual-dtor", | ||
"-Woverloaded-virtual", | ||
"-Wself-assign", | ||
"-Wstring-conversion", | ||
"-Wtautological-overlap-compare", | ||
"-Wthread-safety-analysis", | ||
"-Wthread-safety-beta", | ||
"-Wunused-comparison", | ||
"-Wvla", | ||
# A list of disabled flags coming from internal build system | ||
"-Wno-string-concatenation", | ||
# Exceptions but will be removed | ||
"-Wno-deprecated-declarations", | ||
"-Wno-unused-function", | ||
] | ||
|
||
GRPC_DEFAULT_COPTS = select({ | ||
"//:use_strict_warning": GRPC_LLVM_WARNING_FLAGS, | ||
"//conditions:default": [], | ||
}) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,17 @@ | ||
# Copyright 2019 The gRPC authors. | ||
# | ||
# Licensed under the Apache License, Version 2.0 (the "License"); | ||
# you may not use this file except in compliance with the License. | ||
# You may obtain a copy of the License at | ||
# | ||
# http://www.apache.org/licenses/LICENSE-2.0 | ||
# | ||
# Unless required by applicable law or agreed to in writing, software | ||
# distributed under the License is distributed on an "AS IS" BASIS, | ||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
# See the License for the specific language governing permissions and | ||
# limitations under the License. | ||
|
||
load("@grpc_custom_exec_properties//:constants.bzl", _LARGE_MACHINE = "LARGE_MACHINE") | ||
|
||
LARGE_MACHINE = _LARGE_MACHINE |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,90 @@ | ||
# Copyright 2021 The gRPC Authors | ||
# | ||
# Licensed under the Apache License, Version 2.0 (the "License"); | ||
# you may not use this file except in compliance with the License. | ||
# You may obtain a copy of the License at | ||
# | ||
# http://www.apache.org/licenses/LICENSE-2.0 | ||
# | ||
# Unless required by applicable law or agreed to in writing, software | ||
# distributed under the License is distributed on an "AS IS" BASIS, | ||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
# See the License for the specific language governing permissions and | ||
# limitations under the License. | ||
"""Custom rules for gRPC Python""" | ||
|
||
# Adapted with modifications from | ||
# tensorflow/tensorflow/core/platform/default/build_config.bzl | ||
# Native Bazel rules don't exist yet to compile Cython code, but rules have | ||
# been written at cython/cython and tensorflow/tensorflow. We branch from | ||
# Tensorflow's version as it is more actively maintained and works for gRPC | ||
# Python's needs. | ||
def pyx_library(name, deps = [], py_deps = [], srcs = [], **kwargs): | ||
"""Compiles a group of .pyx / .pxd / .py files. | ||
First runs Cython to create .cpp files for each input .pyx or .py + .pxd | ||
pair. Then builds a shared object for each, passing "deps" to each cc_binary | ||
rule (includes Python headers by default). Finally, creates a py_library rule | ||
with the shared objects and any pure Python "srcs", with py_deps as its | ||
dependencies; the shared objects can be imported like normal Python files. | ||
Args: | ||
name: Name for the rule. | ||
deps: C/C++ dependencies of the Cython (e.g. Numpy headers). | ||
py_deps: Pure Python dependencies of the final library. | ||
srcs: .py, .pyx, or .pxd files to either compile or pass through. | ||
**kwargs: Extra keyword arguments passed to the py_library. | ||
""" | ||
|
||
# First filter out files that should be run compiled vs. passed through. | ||
py_srcs = [] | ||
pyx_srcs = [] | ||
pxd_srcs = [] | ||
for src in srcs: | ||
if src.endswith(".pyx") or (src.endswith(".py") and | ||
src[:-3] + ".pxd" in srcs): | ||
pyx_srcs.append(src) | ||
elif src.endswith(".py"): | ||
py_srcs.append(src) | ||
else: | ||
pxd_srcs.append(src) | ||
if src.endswith("__init__.py"): | ||
pxd_srcs.append(src) | ||
|
||
# Invoke cython to produce the shared object libraries. | ||
for filename in pyx_srcs: | ||
native.genrule( | ||
name = filename + "_cython_translation", | ||
srcs = [filename], | ||
outs = [filename.split(".")[0] + ".cpp"], | ||
# Optionally use PYTHON_BIN_PATH on Linux platforms so that python 3 | ||
# works. Windows has issues with cython_binary so skip PYTHON_BIN_PATH. | ||
cmd = | ||
"PYTHONHASHSEED=0 $(location @cython//:cython_binary) --cplus $(SRCS) --output-file $(OUTS)", | ||
tools = ["@cython//:cython_binary"] + pxd_srcs, | ||
) | ||
|
||
shared_objects = [] | ||
for src in pyx_srcs: | ||
stem = src.split(".")[0] | ||
shared_object_name = stem + ".so" | ||
native.cc_binary( | ||
name = shared_object_name, | ||
srcs = [stem + ".cpp"], | ||
deps = deps + ["@local_config_python//:python_headers"], | ||
linkshared = 1, | ||
) | ||
shared_objects.append(shared_object_name) | ||
|
||
data = shared_objects[:] | ||
data += kwargs.pop("data", []) | ||
|
||
# Now create a py_library with these shared objects as data. | ||
native.py_library( | ||
name = name, | ||
srcs = py_srcs, | ||
deps = py_deps, | ||
srcs_version = "PY2AND3", | ||
data = data, | ||
**kwargs | ||
) |
Oops, something went wrong.