Page MenuHomePhabricator
Differential D4563

[identity] db method to add username to users table in ddb
ClosedPublic
Actions

Authored by varun on Jul 18 2022, 2:24 PM.
Tags
None
Referenced Files
Unknown Object (File)
Mar 3 2025, 8:51 AM
Unknown Object (File)
Mar 3 2025, 8:51 AM
Unknown Object (File)
Mar 3 2025, 8:49 AM
Unknown Object (File)
Jan 25 2025, 9:07 PM
Unknown Object (File)
Jan 25 2025, 9:06 PM
Unknown Object (File)
Jan 25 2025, 9:06 PM
Unknown Object (File)
Jan 25 2025, 9:03 PM
Unknown Object (File)
Jan 23 2025, 6:29 PM
View All 95 Files
Subscribers
• abosh
• adrian
ashoat
atul

Details

Reviewers
tomek
• karol
Commits
rCOMM2677cc8f4c95: [identity] db method to add username to users table in ddb
Summary

We use the helper function for updating the users table to set the username attribute for a given user ID

Depends on D4561

Test Plan

Tested that I could set a username in DynamoDB with this method

Diff Detail

Repository
rCOMM Comm
Branch
eng-1421 (branched from master)
Lint
No Lint Coverage
Unit
No Test Coverage

Revision Contents
Changeset List

PathSize
services/
identity/
src/
1 line
15 lines

Diff 14588

View Options

services/identity/src/constants.rs

// Secrets // Secrets
pub const SECRETS_DIRECTORY: &str = "secrets"; pub const SECRETS_DIRECTORY: &str = "secrets";
pub const SECRETS_FILE_NAME: &str = "secret_key"; pub const SECRETS_FILE_NAME: &str = "secret_key";
pub const SECRETS_FILE_EXTENSION: &str = "txt"; pub const SECRETS_FILE_EXTENSION: &str = "txt";
// DynamoDB // DynamoDB
pub const PAKE_USERS_TABLE: &str = "identity-users"; pub const PAKE_USERS_TABLE: &str = "identity-users";
pub const PAKE_USERS_TABLE_PARTITION_KEY: &str = "userID"; pub const PAKE_USERS_TABLE_PARTITION_KEY: &str = "userID";
pub const PAKE_USERS_TABLE_REGISTRATION_ATTRIBUTE: &str = pub const PAKE_USERS_TABLE_REGISTRATION_ATTRIBUTE: &str =
"pakeRegistrationData"; "pakeRegistrationData";
pub const PAKE_USERS_TABLE_USERNAME_ATTRIBUTE: &str = "username";
pub const ACCESS_TOKEN_TABLE: &str = "identity-tokens"; pub const ACCESS_TOKEN_TABLE: &str = "identity-tokens";
pub const ACCESS_TOKEN_TABLE_PARTITION_KEY: &str = "userID"; pub const ACCESS_TOKEN_TABLE_PARTITION_KEY: &str = "userID";
pub const ACCESS_TOKEN_SORT_KEY: &str = "deviceID"; pub const ACCESS_TOKEN_SORT_KEY: &str = "deviceID";
pub const ACCESS_TOKEN_TABLE_CREATED_ATTRIBUTE: &str = "created"; pub const ACCESS_TOKEN_TABLE_CREATED_ATTRIBUTE: &str = "created";
pub const ACCESS_TOKEN_TABLE_AUTH_TYPE_ATTRIBUTE: &str = "authType"; pub const ACCESS_TOKEN_TABLE_AUTH_TYPE_ATTRIBUTE: &str = "authType";
pub const ACCESS_TOKEN_TABLE_VALID_ATTRIBUTE: &str = "valid"; pub const ACCESS_TOKEN_TABLE_VALID_ATTRIBUTE: &str = "valid";
pub const ACCESS_TOKEN_TABLE_TOKEN_ATTRIBUTE: &str = "token"; pub const ACCESS_TOKEN_TABLE_TOKEN_ATTRIBUTE: &str = "token";
Show All 9 Lines
View Options

services/identity/src/database.rs

Show All 13 Lines
use tracing::{error, info}; use tracing::{error, info};
use crate::constants::{ use crate::constants::{
ACCESS_TOKEN_SORT_KEY, ACCESS_TOKEN_TABLE, ACCESS_TOKEN_SORT_KEY, ACCESS_TOKEN_TABLE,
ACCESS_TOKEN_TABLE_AUTH_TYPE_ATTRIBUTE, ACCESS_TOKEN_TABLE_CREATED_ATTRIBUTE, ACCESS_TOKEN_TABLE_AUTH_TYPE_ATTRIBUTE, ACCESS_TOKEN_TABLE_CREATED_ATTRIBUTE,
ACCESS_TOKEN_TABLE_PARTITION_KEY, ACCESS_TOKEN_TABLE_TOKEN_ATTRIBUTE, ACCESS_TOKEN_TABLE_PARTITION_KEY, ACCESS_TOKEN_TABLE_TOKEN_ATTRIBUTE,
ACCESS_TOKEN_TABLE_VALID_ATTRIBUTE, PAKE_USERS_TABLE, ACCESS_TOKEN_TABLE_VALID_ATTRIBUTE, PAKE_USERS_TABLE,
PAKE_USERS_TABLE_PARTITION_KEY, PAKE_USERS_TABLE_REGISTRATION_ATTRIBUTE, PAKE_USERS_TABLE_PARTITION_KEY, PAKE_USERS_TABLE_REGISTRATION_ATTRIBUTE,
PAKE_USERS_TABLE_USERNAME_ATTRIBUTE,
}; };
use crate::opaque::Cipher; use crate::opaque::Cipher;
use crate::token::{AccessTokenData, AuthType}; use crate::token::{AccessTokenData, AuthType};
#[derive(Clone)] #[derive(Clone)]
pub struct DatabaseClient { pub struct DatabaseClient {
client: Arc<Client>, client: Arc<Client>,
} }
▲ Show 20 Lines • Show All 59 Lines • ▼ Show 20 Lines self
HashMap::from([( HashMap::from([(
":r".into(), ":r".into(),
AttributeValue::B(Blob::new(registration.serialize())), AttributeValue::B(Blob::new(registration.serialize())),
)]), )]),
) )
.await .await
} }
pub async fn put_username(
&self,
user_id: String,
username: String,
) -> Result<UpdateItemOutput, Error> {
self
.update_users_table(
user_id,
format!("SET {} :u", PAKE_USERS_TABLE_USERNAME_ATTRIBUTE),
HashMap::from([(":u".into(), AttributeValue::S(username))]),
)
.await
}
pub async fn get_access_token_data( pub async fn get_access_token_data(
&self, &self,
user_id: String, user_id: String,
device_id: String, device_id: String,
) -> Result<Option<AccessTokenData>, Error> { ) -> Result<Option<AccessTokenData>, Error> {
let primary_key = create_composite_primary_key( let primary_key = create_composite_primary_key(
( (
ACCESS_TOKEN_TABLE_PARTITION_KEY.to_string(), ACCESS_TOKEN_TABLE_PARTITION_KEY.to_string(),
▲ Show 20 Lines • Show All 329 Lines • Show Last 20 Lines