mirror of
https://github.com/containers/podman.git
synced 2025-07-27 02:12:48 +08:00
8489dc4345
With the advent of Podman 2.0.0 we crossed the magical barrier of go modules. While we were able to continue importing all packages inside of the project, the project could not be vendored anymore from the outside. Move the go module to new major version and change all imports to `github.com/containers/libpod/v2`. The renaming of the imports was done via `gomove` [1]. [1] https://github.com/KSubedi/gomove Signed-off-by: Valentin Rothberg <rothberg@redhat.com>
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package parallel
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/containers/libpod/v2/libpod"
|
|
"github.com/pkg/errors"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// ContainerOp performs the given function on the given set of
|
|
// containers, using a number of parallel threads.
|
|
// If no error is returned, each container specified in ctrs will have an entry
|
|
// in the resulting map; containers with no error will be set to nil.
|
|
func ContainerOp(ctx context.Context, ctrs []*libpod.Container, applyFunc func(*libpod.Container) error) (map[*libpod.Container]error, error) {
|
|
jobControlLock.RLock()
|
|
defer jobControlLock.RUnlock()
|
|
|
|
// We could use a sync.Map but given Go's lack of generic I'd rather
|
|
// just use a lock on a normal map...
|
|
// The expectation is that most of the time is spent in applyFunc
|
|
// anyways.
|
|
var (
|
|
errMap = make(map[*libpod.Container]error)
|
|
errLock sync.Mutex
|
|
allDone sync.WaitGroup
|
|
)
|
|
|
|
for _, ctr := range ctrs {
|
|
// Block until a thread is available
|
|
if err := jobControl.Acquire(ctx, 1); err != nil {
|
|
return nil, errors.Wrapf(err, "error acquiring job control semaphore")
|
|
}
|
|
|
|
allDone.Add(1)
|
|
|
|
c := ctr
|
|
go func() {
|
|
logrus.Debugf("Launching job on container %s", c.ID())
|
|
|
|
err := applyFunc(c)
|
|
errLock.Lock()
|
|
errMap[c] = err
|
|
errLock.Unlock()
|
|
|
|
allDone.Done()
|
|
jobControl.Release(1)
|
|
}()
|
|
}
|
|
|
|
allDone.Wait()
|
|
|
|
return errMap, nil
|
|
}
|
|
|
|
// TODO: Add an Enqueue() function that returns a promise
|