Esempio n. 1
0
func NewAwsAsgEbs(maxRetries int) *AwsAsgEbs {
	awsAsgEbs := &AwsAsgEbs{}

	metadata := ec2metadata.New(session.New())

	region, err := metadata.Region()
	if err != nil {
		log.WithFields(log.Fields{"error": err}).Fatal("Failed to get region from instance metadata")
	}
	log.WithFields(log.Fields{"region": region}).Info("Setting region")
	awsAsgEbs.Region = region

	availabilityZone, err := metadata.GetMetadata("placement/availability-zone")
	if err != nil {
		log.WithFields(log.Fields{"error": err}).Fatal("Failed to get availability zone from instance metadata")
	}
	log.WithFields(log.Fields{"az": availabilityZone}).Info("Setting availability zone")
	awsAsgEbs.AvailabilityZone = availabilityZone

	instanceId, err := metadata.GetMetadata("instance-id")
	if err != nil {
		log.WithFields(log.Fields{"error": err}).Fatal("Failed to get instance id from instance metadata")
	}
	log.WithFields(log.Fields{"instance_id": instanceId}).Info("Setting instance id")
	awsAsgEbs.InstanceId = instanceId

	awsAsgEbs.AwsConfig = aws.NewConfig().
		WithRegion(region).
		WithCredentials(ec2rolecreds.NewCredentials(session.New())).
		WithMaxRetries(maxRetries)

	return awsAsgEbs
}
Esempio n. 2
0
func getFromS3() error {

	// creds := &ec2rolecreds.EC2RoleProvider{
	// 	client: &http.Client{Timeout: 10 * time.Second},
	// 	window: 0,
	// }

	s3svc := s3.New(&aws.Config{
		Region:      aws.String(s3Region),
		Credentials: ec2rolecreds.NewCredentials(nil, 5*time.Minute),
	})

	params := &s3.GetObjectInput{
		Bucket: aws.String(p.bucket),
		Key:    aws.String(p.path),
	}
	resp, err := s3svc.GetObject(params)
	if err != nil {
		return err
	}

	outputPath := filepath.Join(p.outputDir, getFilename(p.path))
	artifact, err := os.Create(outputPath)
	if err != nil {
		return err
	}
	defer artifact.Close()

	if _, err := io.Copy(artifact, resp.Body); err != nil {
		return err
	}

	return nil
}