Skip to content
Snippets Groups Projects
Select Git revision
  • f69d4bc80ccd5e984506442c38c52a040d11a97d
  • without_tipselection default
  • develop protected
  • fix/grafana-local-dashboard
  • wasp
  • fix/dashboard-explorer-freeze
  • master
  • feat/timerqueue
  • test/sync_debug_and_650
  • feat/sync_revamp_inv
  • wip/sync
  • tool/db-recovery
  • portcheck/fix
  • fix/synchronization
  • feat/new-dashboard-analysis
  • feat/refactored-analysis-dashboard
  • feat/new-analysis-dashboard
  • test/demo-prometheus-fpc
  • prometheus_metrics
  • wip/analysis-server
  • merge/fpc-test-value-transfer
  • v0.2.2
  • v0.2.1
  • v0.2.0
  • v0.1.3
  • v0.1.2
  • v0.1.1
  • v0.1.0
28 results

plugin.go

Blame
  • database.go 2.15 KiB
    package database
    
    import (
    	"sync"
    
    	"github.com/dgraph-io/badger"
    )
    
    var (
    	ErrKeyNotFound = badger.ErrKeyNotFound
    
    	dbMap = make(map[string]*prefixDb)
    	mu    sync.Mutex
    )
    
    type prefixDb struct {
    	db     *badger.DB
    	name   string
    	prefix []byte
    }
    
    func getPrefix(name string) []byte {
    	return []byte(name + "_")
    }
    
    func Get(name string) (Database, error) {
    	mu.Lock()
    	defer mu.Unlock()
    
    	if db, exists := dbMap[name]; exists {
    		return db, nil
    	}
    
    	badger := GetBadgerInstance()
    	db := &prefixDb{
    		db:     badger,
    		name:   name,
    		prefix: getPrefix(name),
    	}
    
    	dbMap[name] = db
    
    	return db, nil
    }
    
    func (this *prefixDb) Set(key []byte, value []byte) error {
    	err := this.db.Update(func(txn *badger.Txn) error {
    		return txn.Set(append(this.prefix, key...), value)
    	})
    	return err
    }
    
    func (this *prefixDb) Contains(key []byte) (bool, error) {
    	err := this.db.View(func(txn *badger.Txn) error {
    		_, err := txn.Get(append(this.prefix, key...))
    		return err
    	})
    
    	if err == ErrKeyNotFound {
    		return false, nil
    	} else {
    		return err == nil, err
    	}
    }
    
    func (this *prefixDb) Get(key []byte) ([]byte, error) {
    	var result []byte = nil
    
    	err := this.db.View(func(txn *badger.Txn) error {
    		item, err := txn.Get(append(this.prefix, key...))
    		if err != nil {
    			return err
    		}
    
    		return item.Value(func(val []byte) error {
    			result = append([]byte{}, val...)
    
    			return nil
    		})
    	})
    
    	return result, err
    }
    
    func (this *prefixDb) Delete(key []byte) error {
    	err := this.db.Update(func(txn *badger.Txn) error {
    		return txn.Delete(append(this.prefix, key...))
    	})
    	return err
    }
    
    func (this *prefixDb) ForEach(consumer func([]byte, []byte)) error {
    	err := this.db.View(func(txn *badger.Txn) error {
    		iteratorOptions := badger.DefaultIteratorOptions
    		iteratorOptions.Prefix = this.prefix // filter by prefix
    
    		// create an iterator the default options
    		it := txn.NewIterator(iteratorOptions)
    		defer it.Close()
    
    		// loop through every key-value-pair and call the function
    		for it.Rewind(); it.Valid(); it.Next() {
    			item := it.Item()
    
    			value, err := item.ValueCopy(nil)
    			if err != nil {
    				return err
    			}
    
    			consumer(item.Key()[len(this.prefix):], value)
    		}
    		return nil
    	})
    	return err
    }