Timothy Messier
eb61ac6304
This adds a context to the session.RepositoryFactory, allowing errors.NewDeprecated to be replaced with errors.New. It also removes the common.SessionRepoFactory, instead packages just directly import from the session package.
41 lines
1.3 KiB
Go
41 lines
1.3 KiB
Go
package session
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/hashicorp/boundary/internal/db"
|
|
"github.com/hashicorp/boundary/internal/kms"
|
|
"github.com/hashicorp/boundary/internal/scheduler"
|
|
)
|
|
|
|
const deleteTerminatedThreshold = time.Hour
|
|
|
|
// RegisterJobs registers session related jobs with the provided scheduler.
|
|
func RegisterJobs(ctx context.Context, scheduler *scheduler.Scheduler, w db.Writer, r db.Reader, k *kms.Kms, gracePeriod time.Duration) error {
|
|
const op = "session.RegisterJobs"
|
|
|
|
sessionConnectionCleanupJob, err := newSessionConnectionCleanupJob(w, gracePeriod)
|
|
if err != nil {
|
|
return fmt.Errorf("error creating session cleanup job: %w", err)
|
|
}
|
|
if err = scheduler.RegisterJob(ctx, sessionConnectionCleanupJob); err != nil {
|
|
return fmt.Errorf("error registering session cleanup job: %w", err)
|
|
}
|
|
|
|
repo, err := NewRepository(ctx, r, w, k)
|
|
if err != nil {
|
|
return fmt.Errorf("error creating repository: %w", err)
|
|
}
|
|
deleteTerminatedJob, err := newDeleteTerminatedJob(ctx, repo, deleteTerminatedThreshold)
|
|
if err != nil {
|
|
return fmt.Errorf("error creating delete terminated session job: %w", err)
|
|
}
|
|
if err = scheduler.RegisterJob(ctx, deleteTerminatedJob); err != nil {
|
|
return fmt.Errorf("error registering delete terminated session job: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|