[Add] - query cloudwatch insight and get the result

Author: lamhaison

services/logs.sh

@@ -5,7 +5,7 @@ aws_logs_list() {
 	aws_run_commandline 'aws logs describe-log-groups --query "*[].logGroupName"'
 }
 
-aws_logs_tail() {
+function aws_logs_tail() {
 	aws_log_group_name=$1
 
 	# Check aws_log_group_name invalid
@@ -18,8 +18,88 @@ aws_logs_tail() {
 	eval ${aws_cmd}
 }
 
-aws_logs_tail_with_hint() {
+function aws_logs_tail_with_hint() {
 	echo "Your log group name >"
 	aws_log_group_name=$(peco_create_menu 'peco_aws_logs_list')
 	aws_logs_tail $aws_log_group_name $1
 }
+
+# Function to start an AWS CloudWatch Logs query
+aws_logs_run_query() {
+	local log_group_name="$1" # The name of the log group
+	local start_time="$2"     # The start time for the query in milliseconds
+	local end_time="$3"       # The end time for the query in milliseconds
+	local query_string="$4"   # The CloudWatch Logs Insights query string
+	local region="$5"         # The AWS region
+
+	# Ensure required parameters are provided
+	if [[ -z "$log_group_name" || -z "$start_time" || -z "$end_time" || -z "$query_string" || -z "$region" ]]; then
+		echo "Error: Missing required arguments."
+		echo "Usage: aws_logs_run_query <log_group_name> <start_time> <end_time> <query_string> <region>"
+		return 1
+	fi
+
+	# Execute the query
+	local query_id
+	query_id=$(aws logs start-query \
+		--log-group-name "$log_group_name" \
+		--start-time "$start_time" \
+		--end-time "$end_time" \
+		--query-string "$query_string" \
+		--region "$region" \
+		--query "queryId" \
+		--output text 2>&1)
+
+	# Check for errors
+	if [[ $? -ne 0 ]]; then
+		echo "Error starting the query: $query_id"
+		return 1
+	fi
+
+	echo "$query_id"
+}
+
+# Function to get AWS CloudWatch Logs query results
+aws_logs_get_query_results() {
+	local query_id="$1" # The query ID returned by aws_logs_run_query
+	local region="$2"   # The AWS region
+
+	# Ensure required parameters are provided
+	if [[ -z "$query_id" || -z "$region" ]]; then
+		echo "Error: Missing required arguments."
+		echo "Usage: aws_logs_get_query_results <query_id> <region>"
+		return 1
+	fi
+
+	# Wait for the query to complete
+	local query_status="Running"
+	while [[ "$query_status" == "Running" || "$query_status" == "Scheduled" ]]; do
+		echo "Waiting for query results..."
+		sleep 2
+		query_status=$(aws logs get-query-results \
+			--query-id "$query_id" \
+			--region "$region" \
+			--query "status" \
+			--output text 2>&1)
+	done
+
+	# Check if the query was successful
+	if [[ "$query_status" != "Complete" ]]; then
+		echo "Error: Query did not complete successfully. Status: $query_status"
+		return 1
+	fi
+
+	# Retrieve and display query results
+	local results
+	results=$(aws logs get-query-results \
+		--query-id "$query_id" \
+		--region "$region" \
+		--output json 2>&1)
+
+	if [[ $? -ne 0 ]]; then
+		echo "Error retrieving query results: $results"
+		return 1
+	fi
+
+	echo "$results"
+}