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__