rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-05-18 14:00:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: add rate limit status code (#1923)

Browse Source
This commit is contained in:
shuiyisong
2023-07-10 17:41:59 +08:00
committed by GitHub
parent c8ed1bbfae
commit 04ddeffd2a
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/common/error/src/status_code.rs
View File

@@ -66,6 +66,9 @@ pub enum StatusCode {
// ====== Begin of server related status code =====
/// Runtime resources exhausted, like creating threads failed.
RuntimeResourcesExhausted = 6000,
/// Rate limit exceeded
RateLimited = 6001,
// ====== End of server related status code =======
// ====== Begin of auth related status code =====
@@ -111,6 +114,7 @@ impl StatusCode {
| StatusCode::TableColumnNotFound
| StatusCode::TableColumnExists
| StatusCode::DatabaseNotFound
| StatusCode::RateLimited
| StatusCode::UserNotFound
| StatusCode::UnsupportedPasswordType
| StatusCode::UserPasswordMismatch
@@ -141,6 +145,7 @@ impl StatusCode {
| StatusCode::TableColumnNotFound
| StatusCode::TableColumnExists
| StatusCode::DatabaseNotFound
| StatusCode::RateLimited
| StatusCode::UserNotFound
| StatusCode::UnsupportedPasswordType
| StatusCode::UserPasswordMismatch

2
src/servers/src/error.rs
View File

@@ -432,7 +432,7 @@ fn status_to_tonic_code(status_code: StatusCode) -> Code {
| StatusCode::DatabaseNotFound
| StatusCode::UserNotFound => Code::NotFound,
StatusCode::StorageUnavailable => Code::Unavailable,
StatusCode::RuntimeResourcesExhausted => Code::ResourceExhausted,
StatusCode::RuntimeResourcesExhausted | StatusCode::RateLimited => Code::ResourceExhausted,
StatusCode::UnsupportedPasswordType
| StatusCode::UserPasswordMismatch
| StatusCode::AuthHeaderNotFound