🐛 clean up some of the aws resource init functions

mondoohq · Oct 3, 2023 · e701ebf · e701ebf
1 parent a4118a5
commit e701ebf
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 12 deletions.
diff --git a/providers/aws/resources/aws_cloudwatch.go b/providers/aws/resources/aws_cloudwatch.go
@@ -176,7 +176,7 @@ func initAwsCloudwatchMetric(runtime *plugin.Runtime, args map[string]*llx.RawDa
 		return args, nil, err
 	}
 	if len(metrics.Metrics) == 0 {
-		return nil, nil, nil
+		return nil, nil, errors.New("no metrics found")
 	}
 	if len(metrics.Metrics) > 1 {
 		return nil, nil, errors.New("more than one metric found for " + namespace + " " + name + " in region " + region)

diff --git a/providers/aws/resources/aws_ec2.go b/providers/aws/resources/aws_ec2.go
@@ -733,11 +733,11 @@ func (a *mqlAwsEc2) gatherInstanceInfo(instances []ec2types.Reservation, imdsvVe
 			if instance.ImageId != nil {
 				mqlImage, err := NewResource(a.MqlRuntime, "aws.ec2.image",
 					map[string]*llx.RawData{"arn": llx.StringData(fmt.Sprintf(imageArnPattern, regionVal, conn.AccountId(), convert.ToString(instance.ImageId)))})
-				if err != nil {
-					return nil, err
-				}
-				if mqlImage != nil {
+				if err == nil {
 					args["image"] = llx.ResourceData(mqlImage, mqlImage.MqlName())
+				} else {
+					log.Error().Err(err).Msg("cannot find image")
+					args["image"] = llx.NilData
 				}
 			} else {
 				args["image"] = llx.NilData
@@ -766,11 +766,11 @@ func (a *mqlAwsEc2) gatherInstanceInfo(instances []ec2types.Reservation, imdsvVe
 						"region": llx.StringData(regionVal),
 						"name":   llx.StringData(convert.ToString(instance.KeyName)),
 					})
-				if err != nil {
-					return nil, err
-				}
-				if mqlKeyPair != nil {
+				if err == nil {
 					args["keypair"] = llx.ResourceData(mqlKeyPair, mqlKeyPair.MqlName())
+				} else {
+					log.Error().Err(err).Msg("cannot find keypair")
+					args["keypair"] = llx.NilData
 				}
 			} else {
 				args["keypair"] = llx.NilData
@@ -802,7 +802,7 @@ func initAwsEc2Image(runtime *plugin.Runtime, args map[string]*llx.RawData) (map
 	arnVal := args["arn"].Value.(string)
 	arn, err := arn.Parse(arnVal)
 	if err != nil {
-		return nil, nil, nil
+		return nil, nil, err
 	}
 	resource := strings.Split(arn.Resource, "/")
 	conn := runtime.Connection.(*connection.AwsConnection)
@@ -824,7 +824,7 @@ func initAwsEc2Image(runtime *plugin.Runtime, args map[string]*llx.RawData) (map
 		return args, nil, nil
 	}
 
-	return args, nil, nil
+	return nil, nil, errors.New("image not found")
 }
 
 func (a *mqlAwsEc2Securitygroup) id() (string, error) {

diff --git a/providers/aws/resources/aws_sns.go b/providers/aws/resources/aws_sns.go
@@ -60,7 +60,7 @@ func (a *mqlAwsSnsTopic) init(runtime *plugin.Runtime, args map[string]*llx.RawD
 	arnVal := args["arn"].Value.(string)
 	arn, err := arn.Parse(arnVal)
 	if err != nil {
-		return nil, nil, nil
+		return nil, nil, err
 	}
 
 	args["arn"] = llx.StringData(arnVal)