	use std::collections::HashMap;			use std::collections::{HashMap, HashSet};

	use chrono::{DateTime, Utc};			use chrono::{DateTime, Utc};
	use comm_lib::{			use comm_lib::{
	aws::ddb::{			aws::ddb::{
	operation::{get_item::GetItemOutput, query::builders::QueryFluentBuilder},			operation::{get_item::GetItemOutput, query::builders::QueryFluentBuilder},
	types::{			types::{
	error::TransactionCanceledException, AttributeValue, Delete,			error::TransactionCanceledException, AttributeValue, Delete,
	DeleteRequest, Put, TransactWriteItem, Update, WriteRequest,			DeleteRequest, Put, TransactWriteItem, Update, WriteRequest,
	▲ Show 20 Lines • Show All 1,753 Lines • ▼ Show 20 Lines

	/// Deletes all device data for user. Keeps device list rows.			/// Deletes all device data for user. Keeps device list rows.
	/// Returns list of deleted device IDs			/// Returns list of deleted device IDs
	#[tracing::instrument(skip_all)]			#[tracing::instrument(skip_all)]
	pub async fn delete_devices_data_for_user(			pub async fn delete_devices_data_for_user(
	&self,			&self,
	user_id: impl Into<String>,			user_id: impl Into<String>,
	) -> Result<Vec<String>, Error> {			) -> Result<Vec<String>, Error> {
				self.delete_user_devices_data_excluding(user_id, &[]).await
				}

				/// Same as [`DatabaseClient::delete_devices_data_for_user`]
				/// but allows to exclude certain device IDs.
				#[tracing::instrument(skip_all)]
				pub async fn delete_user_devices_data_excluding(
				&self,
				user_id: impl Into<String>,
				excluded_device_ids: &[&String],
				) -> Result<Vec<String>, Error> {
	let user_id: String = user_id.into();			let user_id: String = user_id.into();
				let excluded_device_ids: HashSet<&String> =
				excluded_device_ids.iter().cloned().collect();

	// we project only the primary keys so we can pass these directly to delete requests			// we project only the primary keys so we can pass these directly to delete requests
	let primary_keys =			let primary_keys =
	query_rows_with_prefix(self, &user_id, DEVICE_ITEM_KEY_PREFIX)			query_rows_with_prefix(self, &user_id, DEVICE_ITEM_KEY_PREFIX)
	.projection_expression(format!("{ATTR_USER_ID}, {ATTR_ITEM_ID}"))			.projection_expression(format!("{ATTR_USER_ID}, {ATTR_ITEM_ID}"))
	.send()			.send()
	.await			.await
	.map_err(\|e\| {			.map_err(\|e\| {
	error!(			error!(
	errorType = error_types::DEVICE_LIST_DB_LOG,			errorType = error_types::DEVICE_LIST_DB_LOG,
	"Failed to list user's devices' primary keys: {:?}", e			"Failed to list user's devices' primary keys: {:?}", e
	);			);
	Error::AwsSdk(e.into())			Error::AwsSdk(e.into())
	})?			})?
	.items			.items
	.unwrap_or_default();			.unwrap_or_default();

	let device_ids = primary_keys			let filtered_keys_iter = primary_keys.into_iter().filter_map(\|pk_attrs\| {
	.iter()			let item_id = pk_attrs.get(devices_table::ATTR_ITEM_ID).cloned();
	.map(\|attrs\| {			let device_id_result =
	let attr = attrs.get(devices_table::ATTR_ITEM_ID).cloned();			DeviceIDAttribute::try_from(item_id).map(DeviceIDAttribute::into_inner);
	DeviceIDAttribute::try_from(attr).map(DeviceIDAttribute::into_inner)
	})			match device_id_result {
	.collect::<Result<_, _>>()?;			Err(err) => Some(Err(err)),
				Ok(device_id) if excluded_device_ids.contains(&device_id) => None,
	let delete_requests = primary_keys			Ok(device_id) => {
	.into_iter()
	.map(\|item\| {
	let request = DeleteRequest::builder()			let request = DeleteRequest::builder()
	.set_key(Some(item))			.set_key(Some(pk_attrs))
	.build()			.build()
	.expect("key not set in DeleteRequest builder");			.expect("key not set in DeleteRequest builder");
	WriteRequest::builder().delete_request(request).build()			let request = WriteRequest::builder().delete_request(request).build();
	})
	.collect::<Vec<_>>();			Some(Ok((device_id, request)))
				}
				}
				});

				let (device_ids, delete_requests) =
				filtered_keys_iter.collect::<Result<(Vec<_>, Vec<_>), _>>()?;

	comm_lib::database::batch_operations::batch_write(			comm_lib::database::batch_operations::batch_write(
	&self.client,			&self.client,
	devices_table::NAME,			devices_table::NAME,
	delete_requests,			delete_requests,
	Default::default(),			Default::default(),
	)			)
	.await?;			.await?;
	▲ Show 20 Lines • Show All 683 Lines • Show Last 20 Lines

[lib] refactor logout in `keyserver-reducer` to opsClosedPublicActions

Details

Diff Detail

Event Timeline

Revision ContentsChangeset List

Diff 36398

lib/reducers/keyserver-reducer.js

[lib] refactor logout in `keyserver-reducer` to ops
ClosedPublic
Actions

Revision Contents
Changeset List