Get the artifact allowlist of a certain artifact type. The caller must be a metastore admin. + */ + public ArtifactAllowlistInfo get(GetArtifactAllowlistRequest request) { + return impl.get(request); + } + + public ArtifactAllowlistInfo update(ArtifactMatcher artifactMatchers, ArtifactType artifactType) { + return update( + new SetArtifactAllowlist() + .setArtifactMatchers(artifactMatchers) + .setArtifactType(artifactType)); + } + + /** + * Set an artifact allowlist. + * + *
Set the artifact allowlist of a certain artifact type. The whole artifact allowlist is + * replaced with the new allowlist. The caller must be a metastore admin. + */ + public ArtifactAllowlistInfo update(SetArtifactAllowlist request) { + return impl.update(request); + } + + public ArtifactAllowlistsService impl() { + return impl; + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactAllowlistsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactAllowlistsImpl.java new file mode 100755 index 000000000..b32cd2537 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactAllowlistsImpl.java @@ -0,0 +1,29 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.core.ApiClient; +import com.databricks.sdk.support.Generated; + +/** Package-local implementation of ArtifactAllowlists */ +@Generated +class ArtifactAllowlistsImpl implements ArtifactAllowlistsService { + private final ApiClient apiClient; + + public ArtifactAllowlistsImpl(ApiClient apiClient) { + this.apiClient = apiClient; + } + + @Override + public ArtifactAllowlistInfo get(GetArtifactAllowlistRequest request) { + String path = + String.format("/api/2.1/unity-catalog/artifact-allowlists/%s", request.getArtifactType()); + return apiClient.GET(path, request, ArtifactAllowlistInfo.class); + } + + @Override + public ArtifactAllowlistInfo update(SetArtifactAllowlist request) { + String path = + String.format("/api/2.1/unity-catalog/artifact-allowlists/%s", request.getArtifactType()); + return apiClient.PUT(path, request, ArtifactAllowlistInfo.class); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactAllowlistsService.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactAllowlistsService.java new file mode 100755 index 000000000..7b0302acf --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactAllowlistsService.java @@ -0,0 +1,30 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; + +/** + * In Databricks Runtime 13.3 and above, you can add libraries and init scripts to the `allowlist` + * in UC so that users can leverage these artifacts on compute configured with shared access mode. + * + *
This is the high-level interface, that contains generated methods. + * + *
Evolving: this interface is under development. Method signatures may change. + */ +@Generated +public interface ArtifactAllowlistsService { + /** + * Get an artifact allowlist. + * + *
Get the artifact allowlist of a certain artifact type. The caller must be a metastore admin. + */ + ArtifactAllowlistInfo get(GetArtifactAllowlistRequest getArtifactAllowlistRequest); + + /** + * Set an artifact allowlist. + * + *
Set the artifact allowlist of a certain artifact type. The whole artifact allowlist is + * replaced with the new allowlist. The caller must be a metastore admin. + */ + ArtifactAllowlistInfo update(SetArtifactAllowlist setArtifactAllowlist); +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactMatcher.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactMatcher.java new file mode 100755 index 000000000..22cb8d72c --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactMatcher.java @@ -0,0 +1,58 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +@Generated +public class ArtifactMatcher { + /** The artifact path or maven coordinate */ + @JsonProperty("artifact") + private String artifact; + + /** The pattern matching type of the artifact */ + @JsonProperty("match_type") + private MatchType matchType; + + public ArtifactMatcher setArtifact(String artifact) { + this.artifact = artifact; + return this; + } + + public String getArtifact() { + return artifact; + } + + public ArtifactMatcher setMatchType(MatchType matchType) { + this.matchType = matchType; + return this; + } + + public MatchType getMatchType() { + return matchType; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + ArtifactMatcher that = (ArtifactMatcher) o; + return Objects.equals(artifact, that.artifact) && Objects.equals(matchType, that.matchType); + } + + @Override + public int hashCode() { + return Objects.hash(artifact, matchType); + } + + @Override + public String toString() { + return new ToStringer(ArtifactMatcher.class) + .add("artifact", artifact) + .add("matchType", matchType) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactType.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactType.java new file mode 100755 index 000000000..116937719 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactType.java @@ -0,0 +1,13 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; + +/** The artifact type */ +@Generated +public enum ArtifactType { + INIT_SCRIPT, + LIBRARY_JAR, + LIBRARY_MAVEN, +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CatalogInfo.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CatalogInfo.java index 046a74e29..6358d470f 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CatalogInfo.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CatalogInfo.java @@ -31,12 +31,12 @@ public class CatalogInfo { private String createdBy; /** */ - @JsonProperty("effective_auto_maintenance_flag") - private EffectiveAutoMaintenanceFlag effectiveAutoMaintenanceFlag; + @JsonProperty("effective_predictive_optimization_flag") + private EffectivePredictiveOptimizationFlag effectivePredictiveOptimizationFlag; - /** Whether auto maintenance should be enabled for this object and objects under it. */ - @JsonProperty("enable_auto_maintenance") - private EnableAutoMaintenance enableAutoMaintenance; + /** Whether predictive optimization should be enabled for this object and objects under it. */ + @JsonProperty("enable_predictive_optimization") + private EnablePredictiveOptimization enablePredictiveOptimization; /** * Whether the current securable is accessible from all workspaces or a specific set of @@ -139,23 +139,24 @@ public String getCreatedBy() { return createdBy; } - public CatalogInfo setEffectiveAutoMaintenanceFlag( - EffectiveAutoMaintenanceFlag effectiveAutoMaintenanceFlag) { - this.effectiveAutoMaintenanceFlag = effectiveAutoMaintenanceFlag; + public CatalogInfo setEffectivePredictiveOptimizationFlag( + EffectivePredictiveOptimizationFlag effectivePredictiveOptimizationFlag) { + this.effectivePredictiveOptimizationFlag = effectivePredictiveOptimizationFlag; return this; } - public EffectiveAutoMaintenanceFlag getEffectiveAutoMaintenanceFlag() { - return effectiveAutoMaintenanceFlag; + public EffectivePredictiveOptimizationFlag getEffectivePredictiveOptimizationFlag() { + return effectivePredictiveOptimizationFlag; } - public CatalogInfo setEnableAutoMaintenance(EnableAutoMaintenance enableAutoMaintenance) { - this.enableAutoMaintenance = enableAutoMaintenance; + public CatalogInfo setEnablePredictiveOptimization( + EnablePredictiveOptimization enablePredictiveOptimization) { + this.enablePredictiveOptimization = enablePredictiveOptimization; return this; } - public EnableAutoMaintenance getEnableAutoMaintenance() { - return enableAutoMaintenance; + public EnablePredictiveOptimization getEnablePredictiveOptimization() { + return enablePredictiveOptimization; } public CatalogInfo setIsolationMode(IsolationMode isolationMode) { @@ -276,8 +277,9 @@ public boolean equals(Object o) { && Objects.equals(connectionName, that.connectionName) && Objects.equals(createdAt, that.createdAt) && Objects.equals(createdBy, that.createdBy) - && Objects.equals(effectiveAutoMaintenanceFlag, that.effectiveAutoMaintenanceFlag) - && Objects.equals(enableAutoMaintenance, that.enableAutoMaintenance) + && Objects.equals( + effectivePredictiveOptimizationFlag, that.effectivePredictiveOptimizationFlag) + && Objects.equals(enablePredictiveOptimization, that.enablePredictiveOptimization) && Objects.equals(isolationMode, that.isolationMode) && Objects.equals(metastoreId, that.metastoreId) && Objects.equals(name, that.name) @@ -300,8 +302,8 @@ public int hashCode() { connectionName, createdAt, createdBy, - effectiveAutoMaintenanceFlag, - enableAutoMaintenance, + effectivePredictiveOptimizationFlag, + enablePredictiveOptimization, isolationMode, metastoreId, name, @@ -324,8 +326,8 @@ public String toString() { .add("connectionName", connectionName) .add("createdAt", createdAt) .add("createdBy", createdBy) - .add("effectiveAutoMaintenanceFlag", effectiveAutoMaintenanceFlag) - .add("enableAutoMaintenance", enableAutoMaintenance) + .add("effectivePredictiveOptimizationFlag", effectivePredictiveOptimizationFlag) + .add("enablePredictiveOptimization", enablePredictiveOptimization) .add("isolationMode", isolationMode) .add("metastoreId", metastoreId) .add("name", name) diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EffectivePredictiveOptimizationFlag.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EffectivePredictiveOptimizationFlag.java new file mode 100755 index 000000000..bdb08b43d --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EffectivePredictiveOptimizationFlag.java @@ -0,0 +1,81 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +@Generated +public class EffectivePredictiveOptimizationFlag { + /** + * The name of the object from which the flag was inherited. If there was no inheritance, this + * field is left blank. + */ + @JsonProperty("inherited_from_name") + private String inheritedFromName; + + /** + * The type of the object from which the flag was inherited. If there was no inheritance, this + * field is left blank. + */ + @JsonProperty("inherited_from_type") + private EffectivePredictiveOptimizationFlagInheritedFromType inheritedFromType; + + /** Whether predictive optimization should be enabled for this object and objects under it. */ + @JsonProperty("value") + private EnablePredictiveOptimization value; + + public EffectivePredictiveOptimizationFlag setInheritedFromName(String inheritedFromName) { + this.inheritedFromName = inheritedFromName; + return this; + } + + public String getInheritedFromName() { + return inheritedFromName; + } + + public EffectivePredictiveOptimizationFlag setInheritedFromType( + EffectivePredictiveOptimizationFlagInheritedFromType inheritedFromType) { + this.inheritedFromType = inheritedFromType; + return this; + } + + public EffectivePredictiveOptimizationFlagInheritedFromType getInheritedFromType() { + return inheritedFromType; + } + + public EffectivePredictiveOptimizationFlag setValue(EnablePredictiveOptimization value) { + this.value = value; + return this; + } + + public EnablePredictiveOptimization getValue() { + return value; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + EffectivePredictiveOptimizationFlag that = (EffectivePredictiveOptimizationFlag) o; + return Objects.equals(inheritedFromName, that.inheritedFromName) + && Objects.equals(inheritedFromType, that.inheritedFromType) + && Objects.equals(value, that.value); + } + + @Override + public int hashCode() { + return Objects.hash(inheritedFromName, inheritedFromType, value); + } + + @Override + public String toString() { + return new ToStringer(EffectivePredictiveOptimizationFlag.class) + .add("inheritedFromName", inheritedFromName) + .add("inheritedFromType", inheritedFromType) + .add("value", value) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EffectivePredictiveOptimizationFlagInheritedFromType.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EffectivePredictiveOptimizationFlagInheritedFromType.java new file mode 100755 index 000000000..b0bff3fd6 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EffectivePredictiveOptimizationFlagInheritedFromType.java @@ -0,0 +1,15 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; + +/** + * The type of the object from which the flag was inherited. If there was no inheritance, this field + * is left blank. + */ +@Generated +public enum EffectivePredictiveOptimizationFlagInheritedFromType { + CATALOG, + SCHEMA, +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EnablePredictiveOptimization.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EnablePredictiveOptimization.java new file mode 100755 index 000000000..801f54d44 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EnablePredictiveOptimization.java @@ -0,0 +1,13 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; + +/** Whether predictive optimization should be enabled for this object and objects under it. */ +@Generated +public enum EnablePredictiveOptimization { + DISABLE, + ENABLE, + INHERIT, +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetArtifactAllowlistRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetArtifactAllowlistRequest.java new file mode 100755 index 000000000..d1c96f18b --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GetArtifactAllowlistRequest.java @@ -0,0 +1,43 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import java.util.Objects; + +/** Get an artifact allowlist */ +@Generated +public class GetArtifactAllowlistRequest { + /** The artifact type of the allowlist. */ + private ArtifactType artifactType; + + public GetArtifactAllowlistRequest setArtifactType(ArtifactType artifactType) { + this.artifactType = artifactType; + return this; + } + + public ArtifactType getArtifactType() { + return artifactType; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + GetArtifactAllowlistRequest that = (GetArtifactAllowlistRequest) o; + return Objects.equals(artifactType, that.artifactType); + } + + @Override + public int hashCode() { + return Objects.hash(artifactType); + } + + @Override + public String toString() { + return new ToStringer(GetArtifactAllowlistRequest.class) + .add("artifactType", artifactType) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/MatchType.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/MatchType.java new file mode 100755 index 000000000..4639aa300 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/MatchType.java @@ -0,0 +1,11 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; + +/** The artifact pattern matching type */ +@Generated +public enum MatchType { + PREFIX_MATCH, +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SchemaInfo.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SchemaInfo.java index 8d01aa841..1d010259a 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SchemaInfo.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SchemaInfo.java @@ -31,12 +31,12 @@ public class SchemaInfo { private String createdBy; /** */ - @JsonProperty("effective_auto_maintenance_flag") - private EffectiveAutoMaintenanceFlag effectiveAutoMaintenanceFlag; + @JsonProperty("effective_predictive_optimization_flag") + private EffectivePredictiveOptimizationFlag effectivePredictiveOptimizationFlag; - /** Whether auto maintenance should be enabled for this object and objects under it. */ - @JsonProperty("enable_auto_maintenance") - private EnableAutoMaintenance enableAutoMaintenance; + /** Whether predictive optimization should be enabled for this object and objects under it. */ + @JsonProperty("enable_predictive_optimization") + private EnablePredictiveOptimization enablePredictiveOptimization; /** Full name of schema, in form of __catalog_name__.__schema_name__. */ @JsonProperty("full_name") @@ -119,23 +119,24 @@ public String getCreatedBy() { return createdBy; } - public SchemaInfo setEffectiveAutoMaintenanceFlag( - EffectiveAutoMaintenanceFlag effectiveAutoMaintenanceFlag) { - this.effectiveAutoMaintenanceFlag = effectiveAutoMaintenanceFlag; + public SchemaInfo setEffectivePredictiveOptimizationFlag( + EffectivePredictiveOptimizationFlag effectivePredictiveOptimizationFlag) { + this.effectivePredictiveOptimizationFlag = effectivePredictiveOptimizationFlag; return this; } - public EffectiveAutoMaintenanceFlag getEffectiveAutoMaintenanceFlag() { - return effectiveAutoMaintenanceFlag; + public EffectivePredictiveOptimizationFlag getEffectivePredictiveOptimizationFlag() { + return effectivePredictiveOptimizationFlag; } - public SchemaInfo setEnableAutoMaintenance(EnableAutoMaintenance enableAutoMaintenance) { - this.enableAutoMaintenance = enableAutoMaintenance; + public SchemaInfo setEnablePredictiveOptimization( + EnablePredictiveOptimization enablePredictiveOptimization) { + this.enablePredictiveOptimization = enablePredictiveOptimization; return this; } - public EnableAutoMaintenance getEnableAutoMaintenance() { - return enableAutoMaintenance; + public EnablePredictiveOptimization getEnablePredictiveOptimization() { + return enablePredictiveOptimization; } public SchemaInfo setFullName(String fullName) { @@ -229,8 +230,9 @@ public boolean equals(Object o) { && Objects.equals(comment, that.comment) && Objects.equals(createdAt, that.createdAt) && Objects.equals(createdBy, that.createdBy) - && Objects.equals(effectiveAutoMaintenanceFlag, that.effectiveAutoMaintenanceFlag) - && Objects.equals(enableAutoMaintenance, that.enableAutoMaintenance) + && Objects.equals( + effectivePredictiveOptimizationFlag, that.effectivePredictiveOptimizationFlag) + && Objects.equals(enablePredictiveOptimization, that.enablePredictiveOptimization) && Objects.equals(fullName, that.fullName) && Objects.equals(metastoreId, that.metastoreId) && Objects.equals(name, that.name) @@ -250,8 +252,8 @@ public int hashCode() { comment, createdAt, createdBy, - effectiveAutoMaintenanceFlag, - enableAutoMaintenance, + effectivePredictiveOptimizationFlag, + enablePredictiveOptimization, fullName, metastoreId, name, @@ -271,8 +273,8 @@ public String toString() { .add("comment", comment) .add("createdAt", createdAt) .add("createdBy", createdBy) - .add("effectiveAutoMaintenanceFlag", effectiveAutoMaintenanceFlag) - .add("enableAutoMaintenance", enableAutoMaintenance) + .add("effectivePredictiveOptimizationFlag", effectivePredictiveOptimizationFlag) + .add("enablePredictiveOptimization", enablePredictiveOptimization) .add("fullName", fullName) .add("metastoreId", metastoreId) .add("name", name) diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SetArtifactAllowlist.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SetArtifactAllowlist.java new file mode 100755 index 000000000..724fa4e1d --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SetArtifactAllowlist.java @@ -0,0 +1,58 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.catalog; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +@Generated +public class SetArtifactAllowlist { + /** */ + @JsonProperty("artifact_matchers") + private ArtifactMatcher artifactMatchers; + + /** The artifact type of the allowlist. */ + private ArtifactType artifactType; + + public SetArtifactAllowlist setArtifactMatchers(ArtifactMatcher artifactMatchers) { + this.artifactMatchers = artifactMatchers; + return this; + } + + public ArtifactMatcher getArtifactMatchers() { + return artifactMatchers; + } + + public SetArtifactAllowlist setArtifactType(ArtifactType artifactType) { + this.artifactType = artifactType; + return this; + } + + public ArtifactType getArtifactType() { + return artifactType; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + SetArtifactAllowlist that = (SetArtifactAllowlist) o; + return Objects.equals(artifactMatchers, that.artifactMatchers) + && Objects.equals(artifactType, that.artifactType); + } + + @Override + public int hashCode() { + return Objects.hash(artifactMatchers, artifactType); + } + + @Override + public String toString() { + return new ToStringer(SetArtifactAllowlist.class) + .add("artifactMatchers", artifactMatchers) + .add("artifactType", artifactType) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/TableInfo.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/TableInfo.java index b28f1f07c..7c451fcc6 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/TableInfo.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/TableInfo.java @@ -55,12 +55,12 @@ public class TableInfo { private DeltaRuntimePropertiesKvPairs deltaRuntimePropertiesKvpairs; /** */ - @JsonProperty("effective_auto_maintenance_flag") - private EffectiveAutoMaintenanceFlag effectiveAutoMaintenanceFlag; + @JsonProperty("effective_predictive_optimization_flag") + private EffectivePredictiveOptimizationFlag effectivePredictiveOptimizationFlag; - /** Whether auto maintenance should be enabled for this object and objects under it. */ - @JsonProperty("enable_auto_maintenance") - private EnableAutoMaintenance enableAutoMaintenance; + /** Whether predictive optimization should be enabled for this object and objects under it. */ + @JsonProperty("enable_predictive_optimization") + private EnablePredictiveOptimization enablePredictiveOptimization; /** Encryption options that apply to clients connecting to cloud storage. */ @JsonProperty("encryption_details") @@ -236,23 +236,24 @@ public DeltaRuntimePropertiesKvPairs getDeltaRuntimePropertiesKvpairs() { return deltaRuntimePropertiesKvpairs; } - public TableInfo setEffectiveAutoMaintenanceFlag( - EffectiveAutoMaintenanceFlag effectiveAutoMaintenanceFlag) { - this.effectiveAutoMaintenanceFlag = effectiveAutoMaintenanceFlag; + public TableInfo setEffectivePredictiveOptimizationFlag( + EffectivePredictiveOptimizationFlag effectivePredictiveOptimizationFlag) { + this.effectivePredictiveOptimizationFlag = effectivePredictiveOptimizationFlag; return this; } - public EffectiveAutoMaintenanceFlag getEffectiveAutoMaintenanceFlag() { - return effectiveAutoMaintenanceFlag; + public EffectivePredictiveOptimizationFlag getEffectivePredictiveOptimizationFlag() { + return effectivePredictiveOptimizationFlag; } - public TableInfo setEnableAutoMaintenance(EnableAutoMaintenance enableAutoMaintenance) { - this.enableAutoMaintenance = enableAutoMaintenance; + public TableInfo setEnablePredictiveOptimization( + EnablePredictiveOptimization enablePredictiveOptimization) { + this.enablePredictiveOptimization = enablePredictiveOptimization; return this; } - public EnableAutoMaintenance getEnableAutoMaintenance() { - return enableAutoMaintenance; + public EnablePredictiveOptimization getEnablePredictiveOptimization() { + return enablePredictiveOptimization; } public TableInfo setEncryptionDetails(EncryptionDetails encryptionDetails) { @@ -432,8 +433,9 @@ public boolean equals(Object o) { && Objects.equals(dataSourceFormat, that.dataSourceFormat) && Objects.equals(deletedAt, that.deletedAt) && Objects.equals(deltaRuntimePropertiesKvpairs, that.deltaRuntimePropertiesKvpairs) - && Objects.equals(effectiveAutoMaintenanceFlag, that.effectiveAutoMaintenanceFlag) - && Objects.equals(enableAutoMaintenance, that.enableAutoMaintenance) + && Objects.equals( + effectivePredictiveOptimizationFlag, that.effectivePredictiveOptimizationFlag) + && Objects.equals(enablePredictiveOptimization, that.enablePredictiveOptimization) && Objects.equals(encryptionDetails, that.encryptionDetails) && Objects.equals(fullName, that.fullName) && Objects.equals(metastoreId, that.metastoreId) @@ -467,8 +469,8 @@ public int hashCode() { dataSourceFormat, deletedAt, deltaRuntimePropertiesKvpairs, - effectiveAutoMaintenanceFlag, - enableAutoMaintenance, + effectivePredictiveOptimizationFlag, + enablePredictiveOptimization, encryptionDetails, fullName, metastoreId, @@ -502,8 +504,8 @@ public String toString() { .add("dataSourceFormat", dataSourceFormat) .add("deletedAt", deletedAt) .add("deltaRuntimePropertiesKvpairs", deltaRuntimePropertiesKvpairs) - .add("effectiveAutoMaintenanceFlag", effectiveAutoMaintenanceFlag) - .add("enableAutoMaintenance", enableAutoMaintenance) + .add("effectivePredictiveOptimizationFlag", effectivePredictiveOptimizationFlag) + .add("enablePredictiveOptimization", enablePredictiveOptimization) .add("encryptionDetails", encryptionDetails) .add("fullName", fullName) .add("metastoreId", metastoreId) diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/DeleteRuns.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/DeleteRuns.java new file mode 100755 index 000000000..4cbd65156 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/DeleteRuns.java @@ -0,0 +1,80 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.ml; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +@Generated +public class DeleteRuns { + /** The ID of the experiment containing the runs to delete. */ + @JsonProperty("experiment_id") + private String experimentId; + + /** + * An optional positive integer indicating the maximum number of runs to delete. The maximum + * allowed value for max_runs is 10000. + */ + @JsonProperty("max_runs") + private Long maxRuns; + + /** + * The maximum creation timestamp in milliseconds since the UNIX epoch for deleting runs. Only + * runs created prior to or at this timestamp are deleted. + */ + @JsonProperty("max_timestamp_millis") + private Long maxTimestampMillis; + + public DeleteRuns setExperimentId(String experimentId) { + this.experimentId = experimentId; + return this; + } + + public String getExperimentId() { + return experimentId; + } + + public DeleteRuns setMaxRuns(Long maxRuns) { + this.maxRuns = maxRuns; + return this; + } + + public Long getMaxRuns() { + return maxRuns; + } + + public DeleteRuns setMaxTimestampMillis(Long maxTimestampMillis) { + this.maxTimestampMillis = maxTimestampMillis; + return this; + } + + public Long getMaxTimestampMillis() { + return maxTimestampMillis; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + DeleteRuns that = (DeleteRuns) o; + return Objects.equals(experimentId, that.experimentId) + && Objects.equals(maxRuns, that.maxRuns) + && Objects.equals(maxTimestampMillis, that.maxTimestampMillis); + } + + @Override + public int hashCode() { + return Objects.hash(experimentId, maxRuns, maxTimestampMillis); + } + + @Override + public String toString() { + return new ToStringer(DeleteRuns.class) + .add("experimentId", experimentId) + .add("maxRuns", maxRuns) + .add("maxTimestampMillis", maxTimestampMillis) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/DeleteRunsResponse.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/DeleteRunsResponse.java new file mode 100755 index 000000000..92aa1f824 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/DeleteRunsResponse.java @@ -0,0 +1,42 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.ml; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +@Generated +public class DeleteRunsResponse { + /** The number of runs deleted. */ + @JsonProperty("runs_deleted") + private Long runsDeleted; + + public DeleteRunsResponse setRunsDeleted(Long runsDeleted) { + this.runsDeleted = runsDeleted; + return this; + } + + public Long getRunsDeleted() { + return runsDeleted; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + DeleteRunsResponse that = (DeleteRunsResponse) o; + return Objects.equals(runsDeleted, that.runsDeleted); + } + + @Override + public int hashCode() { + return Objects.hash(runsDeleted); + } + + @Override + public String toString() { + return new ToStringer(DeleteRunsResponse.class).add("runsDeleted", runsDeleted).toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsAPI.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsAPI.java index 479df68b8..54c08295b 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsAPI.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsAPI.java @@ -87,6 +87,21 @@ public void deleteRun(DeleteRun request) { impl.deleteRun(request); } + public DeleteRunsResponse deleteRuns(String experimentId, long maxTimestampMillis) { + return deleteRuns( + new DeleteRuns().setExperimentId(experimentId).setMaxTimestampMillis(maxTimestampMillis)); + } + + /** + * Delete runs by creation time. + * + *
Bulk delete runs in an experiment that were created prior to or at the specified timestamp. + * Deletes at most max_runs per request. + */ + public DeleteRunsResponse deleteRuns(DeleteRuns request) { + return impl.deleteRuns(request); + } + public void deleteTag(String runId, String key) { deleteTag(new DeleteTag().setRunId(runId).setKey(key)); } @@ -355,6 +370,21 @@ public void restoreRun(RestoreRun request) { impl.restoreRun(request); } + public RestoreRunsResponse restoreRuns(String experimentId, long minTimestampMillis) { + return restoreRuns( + new RestoreRuns().setExperimentId(experimentId).setMinTimestampMillis(minTimestampMillis)); + } + + /** + * Restore runs by deletion time. + * + *
Bulk restore runs in an experiment that were deleted no earlier than the specified + * timestamp. Restores at most max_runs per request. + */ + public RestoreRunsResponse restoreRuns(RestoreRuns request) { + return impl.restoreRuns(request); + } + /** * Search experiments. * diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsImpl.java index dc5fb0f8c..1d712d0e4 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsImpl.java @@ -37,6 +37,12 @@ public void deleteRun(DeleteRun request) { apiClient.POST(path, request, Void.class); } + @Override + public DeleteRunsResponse deleteRuns(DeleteRuns request) { + String path = "/api/2.0/mlflow/databricks/runs/delete-runs"; + return apiClient.POST(path, request, DeleteRunsResponse.class); + } + @Override public void deleteTag(DeleteTag request) { String path = "/api/2.0/mlflow/runs/delete-tag"; @@ -136,6 +142,12 @@ public void restoreRun(RestoreRun request) { apiClient.POST(path, request, Void.class); } + @Override + public RestoreRunsResponse restoreRuns(RestoreRuns request) { + String path = "/api/2.0/mlflow/databricks/runs/restore-runs"; + return apiClient.POST(path, request, RestoreRunsResponse.class); + } + @Override public SearchExperimentsResponse searchExperiments(SearchExperiments request) { String path = "/api/2.0/mlflow/experiments/search"; diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsService.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsService.java index 9d3a983e8..3d28056e2 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsService.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsService.java @@ -53,6 +53,14 @@ public interface ExperimentsService { */ void deleteRun(DeleteRun deleteRun); + /** + * Delete runs by creation time. + * + *
Bulk delete runs in an experiment that were created prior to or at the specified timestamp. + * Deletes at most max_runs per request. + */ + DeleteRunsResponse deleteRuns(DeleteRuns deleteRuns); + /** * Delete a tag. * @@ -224,6 +232,14 @@ ExperimentPermissions getExperimentPermissions( */ void restoreRun(RestoreRun restoreRun); + /** + * Restore runs by deletion time. + * + *
Bulk restore runs in an experiment that were deleted no earlier than the specified + * timestamp. Restores at most max_runs per request. + */ + RestoreRunsResponse restoreRuns(RestoreRuns restoreRuns); + /** * Search experiments. * diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/RestoreRuns.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/RestoreRuns.java new file mode 100755 index 000000000..0ce329136 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/RestoreRuns.java @@ -0,0 +1,80 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.ml; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +@Generated +public class RestoreRuns { + /** The ID of the experiment containing the runs to restore. */ + @JsonProperty("experiment_id") + private String experimentId; + + /** + * An optional positive integer indicating the maximum number of runs to restore. The maximum + * allowed value for max_runs is 10000. + */ + @JsonProperty("max_runs") + private Long maxRuns; + + /** + * The minimum deletion timestamp in milliseconds since the UNIX epoch for restoring runs. Only + * runs deleted no earlier than this timestamp are restored. + */ + @JsonProperty("min_timestamp_millis") + private Long minTimestampMillis; + + public RestoreRuns setExperimentId(String experimentId) { + this.experimentId = experimentId; + return this; + } + + public String getExperimentId() { + return experimentId; + } + + public RestoreRuns setMaxRuns(Long maxRuns) { + this.maxRuns = maxRuns; + return this; + } + + public Long getMaxRuns() { + return maxRuns; + } + + public RestoreRuns setMinTimestampMillis(Long minTimestampMillis) { + this.minTimestampMillis = minTimestampMillis; + return this; + } + + public Long getMinTimestampMillis() { + return minTimestampMillis; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + RestoreRuns that = (RestoreRuns) o; + return Objects.equals(experimentId, that.experimentId) + && Objects.equals(maxRuns, that.maxRuns) + && Objects.equals(minTimestampMillis, that.minTimestampMillis); + } + + @Override + public int hashCode() { + return Objects.hash(experimentId, maxRuns, minTimestampMillis); + } + + @Override + public String toString() { + return new ToStringer(RestoreRuns.class) + .add("experimentId", experimentId) + .add("maxRuns", maxRuns) + .add("minTimestampMillis", minTimestampMillis) + .toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/RestoreRunsResponse.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/RestoreRunsResponse.java new file mode 100755 index 000000000..45b2a692f --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/RestoreRunsResponse.java @@ -0,0 +1,42 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.ml; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +@Generated +public class RestoreRunsResponse { + /** The number of runs restored. */ + @JsonProperty("runs_restored") + private Long runsRestored; + + public RestoreRunsResponse setRunsRestored(Long runsRestored) { + this.runsRestored = runsRestored; + return this; + } + + public Long getRunsRestored() { + return runsRestored; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + RestoreRunsResponse that = (RestoreRunsResponse) o; + return Objects.equals(runsRestored, that.runsRestored); + } + + @Override + public int hashCode() { + return Objects.hash(runsRestored); + } + + @Override + public String toString() { + return new ToStringer(RestoreRunsResponse.class).add("runsRestored", runsRestored).toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/ChannelName.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/ChannelName.java index 8e0f647a1..3b9d4cbb0 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/ChannelName.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/ChannelName.java @@ -4,7 +4,6 @@ import com.databricks.sdk.support.Generated; -/** Name of the channel */ @Generated public enum ChannelName { CHANNEL_NAME_CURRENT, diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/GetSecretRequest.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/GetSecretRequest.java new file mode 100755 index 000000000..9eedfcfac --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/GetSecretRequest.java @@ -0,0 +1,56 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.workspace; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.QueryParam; +import com.databricks.sdk.support.ToStringer; +import java.util.Objects; + +/** Get a secret */ +@Generated +public class GetSecretRequest { + /** The key to fetch secret for. */ + @QueryParam("key") + private String key; + + /** The name of the scope to fetch secret information from. */ + @QueryParam("scope") + private String scope; + + public GetSecretRequest setKey(String key) { + this.key = key; + return this; + } + + public String getKey() { + return key; + } + + public GetSecretRequest setScope(String scope) { + this.scope = scope; + return this; + } + + public String getScope() { + return scope; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + GetSecretRequest that = (GetSecretRequest) o; + return Objects.equals(key, that.key) && Objects.equals(scope, that.scope); + } + + @Override + public int hashCode() { + return Objects.hash(key, scope); + } + + @Override + public String toString() { + return new ToStringer(GetSecretRequest.class).add("key", key).add("scope", scope).toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/GetSecretResponse.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/GetSecretResponse.java new file mode 100755 index 000000000..e763f9ea6 --- /dev/null +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/GetSecretResponse.java @@ -0,0 +1,55 @@ +// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. + +package com.databricks.sdk.service.workspace; + +import com.databricks.sdk.support.Generated; +import com.databricks.sdk.support.ToStringer; +import com.fasterxml.jackson.annotation.JsonProperty; +import java.util.Objects; + +@Generated +public class GetSecretResponse { + /** A unique name to identify the secret. */ + @JsonProperty("key") + private String key; + + /** The value of the secret in its byte representation. */ + @JsonProperty("value") + private String value; + + public GetSecretResponse setKey(String key) { + this.key = key; + return this; + } + + public String getKey() { + return key; + } + + public GetSecretResponse setValue(String value) { + this.value = value; + return this; + } + + public String getValue() { + return value; + } + + @Override + public boolean equals(Object o) { + if (this == o) return true; + if (o == null || getClass() != o.getClass()) return false; + GetSecretResponse that = (GetSecretResponse) o; + return Objects.equals(key, that.key) && Objects.equals(value, that.value); + } + + @Override + public int hashCode() { + return Objects.hash(key, value); + } + + @Override + public String toString() { + return new ToStringer(GetSecretResponse.class).add("key", key).add("value", value).toString(); + } +} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/SecretsAPI.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/SecretsAPI.java index d2c4ca1ac..5786ad7e2 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/SecretsAPI.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/SecretsAPI.java @@ -114,6 +114,27 @@ public AclItem getAcl(GetAclRequest request) { return impl.getAcl(request); } + public GetSecretResponse getSecret(String scope, String key) { + return getSecret(new GetSecretRequest().setScope(scope).setKey(key)); + } + + /** + * Get a secret. + * + *
Gets the bytes representation of a secret value for the specified scope and key. + * + *
Users need the READ permission to make this call. + * + *
Note that the secret value returned is in bytes. The interpretation of the bytes is + * determined by the caller in DBUtils and the type the data is decoded into. + * + *
Throws ``PERMISSION_DENIED`` if the user does not have permission to make this API call.
+ * Throws ``RESOURCE_DOES_NOT_EXIST`` if no such secret or secret scope exists.
+ */
+ public GetSecretResponse getSecret(GetSecretRequest request) {
+ return impl.getSecret(request);
+ }
+
public Iterable Gets the bytes representation of a secret value for the specified scope and key.
+ *
+ * Users need the READ permission to make this call.
+ *
+ * Note that the secret value returned is in bytes. The interpretation of the bytes is
+ * determined by the caller in DBUtils and the type the data is decoded into.
+ *
+ * Throws ``PERMISSION_DENIED`` if the user does not have permission to make this API call.
+ * Throws ``RESOURCE_DOES_NOT_EXIST`` if no such secret or secret scope exists.
+ */
+ GetSecretResponse getSecret(GetSecretRequest getSecretRequest);
+
/**
* Lists ACLs.
*
diff --git a/databricks-sdk-java/src/test/java/com/databricks/sdk/integration/SecretsIT.java b/databricks-sdk-java/src/test/java/com/databricks/sdk/integration/SecretsIT.java
new file mode 100644
index 000000000..096d5cd72
--- /dev/null
+++ b/databricks-sdk-java/src/test/java/com/databricks/sdk/integration/SecretsIT.java
@@ -0,0 +1,93 @@
+package com.databricks.sdk.integration;
+
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+import com.databricks.sdk.WorkspaceClient;
+import com.databricks.sdk.integration.framework.EnvContext;
+import com.databricks.sdk.integration.framework.EnvTest;
+import com.databricks.sdk.mixin.SecretsExt;
+import com.databricks.sdk.service.workspace.PutSecret;
+import com.databricks.sdk.service.workspace.SecretMetadata;
+import com.databricks.sdk.service.workspace.SecretScope;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+
+@EnvContext("workspace")
+@ExtendWith(EnvTest.class)
+public class SecretsIT {
+ @Test
+ void createAndGetSecretsTest(WorkspaceClient w) {
+ SecretsExt secretsExt = w.secrets();
+ int randomSuffix = (int) (Math.random() * 1000);
+ String scope = "testScope-" + randomSuffix;
+ String key = "testKey-" + randomSuffix;
+ String value = "testValue-" + randomSuffix;
+
+ try (SecretsTestResource ignored = SecretsTestResource.makeScope(secretsExt, scope)) {
+
+ try (SecretsTestResource ignored1 =
+ SecretsTestResource.makeSecret(secretsExt, scope, key, value)) {
+
+ Iterable