From 7d435671b92f139c28a17a16fb3146cf75dccf3f Mon Sep 17 00:00:00 2001
From: Don Cote <mistercote@gmail.com>
Date: Tue, 29 Jun 2021 12:32:57 -0400
Subject: [PATCH] add docker image to cluster

---
 aws/models/ClusterInfo.go |  1 +
 aws/models/DockerImage.go | 11 +++++++++++
 aws/models/NewCluster.go  |  1 +
 version.go                |  2 +-
 4 files changed, 14 insertions(+), 1 deletion(-)
 create mode 100644 aws/models/DockerImage.go

diff --git a/aws/models/ClusterInfo.go b/aws/models/ClusterInfo.go
index e36ab94..efe1a86 100644
--- a/aws/models/ClusterInfo.go
+++ b/aws/models/ClusterInfo.go
@@ -34,4 +34,5 @@ type ClusterInfo struct {
 	DefaultTags            []ClusterTag      `json:"default_tags,omitempty" url:"default_tags,omitempty"`
 	ClusterLogStatus       *LogSyncStatus    `json:"cluster_log_status,omitempty" url:"cluster_log_status,omitempty"`
 	TerminationReason      *S3StorageInfo    `json:"termination_reason,omitempty" url:"termination_reason,omitempty"`
+	DockerImage            *DockerImage      `json:"docker_image,omitempty" url:"docker_image,omitempty"`
 }
diff --git a/aws/models/DockerImage.go b/aws/models/DockerImage.go
new file mode 100644
index 0000000..01ac8c8
--- /dev/null
+++ b/aws/models/DockerImage.go
@@ -0,0 +1,11 @@
+package models
+
+type DockerImage struct {
+	URL       string     `json:"url,omitempty" url:"url,omitempty"`
+	BasicAuth *BasicAuth `json:"basic_auth,omitempty" url:"basic_auth,omitempty"`
+}
+
+type BasicAuth struct {
+	Username string `json:"username,omitempty" url:"username,omitempty"`
+	Password string `json:"password,omitempty" url:"password,omitempty"`
+}
diff --git a/aws/models/NewCluster.go b/aws/models/NewCluster.go
index 8228d2f..6e8c4b0 100644
--- a/aws/models/NewCluster.go
+++ b/aws/models/NewCluster.go
@@ -15,4 +15,5 @@ type NewCluster struct {
 	InitScripts       []InitScriptInfo  `json:"init_scripts,omitempty" url:"init_scripts,omitempty"`
 	SparkEnvVars      map[string]string `json:"spark_env_vars,omitempty" url:"spark_env_vars,omitempty"`
 	EnableElasticDisk bool              `json:"enable_elastic_disk,omitempty" url:"enable_elastic_disk,omitempty"`
+	DockerImage       *DockerImage      `json:"docker_image,omitempty" url:"docker_image,omitempty"`
 }
diff --git a/version.go b/version.go
index 9c0de16..438eb53 100644
--- a/version.go
+++ b/version.go
@@ -4,5 +4,5 @@ const (
 	// APIVersion is the version of the RESTful API of DataBricks
 	APIVersion = "2.0"
 	// SdkVersion is the version of this library
-	SdkVersion = "0.1.4"
+	SdkVersion = "0.1.5"
 )