Update grpc_getter.py (#157)

* Update grpc_getter.py * Fix formatting * Update main.py
EURODEO · Aug 14, 2024 · 0ab7b8c · 0ab7b8c · github-actions · Aug 14, 2024
1 parent 036aad1
commit 0ab7b8c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/api/grpc_getter.py b/api/grpc_getter.py
@@ -12,7 +12,10 @@
 # Reuse channel and stub as much as possible, see https://grpc.io/docs/guides/performance/
 @cache
 def get_grpc_stub():
-    channel = grpc.aio.insecure_channel(f"{os.getenv('DSHOST', 'localhost')}:{os.getenv('DSPORT', '50050')}")
+    channel = grpc.aio.insecure_channel(
+        f"{os.getenv('DSHOST', 'localhost')}:{os.getenv('DSPORT', '50050')}",
+        options=[("grpc.max_receive_message_length    ", int(os.getenv("GRPC_MAX_MESSAGE_SIZE", 4096)))],
+    )
     return dstore_grpc.DatastoreStub(channel)
 
 

diff --git a/ingest/api/main.py b/ingest/api/main.py
@@ -27,7 +27,7 @@ class Response(BaseModel):
     "username": os.getenv("MQTT_USERNAME"),
     "password": os.getenv("MQTT_PASSWORD"),
     "enable_tls": os.getenv("MQTT_TLS", "False").lower() in ("true", "1", "t"),
-    "port": os.getenv("MQTT_PORT", 8883),
+    "port": int(os.getenv("MQTT_PORT", 8883)),
 }
File	Stmts	Miss	Cover	Missing
\_\_init\_\_.py	0	0	100%
datastore_pb2.py	64	52	19%	24–75
datastore_pb2_grpc.py	43	23	47%	37–52, 85–87, 92–94, 99–101, 106–108, 112–136, 174, 191, 208, 225
grpc_getter.py	20	11	45%	15–19, 23–26, 30–32, 36–38
locustfile.py	15	15	0%	1–31
main.py	41	7	83%	42, 47, 57, 67–68, 78–79
metadata_endpoints.py	55	24	56%	42–51, 55, 82, 97–171, 175
response_classes.py	5	0	100%
utilities.py	80	22	72%	15, 33, 40, 62–65, 73–84, 89–96, 116, 120, 122
custom_geo_json
edr_feature_collection.py	6	0	100%
formatters
\_\_init\_\_.py	11	0	100%
covjson.py	54	1	98%	75
geojson.py	15	2	87%	17, 42
routers
\_\_init\_\_.py	0	0	100%
edr.py	93	4	96%	248–249, 308–309
feature.py	46	19	59%	75–108, 124–129, 135–157
TOTAL	548	180	67%