clean up validation app

Author: yiyuan-he

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

@@ -1,6 +1,9 @@
 name: Build, Test, and Publish ADOT OTLP UDP Exporter
 
 on:
+  push:
+    branches:
+      - "udp-*"
   workflow_dispatch:
     inputs:
       version:
@@ -28,87 +31,58 @@ jobs:
         working-directory: exporters/aws-otel-otlp-udp-exporter
         run: hatch build
 
-      - name: Setup X-Ray daemon
+      # TODO: Add unit tests step
+
+      - name: Download and run 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
+          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 &
 
-          # 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..."
+      - name: Install UDP Exporter
+        run: |
+          pip install ./exporters/aws-otel-otlp-udp-exporter/dist/*.whl
+
+      - name: Install Flask
+        run: |
+          pip install flask
+
+      - name: Run Sample App in Background
+        working-directory: sample-applications/integ-test-app
+        run: |
+          # Start validation app
+          python udp_exporter_validation_app.py &
+          # Wait for validation app to initialize
           sleep 5
-          
-          # Check if process is still running
-          if ps -p $XRAY_PID > /dev/null; then
-            echo "✅ X-Ray daemon process is running"
-          else
-            echo "❌ X-Ray daemon process is not running"
-            echo "Log contents:"
-            cat daemon-logs/xray-daemon.log
-            exit 1
-          fi
-          
-          # Try to connect to the daemon
-          if nc -zv 127.0.0.1 2000 2>&1; then
-            echo "✅ Successfully connected to X-Ray daemon on port 2000"
-          else
-            echo "❌ Cannot connect to X-Ray daemon on port 2000"
-            echo "Log contents:"
-            cat daemon-logs/xray-daemon.log
-            exit 1
-          fi
-          
-          # Extra verification with curl (might not work depending on daemon setup)
-          if curl -s http://localhost:2000/GetDaemonVersion; then
-            echo "✅ X-Ray daemon API responded"
-          else
-            echo "ℹ️ X-Ray daemon doesn't support API or not ready yet"
-            # Don't exit with error as this might not be reliable
-          fi
-          
-          echo "X-Ray daemon setup completed"
 
-      - name: Setup validation app
+      - name: Call Sample App Endpoint
         run: |
-          pip install ./exporters/aws-otel-otlp-udp-exporter/dist/*.whl
+          # Trigger trace generation
+          echo "traceId=$(curl localhost:8080/test)" >> $GITHUB_OUTPUT
+          echo $traceId
 
-      - name: Run validation test
-        working-directory: exporters/aws-otel-otlp-udp-exporter/validation-app
-        run: python app.py
+      - name: Print Daemon Logs
+        run: |
+          sleep 20
+          cat xray-daemon/daemon-logs.log
 
       - name: Verify X-Ray daemon received traces
         run: |
           echo "X-Ray daemon logs:"
-          cat daemon-logs/xray-daemon.log
+          cat xray-daemon/daemon-logs.log
 
           # Check if the daemon received and processed some data
-          if grep -q "sending.*batch" daemon-logs/xray-daemon.log; then
+          if grep -q "sending.*batch" xray-daemon/daemon-logs.log; then
             echo "✅ X-Ray daemon processed trace data (AWS upload errors are expected)"
             exit 0
-          elif grep -q "processor:.*segment" daemon-logs/xray-daemon.log; then
+          elif grep -q "processor:.*segment" xray-daemon/daemon-logs.log; then
             echo "✅ X-Ray daemon processed segment data (AWS upload errors are expected)"
             exit 0
           else
             echo "❌ No evidence of traces being received by X-Ray daemon"
             exit 1
           fi
+
+      # TODO: Steps to publish to PyPI

exporters/aws-otel-otlp-udp-exporter/validation-app/app.py

@@ -0,0 +1,51 @@
+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()
+trace.set_tracer_provider(tracer_provider)
+
+# Set up UDP exporter with batch processor
+exporter = OTLPUdpSpanExporter(endpoint="127.0.0.1:2000")
+span_processor = BatchSpanProcessor(exporter)
+tracer_provider.add_span_processor(span_processor)
+
+# Get tracer
+tracer = trace.get_tracer(__name__)
+
+@app.route('/test', 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)