Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

feat: add aws_instance_types_from_instance_requirements data source #34030

Open
wants to merge 8 commits into
base: main
Choose a base branch
from
+1,635 −0
Open

feat: add aws_instance_types_from_instance_requirements data source #34030

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
06ef19c
feat: add aws_instance_types_from_instance_requirements data source
iandrewt Oct 20, 2023
bac53d6
chore: add changelog
iandrewt Oct 20, 2023
be2e1a4
chore: fix capitalisation, tests and documentation
iandrewt Oct 20, 2023
7ed8831
chore: fix capitalisation
iandrewt Oct 20, 2023
4e0c011
Merge remote-tracking branch 'upstream/main' into f-ec2-instance_type…
iandrewt May 5, 2024
423a0dc
chore: refactor instance_types_from_requirements to use aws sdk go v2
iandrewt May 6, 2024
9fb94e7
chore: fix semgrep failure
iandrewt May 6, 2024
97f1bd8
chore: fix service_package_gen
iandrewt May 6, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .changelog/34030.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
```release-note:new-data-source
aws_ec2_instance_types_from_instance_requirements
```
363 changes: 363 additions & 0 deletions internal/service/ec2/instance_types_from_instance_requirements_data_source.go
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,363 @@
// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0

package ec2

import (
"context"

"github.com/aws/aws-sdk-go-v2/service/ec2"
awstypes "github.com/aws/aws-sdk-go-v2/service/ec2/types"
"github.com/hashicorp/terraform-plugin-framework-validators/listvalidator"
"github.com/hashicorp/terraform-plugin-framework/datasource"
"github.com/hashicorp/terraform-plugin-framework/datasource/schema"
"github.com/hashicorp/terraform-plugin-framework/schema/validator"
"github.com/hashicorp/terraform-plugin-framework/types"
"github.com/hashicorp/terraform-provider-aws/internal/create"
"github.com/hashicorp/terraform-provider-aws/internal/enum"
"github.com/hashicorp/terraform-provider-aws/internal/framework"
"github.com/hashicorp/terraform-provider-aws/internal/framework/flex"
fwtypes "github.com/hashicorp/terraform-provider-aws/internal/framework/types"
"github.com/hashicorp/terraform-provider-aws/names"
)

// @FrameworkDataSource(name="Instance Types From Instance Requirements")
func newDataSourceInstanceTypesFromInstanceRequirements(context.Context) (datasource.DataSourceWithConfigure, error) {
return &dataSourceInstanceTypesFromInstanceRequirements{}, nil
}

const (
DSNameInstanceTypesFromInstanceRequirements = "Instance Types From Instance Requirements Data Source"
)

type dataSourceInstanceTypesFromInstanceRequirements struct {
framework.DataSourceWithConfigure
}

func (d *dataSourceInstanceTypesFromInstanceRequirements) Metadata(_ context.Context, req datasource.MetadataRequest, resp *datasource.MetadataResponse) { // nosemgrep:ci.meta-in-func-name
resp.TypeName = "aws_ec2_instance_types_from_instance_requirements"
}

