ResiLien
/
terraform
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remote/atlas: if `ATLAS_RUN_ID` is set, send it with remote state save 

This detects the presence of ATLAS_RUN_ID in the environment and sends
it if exists with remote state PUT requests with query params.
This commit is contained in:
Jack Pearkes 2015-08-20 13:49:51 -07:00
parent d19ec9d57a
commit eba638d044
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
state/remote/atlas.go
View File

@ -51,6 +51,11 @@ func atlasFactory(conf map[string]string) (Client, error) {
return nil, fmt.Errorf("malformed name '%s', expected format '<account>/<name>'", name) return nil, fmt.Errorf("malformed name '%s', expected format '<account>/<name>'", name)
} }
// If it exists, add the `ATLAS_RUN_ID` environment
// variable as a param, which is injected during Atlas Terraform
// runs. This is completely optional.
client.RunId = os.Getenv("ATLAS_RUN_ID")
client.Server = server client.Server = server
client.ServerURL = url client.ServerURL = url
client.AccessToken = token client.AccessToken = token
@ -67,6 +72,7 @@ type AtlasClient struct {
User string User string
Name string Name string
AccessToken string AccessToken string
RunId string
} }
func (c *AtlasClient) Get() (*Payload, error) { func (c *AtlasClient) Get() (*Payload, error) {
@ -218,10 +224,15 @@ func (c *AtlasClient) readBody(b io.Reader) string {
} }
func (c *AtlasClient) url() *url.URL { func (c *AtlasClient) url() *url.URL {
values := url.Values{}
values.Add("atlas_run_id", c.RunId)
values.Add("access_token", c.AccessToken)
return &url.URL{ return &url.URL{
Scheme: c.ServerURL.Scheme, Scheme: c.ServerURL.Scheme,
Host: c.ServerURL.Host, Host: c.ServerURL.Host,
Path: path.Join("api/v1/terraform/state", c.User, c.Name), Path: path.Join("api/v1/terraform/state", c.User, c.Name),
RawQuery: fmt.Sprintf("access_token=%s", c.AccessToken), RawQuery: values.Encode(),
} }
} }