Skip to content

Commit 6ffff8c

Browse files
authored
Greengrass china endpoint (#198)
* Temporary fix for connection with greengrass china endpoint * replace '-' with '.' * add optional parameter for endpoint * update comment on region usage * add type hint
1 parent 669cb6f commit 6ffff8c

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

awsiot/greengrass_discovery.py

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,8 @@ class DiscoveryClient:
1717
bootstrap: Client bootstrap
1818
socket_options: Socket options
1919
tls_context: Client TLS context
20-
region: AWS region
20+
region: AWS region (not used if gg_server_name is set)
21+
gg_server_name: optional full server name
2122
"""
2223
__slots__ = [
2324
'_bootstrap',
@@ -34,16 +35,23 @@ def __init__(
3435
bootstrap: ClientBootstrap,
3536
socket_options: SocketOptions,
3637
tls_context: ClientTlsContext,
37-
region: str):
38+
region: str,
39+
gg_server_name: str = None):
3840
assert isinstance(bootstrap, ClientBootstrap)
3941
assert isinstance(socket_options, SocketOptions)
4042
assert isinstance(tls_context, ClientTlsContext)
4143
assert isinstance(region, str)
44+
if gg_server_name is not None:
45+
assert isinstance(gg_server_name, str)
4246

4347
self._bootstrap = bootstrap
4448
self._socket_options = socket_options
4549
self._region = region
46-
self._gg_server_name = 'greengrass-ats.iot.{}.amazonaws.com'.format(region)
50+
if gg_server_name is None:
51+
self._gg_server_name = 'greengrass-ats.iot.{}.amazonaws.com'.format(region)
52+
else:
53+
self._gg_server_name = gg_server_name
54+
4755
self._tls_connection_options = tls_context.new_connection_options()
4856
self._tls_connection_options.set_server_name(self._gg_server_name)
4957
self.port = 8443

0 commit comments

Comments
 (0)