1 files changed, 43 insertions, 5 deletions
diff --git a/ethdb/database.go b/ethdb/database.go
index 5faf8c4e0..32ce845a0 100644
--- a/ethdb/database.go
+++ b/ethdb/database.go
@@ -1,9 +1,12 @@
 package ethdb
 
 import (
+	"time"
+
 	"github.com/ethereum/go-ethereum/compression/rle"
 	"github.com/ethereum/go-ethereum/logger"
 	"github.com/ethereum/go-ethereum/logger/glog"
+	"github.com/rcrowley/go-metrics"
 	"github.com/syndtr/goleveldb/leveldb"
 	"github.com/syndtr/goleveldb/leveldb/errors"
 	"github.com/syndtr/goleveldb/leveldb/iterator"
@@ -13,10 +16,15 @@ import (
 var OpenFileLimit = 64
 
 type LDBDatabase struct {
-	// filename for reporting
-	fn string
-	// LevelDB instance
-	db *leveldb.DB
+	fn string      // filename for reporting
+	db *leveldb.DB // LevelDB instance
+
+	GetTimer   metrics.Timer // Timer for measuring the database get request counts and latencies
+	PutTimer   metrics.Timer // Timer for measuring the database put request counts and latencies
+	DelTimer   metrics.Timer // Timer for measuring the database delete request counts and latencies
+	MissMeter  metrics.Meter // MEter for measuring the missed database get requests
+	ReadMeter  metrics.Meter // Meter for measuring the database get request data usage
+	WriteMeter metrics.Meter // Meter for measuring the database put request data usage
 }
 
 // NewLDBDatabase returns a LevelDB wrapped object. LDBDatabase does not persist data by
@@ -43,20 +51,50 @@ func NewLDBDatabase(file string) (*LDBDatabase, error) {
 
 // Put puts the given key / value to the queue
 func (self *LDBDatabase) Put(key []byte, value []byte) error {
-	return self.db.Put(key, rle.Compress(value), nil)
+	// Measure the database put latency, if requested
+	if self.PutTimer != nil {
+		start := time.Now()
+		defer self.PutTimer.UpdateSince(start)
+	}
+	// Generate the data to write to disk, update the meter and write
+	dat := rle.Compress(value)
+
+	if self.WriteMeter != nil {
+		self.WriteMeter.Mark(int64(len(dat)))
+	}
+	return self.db.Put(key, dat, nil)
 }
 
 // Get returns the given key if it's present.
 func (self *LDBDatabase) Get(key []byte) ([]byte, error) {
+	// Measure the database get latency, if requested
+	if self.GetTimer != nil {
+		start := time.Now()
+		defer self.GetTimer.UpdateSince(start)
+	}
+	// Retrieve the key and increment the miss counter if not found
 	dat, err := self.db.Get(key, nil)
 	if err != nil {
+		if self.MissMeter != nil {
+			self.MissMeter.Mark(1)
+		}
 		return nil, err
 	}
+	// Otherwise update the actually retrieved amount of data
+	if self.ReadMeter != nil {
+		self.ReadMeter.Mark(int64(len(dat)))
+	}
 	return rle.Decompress(dat)
 }
 
 // Delete deletes the key from the queue and database
 func (self *LDBDatabase) Delete(key []byte) error {
+	// Measure the database delete latency, if requested
+	if self.DelTimer != nil {
+		start := time.Now()
+		defer self.DelTimer.UpdateSince(start)
+	}
+	// Execute the actual operation
 	return self.db.Delete(key, nil)
 }