Skip to content

Commit bba1822

Browse files
authored
[ENH]: Introduce NUM_REGEX_PREDICATES (#4429)
## Description of changes _Summarize the changes made by this PR._ - Improvements & Bug fixes - Introduces this usage type with a default of 0 - New functionality - ... ## Test plan _How are these changes tested?_ - [x] Tests pass locally with `pytest` for python, `yarn test` for js, `cargo test` for rust ## Documentation Changes None
1 parent aa4fb10 commit bba1822

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

rust/frontend/src/quota/mod.rs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -240,6 +240,7 @@ pub enum UsageType {
240240
NumCollections, // Total number of collections for a tenant
241241
NumDatabases, // Total number of databases for a tenant
242242
NumQueryIDs, // Number of IDs to filter by in a query
243+
NumRegexPredicates, // Number of regex predicates in the where_document
243244
}
244245

245246
impl fmt::Display for UsageType {
@@ -269,6 +270,7 @@ impl fmt::Display for UsageType {
269270
UsageType::NumCollections => write!(f, "Number of collections"),
270271
UsageType::NumDatabases => write!(f, "Number of databases"),
271272
UsageType::NumQueryIDs => write!(f, "Number of IDs to filter by in a query"),
273+
UsageType::NumRegexPredicates => write!(f, "Number of regex predicates"),
272274
}
273275
}
274276
}
@@ -298,6 +300,7 @@ impl TryFrom<&str> for UsageType {
298300
"num_collections" => Ok(UsageType::NumCollections),
299301
"num_databases" => Ok(UsageType::NumDatabases),
300302
"num_query_ids" => Ok(UsageType::NumQueryIDs),
303+
"num_regex_predicates" => Ok(UsageType::NumRegexPredicates),
301304
_ => Err(format!("Invalid UsageType: {}", value)),
302305
}
303306
}
@@ -326,6 +329,7 @@ lazy_static::lazy_static! {
326329
m.insert(UsageType::NumCollections, 1_000_000);
327330
m.insert(UsageType::NumDatabases, 10);
328331
m.insert(UsageType::NumQueryIDs, 1000);
332+
m.insert(UsageType::NumRegexPredicates, 0);
329333
m
330334
};
331335
}

0 commit comments

Comments
 (0)