Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(121)

Side by Side Diff: appengine/swarming/swarming_bot/bot_code/remote_client.py

Issue 2969513002: Add a default Isolate gRPC proxy in config (Closed)
Patch Set: Fix problem in tests Created 3 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 # Copyright 2016 The LUCI Authors. All rights reserved. 1 # Copyright 2016 The LUCI Authors. All rights reserved.
2 # Use of this source code is governed under the Apache License, Version 2.0 2 # Use of this source code is governed under the Apache License, Version 2.0
3 # that can be found in the LICENSE file. 3 # that can be found in the LICENSE file.
4 4
5 import base64 5 import base64
6 import logging 6 import logging
7 import threading 7 import threading
8 import time 8 import time
9 import traceback 9 import traceback
10 import urllib 10 import urllib
(...skipping 23 matching lines...) Expand all
34 # time). On GCE machines it is usually 5 min. 34 # time). On GCE machines it is usually 5 min.
35 AUTH_HEADERS_EXPIRATION_SEC = 4*60+30 35 AUTH_HEADERS_EXPIRATION_SEC = 4*60+30
36 36
37 37
38 # How long to wait for a response from the server. Must not be greater than 38 # How long to wait for a response from the server. Must not be greater than
39 # AUTH_HEADERS_EXPIRATION_SEC, since otherwise there's a chance auth headers 39 # AUTH_HEADERS_EXPIRATION_SEC, since otherwise there's a chance auth headers
40 # will expire while we wait for connection. 40 # will expire while we wait for connection.
41 NET_CONNECTION_TIMEOUT_SEC = 3*60 41 NET_CONNECTION_TIMEOUT_SEC = 3*60
42 42
43 43
44 def createRemoteClient(server, auth, useGrpc): 44 def createRemoteClient(server, auth, grpc_proxy):
45 if useGrpc: 45 if grpc_proxy:
46 import remote_client_grpc 46 import remote_client_grpc
47 return remote_client_grpc.RemoteClientGrpc(server) 47 return remote_client_grpc.RemoteClientGrpc(grpc_proxy)
48 return RemoteClientNative(server, auth) 48 return RemoteClientNative(server, auth)
49 49
50 50
51 class RemoteClientNative(object): 51 class RemoteClientNative(object):
52 """RemoteClientNative knows how to make authenticated calls to the backend. 52 """RemoteClientNative knows how to make authenticated calls to the backend.
53 53
54 It also holds in-memory cache of authentication headers and periodically 54 It also holds in-memory cache of authentication headers and periodically
55 refreshes them (by calling supplied callback, that usually is implemented in 55 refreshes them (by calling supplied callback, that usually is implemented in
56 terms of bot_config.get_authentication_headers() function). 56 terms of bot_config.get_authentication_headers() function).
57 57
(...skipping 255 matching lines...) Expand 10 before | Expand all | Expand 10 after
313 'account_id': account_id, 313 'account_id': account_id,
314 'id': bot_id, 314 'id': bot_id,
315 'scopes': scopes, 315 'scopes': scopes,
316 'task_id': task_id, 316 'task_id': task_id,
317 }) 317 })
318 if not resp: 318 if not resp:
319 raise InternalError('Error when minting the token') 319 raise InternalError('Error when minting the token')
320 if resp.get('error'): 320 if resp.get('error'):
321 raise MintOAuthTokenError(resp['error']) 321 raise MintOAuthTokenError(resp['error'])
322 return resp 322 return resp
OLDNEW
« no previous file with comments | « appengine/swarming/swarming_bot/bot_code/bot_main_test.py ('k') | appengine/swarming/swarming_bot/config/config.json » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698