mirror of
https://github.com/crowdsecurity/crowdsec.git
synced 2025-05-10 20:05:55 +02:00
1133afe58d
* context propagation: pkg/database/lock * QueryAllDecisionsWithFilters(ctx...), QueryExpiredDecisionsWithFilters(ctx...) * more Query...Decision...(ctx..) * rest of decisions * lint
87 lines
2 KiB
Go
87 lines
2 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/crowdsecurity/crowdsec/pkg/database/ent"
|
|
"github.com/crowdsecurity/crowdsec/pkg/database/ent/lock"
|
|
"github.com/crowdsecurity/crowdsec/pkg/types"
|
|
)
|
|
|
|
const (
|
|
CAPIPullLockTimeout = 10
|
|
CapiPullLockName = "pullCAPI"
|
|
)
|
|
|
|
func (c *Client) AcquireLock(ctx context.Context, name string) error {
|
|
log.Debugf("acquiring lock %s", name)
|
|
_, err := c.Ent.Lock.Create().
|
|
SetName(name).
|
|
SetCreatedAt(types.UtcNow()).
|
|
Save(ctx)
|
|
|
|
if ent.IsConstraintError(err) {
|
|
return err
|
|
}
|
|
|
|
if err != nil {
|
|
return errors.Wrapf(InsertFail, "insert lock: %s", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *Client) ReleaseLock(ctx context.Context, name string) error {
|
|
log.Debugf("releasing lock %s", name)
|
|
_, err := c.Ent.Lock.Delete().Where(lock.NameEQ(name)).Exec(ctx)
|
|
if err != nil {
|
|
return errors.Wrapf(DeleteFail, "delete lock: %s", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *Client) ReleaseLockWithTimeout(ctx context.Context, name string, timeout int) error {
|
|
log.Debugf("releasing lock %s with timeout of %d minutes", name, timeout)
|
|
|
|
_, err := c.Ent.Lock.Delete().Where(
|
|
lock.NameEQ(name),
|
|
lock.CreatedAtLT(time.Now().UTC().Add(-time.Duration(timeout)*time.Minute)),
|
|
).Exec(ctx)
|
|
if err != nil {
|
|
return errors.Wrapf(DeleteFail, "delete lock: %s", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *Client) IsLocked(err error) bool {
|
|
return ent.IsConstraintError(err)
|
|
}
|
|
|
|
func (c *Client) AcquirePullCAPILock(ctx context.Context) error {
|
|
// delete orphan "old" lock if present
|
|
err := c.ReleaseLockWithTimeout(ctx, CapiPullLockName, CAPIPullLockTimeout)
|
|
if err != nil {
|
|
log.Errorf("unable to release pullCAPI lock: %s", err)
|
|
}
|
|
|
|
return c.AcquireLock(ctx, CapiPullLockName)
|
|
}
|
|
|
|
func (c *Client) ReleasePullCAPILock(ctx context.Context) error {
|
|
log.Debugf("deleting lock %s", CapiPullLockName)
|
|
|
|
_, err := c.Ent.Lock.Delete().Where(
|
|
lock.NameEQ(CapiPullLockName),
|
|
).Exec(ctx)
|
|
if err != nil {
|
|
return errors.Wrapf(DeleteFail, "delete lock: %s", err)
|
|
}
|
|
|
|
return nil
|
|
}
|