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-11 01:56:02 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     1280828 : DBRequest::DBRequest() : oper(static_cast<DBOperation>(0)) {
      33     1280812 : }
      34             : 
      35     1282378 : 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     1282378 :     assert(key_has_destructor && data_has_destructor);
      41             : #endif
      42             : #endif
      43     1282290 : }
      44             : 
      45      648079 : void DBRequest::Swap(DBRequest *rhs) {
      46      648079 :     swap(oper, rhs->oper);
      47      648075 :     swap(key, rhs->key);
      48      648060 :     swap(data, rhs->data);
      49      648057 : }
      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     1011336 :     AtomicWithCopy(const AtomicWithCopy& other) : std::atomic<_Tp>(other.load()) {}
      65             :     
      66             :     // Custom assignment operator
      67      792330 :     AtomicWithCopy& operator=(const AtomicWithCopy& other) {
      68     1584655 :         this->store(other.load());
      69      792323 :         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     1530190 :     explicit ListenerInfo(const string &table_name) :
      80     1530190 :         db_state_accounting_(true) {
      81     1530128 :         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     1233151 :             db_state_accounting_ = false;
      85             :         }
      86     1530136 :     }
      87             : 
      88      792924 :     DBTableBase::ListenerId Register(ChangeCallback callback,
      89             :         const string &name) {
      90      792924 :         tbb::spin_rw_mutex::scoped_lock write_lock(rw_mutex_, true);
      91      793052 :         size_t i = bmap_.find_first();
      92      792786 :         if (i == bmap_.npos) {
      93      792303 :             i = callbacks_.size();
      94      792298 :             callbacks_.push_back(callback);
      95      792378 :             names_.push_back(name);
      96      792345 :             state_count_.resize(i + 1);
      97      792339 :             state_count_[i] = AtomicWithCopy<uint64_t>(0);
      98             :         } else {
      99         483 :             bmap_.reset(i);
     100         481 :             if (bmap_.none()) {
     101         352 :                 bmap_.clear();
     102             :             }
     103         481 :             callbacks_[i] = callback;
     104         481 :             names_[i] = name;
     105         481 :             state_count_[i] = 0;
     106             :         }
     107      793068 :         return i;
     108      793110 :     }
     109             : 
     110      793047 :     void Unregister(ListenerId listener) {
     111      793047 :         tbb::spin_rw_mutex::scoped_lock write_lock(rw_mutex_, true);
     112      793047 :         callbacks_[listener] = NULL;
     113      793045 :         names_[listener] = "";
     114             :         // During Unregister Listener should have cleaned up,
     115             :         // DB states from all the entries in this table.
     116      793045 :         assert(state_count_[listener] == 0);
     117      793045 :         if ((size_t) listener == callbacks_.size() - 1) {
     118     1257127 :             while (!callbacks_.empty() && callbacks_.back() == NULL) {
     119      792474 :                 callbacks_.pop_back();
     120      792473 :                 names_.pop_back();
     121      792473 :                 state_count_.pop_back();
     122             :             }
     123      464654 :             if (bmap_.size() > callbacks_.size()) {
     124      225789 :                 bmap_.resize(callbacks_.size());
     125             :             }
     126             :         } else {
     127      328391 :             if ((size_t) listener >= bmap_.size()) {
     128      326528 :                 bmap_.resize(listener + 1);
     129             :             }
     130      328392 :             bmap_.set(listener);
     131             :         }
     132      793046 :     }
     133             : 
     134             :     // concurrency: called from DBPartition task.
     135     2501292 :     void RunNotify(DBTablePartBase *tpart, DBEntryBase *entry) {
     136     2501292 :         tbb::spin_rw_mutex::scoped_lock read_lock(rw_mutex_, false);
     137     2502002 :         for (CallbackList::iterator iter = callbacks_.begin();
     138     8637618 :              iter != callbacks_.end(); ++iter) {
     139     6135956 :             if (*iter != NULL) {
     140     6110877 :                 ChangeCallback cb = *iter;
     141     6109952 :                 (cb)(tpart, entry);
     142     6111372 :             }
     143             :         }
     144     2500953 :     }
     145             : 
     146     3828176 :     void AddToDBStateCount(ListenerId listener, int count) {
     147     3828176 :         if (db_state_accounting_ && listener != DBTableBase::kInvalidId) {
     148     3611099 :             tbb::spin_rw_mutex::scoped_lock read_lock(rw_mutex_, false);
     149     3612260 :             state_count_[listener] += count;
     150     3613099 :         }
     151     3829927 :     }
     152             : 
     153     1287406 :     uint64_t GetDBStateCount(ListenerId listener) {
     154     1287406 :         assert(db_state_accounting_ && listener != DBTableBase::kInvalidId);
     155     1287527 :         tbb::spin_rw_mutex::scoped_lock read_lock(rw_mutex_, false);
     156     2575557 :         return state_count_[listener];
     157     1287633 :     }
     158             : 
     159        6654 :     void FillListeners(vector<ShowTableListener> *listeners) const {
     160        6654 :         tbb::spin_rw_mutex::scoped_lock read_lock(rw_mutex_, false);
     161        6732 :         ListenerId id = 0;
     162        6732 :         for (CallbackList::const_iterator iter = callbacks_.begin();
     163       16058 :              iter != callbacks_.end(); ++iter, ++id) {
     164        9362 :             if (*iter != NULL) {
     165        9357 :                 ShowTableListener item;
     166        9352 :                 item.id = id;
     167        9352 :                 item.name = names_[id];
     168        9359 :                 item.state_count = state_count_[id];
     169        9356 :                 listeners->push_back(item);
     170        9333 :             }
     171             :         }
     172        6696 :     }
     173             : 
     174     3755492 :     bool empty() const {
     175     3755492 :         tbb::spin_rw_mutex::scoped_lock read_lock(rw_mutex_, false);
     176     7511793 :         return callbacks_.empty();
     177     3755646 :     }
     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     1530176 : DBTableBase::DBTableBase(DB *db, const string &name)
     194     1530170 :         : db_(db), name_(name), info_(new ListenerInfo(name)),
     195     1530176 :           enqueue_count_(0), input_count_(0), notify_count_(0) {
     196     1530131 :     walker_count_ = 0;
     197     1530241 :     walk_request_count_ = 0;
     198     1530226 :     walk_complete_count_ = 0;
     199     1530238 :     walk_cancel_count_ = 0;
     200     1530236 :     walk_again_count_ = 0;
     201     1530240 :     walk_count_ = 0;
     202     1530228 : }
     203             : 
     204     1530242 : DBTableBase::~DBTableBase() {
     205     1530242 : }
     206             : 
     207      793012 : DBTableBase::ListenerId DBTableBase::Register(ChangeCallback callback,
     208             :     const string &name) {
     209      793012 :     return info_->Register(callback, name);
     210             : }
     211             : 
     212      793047 : void DBTableBase::Unregister(ListenerId listener) {
     213      793047 :     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      793046 :     if (info_->empty())
     217      360774 :         RetryDelete();
     218      793047 : }
     219             : 
     220      646765 : bool DBTableBase::Enqueue(DBRequest *req) {
     221      646765 :     DBTablePartBase *tpart = GetTablePartition(req->key.get());
     222      646760 :     DBPartition *partition = db_->GetPartition(tpart->index());
     223      646762 :     enqueue_count_++;
     224      646762 :     return partition->EnqueueRequest(tpart, NULL, req);
     225             : }
     226             : 
     227      369809 : void DBTableBase::EnqueueRemove(DBEntryBase *db_entry) {
     228      369809 :     DBTablePartBase *tpart = GetTablePartition(db_entry);
     229      369756 :     DBPartition *partition = db_->GetPartition(tpart->index());
     230      369743 :     partition->EnqueueRemove(tpart, db_entry);
     231      369837 : }
     232             : 
     233     2501417 : void DBTableBase::RunNotify(DBTablePartBase *tpart, DBEntryBase *entry) {
     234     2501417 :     notify_count_++;
     235     2501417 :     info_->RunNotify(tpart, entry);
     236     2501954 : }
     237             : 
     238     3828848 : void DBTableBase::AddToDBStateCount(ListenerId listener, int count) {
     239     3828848 :     info_->AddToDBStateCount(listener, count);
     240     3829858 : }
     241             : 
     242     1287603 : uint64_t DBTableBase::GetDBStateCount(ListenerId listener) {
     243     1287603 :     return info_->GetDBStateCount(listener);
     244             : }
     245             : 
     246      288648 : bool DBTableBase::MayDelete() const {
     247      288648 :     if (HasListeners()) {
     248           1 :         return false;
     249             :     }
     250      288647 :     if (HasWalkers()) {
     251           0 :         return false;
     252             :     }
     253      288647 :     if (!empty()) {
     254           0 :         return false;
     255             :     }
     256             : 
     257      288647 :     return true;
     258             : }
     259             : 
     260     2962249 : bool DBTableBase::HasListeners() const {
     261     2962249 :     return !info_->empty();
     262             : }
     263             : 
     264       10317 : size_t DBTableBase::GetListenerCount() const {
     265       10317 :     return info_->size();
     266             : }
     267             : 
     268        6659 : void DBTableBase::FillListeners(vector<ShowTableListener> *listeners) const {
     269        6659 :     info_->FillListeners(listeners);
     270        6729 : }
     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     1530217 :     TableWalker(DBTable *table) : table_(table) {
     293     1530187 :         pending_workers_ = 0;
     294     1530233 :     }
     295             : 
     296             :     void StartWalk();
     297             : 
     298     1051967 :     DBTable *table() {
     299     1051967 :         return table_;
     300             :     }
     301             : 
     302      226195 :     void ClearWalkWorks() {
     303      226195 :         worker_tasks_.clear();
     304      226195 :     }
     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      790531 : bool DBTable::WalkWorker::Run() {
     316      790531 :     int count = 0;
     317      790531 :     DBRequestKey *key_resume = walk_ctx_.get();
     318      790543 :     DBTable *table = walker_->table();
     319      790538 :     int max_walk_entry_count = table->GetWalkIterationToYield();
     320             :     DBEntry *entry;
     321             : 
     322      790539 :     if (key_resume != NULL) {
     323      659938 :         std::unique_ptr<const DBEntryBase> start;
     324      659938 :         start = table->AllocEntry(key_resume);
     325             :         // Find matching or next in sort order
     326      659755 :         entry = tbl_partition_->lower_bound(start.get());
     327      659989 :     } else {
     328      130601 :         entry = tbl_partition_->GetFirst();
     329             :     }
     330      790431 :     if (entry == NULL) {
     331         718 :         goto walk_done;
     332             :     }
     333             : 
     334     1579454 :     for (DBEntry *next = NULL; entry; entry = next) {
     335     1449440 :         next = tbl_partition_->GetNext(entry);
     336     1449459 :         if (count == max_walk_entry_count) {
     337             :             // store the context
     338      659913 :             walk_ctx_ = entry->GetDBRequestKey();
     339      659312 :             return false;
     340             :         }
     341             : 
     342             :         // Invoke walker function
     343      789546 :         bool more = table->InvokeWalkCb(tbl_partition_, entry);
     344      789834 :         if (!more) {
     345         131 :             break;
     346             :         }
     347             : 
     348      789703 :         db_walker_wait();
     349      789741 :         count++;
     350             :     }
     351             : 
     352      130014 : walk_done:
     353             :     // Check whether all other walks on the table is completed
     354      130863 :     long num_walkers_on_tpart = walker_->pending_workers_.fetch_sub(1);
     355      130863 :     if (num_walkers_on_tpart == 1) {
     356      106581 :         table->WalkDone();
     357             :     }
     358      130712 :     return true;
     359             : }
     360             : 
     361      130713 : DBTable::WalkWorker::WalkWorker(TableWalker *walker, int db_partition_id)
     362      130713 :     : Task(walker->table()->GetWalkerTaskId(), db_partition_id), walker_(walker) {
     363      130713 :     tbl_partition_ = static_cast<DBTablePartition *>
     364      130713 :         (walker_->table()->GetTablePartition(db_partition_id));
     365      130713 : }
     366             : 
     367      226195 : void DBTable::TableWalker::StartWalk() {
     368      226195 :     CHECK_CONCURRENCY("db::Walker");
     369      226195 :     assert(pending_workers_ == 0);
     370      737672 :     for (int i = 0; i < table_->PartitionCount(); i++) {
     371             :         DBTablePartition *partition = static_cast<DBTablePartition *>(
     372      511477 :             table_->GetTablePartition(i));
     373      511477 :         if (!partition->size()) continue;
     374      130713 :         worker_tasks_.push_back(new WalkWorker(this, i));
     375      130713 :         pending_workers_++;
     376             :     }
     377      226195 :     if (pending_workers_ == 0) {
     378      119614 :         table_->WalkDone();
     379             :     } else {
     380      106581 :         TaskScheduler *scheduler = TaskScheduler::GetInstance();
     381      237294 :         for (auto *task : worker_tasks_) scheduler->Enqueue(task);
     382             :     }
     383      226195 : }
     384             : 
     385             : ///////////////////////////////////////////////////////////
     386             : // Implementation of DBTable methods
     387             : ///////////////////////////////////////////////////////////
     388     1530180 : DBTable::DBTable(DB *db, const string &name)
     389             :     : DBTableBase(db, name),
     390     1530224 :       walker_(new TableWalker(this)),
     391     3060336 :       walker_task_id_(db->task_id()) {
     392             : 
     393             :     static bool init_ = false;
     394             :     static int iter_to_yield_env_ = 0;
     395             : 
     396     1530137 :     if (!init_) {
     397             :         // XXX To be used for testing purposes only.
     398         159 :         char *count_str = getenv("DB_ITERATION_TO_YIELD");
     399         159 :         if (count_str) {
     400         159 :             iter_to_yield_env_ = strtol(count_str, NULL, 0);
     401             :         } else {
     402           0 :             iter_to_yield_env_ = kIterationToYield;
     403             :         }
     404         159 :         init_ = true;
     405             :     }
     406     1530137 :     max_walk_iteration_to_yield_ = iter_to_yield_env_;
     407     1530137 : }
     408             : 
     409     1530242 : DBTable::~DBTable() {
     410     1530242 :     STLDeleteValues(&partitions_);
     411     1530242 : }
     412             : 
     413     1522075 : void DBTable::Init() {
     414     3597645 :     for (int i = 0; i < PartitionCount(); i++) {
     415     2075750 :         partitions_.push_back(AllocPartition(i));
     416             :     }
     417     1521837 : }
     418             : 
     419     2075689 : DBTablePartition *DBTable::AllocPartition(int index) {
     420     2075689 :     return new DBTablePartition(this, index);
     421             : }
     422             : 
     423      226195 : void DBTable::StartWalk() {
     424      226195 :     CHECK_CONCURRENCY("db::Walker");
     425      226195 :     incr_walk_count();
     426      226195 :     walker_->StartWalk();
     427      226195 : }
     428             : 
     429      212478 : DBEntry *DBTable::Add(const DBRequest *req) {
     430      212478 :     return AllocEntry(req->key.get()).release();
     431             : }
     432             : 
     433        1471 : void DBTable::Change(DBEntryBase *entry) {
     434        1471 :     DBTablePartBase *tpart = GetTablePartition(entry);
     435        1471 :     tpart->Notify(entry);
     436        1471 : }
     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     1412876 : int DBTable::PartitionCount() const {
     447     1412876 :     return DB::PartitionCount();
     448             : }
     449             : 
     450    16124079 : static size_t HashToPartition(size_t hash) {
     451    16124079 :     return hash % DB::PartitionCount();
     452             : }
     453             : 
     454    19581932 : DBTablePartBase *DBTable::GetTablePartition(const int index) {
     455    19581932 :     return partitions_[index];
     456             : }
     457             : 
     458       24100 : const DBTablePartBase *DBTable::GetTablePartition(const int index) const {
     459       24100 :     return partitions_[index];
     460             : }
     461             : 
     462      707942 : DBTablePartBase *DBTable::GetTablePartition(const DBRequestKey *key) {
     463      707942 :     int id = HashToPartition(Hash(key));
     464      707936 :     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    14931469 : DBTablePartBase *DBTable::GetTablePartition(const DBEntryBase *entry) {
     474    14931469 :     const DBEntry *gentry = static_cast<const DBEntry *>(entry);
     475    14931469 :     size_t id = HashToPartition(Hash(gentry));
     476    14923809 :     return GetTablePartition(id);
     477             : }
     478             : 
     479       24100 : const DBTablePartBase *DBTable::GetTablePartition(
     480             :         const DBEntryBase *entry) const {
     481       24100 :     const DBEntry *gentry = static_cast<const DBEntry *>(entry);
     482       24100 :     size_t id = HashToPartition(Hash(gentry));
     483       24100 :     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      383283 : DBEntry *DBTable::Find(const DBEntry *entry) {
     496      383283 :     size_t id = HashToPartition(Hash(entry));
     497             :     DBTablePartition *tbl_partition =
     498      383284 :         static_cast<DBTablePartition *>(GetTablePartition(id));
     499      383284 :     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       85180 : DBEntry *DBTable::Find(const DBRequestKey *key, int id) {
     516       85180 :     if (id == -1)
     517       85152 :         id = HashToPartition(Hash(key));
     518             :     DBTablePartition *tbl_partition =
     519       85180 :     static_cast<DBTablePartition *>(GetTablePartition(id));
     520       85180 :     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     1896051 : size_t DBTable::Size() const {
     534     1896051 :     size_t total = 0;
     535     1896051 :     for (vector<DBTablePartition *>::const_iterator iter = partitions_.begin();
     536     7001868 :          iter != partitions_.end(); iter++) {
     537     5105863 :         total += (*iter)->size();
     538             :     }
     539     1896034 :     return total;
     540             : }
     541             : 
     542        1213 : void DBTable::Input(DBTablePartition *tbl_partition, DBClient *client,
     543             :                     DBRequest *req) {
     544             :     DBRequestKey *key =
     545        1213 :         static_cast<DBRequestKey *>(req->key.get());
     546        1213 :     DBEntry *entry = NULL;
     547             : 
     548        1213 :     entry = tbl_partition->Find(key);
     549        1213 :     if (req->oper == DBRequest::DB_ENTRY_ADD_CHANGE) {
     550         656 :         if (entry) {
     551         347 :             if (OnChange(entry, req) || entry->IsDeleted()) {
     552             :                 // The entry may currently be marked as deleted.
     553          55 :                 entry->ClearDelete();
     554          55 :                 tbl_partition->Change(entry);
     555             :             }
     556             :         } else {
     557         309 :             if ((entry = Add(req)) != NULL) {
     558         309 :                 tbl_partition->Add(entry);
     559             :             }
     560             :         }
     561         557 :     } else if (req->oper == DBRequest::DB_ENTRY_DELETE) {
     562         330 :         if (entry) {
     563         330 :             if (Delete(entry, req)) {
     564         330 :                 tbl_partition->Delete(entry);
     565             :             }
     566             :         }
     567         227 :     } else if (req->oper == DBRequest::DB_ENTRY_NOTIFY) {
     568         227 :         if (entry) {
     569         227 :             tbl_partition->Notify(entry);
     570             :         }
     571             :     } else {
     572           0 :         assert(0);
     573             :     }
     574        1213 : }
     575             : 
     576           2 : void DBTable::DBStateClear(DBTable *table, ListenerId id) {
     577           2 :     DBEntryBase *next = NULL;
     578             : 
     579           4 :     for (int i = 0; i < table->PartitionCount(); ++i) {
     580             :         DBTablePartition *partition = static_cast<DBTablePartition *>(
     581           2 :             table->GetTablePartition(i));
     582             : 
     583           2 :         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           2 : }
     593             : 
     594             : //
     595             : // Callback for table walk triggered by NotifyAllEntries.
     596             : //
     597      129371 : bool DBTable::WalkCallback(DBTablePartBase *tpart, DBEntryBase *entry) {
     598      129371 :     tpart->Notify(entry);
     599      129379 :     return true;
     600             : }
     601             : 
     602             : //
     603             : // Callback for completion of table walk triggered by NotifyAllEntries.
     604             : //
     605       61324 : void DBTable::WalkCompleteCallback(DBTableBase *tbl_base) {
     606       61324 :     walk_ref_.reset();
     607       61324 : }
     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       63509 : void DBTable::NotifyAllEntries() {
     623       63509 :     CHECK_CONCURRENCY("bgp::Config", "bgp::ConfigHelper", "bgp::RTFilter",
     624             :                       "db::DBTable");
     625             : 
     626       63510 :     if (empty())
     627        2177 :         return;
     628             : 
     629       61332 :     if (walk_ref_ == NULL) {
     630             :         walk_ref_ =
     631      122648 :             AllocWalker(boost::bind(&DBTable::WalkCallback, this, _1, _2),
     632       61324 :                     boost::bind(&DBTable::WalkCompleteCallback, this, _2));
     633       61324 :         WalkTable(walk_ref_);
     634             :     } else {
     635           8 :         WalkAgain(walk_ref_);
     636             :     }
     637             : }
     638             : 
     639      219240 : DBTable::DBTableWalkRef DBTable::AllocWalker(WalkFn walk_fn,
     640             :                                              WalkCompleteFn walk_complete) {
     641      219240 :     DBTableWalkMgr *walk_mgr = database()->GetWalkMgr();
     642      219240 :     return walk_mgr->AllocWalker(this, walk_fn, walk_complete);
     643             : }
     644             : 
     645      157639 : void DBTable::ReleaseWalker(DBTable::DBTableWalkRef &walk) {
     646      157639 :     DBTableWalkMgr *walk_mgr = database()->GetWalkMgr();
     647      157639 :     walk_mgr->ReleaseWalker(walk);
     648      157639 :     return;
     649             : }
     650             : 
     651      226783 : void DBTable::WalkTable(DBTable::DBTableWalkRef walk) {
     652      226783 :     DBTableWalkMgr *walk_mgr = database()->GetWalkMgr();
     653      226783 :     walk_mgr->WalkTable(walk);
     654      226783 :     return;
     655             : }
     656             : 
     657       29183 : void DBTable::WalkAgain(DBTable::DBTableWalkRef walk) {
     658       29183 :     DBTableWalkMgr *walk_mgr = database()->GetWalkMgr();
     659       29183 :     walk_mgr->WalkAgain(walk);
     660       29183 :     return;
     661             : }
     662             : 
     663      789808 : bool DBTable::InvokeWalkCb(DBTablePartBase *part, DBEntryBase *entry) {
     664      789808 :     DBTableWalkMgr *walk_mgr = database()->GetWalkMgr();
     665      789692 :     return walk_mgr->InvokeWalkCb(part, entry);
     666             : }
     667             : 
     668      226195 : void DBTable::WalkDone() {
     669      226195 :     incr_walk_complete_count();
     670      226195 :     walker_->ClearWalkWorks();
     671      226195 :     DBTableWalkMgr *walk_mgr = database()->GetWalkMgr();
     672      226195 :     return walk_mgr->WalkDone();
     673             : }

Generated by: LCOV version 1.14