From 0b201d9648e4518afebb80c75b14835d1e07fe00 Mon Sep 17 00:00:00 2001
From: Yarden Sachs <yrdnsa@gmail.com>
Date: Fri, 2 Oct 2020 13:24:07 -0700
Subject: [PATCH] provider: Add TimestreamWrite service client (#15421)
 (#15451)

* Added TimestreamWrite Service

* Update website/allowed-subcategories.txt

Co-authored-by: Brian Flad <bflad417@gmail.com>

Co-authored-by: Brian Flad <bflad417@gmail.com>
---
 .hashibot.hcl                                        | 8 ++++++++
 aws/config.go                                        | 3 +++
 aws/provider.go                                      | 1 +
 infrastructure/repository/labels-service.tf          | 1 +
 website/allowed-subcategories.txt                    | 1 +
 website/docs/guides/custom-service-endpoints.html.md | 1 +
 6 files changed, 15 insertions(+)

diff --git a/.hashibot.hcl b/.hashibot.hcl
index 2b8ce42d2a52..929de477f173 100644
--- a/.hashibot.hcl
+++ b/.hashibot.hcl
@@ -548,6 +548,9 @@ behavior "regexp_issue_labeler_v2" "service_labels" {
     "service/synthetics" = [
       "aws_synthetics_",
     ],
+    "service/timestreamwrite" = [
+      "aws_timestreamwrite_",
+    ],
     "service/transfer" = [
       "aws_transfer_",
     ],
@@ -1428,6 +1431,11 @@ behavior "pull_request_path_labeler" "service_labels" {
       "**/*_synthetics_*",
       "**/synthetics_*"
     ]
+    "service/timestreamwrite" = [
+      "aws/internal/service/timestreamwrite/**/*",
+      "**/*_timestreamwrite_*",
+      "**/timestreamwrite_*"
+    ]
     "service/transfer" = [
       "aws/internal/service/transfer/**/*",
       "**/*_transfer_*",
diff --git a/aws/config.go b/aws/config.go
index 2b2cd6680027..92cbb2cb8e4c 100644
--- a/aws/config.go
+++ b/aws/config.go
@@ -146,6 +146,7 @@ import (
 	"github.com/aws/aws-sdk-go/service/sts"
 	"github.com/aws/aws-sdk-go/service/swf"
 	"github.com/aws/aws-sdk-go/service/synthetics"
+	"github.com/aws/aws-sdk-go/service/timestreamwrite"
 	"github.com/aws/aws-sdk-go/service/transfer"
 	"github.com/aws/aws-sdk-go/service/waf"
 	"github.com/aws/aws-sdk-go/service/wafregional"
@@ -341,6 +342,7 @@ type AWSClient struct {
 	swfconn                             *swf.SWF
 	syntheticsconn                      *synthetics.Synthetics
 	terraformVersion                    string
+	timestreamwriteconn                 *timestreamwrite.TimestreamWrite
 	transferconn                        *transfer.Transfer
 	wafconn                             *waf.WAF
 	wafregionalconn                     *wafregional.WAFRegional
@@ -566,6 +568,7 @@ func (c *Config) Client() (interface{}, error) {
 		swfconn:                             swf.New(sess.Copy(&aws.Config{Endpoint: aws.String(c.Endpoints["swf"])})),
 		syntheticsconn:                      synthetics.New(sess.Copy(&aws.Config{Endpoint: aws.String(c.Endpoints["synthetics"])})),
 		terraformVersion:                    c.terraformVersion,
+		timestreamwriteconn:                 timestreamwrite.New(sess.Copy(&aws.Config{Endpoint: aws.String(c.Endpoints["timestreamwrite"])})),
 		transferconn:                        transfer.New(sess.Copy(&aws.Config{Endpoint: aws.String(c.Endpoints["transfer"])})),
 		wafconn:                             waf.New(sess.Copy(&aws.Config{Endpoint: aws.String(c.Endpoints["waf"])})),
 		wafregionalconn:                     wafregional.New(sess.Copy(&aws.Config{Endpoint: aws.String(c.Endpoints["wafregional"])})),
diff --git a/aws/provider.go b/aws/provider.go
index fa2ca76cef64..4040c614d914 100644
--- a/aws/provider.go
+++ b/aws/provider.go
@@ -1195,6 +1195,7 @@ func init() {
 		"sts",
 		"swf",
 		"synthetics",
+		"timestreamwrite",
 		"transfer",
 		"waf",
 		"wafregional",
diff --git a/infrastructure/repository/labels-service.tf b/infrastructure/repository/labels-service.tf
index 25ee87f67d9b..c5452e30444a 100644
--- a/infrastructure/repository/labels-service.tf
+++ b/infrastructure/repository/labels-service.tf
@@ -185,6 +185,7 @@ variable "service_labels" {
     "synthetics",
     "textract",
     "transcribeservice",
+    "timestreamwrite",
     "transfer",
     "translate",
     "waf",
diff --git a/website/allowed-subcategories.txt b/website/allowed-subcategories.txt
index d558f67296cd..a8e488c776da 100644
--- a/website/allowed-subcategories.txt
+++ b/website/allowed-subcategories.txt
@@ -110,6 +110,7 @@ SimpleDB
 Step Function (SFN)
 Storage Gateway
 Synthetics
+Timestream Write
 Transfer
 Transit Gateway Network Manager
 VPC
diff --git a/website/docs/guides/custom-service-endpoints.html.md b/website/docs/guides/custom-service-endpoints.html.md
index 583ea9b0d558..ddad5754cd90 100644
--- a/website/docs/guides/custom-service-endpoints.html.md
+++ b/website/docs/guides/custom-service-endpoints.html.md
@@ -189,6 +189,7 @@ The Terraform AWS Provider allows the following endpoints to be customized:
   <li><code>sts</code></li>
   <li><code>swf</code></li>
   <li><code>synthetics</code></li>
+  <li><code>timestreamwrite</code></li>
   <li><code>transfer</code></li>
   <li><code>waf</code></li>
   <li><code>wafregional</code></li>