1# Copyright 2016 gRPC authors. 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14"""A gRPC server servicing both Greeter and RouteGuide RPCs.""" 15 16from concurrent import futures 17import logging 18import math 19import time 20 21import grpc 22import helloworld_pb2 23import helloworld_pb2_grpc 24import route_guide_pb2 25import route_guide_pb2_grpc 26import route_guide_resources 27 28 29def _get_feature(feature_db, point): 30 """Returns Feature at given location or None.""" 31 for feature in feature_db: 32 if feature.location == point: 33 return feature 34 return None 35 36 37def _get_distance(start, end): 38 """Distance between two points.""" 39 coord_factor = 10000000.0 40 lat_1 = start.latitude / coord_factor 41 lat_2 = end.latitude / coord_factor 42 lon_1 = start.longitude / coord_factor 43 lon_2 = end.longitude / coord_factor 44 lat_rad_1 = math.radians(lat_1) 45 lat_rad_2 = math.radians(lat_2) 46 delta_lat_rad = math.radians(lat_2 - lat_1) 47 delta_lon_rad = math.radians(lon_2 - lon_1) 48 49 a = pow(math.sin(delta_lat_rad / 2), 2) + ( 50 math.cos(lat_rad_1) 51 * math.cos(lat_rad_2) 52 * pow(math.sin(delta_lon_rad / 2), 2) 53 ) 54 c = 2 * math.atan2(math.sqrt(a), math.sqrt(1 - a)) 55 R = 6371000 56 # metres 57 return R * c 58 59 60class _GreeterServicer(helloworld_pb2_grpc.GreeterServicer): 61 def SayHello(self, request, context): 62 return helloworld_pb2.HelloReply( 63 message="Hello, {}!".format(request.name) 64 ) 65 66 67class _RouteGuideServicer(route_guide_pb2_grpc.RouteGuideServicer): 68 """Provides methods that implement functionality of route guide server.""" 69 70 def __init__(self): 71 self.db = route_guide_resources.read_route_guide_database() 72 73 def GetFeature(self, request, context): 74 feature = _get_feature(self.db, request) 75 if feature is None: 76 return route_guide_pb2.Feature(name="", location=request) 77 else: 78 return feature 79 80 def ListFeatures(self, request, context): 81 left = min(request.lo.longitude, request.hi.longitude) 82 right = max(request.lo.longitude, request.hi.longitude) 83 top = max(request.lo.latitude, request.hi.latitude) 84 bottom = min(request.lo.latitude, request.hi.latitude) 85 for feature in self.db: 86 if ( 87 feature.location.longitude >= left 88 and feature.location.longitude <= right 89 and feature.location.latitude >= bottom 90 and feature.location.latitude <= top 91 ): 92 yield feature 93 94 def RecordRoute(self, request_iterator, context): 95 point_count = 0 96 feature_count = 0 97 distance = 0.0 98 prev_point = None 99 100 start_time = time.time() 101 for point in request_iterator: 102 point_count += 1 103 if _get_feature(self.db, point): 104 feature_count += 1 105 if prev_point: 106 distance += _get_distance(prev_point, point) 107 prev_point = point 108 109 elapsed_time = time.time() - start_time 110 return route_guide_pb2.RouteSummary( 111 point_count=point_count, 112 feature_count=feature_count, 113 distance=int(distance), 114 elapsed_time=int(elapsed_time), 115 ) 116 117 def RouteChat(self, request_iterator, context): 118 prev_notes = [] 119 for new_note in request_iterator: 120 for prev_note in prev_notes: 121 if prev_note.location == new_note.location: 122 yield prev_note 123 prev_notes.append(new_note) 124 125 126def serve(): 127 server = grpc.server(futures.ThreadPoolExecutor(max_workers=10)) 128 helloworld_pb2_grpc.add_GreeterServicer_to_server( 129 _GreeterServicer(), server 130 ) 131 route_guide_pb2_grpc.add_RouteGuideServicer_to_server( 132 _RouteGuideServicer(), server 133 ) 134 server.add_insecure_port("[::]:50051") 135 server.start() 136 server.wait_for_termination() 137 138 139if __name__ == "__main__": 140 logging.basicConfig() 141 serve() 142