Merge pull request awslabs#256 from Bit-Quill/fix-la-influxdb3-migration-plugin-line-protocol-naming

Fix client error checking

Author: forestmvey

tools/python/liveanalytics_influxdb3_migration_plugin/liveanalytics_migration_client/liveanalytics_influxdb3_migration_client.py

@@ -42,6 +42,8 @@ def __init__(
             db_name (str): Timestream for LiveAnalytics database name.
             s3_bucket (str): S3 bucket name.
             resume_migration (bool): Whether to resume an existing migration, skipping unload operations.
+            timeout_seconds (int): The number of seconds to wait for each migration request.
+            region (str): The AWS Region to use.
 
         Returns:
             None
@@ -432,7 +434,7 @@ def bulk_invoke_http_trigger(self, metadata):
                 )
                 trigger_invocation_response.raise_for_status()
                 response_body = trigger_invocation_response.json()
-                if response_body["status"] == "error":
+                if response_body["status"] != 200 and response_body["status"] != 202:
                     raise RuntimeError(
                         f"Migrating {s3_key} failed: {response_body['message']}"
                     )
@@ -455,8 +457,7 @@ def bulk_invoke_http_trigger(self, metadata):
                     f"Final verification failed: {final_invocation_response.json()['message']}"
                 )
         except Exception as e:
-            self.error(f"HTTP invocation failed: {e}")
-            self.error("View processing engine logs for more information")
+            self.error(f"HTTP invocation failed: {e}. View processing engine logs for more information")
             sys.exit(1)
 
     def get_num_completed_and_total_parquet_files(self):

tools/python/liveanalytics_influxdb3_migration_plugin/liveanalytics_migration_plugin/liveanalytics_migration_plugin.py

@@ -513,9 +513,9 @@ def transform_row_to_lp(influxdb3_local, row, table_name, column_types):
     Transforms data into LineBuilder objects for writing to InfluxDB.
 
     Args:
-        influxdb3_local (InfluxDB client): Logging and ingestion client
-        row (string): Row in parquet file
-        table_name (string): Table name
+        influxdb3_local (InfluxDB client): Logging and ingestion client.
+        row (str): Row in parquet file.
+        table_name (str): Table name.
         column_types(dict): Column types and names (only includes types double and int64)
 
     Returns: