[contrib] add orchestrion integration configuration

.github/workflows/orchestrion.yml

@@ -1,6 +1,11 @@
 name: Orchestrion
 on:
   workflow_dispatch: # manually
+    inputs:
+      orchestrion-version:
+        description: Orchestrion version to use for integration testing
+        type: string
+        required: false
   pull_request:
   merge_group:
   push:
@@ -18,9 +23,102 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
-  test:
-    name: 'Run Tests'
-    uses: DataDog/orchestrion/.github/workflows/workflow_call.yml@main # we don't want to pin our own action
-    with:
-      dd-trace-go-ref: ${{ github.sha }}
-      runs-on: ubuntu-latest-16-cores
+  generate:
+    name: Verify generated files are up-to-date
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout Code
+        uses: actions/checkout@v4
+      - name: Setup Go
+        uses: actions/setup-go@v5
+        with:
+          go-version: stable
+          cache: true
+          cache-dependency-path: '**/go.mod'
+      - name: Run generator
+        run: go generate ./internal/orchestrion/...
+      - name: Check for changes
+        run: git diff --exit-code
+
+  integration-test:
+    strategy:
+      fail-fast: false
+      matrix:
+        runs-on:
+          - ubuntu
+          - macos
+          - windows
+        go-version: [oldstable, stable]
+        mode: [DRIVER]
+        include:
+          # Alternate build modes (only on ubuntu with oldstable, so we save up CI time)
+          - runs-on: ubuntu
+            go-version: oldstable
+            mode: TOOLEXEC
+          - runs-on: ubuntu
+            go-version: oldstable
+            mode: GOFLAGS
+    name: Integration Test (${{ matrix.runs-on }} / ${{ matrix.go-version }} / ${{ matrix.mode }})
+    runs-on: ${{ matrix.runs-on == 'ubuntu' && fromJson('{"labels":"ubuntu-16-core-latest","group":"Large Runner Shared Public"}') || (matrix.runs-on == 'windows' && fromJson('{"labels":"windows-shared-8core","group":"LARGE WINDOWS SHARED"}')) || format('{0}-latest', matrix.runs-on) }}
+    env:
+      # Ryuk is problematic with concurrent executions, and unnecessary in ephemeral environments like GHA.
+      TESTCONTAINERS_RYUK_DISABLED: true
+    steps:
+      - name: Checkout Code
+        uses: actions/checkout@v4
+      - name: Setup Go
+        uses: actions/setup-go@v5
+        with:
+          go-version: ${{ matrix.go-version }}
+          cache: true
+          cache-dependency-path: 'internal/orchestrion/_integration/go.mod'
+
+      # ddapm-test-agent is used to observe side effects from the tracer during integration tests.
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: 3.x
+          cache: pip
+          cache-dependency-path: 'internal/orchestrion/_integration/internal/agent/requirements-dev.txt'
+      - name: Install ddapm-test-agent
+        run: pip install -r internal/orchestrion/_integration/requirements-dev.txt
+
+      # If we're in workflow_dispatch, maybe we need to up/downgrade orchestrion
+      - name: Set up orchestrion
+        if: github.event_name == 'workflow_dispatch' && github.event.inputs.orchestrion-version != ''
+        run: |-
+          go get "github.com/DataDog/orchestrion@${VERSION}"
+          go mod tidy
+        working-directory: internal/orchestrion/_integration
+        env:
+          VERSION: ${{ github.event.inputs.orchestrion-version }}
+      # We install the binary to the GOBIN, so it's easy to use
+      - name: Install orchestrion binary
+        run: |-
+          go install "github.com/DataDog/orchestrion"
+        working-directory: internal/orchestrion/_integration
+
+      # Finally, we run the test suite!
+      - name: Run Tests
+        shell: bash
+        run: |-
+          case "${MODE}" in
+          "DRIVER")
+            orchestrion go test -shuffle=on ./...
+            ;;
+          "TOOLEXEC")
+            go test -shuffle=on -toolexec='orchestrion toolexec' ./...
+            ;;
+          "GOFLAGS")
+            export GOFLAGS="${GOFLAGS} '-toolexec=orchestrion toolexec'"
+            go test -shuffle=on ./...
+            ;;
+          *)
+            echo "Unknown mode: ${MODE}"
+            ;;
+          esac
+        working-directory: internal/orchestrion/_integration
+        env:
+          GOFLAGS: -tags=githubci${{ matrix.mode == 'DRIVER' && ',buildtag' || ''}} # The "buildtag" tag is used in //dd:span integration tests
+          MODE: ${{ matrix.mode }}
+          DD_ORCHESTRION_IS_GOMOD_VERSION: true # Prevent auto-respawn, which is problematic with installs from commit SHA

contrib/99designs/gqlgen/orchestrion.yml

