Create exposure SQL model

// Create exposure SQL model returns "Created" response

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"os"

	"github.com/DataDog/datadog-api-client-go/v2/api/datadog"
	"github.com/DataDog/datadog-api-client-go/v2/api/datadogV2"
)

func main() {
	body := datadogV2.ExperimentsCreateExposureSQLModelV2Request{
		Data: datadogV2.ExperimentsCreateExposureSQLModelV2RequestData{
			Attributes: datadogV2.ExperimentsUpdateExposureSQLModelV2RequestDataAttributes{
				DatePartitionColumn: *datadog.NewNullableString(nil),
				ExperimentColumn:    "experiment_id",
				Name:                "Exposure events",
				Properties: []datadogV2.ExperimentsSQLModelPropertyInput{
					{
						ColumnName:  "country",
						ColumnType:  datadogV2.EXPERIMENTSCREATEEXPOSURESQLMODELV2REQUESTDATAATTRIBUTESITEMSCOLUMNTYPE_STRING.Ptr(),
						Description: *datadog.NewNullableString(nil),
						Name:        "country",
					},
				},
				Sql: "SELECT user_id, experiment_id, variant, exposed_at, country FROM analytics.exposures",
				SubjectTypes: []datadogV2.ExperimentsCreateExposureSQLModelV2RequestDataAttributesSubjectTypesItems{
					{
						ColumnName:    "user_id",
						SubjectTypeId: "550e8400-e29b-41d4-a716-446655440010",
					},
				},
				TimestampColumn: "exposed_at",
				VariantColumn:   "variant",
			},
			Type: datadogV2.EXPERIMENTSUPDATEEXPOSURESQLMODELV2REQUESTDATATYPE_EXPOSURE_SQL_MODELS,
		},
	}
	ctx := datadog.NewDefaultContext(context.Background())
	ctx = context.WithValue(ctx, datadog.ContextAccessToken, os.Getenv("DD_BEARER_TOKEN"))
	configuration := datadog.NewConfiguration()
	apiClient := datadog.NewAPIClient(configuration)
	api := datadogV2.NewExperimentsApi(apiClient)
	resp, r, err := api.CreateExposureSQLModel(ctx, body)

	if err != nil {
		fmt.Fprintf(os.Stderr, "Error when calling `ExperimentsApi.CreateExposureSQLModel`: %v\n", err)
		fmt.Fprintf(os.Stderr, "Full HTTP response: %v\n", r)
	}

	responseContent, _ := json.MarshalIndent(resp, "", "  ")
	fmt.Fprintf(os.Stdout, "Response from `ExperimentsApi.CreateExposureSQLModel`:\n%s\n", responseContent)
}

// Create exposure SQL model returns "Created" response

import com.datadog.api.client.ApiClient;
import com.datadog.api.client.ApiException;
import com.datadog.api.client.v2.api.ExperimentsApi;
import com.datadog.api.client.v2.model.ExperimentsCreateExposureSQLModelV2Request;
import com.datadog.api.client.v2.model.ExperimentsCreateExposureSQLModelV2RequestData;
import com.datadog.api.client.v2.model.ExperimentsCreateExposureSQLModelV2RequestDataAttributesItemsColumnType;
import com.datadog.api.client.v2.model.ExperimentsCreateExposureSQLModelV2RequestDataAttributesSubjectTypesItems;
import com.datadog.api.client.v2.model.ExperimentsExposureSQLModelV2DTO;
import com.datadog.api.client.v2.model.ExperimentsSQLModelPropertyInput;
import com.datadog.api.client.v2.model.ExperimentsUpdateExposureSQLModelV2RequestDataAttributes;
import com.datadog.api.client.v2.model.ExperimentsUpdateExposureSQLModelV2RequestDataType;
import java.util.Collections;

