provider/aws: allow local kinesis

This commit is contained in:
Garrett Heel 2015-09-16 09:58:46 -07:00
parent 95832c2fb2
commit 9c2725e0a5
3 changed files with 22 additions and 8 deletions

View File

@ -44,6 +44,7 @@ type Config struct {
ForbiddenAccountIds []interface{}
DynamoDBEndpoint string
KinesisEndpoint string
}
type AWSClient struct {
@ -108,12 +109,6 @@ func (c *Config) Client() (interface{}, error) {
errs = append(errs, err)
}
awsDynamoDBConfig := &aws.Config{
Credentials: creds,
Region: aws.String(c.Region),
MaxRetries: aws.Int(c.MaxRetries),
Endpoint: aws.String(c.DynamoDBEndpoint),
}
// Some services exist only in us-east-1, e.g. because they manage
// resources that can span across multiple regions, or because
// signature format v4 requires region to be us-east-1 for global
@ -125,8 +120,11 @@ func (c *Config) Client() (interface{}, error) {
MaxRetries: aws.Int(c.MaxRetries),
}
awsDynamoDBConfig := *awsConfig
awsDynamoDBConfig.Endpoint = aws.String(c.DynamoDBEndpoint)
log.Println("[INFO] Initializing DynamoDB connection")
client.dynamodbconn = dynamodb.New(awsDynamoDBConfig)
client.dynamodbconn = dynamodb.New(&awsDynamoDBConfig)
log.Println("[INFO] Initializing ELB connection")
client.elbconn = elb.New(awsConfig)
@ -143,8 +141,11 @@ func (c *Config) Client() (interface{}, error) {
log.Println("[INFO] Initializing RDS Connection")
client.rdsconn = rds.New(awsConfig)
awsKinesisConfig := *awsConfig
awsKinesisConfig.Endpoint = aws.String(c.KinesisEndpoint)
log.Println("[INFO] Initializing Kinesis Connection")
client.kinesisconn = kinesis.New(awsConfig)
client.kinesisconn = kinesis.New(&awsKinesisConfig)
authErr := c.ValidateAccountId(client.iamconn)
if authErr != nil {

View File

@ -153,6 +153,13 @@ func Provider() terraform.ResourceProvider {
Default: "",
Description: descriptions["dynamodb_endpoint"],
},
"kinesis_endpoint": &schema.Schema{
Type: schema.TypeString,
Optional: true,
Default: "",
Description: descriptions["kinesis_endpoint"],
},
},
ResourcesMap: map[string]*schema.Resource{
@ -279,6 +286,9 @@ func init() {
"dynamodb_endpoint": "Use this to override the default endpoint URL constructed from the `region`.\n" +
"It's typically used to connect to dynamodb-local.",
"kinesis_endpoint": "Use this to override the default endpoint URL constructed from the `region`.\n" +
"It's typically used to connect to kinesalite.",
}
}
@ -290,6 +300,7 @@ func providerConfigure(d *schema.ResourceData) (interface{}, error) {
Region: d.Get("region").(string),
MaxRetries: d.Get("max_retries").(int),
DynamoDBEndpoint: d.Get("dynamodb_endpoint").(string),
KinesisEndpoint: d.Get("kinesis_endpoint").(string),
}
if v, ok := d.GetOk("allowed_account_ids"); ok {

View File

@ -57,5 +57,7 @@ The following arguments are supported in the `provider` block:
* `dynamodb_endpoint` - (Optional) Use this to override the default endpoint URL constructed from the `region`. It's typically used to connect to dynamodb-local.
* `kinesis_endpoint` - (Optional) Use this to override the default endpoint URL constructed from the `region`. It's typically used to connect to kinesalite.
In addition to the above parameters, the `AWS_SESSION_TOKEN` environmental
variable can be set to set an MFA token.