mirror of https://github.com/prometheus/prometheus
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
|
package ec2
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/aws/aws-sdk-go/aws/awsutil"
|
||
|
|
"github.com/aws/aws-sdk-go/aws/request"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
initRequest = func(r *request.Request) {
|
||
|
|
if r.Operation.Name == opCopySnapshot { // fill the PresignedURL parameter
|
||
|
|
r.Handlers.Build.PushFront(fillPresignedURL)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func fillPresignedURL(r *request.Request) {
|
||
|
|
if !r.ParamsFilled() {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
params := r.Params.(*CopySnapshotInput)
|
||
|
|
|
||
|
|
// Stop if PresignedURL/DestinationRegion is set
|
||
|
|
if params.PresignedUrl != nil || params.DestinationRegion != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// First generate a copy of parameters
|
||
|
|
r.Params = awsutil.CopyOf(r.Params)
|
||
|
|
params = r.Params.(*CopySnapshotInput)
|
||
|
|
|
||
|
|
// Set destination region. Avoids infinite handler loop.
|
||
|
|
// Also needed to sign sub-request.
|
||
|
|
params.DestinationRegion = r.Service.Config.Region
|
||
|
|
|
||
|
|
// Create a new client pointing at source region.
|
||
|
|
// We will use this to presign the CopySnapshot request against
|
||
|
|
// the source region
|
||
|
|
config := r.Service.Config.Copy().
|
||
|
|
WithEndpoint("").
|
||
|
|
WithRegion(*params.SourceRegion)
|
||
|
|
|
||
|
|
client := New(config)
|
||
|
|
|
||
|
|
// Presign a CopySnapshot request with modified params
|
||
|
|
req, _ := client.CopySnapshotRequest(params)
|
||
|
|
url, err := req.Presign(300 * time.Second) // 5 minutes should be enough.
|
||
|
|
|
||
|
|
if err != nil { // bubble error back up to original request
|
||
|
|
r.Error = err
|
||
|
|
}
|
||
|
|
|
||
|
|
// We have our URL, set it on params
|
||
|
|
params.PresignedUrl = &url
|
||
|
|
}
|