| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 | # Copyright the 2019 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."""An example of cancelling requests in gRPC."""from __future__ import absolute_importfrom __future__ import divisionfrom __future__ import print_functionfrom concurrent import futuresimport argparseimport datetimeimport loggingimport timeimport signalimport grpcfrom examples.python.cancellation import hash_name_pb2from examples.python.cancellation import hash_name_pb2_grpc_DESCRIPTION = "A client for finding hashes similar to names."_LOGGER = logging.getLogger(__name__)_TIMEOUT_SECONDS = 0.05def run_unary_client(server_target, name, ideal_distance):    with grpc.insecure_channel(server_target) as channel:        stub = hash_name_pb2_grpc.HashFinderStub(channel)        print("Sending request")        future = stub.Find.future(hash_name_pb2.HashNameRequest(desired_name=name,                                                                  ideal_hamming_distance=ideal_distance))        def cancel_request(unused_signum, unused_frame):            print("Cancelling request.")            future.cancel()        signal.signal(signal.SIGINT, cancel_request)        while True:            try:                result = future.result(timeout=_TIMEOUT_SECONDS)            except grpc.FutureTimeoutError:                continue            except grpc.FutureCancelledError:                break            print("Got response: \n{}".format(result))            breakdef run_streaming_client(target, name, ideal_distance, interesting_distance):    passdef main():    parser = argparse.ArgumentParser(description=_DESCRIPTION)    parser.add_argument("name", type=str, help='The desired name.')    parser.add_argument("--ideal-distance", default=0, nargs='?',                        type=int, help="The desired Hamming distance.")    parser.add_argument(        '--server',        default='localhost:50051',        type=str,        nargs='?',        help='The host-port pair at which to reach the server.')    parser.add_argument(        '--show-inferior',        default=None,        type=int,        nargs='?',        help='Also show candidates with a Hamming distance less than this value.')    args = parser.parse_args()    if args.show_inferior is not None:        run_streaming_client(args.server, args.name, args.ideal_distance, args.interesting_distance)    else:        run_unary_client(args.server, args.name, args.ideal_distance)if __name__ == "__main__":    logging.basicConfig()    main()
 |