@@ -0,0 +1,26 @@
+# Unless explicitly stated otherwise all files in this repository are licensed
+# under the Apache License Version 2.0.
+# This product includes software developed at Datadog (https://www.datadoghq.com/).
+# Copyright 2023-present Datadog, Inc.
+---
+# yaml-language-server: $schema=https://datadoghq.dev/orchestrion/schema.json
+meta:
+  name: gopkg.in/DataDog/dd-trace-go.v1/contrib/99designs/gqlgen
+  description: gqlgen is a Go library for building GraphQL servers without any fuss.
+
+aspects:
+  - id: New
+    join-point:
+      one-of:
+        - function-call: github.com/99designs/gqlgen/graphql/handler.New
+        - function-call: github.com/99designs/gqlgen/graphql/handler.NewDefaultServer
+    advice:
+      - wrap-expression:
+          imports:
+            handler: github.com/99designs/gqlgen/graphql/handler
+            gqlgentrace: gopkg.in/DataDog/dd-trace-go.v1/contrib/99designs/gqlgen
+          template: |-
+            func(s *handler.Server) *handler.Server {
+              s.Use(gqlgentrace.NewTracer())
+              return s
+            }({{ . }})

contrib/IBM/sarama.v1/orchestrion.yml

@@ -0,0 +1,80 @@
+# Unless explicitly stated otherwise all files in this repository are licensed
+# under the Apache License Version 2.0.
+# This product includes software developed at Datadog (https://www.datadoghq.com/).
+# Copyright 2023-present Datadog, Inc.
+---
+# yaml-language-server: $schema=https://datadoghq.dev/orchestrion/schema.json
+meta:
+  name: gopkg.in/DataDog/dd-trace-go.v1/contrib/IBM/sarama.v1
+  description: Sarama is a Go library for Apache Kafka
+
+aspects:
+  - id: NewConsumer
+    join-point:
+      one-of:
+        - function-call: github.com/IBM/sarama.NewConsumer
+        - function-call: github.com/IBM/sarama.NewConsumerClient
+    advice:
+      - wrap-expression:
+          imports:
+            saramatrace: gopkg.in/DataDog/dd-trace-go.v1/contrib/IBM/sarama.v1
+            sarama: github.com/IBM/sarama
+          template: |-
+            func(c sarama.Consumer, err error) (sarama.Consumer, error) {
+              if c != nil {
+                c = saramatrace.WrapConsumer(c)
+              }
+              return c, err
+            }({{ . }})
+
+  - id: NewSyncProducer
+    join-point:
+      one-of:
+        - function-call: github.com/IBM/sarama.NewSyncProducer
+        - function-call: github.com/IBM/sarama.NewSyncProducerFromClient
+    advice:
+      - wrap-expression:
+          imports:
+            saramatrace: gopkg.in/DataDog/dd-trace-go.v1/contrib/IBM/sarama.v1
+            sarama: github.com/IBM/sarama
+          template: |-
+            {{- $cfg := .Function.ArgumentOfType "sarama.Config" -}}
+            func(p sarama.SyncProducer, err error) (sarama.SyncProducer, error) {
+              if p != nil {
+                p = saramatrace.WrapSyncProducer(
+                  {{- if $cfg -}}
+                  {{ $cfg }},
+                  {{- else -}}
+                  nil,
+                  {{- end -}}
+                  p,
+                )
+              }
+              return p, err
+            }({{ . }})
+
+  - id: NewAsyncProducer
+    join-point:
+      one-of:
+        - function-call: github.com/IBM/sarama.NewAsyncProducer
+        - function-call: github.com/IBM/sarama.NewAsyncProducerFromClient
+    advice:
+      - wrap-expression:
+          imports:
+            saramatrace: gopkg.in/DataDog/dd-trace-go.v1/contrib/IBM/sarama.v1
+            sarama: github.com/IBM/sarama
+          template: |-
+            {{- $cfg := .Function.ArgumentOfType "sarama.Config" -}}
+            func(p sarama.AsyncProducer, err error) (sarama.AsyncProducer, error) {
+              if p != nil {
+                p = saramatrace.WrapAsyncProducer(
+                  {{- if $cfg -}}
+                  {{ $cfg }},
+                  {{- else -}}
+                  nil,
+                  {{- end -}}
+                  p,
+                )
+              }
+              return p, err
+            }({{ . }})

contrib/Shopify/sarama/orchestrion.yml