func (d *dataSourceInstanceTypesFromInstanceRequirements) Schema(ctx context.Context, req datasource.SchemaRequest, resp *datasource.SchemaResponse) {
resp.Schema = schema.Schema{
Attributes: map[string]schema.Attribute{
"architecture_types": schema.ListAttribute{
CustomType: fwtypes.ListOfStringType,
ElementType: types.StringType,
Required: true,
Validators: []validator.List{
listvalidator.ValueStringsAre(
enum.FrameworkValidate[awstypes.ArchitectureType](),
),
},
},
"id": framework.IDAttribute(),
"instance_types": schema.ListAttribute{
ElementType: types.StringType,
Computed: true,
},
"virtualization_types": schema.ListAttribute{
CustomType: fwtypes.ListOfStringType,
ElementType: types.StringType,
Required: true,
Validators: []validator.List{
listvalidator.ValueStringsAre(
enum.FrameworkValidate[awstypes.VirtualizationType](),
),
},
},
},
Blocks: map[string]schema.Block{
"instance_requirements": schema.SingleNestedBlock{
CustomType: fwtypes.NewObjectTypeOf[instanceRequirementsData](ctx),
Attributes: map[string]schema.Attribute{
"accelerator_manufacturers": schema.ListAttribute{
CustomType: fwtypes.ListOfStringType,
ElementType: types.StringType,
Optional: true,
Validators: []validator.List{
listvalidator.ValueStringsAre(
enum.FrameworkValidate[awstypes.AcceleratorManufacturer](),
),
},
},
"accelerator_names": schema.ListAttribute{
CustomType: fwtypes.ListOfStringType,
ElementType: types.StringType,
Optional: true,
Validators: []validator.List{
listvalidator.ValueStringsAre(
enum.FrameworkValidate[awstypes.AcceleratorName](),
),
},
},
"accelerator_types": schema.ListAttribute{
CustomType: fwtypes.ListOfStringType,
ElementType: types.StringType,
Optional: true,
Validators: []validator.List{
listvalidator.ValueStringsAre(
enum.FrameworkValidate[awstypes.AcceleratorType](),
),
},
},
"allowed_instance_types": schema.ListAttribute{
CustomType: fwtypes.ListOfStringType,
ElementType: types.StringType,
Optional: true,
},
"bare_metal": schema.StringAttribute{
Optional: true,
Validators: []validator.String{
enum.FrameworkValidate[awstypes.BareMetal](),
},
},
"burstable_performance": schema.StringAttribute{
Optional: true,
Validators: []validator.String{
enum.FrameworkValidate[awstypes.BurstablePerformance](),
},
},
"cpu_manufacturers": schema.ListAttribute{
CustomType: fwtypes.ListOfStringType,
ElementType: types.StringType,
Optional: true,
Validators: []validator.List{
listvalidator.ValueStringsAre(
enum.FrameworkValidate[awstypes.CpuManufacturer](),
),
},
},
"excluded_instance_types": schema.ListAttribute{
CustomType: fwtypes.ListOfStringType,
ElementType: types.StringType,
Optional: true,
},
"instance_generations": schema.ListAttribute{
CustomType: fwtypes.ListOfStringType,
ElementType: types.StringType,
Optional: true,
Validators: []validator.List{
listvalidator.ValueStringsAre(
enum.FrameworkValidate[awstypes.InstanceGeneration](),
),
},
},
"local_storage": schema.StringAttribute{
Optional: true,
Validators: []validator.String{
enum.FrameworkValidate[awstypes.LocalStorage](),
},
},
"local_storage_types": schema.ListAttribute{
CustomType: fwtypes.ListOfStringType,
ElementType: types.StringType,
Optional: true,
Validators: []validator.List{
listvalidator.ValueStringsAre(
enum.FrameworkValidate[awstypes.LocalStorageType](),
),
},
},
"on_demand_max_price_percentage_over_lowest_price": schema.Int64Attribute{
Optional: true,
},
"require_hibernate_support": schema.BoolAttribute{
Optional: true,
},
"spot_max_price_percentage_over_lowest_price": schema.Int64Attribute{
Optional: true,
},
},
Blocks: map[string]schema.Block{
"memory_mib": schema.SingleNestedBlock{
CustomType: fwtypes.NewObjectTypeOf[minMax[types.Int64]](ctx),
Attributes: map[string]schema.Attribute{
"min": schema.Int64Attribute{
Required: true,
},
"max": schema.Int64Attribute{
Optional: true,
},
},
},
"vcpu_count": schema.SingleNestedBlock{
CustomType: fwtypes.NewObjectTypeOf[minMax[types.Int64]](ctx),
Attributes: map[string]schema.Attribute{
"min": schema.Int64Attribute{
Required: true,
},
"max": schema.Int64Attribute{
Optional: true,
},
},
},
"accelerator_count": schema.SingleNestedBlock{
CustomType: fwtypes.NewObjectTypeOf[minMax[types.Int64]](ctx),
Attributes: map[string]schema.Attribute{
"min": schema.Int64Attribute{
Optional: true,
},
"max": schema.Int64Attribute{
Optional: true,
},
},
},
"accelerator_total_memory_mib": schema.SingleNestedBlock{
CustomType: fwtypes.NewObjectTypeOf[minMax[types.Int64]](ctx),
Attributes: map[string]schema.Attribute{
"min": schema.Int64Attribute{
Optional: true,
},
"max": schema.Int64Attribute{
Optional: true,
},
},
},
"baseline_ebs_bandwidth_mbps": schema.SingleNestedBlock{
CustomType: fwtypes.NewObjectTypeOf[minMax[types.Float64]](ctx),
Attributes: map[string]schema.Attribute{
"min": schema.Int64Attribute{
Optional: true,
},
"max": schema.Int64Attribute{
Optional: true,
},
},
},
"memory_gib_per_vcpu": schema.SingleNestedBlock{
CustomType: fwtypes.NewObjectTypeOf[minMax[types.Float64]](ctx),
Attributes: map[string]schema.Attribute{
"min": schema.Float64Attribute{
Optional: true,
},
"max": schema.Float64Attribute{
Optional: true,
},
},
},
"network_bandwidth_gbps": schema.SingleNestedBlock{
CustomType: fwtypes.NewObjectTypeOf[minMax[types.Float64]](ctx),
Attributes: map[string]schema.Attribute{
"min": schema.Float64Attribute{
Optional: true,
},
"max": schema.Float64Attribute{
Optional: true,
},
},
},
"network_interface_count": schema.SingleNestedBlock{
CustomType: fwtypes.NewObjectTypeOf[minMax[types.Int64]](ctx),
Attributes: map[string]schema.Attribute{
"min": schema.Int64Attribute{
Optional: true,
},
"max": schema.Int64Attribute{
Optional: true,
},
},
},
"total_local_storage_gb": schema.SingleNestedBlock{
CustomType: fwtypes.NewObjectTypeOf[minMax[types.Float64]](ctx),
Attributes: map[string]schema.Attribute{
"min": schema.Float64Attribute{
Optional: true,
},
"max": schema.Float64Attribute{
Optional: true,
},
},
},
},
},
},
}
}

