rust/hyperswitch
1
0
Fork 0
mirror of https://github.com/juspay/hyperswitch.git synced 2025-10-30 09:38:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(api_keys): use merchant_id and key_id to query the table (#939)

Browse Source 
Signed-off-by: Chethan <chethan.rao@juspay.in>
This commit is contained in:
Chethan Rao
2023-04-25 01:04:18 +05:30
committed by GitHub
parent ab7fc23a7b
commit 40898c0ac9
7 changed files with 99 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
crates/router/src/core/api_keys.rs
View File

@ -158,10 +158,11 @@ pub async fn create_api_key(
#[instrument(skip_all)]
pub async fn retrieve_api_key(
store: &dyn StorageInterface,
merchant_id: &str,
key_id: &str,
) -> RouterResponse<api::RetrieveApiKeyResponse> {
let api_key = store
.find_api_key_by_key_id_optional(key_id)
.find_api_key_by_merchant_id_key_id_optional(merchant_id, key_id)
.await
.change_context(errors::ApiErrorResponse::InternalServerError) // If retrieve failed
.attach_printable("Failed to retrieve new API key")?
@ -173,11 +174,16 @@ pub async fn retrieve_api_key(
#[instrument(skip_all)]
pub async fn update_api_key(
store: &dyn StorageInterface,
merchant_id: &str,
key_id: &str,
api_key: api::UpdateApiKeyRequest,
) -> RouterResponse<api::RetrieveApiKeyResponse> {
let api_key = store
.update_api_key(key_id.to_owned(), api_key.foreign_into())
.update_api_key(
merchant_id.to_owned(),
key_id.to_owned(),
api_key.foreign_into(),
)
.await
.to_not_found_response(errors::ApiErrorResponse::ApiKeyNotFound)?;
@ -187,16 +193,18 @@ pub async fn update_api_key(
#[instrument(skip_all)]
pub async fn revoke_api_key(
store: &dyn StorageInterface,
merchant_id: &str,
key_id: &str,
) -> RouterResponse<api::RevokeApiKeyResponse> {
let revoked = store
.revoke_api_key(key_id)
.revoke_api_key(merchant_id, key_id)
.await
.to_not_found_response(errors::ApiErrorResponse::ApiKeyNotFound)?;
metrics::API_KEY_REVOKED.add(&metrics::CONTEXT, 1, &[]);
Ok(ApplicationResponse::Json(api::RevokeApiKeyResponse {
merchant_id: merchant_id.to_owned(),
key_id: key_id.to_owned(),
revoked,
}))

50
crates/router/src/db/api_keys.rs
View File

@ -16,14 +16,20 @@ pub trait ApiKeyInterface {
async fn update_api_key(
&self,
merchant_id: String,
key_id: String,
api_key: storage::ApiKeyUpdate,
) -> CustomResult<storage::ApiKey, errors::StorageError>;
async fn revoke_api_key(&self, key_id: &str) -> CustomResult<bool, errors::StorageError>;
async fn find_api_key_by_key_id_optional(
async fn revoke_api_key(
&self,
merchant_id: &str,
key_id: &str,
) -> CustomResult<bool, errors::StorageError>;
async fn find_api_key_by_merchant_id_key_id_optional(
&self,
merchant_id: &str,
key_id: &str,
) -> CustomResult<Option<storage::ApiKey>, errors::StorageError>;
@ -56,30 +62,36 @@ impl ApiKeyInterface for Store {
async fn update_api_key(
&self,
merchant_id: String,
key_id: String,
api_key: storage::ApiKeyUpdate,
) -> CustomResult<storage::ApiKey, errors::StorageError> {
let conn = connection::pg_connection_write(self).await?;
storage::ApiKey::update_by_key_id(&conn, key_id, api_key)
storage::ApiKey::update_by_merchant_id_key_id(&conn, merchant_id, key_id, api_key)
.await
.map_err(Into::into)
.into_report()
}
async fn revoke_api_key(&self, key_id: &str) -> CustomResult<bool, errors::StorageError> {
let conn = connection::pg_connection_write(self).await?;
storage::ApiKey::revoke_by_key_id(&conn, key_id)
.await
.map_err(Into::into)
.into_report()
}
async fn find_api_key_by_key_id_optional(
async fn revoke_api_key(
&self,
merchant_id: &str,
key_id: &str,
) -> CustomResult<bool, errors::StorageError> {
let conn = connection::pg_connection_write(self).await?;
storage::ApiKey::revoke_by_merchant_id_key_id(&conn, merchant_id, key_id)
.await
.map_err(Into::into)
.into_report()
}
async fn find_api_key_by_merchant_id_key_id_optional(
&self,
merchant_id: &str,
key_id: &str,
) -> CustomResult<Option<storage::ApiKey>, errors::StorageError> {
let conn = connection::pg_connection_read(self).await?;
storage::ApiKey::find_optional_by_key_id(&conn, key_id)
storage::ApiKey::find_optional_by_merchant_id_key_id(&conn, merchant_id, key_id)
.await
.map_err(Into::into)
.into_report()
@ -122,6 +134,7 @@ impl ApiKeyInterface for MockDb {
async fn update_api_key(
&self,
_merchant_id: String,
_key_id: String,
_api_key: storage::ApiKeyUpdate,
) -> CustomResult<storage::ApiKey, errors::StorageError> {
@ -129,13 +142,18 @@ impl ApiKeyInterface for MockDb {
Err(errors::StorageError::MockDbError)?
}
async fn revoke_api_key(&self, _key_id: &str) -> CustomResult<bool, errors::StorageError> {
async fn revoke_api_key(
&self,
_merchant_id: &str,
_key_id: &str,
) -> CustomResult<bool, errors::StorageError> {
// [#172]: Implement function for `MockDb`
Err(errors::StorageError::MockDbError)?
}
async fn find_api_key_by_key_id_optional(
async fn find_api_key_by_merchant_id_key_id_optional(
&self,
_merchant_id: &str,
_key_id: &str,
) -> CustomResult<Option<storage::ApiKey>, errors::StorageError> {
// [#172]: Implement function for `MockDb`

24
crates/router/src/routes/api_keys.rs
View File

@ -82,14 +82,16 @@ pub async fn api_key_retrieve(
path: web::Path<(String, String)>,
) -> impl Responder {
let flow = Flow::ApiKeyRetrieve;
let (_merchant_id, key_id) = path.into_inner();
let (merchant_id, key_id) = path.into_inner();
api::server_wrap(
flow,
state.get_ref(),
&req,
&key_id,
|state, _, key_id| api_keys::retrieve_api_key(&*state.store, key_id),
(&merchant_id, &key_id),
|state, _, (merchant_id, key_id)| {
api_keys::retrieve_api_key(&*state.store, merchant_id, key_id)
},
&auth::AdminApiAuth,
)
.await
@ -122,15 +124,17 @@ pub async fn api_key_update(
json_payload: web::Json<api_types::UpdateApiKeyRequest>,
) -> impl Responder {
let flow = Flow::ApiKeyUpdate;
let (_merchant_id, key_id) = path.into_inner();
let (merchant_id, key_id) = path.into_inner();
let payload = json_payload.into_inner();
api::server_wrap(
flow,
state.get_ref(),
&req,
(&key_id, payload),
|state, _, (key_id, payload)| api_keys::update_api_key(&*state.store, key_id, payload),
(&merchant_id, &key_id, payload),
|state, _, (merchant_id, key_id, payload)| {
api_keys::update_api_key(&*state.store, merchant_id, key_id, payload)
},
&auth::AdminApiAuth,
)
.await
@ -162,14 +166,16 @@ pub async fn api_key_revoke(
path: web::Path<(String, String)>,
) -> impl Responder {
let flow = Flow::ApiKeyRevoke;
let (_merchant_id, key_id) = path.into_inner();
let (merchant_id, key_id) = path.into_inner();
api::server_wrap(
flow,
state.get_ref(),
&req,
&key_id,
|state, _, key_id| api_keys::revoke_api_key(&*state.store, key_id),
(&merchant_id, &key_id),
|state, _, (merchant_id, key_id)| {
api_keys::revoke_api_key(&*state.store, merchant_id, key_id)
},
&auth::AdminApiAuth,
)
.await