db_partition.h

Go to the documentation of this file.

/*
 * Copyright (c) 2013 Juniper Networks, Inc. All rights reserved.
 */
 
#ifndef ctrlplane_db_partition_h
#define ctrlplane_db_partition_h
 
#include <boost/function.hpp>
 
#include "base/util.h"
#include "db/db_table.h"
#include "db/db_table_partition.h"
 
class DB;
class DBClient;
class DBTablePartBase;
 
// Database shard interface.
// Each shard handles the full pipeline of DB update processing.
class DBPartition {
public:
    typedef boost::function<void(void)> Callback;
 
    explicit DBPartition(DB *db, int partition_id);
    ~DBPartition();
 
    // Enqueue a request at the start of the DB processing pipeline.
    // Returns false if the client should stop enqueuing updates.
    bool EnqueueRequest(DBTablePartBase *tpart, DBClient *client,
                        DBRequest *req);
 
    void EnqueueRemove(DBTablePartBase *tpart, DBEntryBase *db_entry);
 
    // Enqueue table on change list.
    void OnTableChange(DBTablePartBase *tpart);
    bool IsDBQueueEmpty() const;
    void SetQueueDisable(bool disable);
 
    long request_queue_len() const;
    uint64_t total_request_count() const;
    uint64_t max_request_queue_len() const;
    int task_id() const;
 
private:
    class WorkQueue;
    class QueueRunner;
 
    DB *db_;
    std::unique_ptr<WorkQueue> work_queue_;
    static int db_partition_task_id_;
 
    DISALLOW_COPY_AND_ASSIGN(DBPartition);
};
 
#endif