aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--cmd/update.go15
1 files changed, 5 insertions, 10 deletions
diff --git a/cmd/update.go b/cmd/update.go
index 86e6c7a..8615057 100644
--- a/cmd/update.go
+++ b/cmd/update.go
@@ -7,7 +7,6 @@ import (
"github.com/pablotron/cvez/nvdmirror"
"github.com/rs/zerolog/log"
"github.com/spf13/cobra"
- nvd_feed "github.com/pablotron/cvez/feed"
"os"
"path/filepath"
)
@@ -33,22 +32,16 @@ var updateCmd = &cobra.Command{
defer cache.Close()
// sync data, get updates
+ log.Info().Msg("sync mirror")
updates := nvdmirror.Sync(config, &cache, cacheDir)
if len(updates) > 0 {
// connect to db
ctx := context.Background()
db := getDb()
- // build list of feeds to add
+ // load feeds
log.Info().Msg("load feeds")
- var feeds []nvd_feed.Feed
- for _, row := range(updates) {
- if row.Type == nvdmirror.UpdateCveYear {
- feeds = append(feeds, getFeed(filepath.Join(cacheDir, row.Path)))
- }
- }
-
- if len(feeds) > 0 {
+ if feeds := getFeeds(cacheDir, updates); len(feeds) > 0 {
log.Info().Msg("AddCveFeeds")
if _, err := db.AddCveFeeds(ctx, feeds); err != nil {
// FIXME: failing like this leaves an invalid cache
@@ -88,6 +81,8 @@ var updateCmd = &cobra.Command{
}
}
}
+
+ log.Info().Msg("done")
}
},
}