Merge pull request #156 from nicolasbock/keep

Add option to keep working directory
canonical · May 30, 2024 · 196841e · 196841e
2 parents 5a34474 + a1a9f16
commit 196841e
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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)
 }