@@ -0,0 +1,80 @@
+# Unless explicitly stated otherwise all files in this repository are licensed
+# under the Apache License Version 2.0.
+# This product includes software developed at Datadog (https://www.datadoghq.com/).
+# Copyright 2023-present Datadog, Inc.
+---
+# yaml-language-server: $schema=https://datadoghq.dev/orchestrion/schema.json
+meta:
+  name: gopkg.in/DataDog/dd-trace-go.v1/contrib/Shopify/sarama
+  description: Sarama is a Go library for Apache Kafka
+
+aspects:
+  - id: NewConsumer
+    join-point:
+      one-of:
+        - function-call: github.com/Shopify/sarama.NewConsumer
+        - function-call: github.com/Shopify/sarama.NewConsumerClient
+    advice:
+      - wrap-expression:
+          imports:
+            saramatrace: gopkg.in/DataDog/dd-trace-go.v1/contrib/Shopify/sarama
+            sarama: github.com/Shopify/sarama
+          template: |-
+            func(c sarama.Consumer, err error) (sarama.Consumer, error) {
+              if c != nil {
+                c = saramatrace.WrapConsumer(c)
+              }
+              return c, err
+            }({{ . }})
+
+  - id: NewSyncProducer
+    join-point:
+      one-of:
+        - function-call: github.com/Shopify/sarama.NewSyncProducer
+        - function-call: github.com/Shopify/sarama.NewSyncProducerFromClient
+    advice:
+      - wrap-expression:
+          imports:
+            saramatrace: gopkg.in/DataDog/dd-trace-go.v1/contrib/Shopify/sarama
+            sarama: github.com/Shopify/sarama
+          template: |-
+            {{- $cfg := .Function.ArgumentOfType "sarama.Config" -}}
+            func(p sarama.SyncProducer, err error) (sarama.SyncProducer, error) {
+              if p != nil {
+                p = saramatrace.WrapSyncProducer(
+                  {{- if $cfg -}}
+                  {{ $cfg }},
+                  {{- else -}}
+                  nil,
+                  {{- end -}}
+                  p,
+                )
+              }
+              return p, err
+            }({{ . }})
+
+  - id: NewAsyncProducer
+    join-point:
+      one-of:
+        - function-call: github.com/Shopify/sarama.NewAsyncProducer
+        - function-call: github.com/Shopify/sarama.NewAsyncProducerFromClient
+    advice:
+      - wrap-expression:
+          imports:
+            saramatrace: gopkg.in/DataDog/dd-trace-go.v1/contrib/Shopify/sarama
+            sarama: github.com/Shopify/sarama
+          template: |-
+            {{- $cfg := .Function.ArgumentOfType "sarama.Config" -}}
+            func(p sarama.AsyncProducer, err error) (sarama.AsyncProducer, error) {
+              if p != nil {
+                p = saramatrace.WrapAsyncProducer(
+                  {{- if $cfg -}}
+                  {{ $cfg }},
+                  {{- else -}}
+                  nil,
+                  {{- end -}}
+                  p,
+                )
+              }
+              return p, err
+            }({{ . }})

contrib/aws/aws-sdk-go-v2/aws/orchestrion.yml

@@ -0,0 +1,44 @@
+# Unless explicitly stated otherwise all files in this repository are licensed
+# under the Apache License Version 2.0.
+# This product includes software developed at Datadog (https://www.datadoghq.com/).
+# Copyright 2023-present Datadog, Inc.
+---
+# yaml-language-server: $schema=https://datadoghq.dev/orchestrion/schema.json
+meta:
+  name: gopkg.in/DataDog/dd-trace-go.v1/contrib/aws/aws-sdk-go-v2/aws
+  description: The official AWS SDK for Go
+
+aspects:
+  - id: Config
+    join-point:
+      struct-literal:
+        type: github.com/aws/aws-sdk-go-v2/aws.Config
+        match: value-only
+    advice:
+      - wrap-expression:
+          imports:
+            awstrace: gopkg.in/DataDog/dd-trace-go.v1/contrib/aws/aws-sdk-go-v2/aws
+            aws: github.com/aws/aws-sdk-go-v2/aws
+          template: |-
+            func(cfg aws.Config) (aws.Config) {
+              awstrace.AppendMiddleware(&cfg)
+              return cfg
+            }({{ . }})
+
+  - id: '*Config'
+    join-point:
+      one-of:
+        - struct-literal:
+            type: github.com/aws/aws-sdk-go-v2/aws.Config
+            match: pointer-only
+        - function-call: github.com/aws/aws-sdk-go-v2/aws.NewConfig
+    advice:
+      - wrap-expression:
+          imports:
+            awstrace: gopkg.in/DataDog/dd-trace-go.v1/contrib/aws/aws-sdk-go-v2/aws
+            aws: github.com/aws/aws-sdk-go-v2/aws
+          template: |-
+            func(cfg *aws.Config) (*aws.Config) {
+              awstrace.AppendMiddleware(cfg)
+              return cfg
+            }({{ . }})

contrib/aws/aws-sdk-go/aws/orchestrion.yml

@@ -0,0 +1,27 @@
+# Unless explicitly stated otherwise all files in this repository are licensed
+# under the Apache License Version 2.0.
+# This product includes software developed at Datadog (https://www.datadoghq.com/).
+# Copyright 2023-present Datadog, Inc.
+---
+# yaml-language-server: $schema=https://datadoghq.dev/orchestrion/schema.json
+meta:
+  name: gopkg.in/DataDog/dd-trace-go.v1/contrib/aws/aws-sdk-go/aws
+  description: The official AWS SDK for Go
+
+aspects:
+  - id: session.NewSession
+    join-point:
+      function-call: github.com/aws/aws-sdk-go/aws/session.NewSession
+    advice:
+      - wrap-expression:
+          imports:
+            awstrace: gopkg.in/DataDog/dd-trace-go.v1/contrib/aws/aws-sdk-go/aws
+            session: github.com/aws/aws-sdk-go/aws/session
+          template: |-
+            func(sess *session.Session, err error) (*session.Session, error) {
+              if sess != nil {
+                sess = awstrace.WrapSession(sess)
+              }
+              return sess, err
+            }({{ . }})
+