68183384dc
This change uses an error group to load different DB resources in parallel. Signed-off-by: Gabriel Adrian Samfira <gsamfira@cloudbasesolutions.com>
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package entity
|
|
|
|
import (
|
|
"golang.org/x/sync/errgroup"
|
|
|
|
dbCommon "github.com/cloudbase/garm/database/common"
|
|
"github.com/cloudbase/garm/database/watcher"
|
|
"github.com/cloudbase/garm/params"
|
|
)
|
|
|
|
func composeControllerWatcherFilters() dbCommon.PayloadFilterFunc {
|
|
return watcher.WithAll(
|
|
watcher.WithAny(
|
|
watcher.WithEntityTypeFilter(dbCommon.RepositoryEntityType),
|
|
watcher.WithEntityTypeFilter(dbCommon.OrganizationEntityType),
|
|
watcher.WithEntityTypeFilter(dbCommon.EnterpriseEntityType),
|
|
),
|
|
watcher.WithAny(
|
|
watcher.WithOperationTypeFilter(dbCommon.CreateOperation),
|
|
watcher.WithOperationTypeFilter(dbCommon.DeleteOperation),
|
|
),
|
|
)
|
|
}
|
|
|
|
func composeWorkerWatcherFilters(entity params.GithubEntity) dbCommon.PayloadFilterFunc {
|
|
return watcher.WithAny(
|
|
watcher.WithAll(
|
|
watcher.WithEntityFilter(entity),
|
|
watcher.WithOperationTypeFilter(dbCommon.UpdateOperation),
|
|
),
|
|
// Watch for credentials updates.
|
|
watcher.WithAll(
|
|
watcher.WithGithubCredentialsFilter(entity.Credentials),
|
|
watcher.WithOperationTypeFilter(dbCommon.UpdateOperation),
|
|
),
|
|
)
|
|
}
|
|
|
|
func (c *Controller) waitForErrorGroupOrContextCancelled(g *errgroup.Group) error {
|
|
if g == nil {
|
|
return nil
|
|
}
|
|
|
|
done := make(chan error, 1)
|
|
go func() {
|
|
waitErr := g.Wait()
|
|
done <- waitErr
|
|
}()
|
|
|
|
select {
|
|
case err := <-done:
|
|
return err
|
|
case <-c.ctx.Done():
|
|
return c.ctx.Err()
|
|
case <-c.quit:
|
|
return nil
|
|
}
|
|
}
|