Line data    Source code

       1             : /*
       2             :  * Copyright (c) 2016 Juniper Networks, Inc. All rights reserved.
       3             :  */
       4             : 
       5             : #include "db/db_table_walk_mgr.h"
       6             : 
       7             : #include <boost/bind/bind.hpp>
       8             : #include <boost/foreach.hpp>
       9             : 
      10             : #include "base/logging.h"
      11             : #include "base/task.h"
      12             : #include "base/task_annotations.h"
      13             : #include "db/db.h"
      14             : #include "db/db_partition.h"
      15             : #include "db/db_table.h"
      16             : #include "db/db_table_partition.h"
      17             : 
      18             : using namespace boost::placeholders;
      19             : 
      20       27401 : DBTableWalkMgr::DBTableWalkMgr()
      21       54802 :     : walk_request_trigger_(new TaskTrigger(
      22             :         boost::bind(&DBTableWalkMgr::ProcessWalkRequestList, this),
      23       54802 :         TaskScheduler::GetInstance()->GetTaskId("db::Walker"), 0)),
      24       54802 :       walk_done_trigger_(new TaskTrigger(
      25             :         boost::bind(&DBTableWalkMgr::ProcessWalkDone, this),
      26       54802 :         TaskScheduler::GetInstance()->GetTaskId("db::Walker"), 0)) {
      27       27401 : }
      28             : 
      29      327903 : bool DBTableWalkMgr::ProcessWalkRequestList() {
      30      327903 :     CHECK_CONCURRENCY("db::Walker");
      31      327903 :     std::scoped_lock lock(mutex_);
      32      327903 :     if (!current_table_walk_.empty()) return true;
      33             :     while (true) {
      34      311523 :         if (walk_request_list_.empty()) break;
      35      226196 :         WalkRequestInfoPtr info = walk_request_list_.front();
      36      226196 :         walk_request_set_.erase(info.get());
      37      226196 :         walk_request_list_.pop_front();
      38      226196 :         current_table_walk_.swap(info->pending_requests);
      39      226196 :         DBTable *table = info->table;
      40      226196 :         bool walk_table = false;
      41      453900 :         for (auto walker : current_table_walk_) {
      42      227704 :             if (walker->stopped()) continue;
      43      227702 :             walker->set_in_progress();
      44      227702 :             walker->reset_walk_again();
      45      227702 :             walk_table = true;
      46      227704 :         }
      47      226196 :         if (walk_table) {
      48             :             // start the walk
      49      226195 :             table->StartWalk();
      50      226195 :             break;
      51             :         } else {
      52           1 :             current_table_walk_.clear();
      53             :         }
      54      226197 :     }
      55      311522 :     return true;
      56      327903 : }
      57             : 
      58      226195 : bool DBTableWalkMgr::ProcessWalkDone() {
      59      226195 :     CHECK_CONCURRENCY("db::Walker");
      60      226195 :     assert(!current_table_walk_.empty());
      61      453898 :     for (auto walker : current_table_walk_) {
      62      227703 :         if (walker->walk_again())
      63         238 :             walker->set_walk_requested();
      64      227465 :         else if (!walker->stopped())
      65      227462 :             walker->set_walk_done();
      66      227703 :         if (walker->stopped() || walker->walk_again()) continue;
      67      227462 :         walker->walk_complete()(walker, walker->table());
      68      227703 :     }
      69      226195 :     current_table_walk_.clear();
      70      226195 :     walk_request_trigger_->Set();
      71      226195 :     return true;
      72             : }
      73             : 
      74      219240 : DBTable::DBTableWalkRef DBTableWalkMgr::AllocWalker(DBTable *table,
      75             :                DBTable::WalkFn walk_fn, DBTable::WalkCompleteFn walk_complete) {
      76      219240 :     table->incr_walker_count();
      77      219240 :     DBTableWalk *walker = new DBTableWalk(table, walk_fn, walk_complete);
      78      219240 :     return DBTable::DBTableWalkRef(walker);
      79             : }
      80             : 
      81      157639 : void DBTableWalkMgr::ReleaseWalker(DBTable::DBTableWalkRef &ref) {
      82      157639 :     ref->set_walk_stopped();
      83      157639 :     ref.reset();
      84      157639 : }
      85             : 
      86       29183 : void DBTableWalkMgr::WalkAgain(DBTable::DBTableWalkRef ref) {
      87       29183 :     WalkTable(ref);
      88       29183 : }
      89             : 
      90      255966 : void DBTableWalkMgr::WalkTable(DBTable::DBTableWalkRef walk) {
      91      255966 :     std::scoped_lock lock(mutex_);
      92      255966 :     DBTable *table = walk->table();
      93             : 
      94      255966 :     if (walk->in_progress()) {
      95         447 :         table->incr_walk_again_count();
      96         447 :         walk->set_walk_again();
      97             :     } else {
      98      255519 :         table->incr_walk_request_count();
      99      255519 :         walk->set_walk_requested();
     100             :     }
     101             : 
     102      255966 :     WalkRequestInfo tmp_info = WalkRequestInfo(table);
     103      255966 :     WalkRequestInfoSet::iterator it = walk_request_set_.find(&tmp_info);
     104      255966 :     if (it != walk_request_set_.end()) {
     105       29770 :         (*it)->AppendWalkReq(walk);
     106       29770 :         return;
     107             :     }
     108             : 
     109      226196 :     WalkRequestInfo *new_info = new WalkRequestInfo(table);
     110      226196 :     new_info->AppendWalkReq(walk);
     111      226196 :     walk_request_list_.push_back(WalkRequestInfoPtr(new_info));
     112      226196 :     walk_request_set_.insert(new_info);
     113      226196 :     walk_request_trigger_->Set();
     114      285736 : }
     115             : 
     116      226195 : void DBTableWalkMgr::WalkDone() {
     117      226195 :     walk_done_trigger_->Set();
     118      226195 : }
     119             : 
     120      789667 : bool DBTableWalkMgr::InvokeWalkCb(DBTablePartBase *part, DBEntryBase *entry) {
     121      789667 :     uint32_t skip_walk_count = 0;
     122     1581700 :     for (auto walker : current_table_walk_) {
     123      791903 :         if (walker->done() || walker->stopped() || walker->walk_again()) {
     124         383 :             skip_walk_count++;
     125         383 :             continue;
     126             :         }
     127      791029 :         bool more = walker->walk_fn()(part, entry);
     128      791232 :         if (!more) {
     129           3 :             skip_walk_count++;
     130           3 :             if (!walker->stopped()) walker->set_walk_done();
     131             :         }
     132      791616 :     }
     133      789749 :     return (skip_walk_count < current_table_walk_.size());
     134             : }