pdk-rate-limit-lib 1.7.0

PDK Rate Limit Library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

// Copyright (c) 2026, Salesforce, Inc.,
// All rights reserved.
// For full license text, see the LICENSE.txt file

use super::bucket::cluster::ClusterLimit;

/// Distribution formula for the rate limiting.
pub enum DistributionFormula {
    /// As less requests are available in the cluster, less requests are given.
    RemainingPercentage(f32),
    /// Given requests is always a predetermined percentage of the max requests.
    #[allow(dead_code)]
    TotalPercentage(f32),
}

impl DistributionFormula {
    /// Get the portion of the requests to be given to the cluster.
    pub fn portion(&self, limit: &ClusterLimit) -> u64 {
        match self {
            DistributionFormula::RemainingPercentage(percentage) => {
                ((limit.cluster_remaining as f32) * percentage) as u64
            }
            DistributionFormula::TotalPercentage(percentage) => {
                ((limit.max_requests as f32) * percentage) as u64
            }
        }
    }
}