public class Example {
  public static void main(String[] args) {
    ApiClient defaultClient = ApiClient.getDefaultApiClient();
    defaultClient.setAccessToken(System.getenv("DD_BEARER_TOKEN"));
    ExperimentsApi apiInstance = new ExperimentsApi(defaultClient);

    ExperimentsCreateExposureSQLModelV2Request body =
        new ExperimentsCreateExposureSQLModelV2Request()
            .data(
                new ExperimentsCreateExposureSQLModelV2RequestData()
                    .attributes(
                        new ExperimentsUpdateExposureSQLModelV2RequestDataAttributes()
                            .datePartitionColumn(null)
                            .experimentColumn("experiment_id")
                            .name("Exposure events")
                            .properties(
                                Collections.singletonList(
                                    new ExperimentsSQLModelPropertyInput()
                                        .columnName("country")
                                        .columnType(
                                            ExperimentsCreateExposureSQLModelV2RequestDataAttributesItemsColumnType
                                                .STRING)
                                        .description(null)
                                        .name("country")))
                            .sql(
                                "SELECT user_id, experiment_id, variant, exposed_at, country FROM"
                                    + " analytics.exposures")
                            .subjectTypes(
                                Collections.singletonList(
                                    new ExperimentsCreateExposureSQLModelV2RequestDataAttributesSubjectTypesItems()
                                        .columnName("user_id")
                                        .subjectTypeId("550e8400-e29b-41d4-a716-446655440010")))
                            .timestampColumn("exposed_at")
                            .variantColumn("variant"))
                    .type(ExperimentsUpdateExposureSQLModelV2RequestDataType.EXPOSURE_SQL_MODELS));

    try {
      ExperimentsExposureSQLModelV2DTO result = apiInstance.createExposureSQLModel(body);
      System.out.println(result);
    } catch (ApiException e) {
      System.err.println("Exception when calling ExperimentsApi#createExposureSQLModel");
      System.err.println("Status code: " + e.getCode());
      System.err.println("Reason: " + e.getResponseBody());
      System.err.println("Response headers: " + e.getResponseHeaders());
      e.printStackTrace();
    }
  }
}

"""
Create exposure SQL model returns "Created" response
"""

from os import environ
from datadog_api_client import ApiClient, Configuration
from datadog_api_client.v2.api.experiments_api import ExperimentsApi
from datadog_api_client.v2.model.experiments_create_exposure_sql_model_v2_request import (
    ExperimentsCreateExposureSQLModelV2Request,
)
from datadog_api_client.v2.model.experiments_create_exposure_sql_model_v2_request_data import (
    ExperimentsCreateExposureSQLModelV2RequestData,
)
from datadog_api_client.v2.model.experiments_create_exposure_sql_model_v2_request_data_attributes_items_column_type import (
    ExperimentsCreateExposureSQLModelV2RequestDataAttributesItemsColumnType,
)
from datadog_api_client.v2.model.experiments_create_exposure_sql_model_v2_request_data_attributes_subject_types_items import (
    ExperimentsCreateExposureSQLModelV2RequestDataAttributesSubjectTypesItems,
)
from datadog_api_client.v2.model.experiments_sql_model_property_input import ExperimentsSQLModelPropertyInput
from datadog_api_client.v2.model.experiments_update_exposure_sql_model_v2_request_data_attributes import (
    ExperimentsUpdateExposureSQLModelV2RequestDataAttributes,
)
from datadog_api_client.v2.model.experiments_update_exposure_sql_model_v2_request_data_type import (
    ExperimentsUpdateExposureSQLModelV2RequestDataType,
)

body = ExperimentsCreateExposureSQLModelV2Request(
    data=ExperimentsCreateExposureSQLModelV2RequestData(
        attributes=ExperimentsUpdateExposureSQLModelV2RequestDataAttributes(
            date_partition_column=None,
            experiment_column="experiment_id",
            name="Exposure events",
            properties=[
                ExperimentsSQLModelPropertyInput(
                    column_name="country",
                    column_type=ExperimentsCreateExposureSQLModelV2RequestDataAttributesItemsColumnType.STRING,
                    description=None,
                    name="country",
                ),
            ],
            sql="SELECT user_id, experiment_id, variant, exposed_at, country FROM analytics.exposures",
            subject_types=[
                ExperimentsCreateExposureSQLModelV2RequestDataAttributesSubjectTypesItems(
                    column_name="user_id",
                    subject_type_id="550e8400-e29b-41d4-a716-446655440010",
                ),
            ],
            timestamp_column="exposed_at",
            variant_column="variant",
        ),
        type=ExperimentsUpdateExposureSQLModelV2RequestDataType.EXPOSURE_SQL_MODELS,
    ),
)

configuration = Configuration()
configuration.access_token = environ["DD_BEARER_TOKEN"]
with ApiClient(configuration) as api_client:
    api_instance = ExperimentsApi(api_client)
    response = api_instance.create_exposure_sql_model(body=body)

    print(response)

