LCOV - code coverage report
Current view: top level - db - db_table.cc (source / functions) Hit Total Coverage
Test: OpenSDN C/C++ coverage (all TARGET_SET jobs) Lines: 335 354 94.6 %
Date: 2026-06-08 02:02:55 Functions: 64 71 90.1 %
Legend: Lines: hit not hit

          Line data    Source code
       1             : /*
       2             :  * Copyright (c) 2013 Juniper Networks, Inc. All rights reserved.
       3             :  */
       4             : 
       5             : #include <vector>
       6             : #include <atomic>
       7             : 
       8             : #include <tbb/spin_rw_mutex.h>
       9             : 
      10             : #include <boost/bind/bind.hpp>
      11             : #include <boost/foreach.hpp>
      12             : #include <boost/dynamic_bitset.hpp>
      13             : #include <boost/type_traits.hpp>
      14             : 
      15             : #include "base/compiler.h"
      16             : #include "base/logging.h"
      17             : #include "base/task_annotations.h"
      18             : #include "base/time_util.h"
      19             : #include "db/db.h"
      20             : #include "db/db_partition.h"
      21             : #include "db/db_table.h"
      22             : #include "db/db_table_partition.h"
      23             : #include "db/db_table_walk_mgr.h"
      24             : #include "db/db_types.h"
      25             : 
      26             : class DBEntry;
      27             : class DBEntryBase;
      28             : 
      29             : using namespace std;
      30             : using namespace boost::placeholders;
      31             : 
      32     1281633 : DBRequest::DBRequest() : oper(static_cast<DBOperation>(0)) {
      33     1281608 : }
      34             : 
      35     1282901 : DBRequest::~DBRequest() {
      36             : #if defined(__GNUC__)
      37             : #if (__GNUC_PREREQ(4, 2) > 0)
      38             :     boost::has_virtual_destructor<DBRequestKey>::type key_has_destructor;
      39             :     boost::has_virtual_destructor<DBRequestData>::type data_has_destructor;
      40     1282901 :     assert(key_has_destructor && data_has_destructor);
      41             : #endif
      42             : #endif
      43     1282807 : }
      44             : 
      45      648453 : void DBRequest::Swap(DBRequest *rhs) {
      46      648453 :     swap(oper, rhs->oper);
      47      648455 :     swap(key, rhs->key);
      48      648448 :     swap(data, rhs->data);
      49      648431 : }
      50             : 
      51             : // we need copy to be able to resize vector of atomics
      52             : // therefore we don't need the same value in both instances
      53             : template<typename _Tp>
      54             : struct AtomicWithCopy : public std::atomic<_Tp> {
      55             :     // Inherit constructors
      56             :     using std::atomic<_Tp>::atomic;
      57             : 
      58             :     // Bring in base class operators and methods
      59             :     using std::atomic<_Tp>::operator=;
      60             :     using std::atomic<_Tp>::load;
      61             :     using std::atomic<_Tp>::store;
      62             : 
      63             :     // Custom constructor to handle copy from another AtomicWithCopy (must perform a load/store)
      64     1011209 :     AtomicWithCopy(const AtomicWithCopy& other) : std::atomic<_Tp>(other.load()) {}
      65             :     
      66             :     // Custom assignment operator
      67      792147 :     AtomicWithCopy& operator=(const AtomicWithCopy& other) {
      68     1584292 :         this->store(other.load());
      69      792143 :         return *this;
      70             :     }
      71             : };
      72             : 
      73             : class DBTableBase::ListenerInfo {
      74             : public:
      75             :     typedef vector<ChangeCallback> CallbackList;
      76             :     typedef vector<string> NameList;
      77             :     typedef vector<AtomicWithCopy<uint64_t>> StateCountList;
      78             : 
      79     1529827 :     explicit ListenerInfo(const string &table_name) :
      80     1529827 :         db_state_accounting_(true) {
      81     1529756 :         if (table_name.find("__ifmap_") != string::npos) {
      82             :             // TODO need to have unconditional DB state accounting
      83             :             // for now skipp DB State accounting for ifmap tables
      84     1232851 :             db_state_accounting_ = false;
      85             :         }
      86     1529764 :     }
      87             : 
      88      792624 :     DBTableBase::ListenerId Register(ChangeCallback callback,
      89             :         const string &name) {
      90      792624 :         tbb::spin_rw_mutex::scoped_lock write_lock(rw_mutex_, true);
      91      792795 :         size_t i = bmap_.find_first();
      92      792517 :         if (i == bmap_.npos) {
      93      792077 :             i = callbacks_.size();
      94      792068 :             callbacks_.push_back(callback);
      95      792175 :             names_.push_back(name);
      96      792149 :             state_count_.resize(i + 1);
      97      792149 :             state_count_[i] = AtomicWithCopy<uint64_t>(0);
      98             :         } else {
      99         440 :             bmap_.reset(i);
     100         440 :             if (bmap_.none()) {
     101         320 :                 bmap_.clear();
     102             :             }
     103         440 :             callbacks_[i] = callback;
     104         440 :             names_[i] = name;
     105         440 :             state_count_[i] = 0;
     106             :         }
     107      792801 :         return i;
     108      792825 :     }
     109             : 
     110      792795 :     void Unregister(ListenerId listener) {
     111      792795 :         tbb::spin_rw_mutex::scoped_lock write_lock(rw_mutex_, true);
     112      792794 :         callbacks_[listener] = NULL;
     113      792793 :         names_[listener] = "";
     114             :         // During Unregister Listener should have cleaned up,
     115             :         // DB states from all the entries in this table.
     116      792793 :         assert(state_count_[listener] == 0);
     117      792793 :         if ((size_t) listener == callbacks_.size() - 1) {
     118     1256724 :             while (!callbacks_.empty() && callbacks_.back() == NULL) {
     119      792300 :                 callbacks_.pop_back();
     120      792299 :                 names_.pop_back();
     121      792300 :                 state_count_.pop_back();
     122             :             }
     123      464425 :             if (bmap_.size() > callbacks_.size()) {
     124      225779 :                 bmap_.resize(callbacks_.size());
     125             :             }
     126             :         } else {
     127      328368 :             if ((size_t) listener >= bmap_.size()) {
     128      326587 :                 bmap_.resize(listener + 1);
     129             :             }
     130      328368 :             bmap_.set(listener);
     131             :         }
     132      792794 :     }
     133             : 
     134             :     // concurrency: called from DBPartition task.
     135     2517239 :     void RunNotify(DBTablePartBase *tpart, DBEntryBase *entry) {
     136     2517239 :         tbb::spin_rw_mutex::scoped_lock read_lock(rw_mutex_, false);
     137     2518187 :         for (CallbackList::iterator iter = callbacks_.begin();
     138     8703404 :              iter != callbacks_.end(); ++iter) {
     139     6185283 :             if (*iter != NULL) {
     140     6160374 :                 ChangeCallback cb = *iter;
     141     6159567 :                 (cb)(tpart, entry);
     142     6160971 :             }
     143             :         }
     144     2516283 :     }
     145             : 
     146     3853923 :     void AddToDBStateCount(ListenerId listener, int count) {
     147     3853923 :         if (db_state_accounting_ && listener != DBTableBase::kInvalidId) {
     148     3637500 :             tbb::spin_rw_mutex::scoped_lock read_lock(rw_mutex_, false);
     149     3639173 :             state_count_[listener] += count;
     150     3640072 :         }
     151     3856241 :     }
     152             : 
     153     1299176 :     uint64_t GetDBStateCount(ListenerId listener) {
     154     1299176 :         assert(db_state_accounting_ && listener != DBTableBase::kInvalidId);
     155     1299189 :         tbb::spin_rw_mutex::scoped_lock read_lock(rw_mutex_, false);
     156     2599665 :         return state_count_[listener];
     157     1299554 :     }
     158             : 
     159        6539 :     void FillListeners(vector<ShowTableListener> *listeners) const {
     160        6539 :         tbb::spin_rw_mutex::scoped_lock read_lock(rw_mutex_, false);
     161        6709 :         ListenerId id = 0;
     162        6709 :         for (CallbackList::const_iterator iter = callbacks_.begin();
     163       15854 :              iter != callbacks_.end(); ++iter, ++id) {
     164        9278 :             if (*iter != NULL) {
     165        9265 :                 ShowTableListener item;
     166        9239 :                 item.id = id;
     167        9239 :                 item.name = names_[id];
     168        9244 :                 item.state_count = state_count_[id];
     169        9224 :                 listeners->push_back(item);
     170        9153 :             }
     171             :         }
     172        6584 :     }
     173             : 
     174     3755619 :     bool empty() const {
     175     3755619 :         tbb::spin_rw_mutex::scoped_lock read_lock(rw_mutex_, false);
     176     7512771 :         return callbacks_.empty();
     177     3755757 :     }
     178             : 
     179       10317 :     size_t size() const {
     180       10317 :         tbb::spin_rw_mutex::scoped_lock read_lock(rw_mutex_, false);
     181       20634 :         return (callbacks_.size() - bmap_.count());
     182       10317 :     }
     183             : 
     184             : private:
     185             :     bool db_state_accounting_;
     186             :     CallbackList callbacks_;
     187             :     NameList names_;
     188             :     StateCountList state_count_;
     189             :     mutable tbb::spin_rw_mutex rw_mutex_;
     190             :     boost::dynamic_bitset<> bmap_;      // free list.
     191             : };
     192             : 
     193     1529822 : DBTableBase::DBTableBase(DB *db, const string &name)
     194     1529802 :         : db_(db), name_(name), info_(new ListenerInfo(name)),
     195     1529822 :           enqueue_count_(0), input_count_(0), notify_count_(0) {
     196     1529765 :     walker_count_ = 0;
     197     1529884 :     walk_request_count_ = 0;
     198     1529856 :     walk_complete_count_ = 0;
     199     1529886 :     walk_cancel_count_ = 0;
     200     1529884 :     walk_again_count_ = 0;
     201     1529883 :     walk_count_ = 0;
     202     1529873 : }
     203             : 
     204     1529892 : DBTableBase::~DBTableBase() {
     205     1529892 : }
     206             : 
     207      792740 : DBTableBase::ListenerId DBTableBase::Register(ChangeCallback callback,
     208             :     const string &name) {
     209      792740 :     return info_->Register(callback, name);
     210             : }
     211             : 
     212      792795 : void DBTableBase::Unregister(ListenerId listener) {
     213      792795 :     info_->Unregister(listener);
     214             :     // If a table is marked for deletion, then we may trigger the deletion
     215             :     // process when the last client is removed
     216      792794 :     if (info_->empty())
     217      360684 :         RetryDelete();
     218      792795 : }
     219             : 
     220      647185 : bool DBTableBase::Enqueue(DBRequest *req) {
     221      647185 :     DBTablePartBase *tpart = GetTablePartition(req->key.get());
     222      647174 :     DBPartition *partition = db_->GetPartition(tpart->index());
     223      647172 :     enqueue_count_++;
     224      647172 :     return partition->EnqueueRequest(tpart, NULL, req);
     225             : }
     226             : 
     227      370968 : void DBTableBase::EnqueueRemove(DBEntryBase *db_entry) {
     228      370968 :     DBTablePartBase *tpart = GetTablePartition(db_entry);
     229      370905 :     DBPartition *partition = db_->GetPartition(tpart->index());
     230      370893 :     partition->EnqueueRemove(tpart, db_entry);
     231      371048 : }
     232             : 
     233     2517437 : void DBTableBase::RunNotify(DBTablePartBase *tpart, DBEntryBase *entry) {
     234     2517437 :     notify_count_++;
     235     2517437 :     info_->RunNotify(tpart, entry);
     236     2518082 : }
     237             : 
     238     3854983 : void DBTableBase::AddToDBStateCount(ListenerId listener, int count) {
     239     3854983 :     info_->AddToDBStateCount(listener, count);
     240     3856252 : }
     241             : 
     242     1299437 : uint64_t DBTableBase::GetDBStateCount(ListenerId listener) {
     243     1299437 :     return info_->GetDBStateCount(listener);
     244             : }
     245             : 
     246      288622 : bool DBTableBase::MayDelete() const {
     247      288622 :     if (HasListeners()) {
     248           1 :         return false;
     249             :     }
     250      288621 :     if (HasWalkers()) {
     251           0 :         return false;
     252             :     }
     253      288621 :     if (!empty()) {
     254           0 :         return false;
     255             :     }
     256             : 
     257      288621 :     return true;
     258             : }
     259             : 
     260     2963017 : bool DBTableBase::HasListeners() const {
     261     2963017 :     return !info_->empty();
     262             : }
     263             : 
     264       10317 : size_t DBTableBase::GetListenerCount() const {
     265       10317 :     return info_->size();
     266             : }
     267             : 
     268        6545 : void DBTableBase::FillListeners(vector<ShowTableListener> *listeners) const {
     269        6545 :     info_->FillListeners(listeners);
     270        6711 : }
     271             : 
     272             : class DBTable::WalkWorker : public Task {
     273             : public:
     274             :     WalkWorker(TableWalker *walker, int db_partition_id);
     275             : 
     276             :     virtual bool Run();
     277             : 
     278           0 :     std::string Description() const { return "DBTable::WalkWorker"; }
     279             : 
     280             : private:
     281             :     // Store the last visited node to continue walk
     282             :     std::unique_ptr<DBRequestKey> walk_ctx_;
     283             : 
     284             :     // Table partition for which this worker was created
     285             :     DBTablePartition *tbl_partition_;
     286             : 
     287             :     TableWalker *walker_;
     288             : };
     289             : 
     290             : class DBTable::TableWalker {
     291             : public:
     292     1529864 :     TableWalker(DBTable *table) : table_(table) {
     293     1529825 :         pending_workers_ = 0;
     294     1529886 :     }
     295             : 
     296             :     void StartWalk();
     297             : 
     298     1053924 :     DBTable *table() {
     299     1053924 :         return table_;
     300             :     }
     301             : 
     302      226254 :     void ClearWalkWorks() {
     303      226254 :         worker_tasks_.clear();
     304      226254 :     }
     305             : 
     306             :     DBTable *table_;
     307             :     // check whether iteration is completed on all Table Partition
     308             :     std::atomic<uint16_t> pending_workers_;
     309             :     // For debugging purpose. Few of the tasks in this list could has finished
     310             :     // executing and destroyed. List of workers are useful in debugging with
     311             :     // gdb/gcore to see the current state of the walk and walk_context
     312             :     std::list<Task *> worker_tasks_;
     313             : };
     314             : 
     315      793023 : bool DBTable::WalkWorker::Run() {
     316      793023 :     int count = 0;
     317      793023 :     DBRequestKey *key_resume = walk_ctx_.get();
     318      793021 :     DBTable *table = walker_->table();
     319      793013 :     int max_walk_entry_count = table->GetWalkIterationToYield();
     320             :     DBEntry *entry;
     321             : 
     322      793007 :     if (key_resume != NULL) {
     323      662631 :         std::unique_ptr<const DBEntryBase> start;
     324      662631 :         start = table->AllocEntry(key_resume);
     325             :         // Find matching or next in sort order
     326      662426 :         entry = tbl_partition_->lower_bound(start.get());
     327      662691 :     } else {
     328      130376 :         entry = tbl_partition_->GetFirst();
     329             :     }
     330      792909 :     if (entry == NULL) {
     331         665 :         goto walk_done;
     332             :     }
     333             : 
     334     1584516 :     for (DBEntry *next = NULL; entry; entry = next) {
     335     1454724 :         next = tbl_partition_->GetNext(entry);
     336     1454798 :         if (count == max_walk_entry_count) {
     337             :             // store the context
     338      662628 :             walk_ctx_ = entry->GetDBRequestKey();
     339      661976 :             return false;
     340             :         }
     341             : 
     342             :         // Invoke walker function
     343      792170 :         bool more = table->InvokeWalkCb(tbl_partition_, entry);
     344      792250 :         if (!more) {
     345         134 :             break;
     346             :         }
     347             : 
     348      792116 :         db_walker_wait();
     349      792272 :         count++;
     350             :     }
     351             : 
     352      129792 : walk_done:
     353             :     // Check whether all other walks on the table is completed
     354      130591 :     long num_walkers_on_tpart = walker_->pending_workers_.fetch_sub(1);
     355      130591 :     if (num_walkers_on_tpart == 1) {
     356      106579 :         table->WalkDone();
     357             :     }
     358      130455 :     return true;
     359             : }
     360             : 
     361      130455 : DBTable::WalkWorker::WalkWorker(TableWalker *walker, int db_partition_id)
     362      130455 :     : Task(walker->table()->GetWalkerTaskId(), db_partition_id), walker_(walker) {
     363      130455 :     tbl_partition_ = static_cast<DBTablePartition *>
     364      130455 :         (walker_->table()->GetTablePartition(db_partition_id));
     365      130455 : }
     366             : 
     367      226254 : void DBTable::TableWalker::StartWalk() {
     368      226254 :     CHECK_CONCURRENCY("db::Walker");
     369      226254 :     assert(pending_workers_ == 0);
     370      737832 :     for (int i = 0; i < table_->PartitionCount(); i++) {
     371             :         DBTablePartition *partition = static_cast<DBTablePartition *>(
     372      511578 :             table_->GetTablePartition(i));
     373      511578 :         if (!partition->size()) continue;
     374      130455 :         worker_tasks_.push_back(new WalkWorker(this, i));
     375      130455 :         pending_workers_++;
     376             :     }
     377      226254 :     if (pending_workers_ == 0) {
     378      119675 :         table_->WalkDone();
     379             :     } else {
     380      106579 :         TaskScheduler *scheduler = TaskScheduler::GetInstance();
     381      237034 :         for (auto *task : worker_tasks_) scheduler->Enqueue(task);
     382             :     }
     383      226254 : }
     384             : 
     385             : ///////////////////////////////////////////////////////////
     386             : // Implementation of DBTable methods
     387             : ///////////////////////////////////////////////////////////
     388     1529819 : DBTable::DBTable(DB *db, const string &name)
     389             :     : DBTableBase(db, name),
     390     1529875 :       walker_(new TableWalker(this)),
     391     3059593 :       walker_task_id_(db->task_id()) {
     392             : 
     393             :     static bool init_ = false;
     394             :     static int iter_to_yield_env_ = 0;
     395             : 
     396     1529763 :     if (!init_) {
     397             :         // XXX To be used for testing purposes only.
     398         158 :         char *count_str = getenv("DB_ITERATION_TO_YIELD");
     399         158 :         if (count_str) {
     400         158 :             iter_to_yield_env_ = strtol(count_str, NULL, 0);
     401             :         } else {
     402           0 :             iter_to_yield_env_ = kIterationToYield;
     403             :         }
     404         158 :         init_ = true;
     405             :     }
     406     1529763 :     max_walk_iteration_to_yield_ = iter_to_yield_env_;
     407     1529763 : }
     408             : 
     409     1529892 : DBTable::~DBTable() {
     410     1529892 :     STLDeleteValues(&partitions_);
     411     1529892 : }
     412             : 
     413     1521722 : void DBTable::Init() {
     414     3596803 :     for (int i = 0; i < PartitionCount(); i++) {
     415     2075260 :         partitions_.push_back(AllocPartition(i));
     416             :     }
     417     1521506 : }
     418             : 
     419     2075192 : DBTablePartition *DBTable::AllocPartition(int index) {
     420     2075192 :     return new DBTablePartition(this, index);
     421             : }
     422             : 
     423      226254 : void DBTable::StartWalk() {
     424      226254 :     CHECK_CONCURRENCY("db::Walker");
     425      226254 :     incr_walk_count();
     426      226254 :     walker_->StartWalk();
     427      226254 : }
     428             : 
     429      212959 : DBEntry *DBTable::Add(const DBRequest *req) {
     430      212959 :     return AllocEntry(req->key.get()).release();
     431             : }
     432             : 
     433        1445 : void DBTable::Change(DBEntryBase *entry) {
     434        1445 :     DBTablePartBase *tpart = GetTablePartition(entry);
     435        1445 :     tpart->Notify(entry);
     436        1445 : }
     437             : 
     438           0 : bool DBTable::OnChange(DBEntry *entry, const DBRequest *req) {
     439           0 :     return true;
     440             : }
     441             : 
     442           0 : bool DBTable::Delete(DBEntry *entry, const DBRequest *req) {
     443           0 :     return true;
     444             : }
     445             : 
     446     1412735 : int DBTable::PartitionCount() const {
     447     1412735 :     return DB::PartitionCount();
     448             : }
     449             : 
     450    16220222 : static size_t HashToPartition(size_t hash) {
     451    16220222 :     return hash % DB::PartitionCount();
     452             : }
     453             : 
     454    19676555 : DBTablePartBase *DBTable::GetTablePartition(const int index) {
     455    19676555 :     return partitions_[index];
     456             : }
     457             : 
     458       24086 : const DBTablePartBase *DBTable::GetTablePartition(const int index) const {
     459       24086 :     return partitions_[index];
     460             : }
     461             : 
     462      708273 : DBTablePartBase *DBTable::GetTablePartition(const DBRequestKey *key) {
     463      708273 :     int id = HashToPartition(Hash(key));
     464      708255 :     return GetTablePartition(id);
     465             : }
     466             : 
     467           0 : const DBTablePartBase *DBTable::GetTablePartition(
     468             :         const DBRequestKey *key) const {
     469           0 :     int id = HashToPartition(Hash(key));
     470           0 :     return GetTablePartition(id);
     471             : }
     472             : 
     473    15029636 : DBTablePartBase *DBTable::GetTablePartition(const DBEntryBase *entry) {
     474    15029636 :     const DBEntry *gentry = static_cast<const DBEntry *>(entry);
     475    15029636 :     size_t id = HashToPartition(Hash(gentry));
     476    15020506 :     return GetTablePartition(id);
     477             : }
     478             : 
     479       24086 : const DBTablePartBase *DBTable::GetTablePartition(
     480             :         const DBEntryBase *entry) const {
     481       24086 :     const DBEntry *gentry = static_cast<const DBEntry *>(entry);
     482       24086 :     size_t id = HashToPartition(Hash(gentry));
     483       24086 :     return GetTablePartition(id);
     484             : }
     485             : 
     486             : // Find DB Entry without taking lock. Calling routine must ensure its
     487             : // running in exclusion with DB task
     488          98 : DBEntry *DBTable::FindNoLock(const DBEntry *entry) {
     489          98 :     size_t id = HashToPartition(Hash(entry));
     490             :     DBTablePartition *tbl_partition =
     491          98 :         static_cast<DBTablePartition *>(GetTablePartition(id));
     492          98 :     return tbl_partition->FindNoLock(entry);
     493             : }
     494             : 
     495      383151 : DBEntry *DBTable::Find(const DBEntry *entry) {
     496      383151 :     size_t id = HashToPartition(Hash(entry));
     497             :     DBTablePartition *tbl_partition =
     498      383150 :         static_cast<DBTablePartition *>(GetTablePartition(id));
     499      383150 :     return tbl_partition->Find(entry);
     500             : }
     501             : 
     502           0 : const DBEntry *DBTable::Find(const DBEntry *entry) const {
     503           0 :     return const_cast<DBTable *>(this)->Find(entry);
     504             : }
     505             : 
     506             : // Find DB Entry without taking lock. Calling routine must ensure its
     507             : // running in exclusion with DB task
     508          17 : DBEntry *DBTable::FindNoLock(const DBRequestKey *key) {
     509          17 :     int id = HashToPartition(Hash(key));
     510             :     DBTablePartition *tbl_partition =
     511          17 :     static_cast<DBTablePartition *>(GetTablePartition(id));
     512          17 :     return tbl_partition->FindNoLock(key);
     513             : }
     514             : 
     515       84504 : DBEntry *DBTable::Find(const DBRequestKey *key, int id) {
     516       84504 :     if (id == -1)
     517       84476 :         id = HashToPartition(Hash(key));
     518             :     DBTablePartition *tbl_partition =
     519       84504 :     static_cast<DBTablePartition *>(GetTablePartition(id));
     520       84504 :     return tbl_partition->Find(key);
     521             : }
     522             : 
     523         260 : const DBEntry *DBTable::Find(const DBRequestKey *key, int id) const {
     524         260 :     return const_cast<DBTable *>(this)->Find(key, id);
     525             : }
     526             : 
     527             : //
     528             : // Concurrency: called from task that's mutually exclusive with db::DBTable
     529             : // or db::IFMapTable as applicable.
     530             : //
     531             : // Calculate the size across all partitions.
     532             : //
     533     1894931 : size_t DBTable::Size() const {
     534     1894931 :     size_t total = 0;
     535     1894931 :     for (vector<DBTablePartition *>::const_iterator iter = partitions_.begin();
     536     6997535 :          iter != partitions_.end(); iter++) {
     537     5102660 :         total += (*iter)->size();
     538             :     }
     539     1894892 :     return total;
     540             : }
     541             : 
     542        1111 : void DBTable::Input(DBTablePartition *tbl_partition, DBClient *client,
     543             :                     DBRequest *req) {
     544             :     DBRequestKey *key =
     545        1111 :         static_cast<DBRequestKey *>(req->key.get());
     546        1111 :     DBEntry *entry = NULL;
     547             : 
     548        1111 :     entry = tbl_partition->Find(key);
     549        1111 :     if (req->oper == DBRequest::DB_ENTRY_ADD_CHANGE) {
     550         613 :         if (entry) {
     551         331 :             if (OnChange(entry, req) || entry->IsDeleted()) {
     552             :                 // The entry may currently be marked as deleted.
     553          52 :                 entry->ClearDelete();
     554          52 :                 tbl_partition->Change(entry);
     555             :             }
     556             :         } else {
     557         282 :             if ((entry = Add(req)) != NULL) {
     558         282 :                 tbl_partition->Add(entry);
     559             :             }
     560             :         }
     561         498 :     } else if (req->oper == DBRequest::DB_ENTRY_DELETE) {
     562         304 :         if (entry) {
     563         304 :             if (Delete(entry, req)) {
     564         304 :                 tbl_partition->Delete(entry);
     565             :             }
     566             :         }
     567         194 :     } else if (req->oper == DBRequest::DB_ENTRY_NOTIFY) {
     568         194 :         if (entry) {
     569         194 :             tbl_partition->Notify(entry);
     570             :         }
     571             :     } else {
     572           0 :         assert(0);
     573             :     }
     574        1111 : }
     575             : 
     576           1 : void DBTable::DBStateClear(DBTable *table, ListenerId id) {
     577           1 :     DBEntryBase *next = NULL;
     578             : 
     579           2 :     for (int i = 0; i < table->PartitionCount(); ++i) {
     580             :         DBTablePartition *partition = static_cast<DBTablePartition *>(
     581           1 :             table->GetTablePartition(i));
     582             : 
     583           1 :         for (DBEntryBase *entry = partition->GetFirst(); entry; entry = next) {
     584           0 :             next = partition->GetNext(entry);
     585           0 :             DBState *state = entry->GetState(table, id);
     586           0 :             if (state) {
     587           0 :                 entry->ClearState(table, id);
     588           0 :                 delete state;
     589             :             }
     590             :         }
     591             :     }
     592           1 : }
     593             : 
     594             : //
     595             : // Callback for table walk triggered by NotifyAllEntries.
     596             : //
     597      129494 : bool DBTable::WalkCallback(DBTablePartBase *tpart, DBEntryBase *entry) {
     598      129494 :     tpart->Notify(entry);
     599      129498 :     return true;
     600             : }
     601             : 
     602             : //
     603             : // Callback for completion of table walk triggered by NotifyAllEntries.
     604             : //
     605       61319 : void DBTable::WalkCompleteCallback(DBTableBase *tbl_base) {
     606       61319 :     walk_ref_.reset();
     607       61319 : }
     608             : 
     609             : //
     610             : // Concurrency: called from task that's mutually exclusive with db::DBTable
     611             : // or db::IFMapTable as applicable.
     612             : //
     613             : // Trigger notification of all entries to all listeners.
     614             : // Should be used sparingly e.g. to handle significant configuration change.
     615             : //
     616             : // The walk callback just turns around and puts the DBentryBase on the change
     617             : // list.
     618             : //
     619             : // If the walk is already running, it is allowed to complete and WalkAgain API
     620             : // is invoked to trigger walk on current walk completion.
     621             : //
     622       63772 : void DBTable::NotifyAllEntries() {
     623       63772 :     CHECK_CONCURRENCY("bgp::Config", "bgp::ConfigHelper", "bgp::RTFilter",
     624             :                       "db::DBTable");
     625             : 
     626       63773 :     if (empty())
     627        2438 :         return;
     628             : 
     629       61332 :     if (walk_ref_ == NULL) {
     630             :         walk_ref_ =
     631      122638 :             AllocWalker(boost::bind(&DBTable::WalkCallback, this, _1, _2),
     632       61319 :                     boost::bind(&DBTable::WalkCompleteCallback, this, _2));
     633       61319 :         WalkTable(walk_ref_);
     634             :     } else {
     635          13 :         WalkAgain(walk_ref_);
     636             :     }
     637             : }
     638             : 
     639      219262 : DBTable::DBTableWalkRef DBTable::AllocWalker(WalkFn walk_fn,
     640             :                                              WalkCompleteFn walk_complete) {
     641      219262 :     DBTableWalkMgr *walk_mgr = database()->GetWalkMgr();
     642      219262 :     return walk_mgr->AllocWalker(this, walk_fn, walk_complete);
     643             : }
     644             : 
     645      157707 : void DBTable::ReleaseWalker(DBTable::DBTableWalkRef &walk) {
     646      157707 :     DBTableWalkMgr *walk_mgr = database()->GetWalkMgr();
     647      157707 :     walk_mgr->ReleaseWalker(walk);
     648      157707 :     return;
     649             : }
     650             : 
     651      226886 : void DBTable::WalkTable(DBTable::DBTableWalkRef walk) {
     652      226886 :     DBTableWalkMgr *walk_mgr = database()->GetWalkMgr();
     653      226886 :     walk_mgr->WalkTable(walk);
     654      226886 :     return;
     655             : }
     656             : 
     657       29106 : void DBTable::WalkAgain(DBTable::DBTableWalkRef walk) {
     658       29106 :     DBTableWalkMgr *walk_mgr = database()->GetWalkMgr();
     659       29106 :     walk_mgr->WalkAgain(walk);
     660       29106 :     return;
     661             : }
     662             : 
     663      792360 : bool DBTable::InvokeWalkCb(DBTablePartBase *part, DBEntryBase *entry) {
     664      792360 :     DBTableWalkMgr *walk_mgr = database()->GetWalkMgr();
     665      792230 :     return walk_mgr->InvokeWalkCb(part, entry);
     666             : }
     667             : 
     668      226254 : void DBTable::WalkDone() {
     669      226254 :     incr_walk_complete_count();
     670      226254 :     walker_->ClearWalkWorks();
     671      226254 :     DBTableWalkMgr *walk_mgr = database()->GetWalkMgr();
     672      226254 :     return walk_mgr->WalkDone();
     673             : }

Generated by: LCOV version 1.14