[IndexedDB] Adding txn, value, and key size metrics

content/browser/indexed_db/leveldb/leveldb_transaction.cc

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/browser/indexed_db/leveldb/leveldb_transaction.h"
 
 #include "base/logging.h"
 #include "base/memory/ptr_util.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/time/time.h"
 #include "content/browser/indexed_db/indexed_db_tracing.h"
 #include "content/browser/indexed_db/leveldb/leveldb_database.h"
 #include "content/browser/indexed_db/leveldb/leveldb_write_batch.h"
 #include "third_party/leveldatabase/src/include/leveldb/db.h"
 
 using base::StringPiece;
 
 namespace {
 
 // Tests if the given key is before the end of a range, which
 // may have an open (exclusive) or closed (inclusive) bound.
 bool IsKeyBeforeEndOfRange(const content::LevelDBComparator* comparator,
                            const StringPiece& key,
                            const StringPiece& end,
                            bool open) {
   return (open ? comparator->Compare(key, end) < 0
                : comparator->Compare(key, end) <= 0);
 }
-
 }  // namespace
 
 namespace content {
 
 LevelDBTransaction::LevelDBTransaction(LevelDBDatabase* db)
     : db_(db),
       snapshot_(db),
       comparator_(db->Comparator()),
       data_comparator_(comparator_),
       data_(data_comparator_) {}
 
 LevelDBTransaction::Record::Record() {}
 LevelDBTransaction::Record::~Record() {}
 
 LevelDBTransaction::~LevelDBTransaction() {}
 
 void LevelDBTransaction::Set(const StringPiece& key,
                              std::string* value,
                              bool deleted) {
   DCHECK(!finished_);
   DataType::iterator it = data_.find(key);
 
   if (it == data_.end()) {
     std::unique_ptr<Record> record = base::MakeUnique<Record>();
+    size_ += SizeOfRecordInMap(key.size()) + value->size();
     record->key.assign(key.begin(), key.end() - key.begin());
     record->value.swap(*value);
     record->deleted = deleted;
     data_[record->key] = std::move(record);
     NotifyIterators();
     return;
   }
+  size_ += value->size();
   it->second->value.swap(*value);
+  size_ -= value->size();
   it->second->deleted = deleted;
 }
 
 void LevelDBTransaction::Put(const StringPiece& key, std::string* value) {
   Set(key, value, false);
 }
 
 void LevelDBTransaction::Remove(const StringPiece& key) {
   std::string empty;
   Set(key, &empty, true);
@@ skipping 84 matching lines @@
 
 std::unique_ptr<LevelDBIterator> LevelDBTransaction::CreateIterator() {
   return TransactionIterator::Create(this);
 }
 
 std::unique_ptr<LevelDBTransaction::DataIterator>
 LevelDBTransaction::DataIterator::Create(LevelDBTransaction* transaction) {
   return base::WrapUnique(new DataIterator(transaction));
 }
 
+constexpr uint64_t LevelDBTransaction::SizeOfRecordInMap(size_t key_size) {
+  return sizeof(Record) + key_size * 2;
+}
+
 bool LevelDBTransaction::DataIterator::IsValid() const {
   return iterator_ != data_->end();
 }
 
 leveldb::Status LevelDBTransaction::DataIterator::SeekToLast() {
   iterator_ = data_->end();
   if (iterator_ != data_->begin())
     --iterator_;
   return leveldb::Status::OK();
 }
@@ skipping 191 matching lines @@
   return db_iterator_->IsDetached();
 }
 
 void LevelDBTransaction::TransactionIterator::DataChanged() {
   data_changed_ = true;
 }
 
 void LevelDBTransaction::TransactionIterator::Delete() {
   DCHECK(IsValid());
   if (current_ == data_iterator_.get()) {
+    transaction_->size_ -= data_iterator_->Value().size();
     data_iterator_->Delete();
   } else {
     std::unique_ptr<Record> record = base::MakeUnique<Record>();
     record->key = Key().as_string();
     record->deleted = true;
+    transaction_->size_ +=
+        LevelDBTransaction::SizeOfRecordInMap(record->key.size());
     transaction_->data_[record->key] = std::move(record);
   }
 }
 
 void LevelDBTransaction::TransactionIterator::RefreshDataIterator() const {
   DCHECK(data_changed_);
 
   data_changed_ = false;
 
   if (data_iterator_->IsValid() && data_iterator_.get() == current_) {
@@ skipping 147 matching lines @@
 
   leveldb::Status s = db_->Write(*write_batch_);
   if (s.ok()) {
     finished_ = true;
     write_batch_->Clear();
   }
   return s;
 }
 
 }  // namespace content