r/sagemaker_domain - new resource

aws/internal/service/sagemaker/finder/finder.go

@@ -42,3 +42,22 @@ func ImageByName(conn *sagemaker.SageMaker, name string) (*sagemaker.DescribeIma
 
 	return output, nil
 }
+
+// DomainByName returns the domain corresponding to the specified domain id.
+// Returns nil if no domain is found.
+func DomainByName(conn *sagemaker.SageMaker, domainID string) (*sagemaker.DescribeDomainOutput, error) {
+	input := &sagemaker.DescribeDomainInput{
+		DomainId: aws.String(domainID),
+	}
+
+	output, err := conn.DescribeDomain(input)
+	if err != nil {
+		return nil, err
+	}
+
+	if output == nil {
+		return nil, nil
+	}
+
+	return output, nil
+}

aws/internal/service/sagemaker/waiter/status.go

@@ -13,6 +13,7 @@ const (
 	SagemakerNotebookInstanceStatusNotFound = "NotFound"
 	SagemakerImageStatusNotFound            = "NotFound"
 	SagemakerImageStatusFailed              = "Failed"
+	SagemakerDomainStatusNotFound           = "NotFound"
 )
 
 // NotebookInstanceStatus fetches the NotebookInstance and its Status
@@ -68,3 +69,28 @@ func ImageStatus(conn *sagemaker.SageMaker, name string) resource.StateRefreshFu
 		return output, aws.StringValue(output.ImageStatus), nil
 	}
 }
+
+// DomainStatus fetches the Domain and its Status
+func DomainStatus(conn *sagemaker.SageMaker, domainID string) resource.StateRefreshFunc {
+	return func() (interface{}, string, error) {
+		input := &sagemaker.DescribeDomainInput{
+			DomainId: aws.String(domainID),
+		}
+
+		output, err := conn.DescribeDomain(input)
+
+		if tfawserr.ErrMessageContains(err, "ValidationException", "RecordNotFound") {
+			return nil, SagemakerDomainStatusNotFound, nil
+		}
+
+		if err != nil {
+			return nil, sagemaker.DomainStatusFailed, err
+		}
+
+		if output == nil {
+			return nil, SagemakerDomainStatusNotFound, nil
+		}
+
+		return output, aws.StringValue(output.Status), nil
+	}
+}

aws/internal/service/sagemaker/waiter/waiter.go

@@ -13,6 +13,8 @@ const (
 	NotebookInstanceDeletedTimeout   = 10 * time.Minute
 	ImageCreatedTimeout              = 10 * time.Minute
 	ImageDeletedTimeout              = 10 * time.Minute
+	DomainInServiceTimeout           = 10 * time.Minute
+	DomainDeletedTimeout             = 10 * time.Minute
 )
 
 // NotebookInstanceInService waits for a NotebookInstance to return InService
@@ -117,3 +119,44 @@ func ImageDeleted(conn *sagemaker.SageMaker, name string) (*sagemaker.DescribeIm
 
 	return nil, err
 }
+
+// DomainInService waits for a Domain to return InService
+func DomainInService(conn *sagemaker.SageMaker, domainID string) (*sagemaker.DescribeDomainOutput, error) {
+	stateConf := &resource.StateChangeConf{
+		Pending: []string{
+			SagemakerDomainStatusNotFound,
+			sagemaker.DomainStatusPending,
+		},
+		Target:  []string{sagemaker.DomainStatusInService},
+		Refresh: DomainStatus(conn, domainID),
+		Timeout: DomainInServiceTimeout,
+	}
+
+	outputRaw, err := stateConf.WaitForState()
+
+	if output, ok := outputRaw.(*sagemaker.DescribeDomainOutput); ok {
+		return output, err
+	}
+
+	return nil, err
+}
+
+// DomainDeleted waits for a Domain to return Deleted
+func DomainDeleted(conn *sagemaker.SageMaker, domainID string) (*sagemaker.DescribeDomainOutput, error) {
+	stateConf := &resource.StateChangeConf{
+		Pending: []string{
+			sagemaker.DomainStatusDeleting,
+		},
+		Target:  []string{},
+		Refresh: DomainStatus(conn, domainID),
+		Timeout: DomainDeletedTimeout,
+	}
+
+	outputRaw, err := stateConf.WaitForState()
+
+	if output, ok := outputRaw.(*sagemaker.DescribeDomainOutput); ok {
+		return output, err
+	}
+
+	return nil, err
+}

aws/provider.go

@@ -870,6 +870,7 @@ func Provider() *schema.Provider {
 			"aws_default_route_table":                                 resourceAwsDefaultRouteTable(),
 			"aws_route_table_association":                             resourceAwsRouteTableAssociation(),
 			"aws_sagemaker_code_repository":                           resourceAwsSagemakerCodeRepository(),
+			"aws_sagemaker_domain":                                    resourceAwsSagemakerDomain(),
 			"aws_sagemaker_endpoint_configuration":                    resourceAwsSagemakerEndpointConfiguration(),
 			"aws_sagemaker_image":                                     resourceAwsSagemakerImage(),
 			"aws_sagemaker_endpoint":                                  resourceAwsSagemakerEndpoint(),