feat(saas): delete json dir automatically after upload (#1212)

* feat(saas): delete json dir automatically after upload

* fix lint err
This commit is contained in:
Kota Kanbe
2021-04-15 05:58:41 +09:00
committed by GitHub
parent 740781af56
commit 27b3e17b79
4 changed files with 14 additions and 3 deletions

View File

@@ -22,7 +22,7 @@ func FillWithMetasploit(r *models.ScanResult, cnf config.MetasploitConf) (nMetas
}
defer func() {
if err := driver.CloseDB(); err != nil {
logging.Log.Errorf("Failed to close DB. err: %+v")
logging.Log.Errorf("Failed to close DB. err: %+v", err)
}
}()

View File

@@ -33,7 +33,7 @@ func (o RedHatBase) FillWithOval(r *models.ScanResult) (nCVEs int, err error) {
}
defer func() {
if err := driver.CloseDB(); err != nil {
logging.Log.Errorf("Failed to close DB. err: %+v")
logging.Log.Errorf("Failed to close DB. err: %+v", err)
}
}()

View File

@@ -40,7 +40,7 @@ func (o SUSE) FillWithOval(r *models.ScanResult) (nCVEs int, err error) {
}
defer func() {
if err := driver.CloseDB(); err != nil {
logging.Log.Errorf("Failed to close DB. err: %+v")
logging.Log.Errorf("Failed to close DB. err: %+v", err)
}
}()

View File

@@ -125,5 +125,16 @@ func (p *SaaSCmd) Execute(_ context.Context, f *flag.FlagSet, _ ...interface{})
return subcommands.ExitFailure
}
if !config.Conf.Debug {
if err := os.RemoveAll(dir); err != nil {
logging.Log.Warnf("Failed to remove %s. err: %+v", dir, err)
}
symlink := filepath.Join(config.Conf.ResultsDir, "current")
err := os.Remove(symlink)
if err != nil {
logging.Log.Warnf("Failed to remove %s. err: %+v", dir, err)
}
}
return subcommands.ExitSuccess
}