Generate Log Restriction Query API Spec

examples/v2/logs-restriction-queries/AddRoleToRestrictionQuery.java

@@ -0,0 +1,34 @@
+// Grant role to a restriction query returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LogsRestrictionQueriesApi;
+import com.datadog.api.client.v2.model.RelationshipToRole;
+import com.datadog.api.client.v2.model.RelationshipToRoleData;
+import com.datadog.api.client.v2.model.RolesType;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.addRoleToRestrictionQuery", true);
+    LogsRestrictionQueriesApi apiInstance = new LogsRestrictionQueriesApi(defaultClient);
+
+    RelationshipToRole body =
+        new RelationshipToRole()
+            .data(
+                new RelationshipToRoleData()
+                    .id("3653d3c6-0c75-11ea-ad28-fb5701eabc7d")
+                    .type(RolesType.ROLES));
+
+    try {
+      apiInstance.addRoleToRestrictionQuery("restriction_query_id", body);
+    } catch (ApiException e) {
+      System.err.println(
+          "Exception when calling LogsRestrictionQueriesApi#addRoleToRestrictionQuery");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/logs-restriction-queries/CreateRestrictionQuery.java

@@ -0,0 +1,38 @@
+// Create a restriction query returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LogsRestrictionQueriesApi;
+import com.datadog.api.client.v2.model.LogsRestrictionQueriesType;
+import com.datadog.api.client.v2.model.RestrictionQueryCreateAttributes;
+import com.datadog.api.client.v2.model.RestrictionQueryCreateData;
+import com.datadog.api.client.v2.model.RestrictionQueryCreatePayload;
+import com.datadog.api.client.v2.model.RestrictionQueryWithoutRelationshipsResponse;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.createRestrictionQuery", true);
+    LogsRestrictionQueriesApi apiInstance = new LogsRestrictionQueriesApi(defaultClient);
+
+    RestrictionQueryCreatePayload body =
+        new RestrictionQueryCreatePayload()
+            .data(
+                new RestrictionQueryCreateData()
+                    .attributes(
+                        new RestrictionQueryCreateAttributes().restrictionQuery("env:sandbox"))
+                    .type(LogsRestrictionQueriesType.LOGS_RESTRICTION_QUERIES));
+
+    try {
+      RestrictionQueryWithoutRelationshipsResponse result =
+          apiInstance.createRestrictionQuery(body);
+      System.out.println(result);
+    } catch (ApiException e) {
+      System.err.println("Exception when calling LogsRestrictionQueriesApi#createRestrictionQuery");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/logs-restriction-queries/DeleteRestrictionQuery.java

@@ -0,0 +1,23 @@
+// Delete a restriction query returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LogsRestrictionQueriesApi;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.deleteRestrictionQuery", true);
+    LogsRestrictionQueriesApi apiInstance = new LogsRestrictionQueriesApi(defaultClient);
+
+    try {
+      apiInstance.deleteRestrictionQuery("restriction_query_id");
+    } catch (ApiException e) {
+      System.err.println("Exception when calling LogsRestrictionQueriesApi#deleteRestrictionQuery");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/logs-restriction-queries/GetRestrictionQuery.java

@@ -0,0 +1,26 @@
+// Get a restriction query returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LogsRestrictionQueriesApi;
+import com.datadog.api.client.v2.model.RestrictionQueryWithRelationshipsResponse;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.getRestrictionQuery", true);
+    LogsRestrictionQueriesApi apiInstance = new LogsRestrictionQueriesApi(defaultClient);
+
+    try {
+      RestrictionQueryWithRelationshipsResponse result =
+          apiInstance.getRestrictionQuery("restriction_query_id");
+      System.out.println(result);
+    } catch (ApiException e) {
+      System.err.println("Exception when calling LogsRestrictionQueriesApi#getRestrictionQuery");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/logs-restriction-queries/GetRoleRestrictionQuery.java

@@ -0,0 +1,26 @@
+// Get restriction query for a given role returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LogsRestrictionQueriesApi;
+import com.datadog.api.client.v2.model.RestrictionQueryListResponse;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.getRoleRestrictionQuery", true);
+    LogsRestrictionQueriesApi apiInstance = new LogsRestrictionQueriesApi(defaultClient);
+
+    try {
+      RestrictionQueryListResponse result = apiInstance.getRoleRestrictionQuery("role_id");
+      System.out.println(result);
+    } catch (ApiException e) {
+      System.err.println(
+          "Exception when calling LogsRestrictionQueriesApi#getRoleRestrictionQuery");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/logs-restriction-queries/ListRestrictionQueries.java

@@ -0,0 +1,25 @@
+// List restriction queries returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LogsRestrictionQueriesApi;
+import com.datadog.api.client.v2.model.RestrictionQueryListResponse;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.listRestrictionQueries", true);
+    LogsRestrictionQueriesApi apiInstance = new LogsRestrictionQueriesApi(defaultClient);
+
+    try {
+      RestrictionQueryListResponse result = apiInstance.listRestrictionQueries();
+      System.out.println(result);
+    } catch (ApiException e) {
+      System.err.println("Exception when calling LogsRestrictionQueriesApi#listRestrictionQueries");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/logs-restriction-queries/ListRestrictionQueryRoles.java

@@ -0,0 +1,27 @@
+// List roles for a restriction query returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LogsRestrictionQueriesApi;
+import com.datadog.api.client.v2.model.RestrictionQueryRolesResponse;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.listRestrictionQueryRoles", true);
+    LogsRestrictionQueriesApi apiInstance = new LogsRestrictionQueriesApi(defaultClient);
+
+    try {
+      RestrictionQueryRolesResponse result =
+          apiInstance.listRestrictionQueryRoles("restriction_query_id");
+      System.out.println(result);
+    } catch (ApiException e) {
+      System.err.println(
+          "Exception when calling LogsRestrictionQueriesApi#listRestrictionQueryRoles");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/logs-restriction-queries/ListUserRestrictionQueries.java

@@ -0,0 +1,26 @@
+// Get all restriction queries for a given user returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LogsRestrictionQueriesApi;
+import com.datadog.api.client.v2.model.RestrictionQueryListResponse;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.listUserRestrictionQueries", true);
+    LogsRestrictionQueriesApi apiInstance = new LogsRestrictionQueriesApi(defaultClient);
+
+    try {
+      RestrictionQueryListResponse result = apiInstance.listUserRestrictionQueries("user_id");
+      System.out.println(result);
+    } catch (ApiException e) {
+      System.err.println(
+          "Exception when calling LogsRestrictionQueriesApi#listUserRestrictionQueries");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/logs-restriction-queries/RemoveRoleFromRestrictionQuery.java

@@ -0,0 +1,34 @@
+// Revoke role from a restriction query returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LogsRestrictionQueriesApi;
+import com.datadog.api.client.v2.model.RelationshipToRole;
+import com.datadog.api.client.v2.model.RelationshipToRoleData;
+import com.datadog.api.client.v2.model.RolesType;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.removeRoleFromRestrictionQuery", true);
+    LogsRestrictionQueriesApi apiInstance = new LogsRestrictionQueriesApi(defaultClient);
+
+    RelationshipToRole body =
+        new RelationshipToRole()
+            .data(
+                new RelationshipToRoleData()
+                    .id("3653d3c6-0c75-11ea-ad28-fb5701eabc7d")
+                    .type(RolesType.ROLES));
+
+    try {
+      apiInstance.removeRoleFromRestrictionQuery("restriction_query_id", body);
+    } catch (ApiException e) {
+      System.err.println(
+          "Exception when calling LogsRestrictionQueriesApi#removeRoleFromRestrictionQuery");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

examples/v2/logs-restriction-queries/UpdateRestrictionQuery.java

@@ -0,0 +1,38 @@
+// Update a restriction query returns "OK" response
+
+import com.datadog.api.client.ApiClient;
+import com.datadog.api.client.ApiException;
+import com.datadog.api.client.v2.api.LogsRestrictionQueriesApi;
+import com.datadog.api.client.v2.model.LogsRestrictionQueriesType;
+import com.datadog.api.client.v2.model.RestrictionQueryUpdateAttributes;
+import com.datadog.api.client.v2.model.RestrictionQueryUpdateData;
+import com.datadog.api.client.v2.model.RestrictionQueryUpdatePayload;
+import com.datadog.api.client.v2.model.RestrictionQueryWithoutRelationshipsResponse;
+
+public class Example {
+  public static void main(String[] args) {
+    ApiClient defaultClient = ApiClient.getDefaultApiClient();
+    defaultClient.setUnstableOperationEnabled("v2.updateRestrictionQuery", true);
+    LogsRestrictionQueriesApi apiInstance = new LogsRestrictionQueriesApi(defaultClient);
+
+    RestrictionQueryUpdatePayload body =
+        new RestrictionQueryUpdatePayload()
+            .data(
+                new RestrictionQueryUpdateData()
+                    .attributes(
+                        new RestrictionQueryUpdateAttributes().restrictionQuery("env:sandbox"))
+                    .type(LogsRestrictionQueriesType.LOGS_RESTRICTION_QUERIES));
+
+    try {
+      RestrictionQueryWithoutRelationshipsResponse result =
+          apiInstance.updateRestrictionQuery("restriction_query_id", body);
+      System.out.println(result);
+    } catch (ApiException e) {
+      System.err.println("Exception when calling LogsRestrictionQueriesApi#updateRestrictionQuery");
+      System.err.println("Status code: " + e.getCode());
+      System.err.println("Reason: " + e.getResponseBody());
+      System.err.println("Response headers: " + e.getResponseHeaders());
+      e.printStackTrace();
+    }
+  }
+}

src/main/java/com/datadog/api/client/ApiClient.java

@@ -769,6 +769,16 @@ public class ApiClient {
           put("v2.updateIncidentNotificationTemplate", false);
           put("v2.updateIncidentTodo", false);
           put("v2.updateIncidentType", false);
+          put("v2.addRoleToRestrictionQuery", false);
+          put("v2.createRestrictionQuery", false);
+          put("v2.deleteRestrictionQuery", false);
+          put("v2.getRestrictionQuery", false);
+          put("v2.getRoleRestrictionQuery", false);
+          put("v2.listRestrictionQueries", false);
+          put("v2.listRestrictionQueryRoles", false);
+          put("v2.listUserRestrictionQueries", false);
+          put("v2.removeRoleFromRestrictionQuery", false);
+          put("v2.updateRestrictionQuery", false);
           put("v2.createMonitorUserTemplate", false);
           put("v2.deleteMonitorUserTemplate", false);
           put("v2.getMonitorUserTemplate", false);