mirror of https://github.com/stashapp/stash.git
57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package manager
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/stashapp/stash/pkg/job"
|
|
"github.com/stashapp/stash/pkg/logger"
|
|
)
|
|
|
|
type Optimiser interface {
|
|
Analyze(ctx context.Context) error
|
|
Vacuum(ctx context.Context) error
|
|
}
|
|
|
|
type OptimiseDatabaseJob struct {
|
|
Optimiser Optimiser
|
|
}
|
|
|
|
func (j *OptimiseDatabaseJob) Execute(ctx context.Context, progress *job.Progress) {
|
|
logger.Info("Optimising database")
|
|
progress.SetTotal(2)
|
|
|
|
start := time.Now()
|
|
|
|
var err error
|
|
|
|
progress.ExecuteTask("Analyzing database", func() {
|
|
err = j.Optimiser.Analyze(ctx)
|
|
progress.Increment()
|
|
})
|
|
if job.IsCancelled(ctx) {
|
|
logger.Info("Stopping due to user request")
|
|
return
|
|
}
|
|
if err != nil {
|
|
logger.Errorf("Error analyzing database: %v", err)
|
|
return
|
|
}
|
|
|
|
progress.ExecuteTask("Vacuuming database", func() {
|
|
err = j.Optimiser.Vacuum(ctx)
|
|
progress.Increment()
|
|
})
|
|
if job.IsCancelled(ctx) {
|
|
logger.Info("Stopping due to user request")
|
|
return
|
|
}
|
|
if err != nil {
|
|
logger.Errorf("Error vacuuming database: %v", err)
|
|
return
|
|
}
|
|
|
|
elapsed := time.Since(start)
|
|
logger.Infof("Finished optimising database after %s", elapsed)
|
|
}
|