70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
package pkg
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/cloudskiff/driftctl/pkg/alerter"
|
|
"github.com/cloudskiff/driftctl/pkg/parallel"
|
|
"github.com/cloudskiff/driftctl/pkg/remote"
|
|
"github.com/cloudskiff/driftctl/pkg/resource"
|
|
"github.com/pkg/errors"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type Scanner struct {
|
|
resourceSuppliers []resource.Supplier
|
|
runner *parallel.ParallelRunner
|
|
alerter *alerter.Alerter
|
|
}
|
|
|
|
func NewScanner(resourceSuppliers []resource.Supplier, alerter *alerter.Alerter) *Scanner {
|
|
return &Scanner{
|
|
resourceSuppliers: resourceSuppliers,
|
|
runner: parallel.NewParallelRunner(context.TODO(), 10),
|
|
alerter: alerter,
|
|
}
|
|
}
|
|
|
|
func (s *Scanner) Resources() ([]resource.Resource, error) {
|
|
for _, resourceProvider := range s.resourceSuppliers {
|
|
supplier := resourceProvider
|
|
s.runner.Run(func() (interface{}, error) {
|
|
res, err := supplier.Resources()
|
|
if err != nil {
|
|
err := remote.HandleResourceEnumerationError(err, s.alerter)
|
|
if err == nil {
|
|
return []resource.Resource{}, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
for _, resource := range res {
|
|
logrus.WithFields(logrus.Fields{
|
|
"id": resource.TerraformId(),
|
|
"type": resource.TerraformType(),
|
|
}).Debug("Found cloud resource")
|
|
}
|
|
return res, nil
|
|
})
|
|
}
|
|
|
|
results := make([]resource.Resource, 0)
|
|
loop:
|
|
for {
|
|
select {
|
|
case resources, ok := <-s.runner.Read():
|
|
if !ok || resources == nil {
|
|
break loop
|
|
}
|
|
results = append(results, resources.([]resource.Resource)...)
|
|
case <-s.runner.DoneChan():
|
|
break loop
|
|
}
|
|
}
|
|
return results, s.runner.Err()
|
|
}
|
|
|
|
func (s *Scanner) Stop() {
|
|
logrus.Debug("Stopping scanner")
|
|
s.runner.Stop(errors.New("interrupted"))
|
|
}
|