mirror of https://github.com/grpc/grpc.git
Merge pull request #22638 from tgalkovskyi/master
Expose ALTS client/server credentials in Python APIpull/22857/head
commit
4efeb8f1fd
15 changed files with 263 additions and 28 deletions
@ -0,0 +1,49 @@ |
||||
# Copyright 2020 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. |
||||
|
||||
licenses(["notice"]) # 3-clause BSD |
||||
|
||||
load("@grpc_python_dependencies//:requirements.bzl", "requirement") |
||||
|
||||
py_binary( |
||||
name = "alts_server", |
||||
srcs = [ |
||||
"alts_server.py", |
||||
"demo_pb2.py", |
||||
"demo_pb2_grpc.py", |
||||
"server.py", |
||||
], |
||||
main = "alts_server.py", |
||||
python_version = "PY3", |
||||
srcs_version = "PY2AND3", |
||||
deps = [ |
||||
"//src/python/grpcio/grpc:grpcio", |
||||
], |
||||
) |
||||
|
||||
py_binary( |
||||
name = "alts_client", |
||||
srcs = [ |
||||
"alts_client.py", |
||||
"client.py", |
||||
"demo_pb2.py", |
||||
"demo_pb2_grpc.py", |
||||
], |
||||
main = "alts_client.py", |
||||
python_version = "PY3", |
||||
srcs_version = "PY2AND3", |
||||
deps = [ |
||||
"//src/python/grpcio/grpc:grpcio", |
||||
], |
||||
) |
@ -0,0 +1,39 @@ |
||||
# Copyright 2020 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. |
||||
"""The example of using ALTS credentials to setup gRPC client. |
||||
|
||||
The example would only successfully run in GCP environment.""" |
||||
|
||||
import grpc |
||||
|
||||
import demo_pb2_grpc |
||||
from client import (bidirectional_streaming_method, client_streaming_method, |
||||
server_streaming_method, simple_method) |
||||
|
||||
SERVER_ADDRESS = "localhost:23333" |
||||
|
||||
|
||||
def main(): |
||||
with grpc.secure_channel( |
||||
SERVER_ADDRESS, |
||||
credentials=grpc.alts_channel_credentials()) as channel: |
||||
stub = demo_pb2_grpc.GRPCDemoStub(channel) |
||||
simple_method(stub) |
||||
client_streaming_method(stub) |
||||
server_streaming_method(stub) |
||||
bidirectional_streaming_method(stub) |
||||
|
||||
|
||||
if __name__ == '__main__': |
||||
main() |
@ -0,0 +1,39 @@ |
||||
# Copyright 2020 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. |
||||
"""The example of using ALTS credentials to setup gRPC server in python. |
||||
|
||||
The example would only successfully run in GCP environment.""" |
||||
|
||||
from concurrent import futures |
||||
|
||||
import grpc |
||||
|
||||
import demo_pb2_grpc |
||||
from server import DemoServer |
||||
|
||||
SERVER_ADDRESS = 'localhost:23333' |
||||
|
||||
|
||||
def main(): |
||||
svr = grpc.server(futures.ThreadPoolExecutor()) |
||||
demo_pb2_grpc.add_GRPCDemoServicer_to_server(DemoServer(), svr) |
||||
svr.add_secure_port(SERVER_ADDRESS, |
||||
server_credentials=grpc.alts_server_credentials()) |
||||
print("------------------start Python GRPC server with ALTS encryption") |
||||
svr.start() |
||||
svr.wait_for_termination() |
||||
|
||||
|
||||
if __name__ == '__main__': |
||||
main() |
Loading…
Reference in new issue