From a1a9f16eb5861f8eb1211e41d2fec890a8c1b816 Mon Sep 17 00:00:00 2001 From: Nicolas Bock Date: Thu, 30 May 2024 09:24:31 -0600 Subject: [PATCH] Add option to keep working directory For debugging. This option is defaulting to `false`. Signed-off-by: Nicolas Bock --- athena-processor.yaml | 1 + pkg/config/config.go | 9 +++++---- pkg/processor/processor.go | 4 ++++ 3 files changed, 10 insertions(+), 4 deletions(-) diff --git a/athena-processor.yaml b/athena-processor.yaml index ebbaec2..b9137c0 100644 --- a/athena-processor.yaml +++ b/athena-processor.yaml @@ -1,6 +1,7 @@ processor: batch-comments-every: 1m base-tmpdir: "/tmp/athena" + keep-processing-output: true subscribers: sosreports: sf-comment-enabled: true diff --git a/pkg/config/config.go b/pkg/config/config.go index c072d52..555894d 100644 --- a/pkg/config/config.go +++ b/pkg/config/config.go @@ -57,10 +57,11 @@ func NewMonitor() Monitor { } type Processor struct { - ReportsUploadPath string `yaml:"reports-upload-dir"` - BatchCommentsEvery string `yaml:"batch-comments-every"` - BaseTmpDir string `yaml:"base-tmpdir"` - SubscribeTo map[string]Subscriber `yaml:"subscribers,omitempty"` + ReportsUploadPath string `yaml:"reports-upload-dir"` + BatchCommentsEvery string `yaml:"batch-comments-every"` + BaseTmpDir string `yaml:"base-tmpdir"` + KeepProcessingOutput bool `yaml:"keep-processing-output"` + SubscribeTo map[string]Subscriber `yaml:"subscribers,omitempty"` } func NewProcessor() Processor { diff --git a/pkg/processor/processor.go b/pkg/processor/processor.go index 9f5d642..b5ce03a 100644 --- a/pkg/processor/processor.go +++ b/pkg/processor/processor.go @@ -327,6 +327,10 @@ func NewReportRunner(cfg *config.Config, dbConn *gorm.DB, sf common.SalesforceCl } func (runner *ReportRunner) Clean() error { + if runner.Config.Processor.KeepProcessingOutput { + log.Infof("Keeping base direcotry %s for report %s", runner.Basedir, runner.Name) + return nil + } log.Infof("Removing base directory: %s for report: %s", runner.Basedir, runner.Name) return os.RemoveAll(runner.Basedir) }