Skip to content

Commit

Permalink
Fix: silence badger logger
Browse files Browse the repository at this point in the history
  • Loading branch information
adedayo committed Jul 28, 2019
1 parent 37421ab commit 2268fa0
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions persistence.go
Original file line number Diff line number Diff line change
Expand Up @@ -21,8 +21,17 @@ var (
psrCache = make(map[string]PersistedScanRequest)
lock = sync.RWMutex{}
scanCacheLock = sync.RWMutex{}
myLogger = logger{}
)

type logger struct {
}

func (log logger) Errorf(format string, params ...interface{}) {}
func (log logger) Warningf(format string, params ...interface{}) {}
func (log logger) Infof(format string, params ...interface{}) {}
func (log logger) Debugf(format string, params ...interface{}) {}

//ListScans returns the ScanID list of persisted scans
func ListScans(rewindDays int, completed bool) (result []ScanRequest) {
if rewindDays < 0 {
Expand Down Expand Up @@ -71,6 +80,7 @@ func ListScans(rewindDays int, completed bool) (result []ScanRequest) {
func PersistScans(psr PersistedScanRequest, server string, scans []PortACK) {
dbDir := filepath.Join(baseScanDBDirectory, psr.Request.Day, psr.Request.ScanID)
opts := badger.DefaultOptions(dbDir)
opts.Logger = myLogger
opts.NumVersionsToKeep = 0
db, err := badger.Open(opts)
if err != nil {
Expand All @@ -94,6 +104,7 @@ func LoadScanRequest(dir, scanID string) (psr PersistedScanRequest, e error) {
lock.Unlock()
dbDir := filepath.Join(baseScanDBDirectory, dir, scanID, "request")
opts := badger.DefaultOptions(dbDir)
opts.Logger = myLogger
opts.ReadOnly = true
db, err := badger.Open(opts)
if err != nil {
Expand Down Expand Up @@ -163,6 +174,7 @@ func marshallPortAcks(s []PortACK) []byte {
func PersistScanRequest(psr PersistedScanRequest) {
dbDir := filepath.Join(baseScanDBDirectory, psr.Request.Day, psr.Request.ScanID, "request")
opts := badger.DefaultOptions(dbDir)
opts.Logger = myLogger
opts.NumVersionsToKeep = 0
db, err := badger.Open(opts)
if err != nil {
Expand All @@ -181,6 +193,7 @@ func CompactDB(dayPath, scanID string) {
//compact the scan requests
dbDir := filepath.Join(baseScanDBDirectory, dayPath, scanID, "request")
opts := badger.DefaultOptions(dbDir)
opts.Logger = myLogger
opts.NumVersionsToKeep = 0
db, err := badger.Open(opts)
if err != nil {
Expand Down

0 comments on commit 2268fa0

Please sign in to comment.