mirrors/mCaptcha-cache
1
0
Fork 0
mirror of https://github.com/mCaptcha/cache.git synced 2024-11-24 09:25:30 +03:00
Code Issues Projects Releases Packages Wiki Activity

pgk version and pre-alloc config

Browse source
This commit is contained in:
realaravinth 2021-06-05 15:44:41 +05:30
parent 8e7a7dd093
commit bef36d2013
No known key found for this signature in database
GPG key ID: AD9F0F08E855ED88
2 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/lib.rs
View file

@ -15,7 +15,6 @@
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
use lazy_static::lazy_static;
use redis_module::raw::KeyType;
use redis_module::{redis_command, redis_event_handler, redis_module};
use redis_module::{Context, NextArg, RedisResult, REDIS_OK};
@ -26,9 +25,14 @@ mod utils;
use pocket::MCAPTCHA_POCKET_TYPE;
/// Initial allocation ammount of pocket[pocket::Pocket]
pub const HIT_PER_SECOND: usize = 100;
/// Pocket[pocket::Pocket] type version
pub const REDIS_MCAPTCHA_POCKET_TYPE_VERSION: i32 = 0;
pub const REDIS_MCAPTCHA_POCKET_TYPE_VERSION: i32 = 1;
pub const PKG_NAME: &str = "mcap";
pub const PKG_VERSION: usize = 1;
/// pocket timer key prefix
// PREFIX_POCKET_TIMER is used like this:
@ -78,11 +82,9 @@ fn counter_create(ctx: &Context, args: Vec<String>) -> RedisResult {
REDIS_OK
}
//////////////////////////////////////////////////////
redis_module! {
name: "mcaptcha_cahce",
version: 1,
version: PKG_VERSION,
data_types: [MCAPTCHA_POCKET_TYPE,],
commands: [
["mcaptcha_cache.count", counter_create, "write", 1, 1, 1],

4
src/pocket.rs
View file

@ -79,7 +79,7 @@ impl Pocket {
/// creates new pocket and sets off timer to go off at `duration`
#[inline]
pub fn new(ctx: &Context, duration: u64) -> Result<Self, RedisError> {
let decrement = HashMap::with_capacity(1);
let decrement = HashMap::with_capacity(HIT_PER_SECOND);
let pocket_instant = get_pocket_instant(duration)?;
let timer = ctx.create_timer(
@ -156,7 +156,7 @@ impl Pocket {
match val {
Some(pocket) => {
ctx.log_debug(&format!("entering loop hashmap "));
for (captcha, count) in pocket.decrement.iter() {
for (captcha, count) in pocket.decrement.drain() {
ctx.log_debug(&format!(
"reading captcha: {} with decr count {}",
&captcha, count