58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
package resource
|
|
|
|
import (
|
|
"context"
|
|
"runtime"
|
|
|
|
"github.com/cloudskiff/driftctl/pkg/parallel"
|
|
)
|
|
|
|
type ChainSupplier struct {
|
|
suppliers []Supplier
|
|
runner *parallel.ParallelRunner
|
|
}
|
|
|
|
func NewChainSupplier() *ChainSupplier {
|
|
return &ChainSupplier{
|
|
runner: parallel.NewParallelRunner(context.TODO(), int64(runtime.NumCPU())),
|
|
}
|
|
}
|
|
|
|
func (r *ChainSupplier) AddSupplier(supplier Supplier) {
|
|
r.suppliers = append(r.suppliers, supplier)
|
|
}
|
|
|
|
func (r *ChainSupplier) Resources() ([]Resource, error) {
|
|
|
|
for _, supplier := range r.suppliers {
|
|
sup := supplier
|
|
r.runner.Run(func() (interface{}, error) {
|
|
return sup.Resources()
|
|
})
|
|
}
|
|
|
|
results := make([]Resource, 0)
|
|
|
|
ReadLoop:
|
|
for {
|
|
select {
|
|
case supplierResult, ok := <-r.runner.Read():
|
|
if !ok || supplierResult == nil {
|
|
break ReadLoop
|
|
}
|
|
// Type cannot be invalid as return type is enforced
|
|
// by Supplier interface
|
|
resources, _ := supplierResult.([]Resource)
|
|
results = append(results, resources...)
|
|
case <-r.runner.DoneChan():
|
|
break ReadLoop
|
|
}
|
|
}
|
|
|
|
if r.runner.Err() != nil {
|
|
return nil, r.runner.Err()
|
|
}
|
|
|
|
return results, nil
|
|
}
|