mirror of
https://github.com/yzernik/squeaknode.git
synced 2026-08-16 13:01:04 +02:00
2b167948e5
* Save and query squeak in client db. * Save and query squeak in client db. * MakeSqueak and GetSqueak RPC calls both return full squeak info.
143 lines
4.8 KiB
Python
143 lines
4.8 KiB
Python
# Copyright 2015 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 Python implementation of the gRPC route guide client."""
|
|
from __future__ import print_function
|
|
|
|
import logging
|
|
import random
|
|
import time
|
|
|
|
import grpc
|
|
import route_guide_pb2
|
|
import route_guide_pb2_grpc
|
|
|
|
|
|
def make_route_note(message, latitude, longitude):
|
|
return route_guide_pb2.RouteNote(
|
|
message=message,
|
|
location=route_guide_pb2.Point(latitude=latitude, longitude=longitude))
|
|
|
|
|
|
def guide_get_one_feature(stub, point):
|
|
feature = stub.GetFeature(point)
|
|
if not feature.location:
|
|
print("Server returned incomplete feature")
|
|
return
|
|
|
|
if feature.name:
|
|
print("Feature called %s at %s" % (feature.name, feature.location))
|
|
else:
|
|
print("Found no feature at %s" % feature.location)
|
|
|
|
|
|
def guide_get_feature(stub):
|
|
guide_get_one_feature(
|
|
stub,
|
|
route_guide_pb2.Point(
|
|
latitude=409146138,
|
|
longitude=-746188906,
|
|
),
|
|
)
|
|
guide_get_one_feature(
|
|
stub,
|
|
route_guide_pb2.Point(
|
|
latitude=0,
|
|
longitude=0,
|
|
),
|
|
)
|
|
|
|
|
|
def guide_list_features(stub):
|
|
rectangle = route_guide_pb2.Rectangle(
|
|
lo=route_guide_pb2.Point(latitude=400000000, longitude=-750000000),
|
|
hi=route_guide_pb2.Point(latitude=420000000, longitude=-730000000))
|
|
print("Looking for features between 40, -75 and 42, -73")
|
|
|
|
features = stub.ListFeatures(rectangle)
|
|
|
|
for feature in features:
|
|
print("Feature called %s at %s" % (feature.name, feature.location))
|
|
|
|
|
|
def generate_route(feature_list):
|
|
for _ in range(0, 10):
|
|
random_feature = feature_list[random.randint(0, len(feature_list) - 1)]
|
|
print("Visiting point %s" % random_feature.location)
|
|
yield random_feature.location
|
|
|
|
|
|
def generate_messages():
|
|
messages = [
|
|
make_route_note("First message", 0, 0),
|
|
make_route_note("Second message", 0, 1),
|
|
make_route_note("Third message", 1, 0),
|
|
make_route_note("Fourth message", 0, 0),
|
|
make_route_note("Fifth message", 1, 0),
|
|
]
|
|
for msg in messages:
|
|
print("Sending %s at %s" % (msg.message, msg.location))
|
|
yield msg
|
|
|
|
|
|
def guide_route_chat(stub):
|
|
responses = stub.RouteChat(generate_messages())
|
|
for response in responses:
|
|
print("Received message %s at %s" % (response.message,
|
|
response.location))
|
|
|
|
|
|
def run():
|
|
# NOTE(gRPC Python Team): .close() is possible on a channel and should be
|
|
# used in circumstances in which the with statement does not fit the needs
|
|
# of the code.
|
|
with grpc.insecure_channel('sqkclient_alice:50051') as alice_channel, \
|
|
grpc.insecure_channel('sqkclient_bob:50051') as bob_channel, \
|
|
grpc.insecure_channel('sqkclient_carol:50051') as carol_channel:
|
|
|
|
# Make the stubs
|
|
alice_stub = route_guide_pb2_grpc.RouteGuideStub(alice_channel)
|
|
bob_stub = route_guide_pb2_grpc.RouteGuideStub(bob_channel)
|
|
carol_stub = route_guide_pb2_grpc.RouteGuideStub(carol_channel)
|
|
|
|
print("-------------- GetFeature --------------")
|
|
guide_get_feature(alice_stub)
|
|
print("-------------- ListFeatures --------------")
|
|
guide_list_features(alice_stub)
|
|
print("-------------- RouteChat --------------")
|
|
guide_route_chat(alice_stub)
|
|
|
|
print("-------------- WalletBalance --------------")
|
|
balance = alice_stub.WalletBalance(route_guide_pb2.WalletBalanceRequest())
|
|
print("Balance: %s" % balance)
|
|
print("Balance confirmed %s %s" % (balance.total_balance, balance.total_balance))
|
|
assert balance.total_balance == 1505000000000
|
|
|
|
print("-------------- MakeSqueak --------------")
|
|
squeak_resp = alice_stub.MakeSqueak(route_guide_pb2.MakeSqueakRequest(
|
|
content='hello squeak.',
|
|
))
|
|
print("squeak: %s" % squeak_resp.squeak)
|
|
assert squeak_resp.squeak.content == 'hello squeak.'
|
|
|
|
print("-------------- GetSqueak --------------")
|
|
get_squeak_resp = alice_stub.GetSqueak(route_guide_pb2.GetSqueakRequest(
|
|
hash=squeak_resp.squeak.hash,
|
|
))
|
|
print("squeak: %s" % get_squeak_resp)
|
|
assert get_squeak_resp.squeak.content == 'hello squeak.'
|
|
|
|
|
|
if __name__ == '__main__':
|
|
logging.basicConfig()
|
|
run()
|