flow_token.h

Go to the documentation of this file.

#ifndef __AGENT_PKT_FLOW_TOKEN_H__
#define __AGENT_PKT_FLOW_TOKEN_H__
 
#include <memory>
#include <tbb/atomic.h>
#include <base/util.h>
 
class Token;
class TokenPool;
class FlowEntry;
class Proto;
 
typedef boost::shared_ptr<Token> TokenPtr;
 
class Token {
public:
    Token(TokenPool *pool);
    virtual ~Token();
 
protected:
    TokenPool *pool_;
    DISALLOW_COPY_AND_ASSIGN(Token);
};
 
class TokenPool {
public:
    TokenPool(const std::string &name, Proto *proto, int count);
    virtual ~TokenPool();
 
    virtual TokenPtr GetToken();
    int token_count() const { return token_count_; }
    bool TokenCheck() const;
    uint64_t failures() const { return failures_; }
    void IncrementRestarts() { restarts_++; }
    uint64_t restarts() const { return restarts_; }
protected:
    friend class Token;
 
    // Token destructor invokes this
    void FreeToken();
 
    std::string name_;
    int max_tokens_;
    int min_tokens_;
    int low_water_mark_;
    tbb::atomic<int> token_count_;
    mutable uint64_t failures_;
    uint64_t restarts_;
    Proto *proto_;
    DISALLOW_COPY_AND_ASSIGN(TokenPool);
};
 
class FlowTokenPool : public TokenPool {
public:
    FlowTokenPool(const std::string &name, Proto *proto, int count):
        TokenPool(name, proto, count) {}
    virtual ~FlowTokenPool() {}
    TokenPtr GetToken(FlowEntry *entry);
};
 
class FlowToken : public Token {
public:
    FlowToken(TokenPool *pool, FlowEntry *entry):
        Token(pool), flow_entry_(entry) {}
    virtual ~FlowToken() {}
private:
    FlowEntry *flow_entry_;
};
#endif //  __AGENT_PKT_FLOW_TOKEN_H__