Golang EC2.DescribeVPCs方法代码示例

// VPCStateRefreshFunc returns a resource.StateRefreshFunc that is used to watch
// a VPC.
func VPCStateRefreshFunc(conn *ec2.EC2, id string) resource.StateRefreshFunc {
	return func() (interface{}, string, error) {
		DescribeVpcOpts := &ec2.DescribeVPCsInput{
			VPCIDs: []*string{aws.String(id)},
		}
		resp, err := conn.DescribeVPCs(DescribeVpcOpts)
		if err != nil {
			if ec2err, ok := err.(awserr.Error); ok && ec2err.Code() == "InvalidVpcID.NotFound" {
				resp = nil
			} else {
				log.Printf("Error on VPCStateRefresh: %s", err)
				return nil, "", err
			}
		}

		if resp == nil {
			// Sometimes AWS just has consistency issues and doesn't see
			// our instance yet. Return an empty state.
			return nil, "", nil
		}

		vpc := resp.VPCs[0]
		return vpc, *vpc.State, nil
	}
}

// Get all of the VPCs configured in the environment
func getAllVPCs(ec2client *ec2.EC2) ([]*ec2.VPC, error) {
	//Get all of the VPCs
	vpcs, err := ec2client.DescribeVPCs(&ec2.DescribeVPCsInput{})

	//If we had an error, return it
	if err != nil {
		return []*ec2.VPC{}, err
	}

	//Otherwise, return all of our VPCs
	return vpcs.VPCs, nil
}

/*
validateVPC will query ec2 and see if the VPC exists. If it does not exist, then if
createIfNotFound is true, it will attempt to create it
*/
func validateVPC(ec2Client *ec2.EC2, providerConfig *providers.JSONObject, createIfNotFound bool) (success bool, warnings []string, err []error) {
	vpcName := providerConfig.Provider.ProviderConfig.Aws.Vpc.Name
	log.WithFields(log.Fields{
		"VPC Name": vpcName,
	}).Info("Validating if VPC exists")
	//See if VPC exists
	filters := []*ec2.Filter{newEc2Filter("tag:"+nameTag, vpcName)}
	params := &ec2.DescribeVPCsInput{Filters: filters}
	vpcQueryOutput, callErr := ec2Client.DescribeVPCs(params)

	if err != nil {
		err = append(err, callErr)
		log.WithFields(log.Fields{
			"AWS Error": err,
		}).Error("Error Querying VPC")
	} else {
		if len(vpcQueryOutput.VPCs) > 0 {
			//Found VPC
			log.WithFields(log.Fields{
				"VpcID": providerConfig.Provider.ProviderConfig.Aws.Vpc.VpcID,
			}).Info("VPC Found")
			providerConfig.Provider.ProviderConfig.Aws.Vpc.VpcID = *vpcQueryOutput.VPCs[0].VPCID
			success = true
		} else {
			if createIfNotFound {
				//create new VPC
				createSuccess, createWarning, createError := addVPC(ec2Client, providerConfig)
				if createSuccess {
					success = true
					for w := range createWarning {
						warnings = append(warnings, createWarning[w])
					}
				} else {
					success = false
					for e := range createError {
						err = append(err, createError[e])
					}
				}
			} else {
				//VPC not found and createIfNotFound is false
				err = append(err, errors.New("VPC does not exist and create if not found is false"))
				success = false
			}
		}
	}

	return success, warnings, err
}

func findVPCsByDHCPOptionsID(conn *ec2.EC2, id string) ([]*ec2.VPC, error) {
	req := &ec2.DescribeVPCsInput{
		Filters: []*ec2.Filter{
			&ec2.Filter{
				Name: aws.String("dhcp-options-id"),
				Values: []*string{
					aws.String(id),
				},
			},
		},
	}

	resp, err := conn.DescribeVPCs(req)
	if err != nil {
		if ec2err, ok := err.(awserr.Error); ok && ec2err.Code() == "InvalidVpcID.NotFound" {
			return nil, nil
		}
		return nil, err
	}

	return resp.VPCs, nil
}