func (d *dataSourceInstanceTypesFromInstanceRequirements) Read(ctx context.Context, req datasource.ReadRequest, resp *datasource.ReadResponse) {
conn := d.Meta().EC2Client(ctx)

var data dataSourceInstanceTypesFromInstanceRequirementsData
resp.Diagnostics.Append(req.Config.Get(ctx, &data)...)
if resp.Diagnostics.HasError() {
return
}

input := &ec2.GetInstanceTypesFromInstanceRequirementsInput{}
resp.Diagnostics.Append(flex.Expand(ctx, data, input)...)
if resp.Diagnostics.HasError() {
return
}

out, err := findInstanceTypesFromInstanceRequirements(ctx, conn, input)
if err != nil {
resp.Diagnostics.AddError(
create.ProblemStandardMessage(names.EC2, create.ErrActionReading, DSNameInstanceTypesFromInstanceRequirements, d.Meta().Region, err),
err.Error(),
)
return
}

data.ID = types.StringValue(d.Meta().Region)
data.InstanceTypes = flex.FlattenFrameworkStringList(ctx, out)

resp.Diagnostics.Append(resp.State.Set(ctx, &data)...)
}

func findInstanceTypesFromInstanceRequirements(ctx context.Context, conn *ec2.Client, input *ec2.GetInstanceTypesFromInstanceRequirementsInput) ([]*string, error) {
var output []*string

paginator := ec2.NewGetInstanceTypesFromInstanceRequirementsPaginator(conn, input)

for paginator.HasMorePages() {
page, err := paginator.NextPage(ctx)
if err != nil {
return nil, err
}
for _, v := range page.InstanceTypes {
output = append(output, v.InstanceType)
}
}

return output, nil
}

type dataSourceInstanceTypesFromInstanceRequirementsData struct {
ArchitectureTypes fwtypes.ListValueOf[types.String] `tfsdk:"architecture_types"`
InstanceRequirements fwtypes.ObjectValueOf[instanceRequirementsData] `tfsdk:"instance_requirements"`
InstanceTypes types.List `tfsdk:"instance_types"`
ID types.String `tfsdk:"id"`
VirtualizationTypes fwtypes.ListValueOf[types.String] `tfsdk:"virtualization_types"`
}

type instanceRequirementsData struct {
MemoryMiB fwtypes.ObjectValueOf[minMax[types.Int64]] `tfsdk:"memory_mib"`
VCPUCount fwtypes.ObjectValueOf[minMax[types.Int64]] `tfsdk:"vcpu_count"`
AcceleratorCount fwtypes.ObjectValueOf[minMax[types.Int64]] `tfsdk:"accelerator_count"`
AcceleratorManufacturers fwtypes.ListValueOf[types.String] `tfsdk:"accelerator_manufacturers"`
AcceleratorNames fwtypes.ListValueOf[types.String] `tfsdk:"accelerator_names"`
AcceleratorTotalMemoryMiB fwtypes.ObjectValueOf[minMax[types.Int64]] `tfsdk:"accelerator_total_memory_mib"`
AcceleratorTypes fwtypes.ListValueOf[types.String] `tfsdk:"accelerator_types"`
AllowedInstanceTypes fwtypes.ListValueOf[types.String] `tfsdk:"allowed_instance_types"`
BareMetal types.String `tfsdk:"bare_metal"`
BaselineEBSBandwidthMbps fwtypes.ObjectValueOf[minMax[types.Float64]] `tfsdk:"baseline_ebs_bandwidth_mbps"`
BurstablePerformance types.String `tfsdk:"burstable_performance"`
CPUManufacturers fwtypes.ListValueOf[types.String] `tfsdk:"cpu_manufacturers"`
ExcludedInstanceTypes fwtypes.ListValueOf[types.String] `tfsdk:"excluded_instance_types"`
InstanceGenerations fwtypes.ListValueOf[types.String] `tfsdk:"instance_generations"`
LocalStorage types.String `tfsdk:"local_storage"`
LocalStorageTypes fwtypes.ListValueOf[types.String] `tfsdk:"local_storage_types"`
MemoryGiBPerVCPU fwtypes.ObjectValueOf[minMax[types.Float64]] `tfsdk:"memory_gib_per_vcpu"`
NetworkBandwidthGbps fwtypes.ObjectValueOf[minMax[types.Float64]] `tfsdk:"network_bandwidth_gbps"`
NetworkInterfaceCount fwtypes.ObjectValueOf[minMax[types.Int64]] `tfsdk:"network_interface_count"`
OnDemandMaxPricePercentageOverLowestPrice types.Int64 `tfsdk:"on_demand_max_price_percentage_over_lowest_price"`
RequireHibernateSupport types.Bool `tfsdk:"require_hibernate_support"`
SpotMaxPricePercentageOverLowestPrice types.Int64 `tfsdk:"spot_max_price_percentage_over_lowest_price"`
TotalLocalStorageGB fwtypes.ObjectValueOf[minMax[types.Float64]] `tfsdk:"total_local_storage_gb"`
}

type minMax[T comparable] struct {
Min T `tfsdk:"min"`
Max T `tfsdk:"max"`
}
Loading
Loading