# Create exposure SQL model returns "Created" response

require "datadog_api_client"
DatadogAPIClient.configure do |config|
  config.access_token = ENV["DD_BEARER_TOKEN"]
end
api_instance = DatadogAPIClient::V2::ExperimentsAPI.new

body = DatadogAPIClient::V2::ExperimentsCreateExposureSQLModelV2Request.new({
  data: DatadogAPIClient::V2::ExperimentsCreateExposureSQLModelV2RequestData.new({
    attributes: DatadogAPIClient::V2::ExperimentsUpdateExposureSQLModelV2RequestDataAttributes.new({
      date_partition_column: nil,
      experiment_column: "experiment_id",
      name: "Exposure events",
      properties: [
        DatadogAPIClient::V2::ExperimentsSQLModelPropertyInput.new({
          column_name: "country",
          column_type: DatadogAPIClient::V2::ExperimentsCreateExposureSQLModelV2RequestDataAttributesItemsColumnType::STRING,
          description: nil,
          name: "country",
        }),
      ],
      sql: "SELECT user_id, experiment_id, variant, exposed_at, country FROM analytics.exposures",
      subject_types: [
        DatadogAPIClient::V2::ExperimentsCreateExposureSQLModelV2RequestDataAttributesSubjectTypesItems.new({
          column_name: "user_id",
          subject_type_id: "550e8400-e29b-41d4-a716-446655440010",
        }),
      ],
      timestamp_column: "exposed_at",
      variant_column: "variant",
    }),
    type: DatadogAPIClient::V2::ExperimentsUpdateExposureSQLModelV2RequestDataType::EXPOSURE_SQL_MODELS,
  }),
})
p api_instance.create_exposure_sql_model(body)

// Create exposure SQL model returns "Created" response
use datadog_api_client::datadog;
use datadog_api_client::datadogV2::api_experiments::ExperimentsAPI;
use datadog_api_client::datadogV2::model::ExperimentsCreateExposureSQLModelV2Request;
use datadog_api_client::datadogV2::model::ExperimentsCreateExposureSQLModelV2RequestData;
use datadog_api_client::datadogV2::model::ExperimentsCreateExposureSQLModelV2RequestDataAttributesItemsColumnType;
use datadog_api_client::datadogV2::model::ExperimentsCreateExposureSQLModelV2RequestDataAttributesSubjectTypesItems;
use datadog_api_client::datadogV2::model::ExperimentsSQLModelPropertyInput;
use datadog_api_client::datadogV2::model::ExperimentsUpdateExposureSQLModelV2RequestDataAttributes;
use datadog_api_client::datadogV2::model::ExperimentsUpdateExposureSQLModelV2RequestDataType;

#[tokio::main]
async fn main() {
    let body =
        ExperimentsCreateExposureSQLModelV2Request::new(
            ExperimentsCreateExposureSQLModelV2RequestData::new(
                ExperimentsUpdateExposureSQLModelV2RequestDataAttributes::new(
                    "experiment_id".to_string(),
                    "Exposure events".to_string(),
                    "SELECT user_id, experiment_id, variant, exposed_at, country FROM analytics.exposures".to_string(),
                    vec![
                        ExperimentsCreateExposureSQLModelV2RequestDataAttributesSubjectTypesItems::new(
                            "user_id".to_string(),
                            "550e8400-e29b-41d4-a716-446655440010".to_string(),
                        )
                    ],
                    "exposed_at".to_string(),
                    "variant".to_string(),
                )
                    .date_partition_column(None)
                    .properties(
                        vec![
                            ExperimentsSQLModelPropertyInput::new("country".to_string(), "country".to_string())
                                .column_type(
                                    ExperimentsCreateExposureSQLModelV2RequestDataAttributesItemsColumnType::STRING,
                                )
                                .description(None)
                        ],
                    ),
                ExperimentsUpdateExposureSQLModelV2RequestDataType::EXPOSURE_SQL_MODELS,
            ),
        );
    let configuration = datadog::Configuration::new();
    let api = ExperimentsAPI::with_config(configuration);
    let resp = api.create_exposure_sql_model(body).await;
    if let Ok(value) = resp {
        println!("{:#?}", value);
    } else {
        println!("{:#?}", resp.unwrap_err());
    }
}