00001 /* 00002 * IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING. By 00003 * downloading, copying, installing or using the software you agree to 00004 * this license. If you do not agree to this license, do not download, 00005 * install, copy or use the software. 00006 * 00007 * Intel Open Source License 00008 * 00009 * Copyright (c) 2006 Intel Corporation. All rights reserved. 00010 * 00011 * Redistribution and use in source and binary forms, with or without 00012 * modification, are permitted provided that the following conditions are 00013 * met: 00014 * 00015 * Redistributions of source code must retain the above copyright 00016 * notice, this list of conditions and the following disclaimer. 00017 * 00018 * Redistributions in binary form must reproduce the above copyright 00019 * notice, this list of conditions and the following disclaimer in the 00020 * documentation and/or other materials provided with the distribution. 00021 * 00022 * Neither the name of the Intel Corporation nor the names of its 00023 * contributors may be used to endorse or promote products derived from 00024 * this software without specific prior written permission. 00025 * 00026 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 00027 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 00028 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 00029 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE INTEL OR 00030 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 00031 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 00032 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 00033 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 00034 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 00035 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 00036 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 00037 */ 00038 #ifndef _OASYS_TOKEN_BUCKET_H_ 00039 #define _OASYS_TOKEN_BUCKET_H_ 00040 00041 #include "Time.h" 00042 #include "../debug/Logger.h" 00043 00044 namespace oasys { 00045 00049 class TokenBucket : public Logger { 00050 public: 00054 TokenBucket(const char* logpath, 00055 u_int32_t depth, /* in tokens */ 00056 u_int32_t rate /* in seconds */); 00057 00067 bool drain(u_int32_t length); 00068 00074 void update(); 00075 00080 u_int32_t time_to_fill(); 00081 00083 u_int32_t depth() const { return depth_; } 00084 u_int32_t rate() const { return rate_; } 00085 u_int32_t tokens() const { return tokens_; } 00087 00089 void set_depth(u_int32_t depth) { depth_ = depth; update(); } 00090 void set_rate(u_int32_t rate) { rate_ = rate; update(); } 00092 00096 void empty(); 00097 00098 protected: 00099 u_int32_t depth_; 00100 u_int32_t rate_; 00101 u_int32_t tokens_; 00102 Time last_update_; 00103 }; 00104 00105 } // namespace oasys 00106 00107 #endif /* _OASYS_TOKEN_BUCKET_H_ */