fix "Init k-v db 不能多个进程同时开启"

Author: x51pwn

nuclei_Yaml/internal/runner/options.go

@@ -47,7 +47,8 @@ func ParseOptions(options *types.Options) {
 	}
 	if options.Version {
 		gologger.Info().Msgf("Current Version: %s\n", config.Version)
-		os.Exit(0)
+		//os.Exit(0)
+		return
 	}
 	if options.TemplatesVersion {
 		configuration, err := config.ReadConfiguration()

pkg/kvDb.go

@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"github.com/dgraph-io/badger"
 	"log"
+	"os"
 	"sync"
 )
 
@@ -33,8 +34,10 @@ func (r *KvDbOp) SetExpiresAt(ExpiresAt uint64) {
 }
 
 func (r *KvDbOp) Init(szDb string) error {
+	os.RemoveAll(szDb)
 	opts := badger.DefaultOptions(szDb)
 	opts.CompactL0OnClose = true
+	opts.EventLogging = false
 	opts.LevelOneSize = 256 << 10
 	opts.LevelSizeMultiplier = 20
 	db, err := badger.Open(opts)