grpc/examples/python/multiplex/multiplex_client.py

# Copyright 2016, Google Inc.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
#     * Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
#     * Redistributions in binary form must reproduce the above
# copyright notice, this list of conditions and the following disclaimer
# in the documentation and/or other materials provided with the
# distribution.
#     * Neither the name of Google Inc. nor the names of its
# contributors may be used to endorse or promote products derived from
# this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

"""A client that makes both Greeter and RouteGuide RPCs."""

from __future__ import print_function

import random
import time

import grpc

import helloworld_pb2
import route_guide_pb2
import route_guide_resources


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(route_guide_stub, point):
  feature = route_guide_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(route_guide_stub):
  guide_get_one_feature(
      route_guide_stub,
      route_guide_pb2.Point(latitude=409146138, longitude=-746188906))
  guide_get_one_feature(
      route_guide_stub, route_guide_pb2.Point(latitude=0, longitude=0))


def guide_list_features(route_guide_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 = route_guide_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
    time.sleep(random.uniform(0.5, 1.5))


def guide_record_route(route_guide_stub):
  feature_list = route_guide_resources.read_route_guide_database()

  route_iterator = generate_route(feature_list)
  route_summary = route_guide_stub.RecordRoute(route_iterator)
  print("Finished trip with %s points " % route_summary.point_count)
  print("Passed %s features " % route_summary.feature_count)
  print("Travelled %s meters " % route_summary.distance)
  print("It took %s seconds " % route_summary.elapsed_time)


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
    time.sleep(random.uniform(0.5, 1.0))


def guide_route_chat(route_guide_stub):
  responses = route_guide_stub.RouteChat(generate_messages())
  for response in responses:
    print("Received message %s at %s" % (response.message, response.location))


def run():
  channel = grpc.insecure_channel('localhost:50051')
  greeter_stub = helloworld_pb2.GreeterStub(channel)
  route_guide_stub = route_guide_pb2.RouteGuideStub(channel)
  greeter_response = greeter_stub.SayHello(
      helloworld_pb2.HelloRequest(name='you'))
  print("Greeter client received: " + greeter_response.message)
  print("-------------- GetFeature --------------")
  guide_get_feature(route_guide_stub)
  print("-------------- ListFeatures --------------")
  guide_list_features(route_guide_stub)
  print("-------------- RecordRoute --------------")
  guide_record_route(route_guide_stub)
  print("-------------- RouteChat --------------")
  guide_route_chat(route_guide_stub)


if __name__ == '__main__':
  run()