convert validation app to flask app

Author: yiyuan-he

.github/workflows/release-udp-exporter.yml

@@ -31,45 +31,37 @@ jobs:
         working-directory: exporters/aws-otel-otlp-udp-exporter
         run: hatch build
 
-      - name: Setup X-Ray daemon
-        run: |
-          # Download X-Ray daemon
-          wget https://s3.us-east-2.amazonaws.com/aws-xray-assets.us-east-2/xray-daemon/aws-xray-daemon-linux-3.x.zip
-          unzip -o aws-xray-daemon-linux-3.x.zip
-          
-          # Create config file
-          echo '{
-            "Version": 2,
-            "TotalBufferSizeMB": 10,
-            "Logging": {
-              "LogLevel": "debug"
-            },
-          }' > xray-daemon-config.json
+      # TODO: Add unit tests step
 
-          # Make sure xray is executable
-          chmod +x ./xray
-          
-          # Create logs directory
-          mkdir -p daemon-logs
-          
-          # Start X-Ray daemon
-          ./xray -o -n us-west-2 -c xray-daemon-config.json > daemon-logs/xray-daemon.log 2>&1 &
-          XRAY_PID=$!
-          echo "X-Ray daemon started with PID $XRAY_PID"
-          
-          # Wait for daemon to be ready
-          echo "Waiting for X-Ray daemon to start..."
-          sleep 5
-          
-          echo "X-Ray daemon setup completed"
+      - name: Download and run X-Ray Daemon
+        run: |
+          mkdir xray-daemon
+          cd xray-daemon
+          wget https://s3.us-west-2.amazonaws.com/aws-xray-assets.us-west-2/xray-daemon/aws-xray-daemon-linux-3.x.zip
+          unzip aws-xray-daemon-linux-3.x.zip
+          ./xray -o -n us-west-2 -f ./daemon-logs.log --log-level debug &
 
-      - name: Setup validation app
+      - name: Install UDP Exporter
         run: |
           pip install ./exporters/aws-otel-otlp-udp-exporter/dist/*.whl
 
-      - name: Run validation test
+      - name: Run Sample App in Background
         working-directory: sample-applications/integ-test-app
-        run: python udp_exporter_validation_app.py
+        run: |
+          # Start validation app
+          python udp_exporter_validation_app.py &
+          # Wait for validation app to initialize
+          sleep 5
+
+      -name: Call Sample App Endpoint
+          # Trigger trace generation
+          echo "traceId=$(curl localhost:8080/test)" >> $GITHUB_OUTPUT
+          echo $traceId
+
+      - name: Print Daemon Logs
+        run: |
+          sleep 20
+          cat xray-daemon/daemon-logs.log
 
       - name: Verify X-Ray daemon received traces
         run: |

sample-applications/integ-test-app/udp_exporter_validation_app.py

@@ -1,12 +1,14 @@
 import socket
 import threading
 import time
+from flask import Flask, jsonify
 
 from amazon.opentelemetry.exporters.otlp.udp import OTLPUdpSpanExporter
 from opentelemetry import trace
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor
 
+app = Flask(__name__)
 
 # Set up tracer provider
 tracer_provider = TracerProvider()
@@ -17,24 +19,33 @@
 span_processor = BatchSpanProcessor(exporter)
 tracer_provider.add_span_processor(span_processor)
 
-# Create a span for testing with various attributes
+# Get tracer
 tracer = trace.get_tracer(__name__)
-with tracer.start_as_current_span("test_parent_span") as parent:
-    parent.set_attribute("service.name", "validation-app")
-    parent.set_attribute("test.attribute", "test_value")
-    parent.add_event("test-event", {"event.data": "some data"})
 
-    # Add a child span
-    with tracer.start_as_current_span("test_child_span") as child:
-        child.set_attribute("child.attribute", "child_value")
-        print("Created spans with attributes and events")
-
-# Force flush to ensure spans are exported immediately
-success = tracer_provider.force_flush()
-print(f"Force flush {'succeeded' if success else 'failed'}")
-
-# Give some time for the UDP packet to be processed
-time.sleep(2)
-
-# Shutdown
-tracer_provider.shutdown()
+@app.route('/trace', methods=['GET'])
+def create_trace():
+    # Create a span for testing with various attributes
+    tracer = trace.get_tracer(__name__)
+    with tracer.start_as_current_span("test_parent_span") as parent:
+        parent.set_attribute("service.name", "validation-app")
+        parent.set_attribute("test.attribute", "test_value")
+        parent.add_event("test-event", {"event.data": "some data"})
+
+        # Get the trace ID
+        trace_id = format(parent.get_span_context().trace_id, '032x')
+
+        # Add a child span
+        with tracer.start_as_current_span("test_child_span") as child:
+            child.set_attribute("child.attribute", "child_value")
+            print("Created spans with attributes and events")
+
+    # Force flush to ensure spans are exported immediately
+    success = tracer_provider.force_flush()
+    print(f"Force flush {'succeeded' if success else 'failed'}")
+
+    return jsonify({
+        "trace_id": trace_id
+    })
+
+if __name__ == '__main__':
+    app.run(port=8080)