driftctl/pkg/remote/aws/lambda_function_supplier.go

73 lines
2.1 KiB
Go
Raw Normal View History

package aws
import (
"github.com/cloudskiff/driftctl/pkg/remote/aws/repository"
remoteerror "github.com/cloudskiff/driftctl/pkg/remote/error"
"github.com/cloudskiff/driftctl/pkg/remote/deserializer"
"github.com/cloudskiff/driftctl/pkg/resource"
resourceaws "github.com/cloudskiff/driftctl/pkg/resource/aws"
awsdeserializer "github.com/cloudskiff/driftctl/pkg/resource/aws/deserializer"
"github.com/cloudskiff/driftctl/pkg/terraform"
"github.com/aws/aws-sdk-go/service/lambda"
"github.com/sirupsen/logrus"
"github.com/zclconf/go-cty/cty"
)
type LambdaFunctionSupplier struct {
reader terraform.ResourceReader
deserializer deserializer.CTYDeserializer
client repository.LambdaRepository
runner *terraform.ParallelResourceReader
}
func NewLambdaFunctionSupplier(provider *AWSTerraformProvider) *LambdaFunctionSupplier {
return &LambdaFunctionSupplier{
provider,
awsdeserializer.NewLambdaFunctionDeserializer(),
repository.NewLambdaRepository(provider.session),
terraform.NewParallelResourceReader(provider.Runner().SubRunner()),
}
}
func (s LambdaFunctionSupplier) Resources() ([]resource.Resource, error) {
functions, err := s.client.ListAllLambdaFunctions()
if err != nil {
return nil, remoteerror.NewResourceEnumerationError(err, resourceaws.AwsLambdaFunctionResourceType)
}
results := make([]cty.Value, 0)
if len(functions) > 0 {
for _, function := range functions {
fun := *function
s.runner.Run(func() (cty.Value, error) {
return s.readLambda(fun)
})
}
results, err = s.runner.Wait()
if err != nil {
return nil, err
}
}
return s.deserializer.Deserialize(results)
}
func (s LambdaFunctionSupplier) readLambda(function lambda.FunctionConfiguration) (cty.Value, error) {
name := *function.FunctionName
resFunction, err := s.reader.ReadResource(
terraform.ReadResourceArgs{
Ty: resourceaws.AwsLambdaFunctionResourceType,
ID: name,
Attributes: map[string]string{
"function_name": name,
},
},
)
if err != nil {
logrus.Warnf("Error reading function %s[%s]: %+v", name, resourceaws.AwsLambdaFunctionResourceType, err)
return cty.NilVal, err
}
return *resFunction, nil
}