Infracost项目：如何为云资源添加成本估算支持-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_01018/article/details/148393749

Infracost项目：如何为云资源添加成本估算支持

infracost Infracost能够读取Terraform文件并预估即将部署的云资源的成本，帮助开发者在规划阶段就进行成本优化决策，防止意外的高昂费用支出。 $infracost$ 项目地址: https://gitcode.com/gh_mirrors/in/infracost

作为一款强大的云成本估算工具，Infracost能够帮助开发者在基础设施即代码(IaC)阶段就准确预测云资源的使用成本。本文将详细介绍如何为Infracost添加新的云资源支持，使工具能够估算更多类型的云服务成本。

理解Infracost的资源模型

在开始添加新资源前，我们需要了解Infracost如何处理云资源：

资源(Resource)：对应IaC提供者(如Terraform)中的一个具体资源，例如AWS的NAT网关
成本组件(Cost Component)：资源的一个具体计费项，包含价格和使用量计算逻辑

以AWS NAT网关为例，它包含两个成本组件：

每小时基础费用
数据处理费用

准备工作

在添加新资源前，请确保：

已设置好Infracost开发环境
能够访问云定价API数据库(PostgreSQL)
已有明确的需求说明，包括：
- 要添加的资源类型
- 该资源的所有成本组件
- 相关文档链接

资源添加流程详解

1. 生成资源文件结构

Infracost提供了便捷的命令行工具自动生成资源模板：

go run ./cmd/resourcegen/main.go -cloud-provider aws -resource-name transfer_server

该命令会生成以下关键文件：

资源映射文件：将Terraform属性映射到通用资源
测试文件：包含集成测试用例
示例文件：用于测试的Terraform代码
注册文件：将资源添加到支持列表

2. 编写测试用例

在生成的测试文件中添加真实的Terraform资源配置。例如，对于AWS Transfer Server：

resource "aws_transfer_server" "example" {
  tags = {
    Name = "Example"
  }
}

运行测试确保基础结构正常工作：

ARGS="--run TestTransferServer -v -update" make test_aws

3. 添加成本组件

成本组件是资源的核心部分，需要仔细定义。以下是关键步骤：

3.1 定义成本组件结构

在资源文件中添加成本组件方法：

func (r *TransferServer) protocolEnabledCostComponent() *schema.CostComponent {
    return &schema.CostComponent{
        Name:           "FTP protocol enabled",
        Unit:           "hours",
        UnitMultiplier: decimal.NewFromInt(1),
        HourlyQuantity: decimalPtr(decimal.NewFromInt(1)),
        ProductFilter: &schema.ProductFilter{
            VendorName:    strPtr("aws"),
            Region:        strPtr(r.Region),
            Service:       strPtr("<SERVICE>"),
            ProductFamily: strPtr("<PRODUCT FAMILY>"),
            AttributeFilters: []*schema.AttributeFilter{
                {Key: "<ATTRIBUTE KEY>", Value: strPtr("<STRING FILTER>")},
            },
        },
    }
}

3.2 查询定价数据

使用PostgreSQL查询云定价数据库，找到准确的定价信息：

SELECT service, productFamily, attributes 
FROM products 
WHERE vendorName = 'aws' 
AND service LIKE '%Transfer%';

通过分析查询结果，确定正确的服务名称、产品系列和属性过滤器。

3.3 映射资源属性

如果成本组件依赖Terraform资源配置，需要在映射文件中添加属性解析逻辑：

func NewTransferServer(d *schema.ResourceData) *resources.TransferServer {
    return &resources.TransferServer{
        Address: d.Address,
        Region:  d.Get("region").String(),
        // 其他属性映射
    }
}

3.4 处理使用量数据

对于基于使用量的计费项，需要定义使用量模式：

var TransferServerUsageSchema = []*schema.UsageItem{
    {Key: "monthly_data_processed", DefaultValue: 0},
}

func (r *TransferServer) PopulateUsage(u *schema.UsageData) {
    resources.PopulateArgsWithUsage(r, u)
}