blob: e7f802c21e618018cf1c4238efee9e8b4dca6b7d [file] [log] [blame]
#!/usr/bin/env ruby
# Copyright 2015, 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.
# Sample app that connects to a Route Guide service.
#
# Usage: $ path/to/route_guide_client.rb path/to/route_guide_db.json &
this_dir = File.expand_path(File.dirname(__FILE__))
lib_dir = File.join(File.dirname(this_dir), 'lib')
$LOAD_PATH.unshift(lib_dir) unless $LOAD_PATH.include?(lib_dir)
require 'grpc'
require 'multi_json'
require 'route_guide_services'
include Routeguide
GET_FEATURE_POINTS = [
Point.new(latitude: 409_146_138, longitude: -746_188_906),
Point.new(latitude: 0, longitude: 0)
]
# runs a GetFeature rpc.
#
# - once with a point known to be present in the sample route database
# - once with a point that is not in the sample database
def run_get_feature(stub)
p 'GetFeature'
p '----------'
GET_FEATURE_POINTS.each do |pt|
resp = stub.get_feature(pt)
if resp.name != ''
p "- found '#{resp.name}' at #{pt.inspect}"
else
p "- found nothing at #{pt.inspect}"
end
end
end
LIST_FEATURES_RECT = Rectangle.new(
lo: Point.new(latitude: 400_000_000, longitude: -750_000_000),
hi: Point.new(latitude: 420_000_000, longitude: -730_000_000))
# runs a ListFeatures rpc.
#
# - the rectangle to chosen to include most of the known features
# in the sample db.
def run_list_features(stub)
p 'ListFeatures'
p '------------'
resps = stub.list_features(LIST_FEATURES_RECT)
resps.each do |r|
p "- found '#{r.name}' at #{r.location.inspect}"
end
end
# RandomRoute provides an Enumerable that yields a random 'route' of points
# from a list of Features.
class RandomRoute
def initialize(features, size)
@features = features
@size = size
end
# yields a point, waiting between 0 and 1 seconds between each yield
#
# @return an Enumerable that yields a random point
def each
return enum_for(:each) unless block_given?
@size.times do
json_feature = @features[rand(0..@features.length)]
next if json_feature.nil?
location = json_feature['location']
pt = Point.new(
Hash[location.each_pair.map { |k, v| [k.to_sym, v] }])
p "- next point is #{pt.inspect}"
yield pt
sleep(rand(0..1))
end
end
end
# runs a RecordRoute rpc.
#
# - the rectangle to chosen to include most of the known features
# in the sample db.
def run_record_route(stub, features)
p 'RecordRoute'
p '-----------'
points_on_route = 10 # arbitrary
reqs = RandomRoute.new(features, points_on_route)
resp = stub.record_route(reqs.each)
p "summary: #{resp.inspect}"
end
ROUTE_CHAT_NOTES = [
RouteNote.new(message: 'doh - a deer',
location: Point.new(latitude: 0, longitude: 0)),
RouteNote.new(message: 'ray - a drop of golden sun',
location: Point.new(latitude: 0, longitude: 1)),
RouteNote.new(message: 'me - the name I call myself',
location: Point.new(latitude: 1, longitude: 0)),
RouteNote.new(message: 'fa - a longer way to run',
location: Point.new(latitude: 1, longitude: 1)),
RouteNote.new(message: 'soh - with needle and a thread',
location: Point.new(latitude: 0, longitude: 1))
]
# runs a RouteChat rpc.
#
# sends a canned set of route notes and prints out the responses.
def run_route_chat(stub)
p 'Route Chat'
p '----------'
# TODO: decouple sending and receiving, i.e have the response enumerator run
# on its own thread.
resps = stub.route_chat(ROUTE_CHAT_NOTES)
resps.each { |r| p "received #{r.inspect}" }
end
def main
stub = RouteGuide::Stub.new('localhost:50051', :this_channel_is_insecure)
run_get_feature(stub)
run_list_features(stub)
run_route_chat(stub)
if ARGV.length == 0
p 'no feature database; skipping record_route'
exit
end
raw_data = []
File.open(ARGV[0]) do |f|
raw_data = MultiJson.load(f.read)
end
run_record_route(stub, raw_data)
end
main