chore: collect structured GetUserMusicApiResp
This commit is contained in:
@@ -5,21 +5,21 @@ use crate::{
|
||||
title::{
|
||||
MaiVersionExt as _, Sdgb1_50,
|
||||
methods::APIMethod,
|
||||
model::{GetUserMusicApi, GetUserMusicApiResp, UserMusicDetail},
|
||||
model::{GetUserMusicApi, GetUserMusicApiResp},
|
||||
},
|
||||
};
|
||||
|
||||
pub async fn get_user_all_music(
|
||||
client: &AsyncClient,
|
||||
user_id: u32,
|
||||
) -> Result<Vec<UserMusicDetail>, ApiError> {
|
||||
let mut music_detail = Vec::new();
|
||||
) -> Result<GetUserMusicApiResp, ApiError> {
|
||||
let mut user_music_list = Vec::new();
|
||||
let mut index = None;
|
||||
|
||||
loop {
|
||||
let GetUserMusicApiResp {
|
||||
next_index,
|
||||
mut user_music_list,
|
||||
user_music_list: mut new_list,
|
||||
..
|
||||
} = Sdgb1_50::request::<_, GetUserMusicApiResp>(
|
||||
&client,
|
||||
@@ -33,13 +33,23 @@ pub async fn get_user_all_music(
|
||||
)
|
||||
.await?;
|
||||
|
||||
for list in &mut user_music_list {
|
||||
music_detail.append(&mut list.user_music_detail_list);
|
||||
if new_list.is_empty() {
|
||||
break;
|
||||
}
|
||||
|
||||
if next_index == 0 || user_music_list.is_empty() {
|
||||
break Ok(music_detail);
|
||||
user_music_list.append(&mut new_list);
|
||||
|
||||
if next_index == 0 {
|
||||
break;
|
||||
}
|
||||
|
||||
index = Some(next_index);
|
||||
}
|
||||
|
||||
Ok(GetUserMusicApiResp {
|
||||
user_id,
|
||||
next_index: 0,
|
||||
length: user_music_list.len() as _,
|
||||
user_music_list,
|
||||
})
|
||||
}
|
||||
|
||||
@@ -17,7 +17,7 @@ pub struct GetUserMusicApi {
|
||||
pub max_count: u32,
|
||||
}
|
||||
|
||||
#[derive(Default, Debug, Clone, PartialEq, Serialize, Deserialize)]
|
||||
#[derive(Default, Debug, Clone, PartialEq, Serialize, Deserialize, Encode, Decode)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
pub struct GetUserMusicApiResp {
|
||||
pub user_id: u32,
|
||||
@@ -26,7 +26,7 @@ pub struct GetUserMusicApiResp {
|
||||
pub user_music_list: Vec<UserMusic>,
|
||||
}
|
||||
|
||||
#[derive(Default, Debug, Clone, PartialEq, Serialize, Deserialize)]
|
||||
#[derive(Default, Debug, Clone, PartialEq, Serialize, Deserialize, Encode, Decode)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
pub struct UserMusic {
|
||||
pub user_music_detail_list: Vec<UserMusicDetail>,
|
||||
|
||||
@@ -80,21 +80,27 @@ async fn main() -> Result<(), Box<dyn snafu::Error>> {
|
||||
match command {
|
||||
Commands::MusicDetail { user_id, format } => {
|
||||
let music_detail = get_user_all_music(&client, user_id).await?;
|
||||
let details = music_detail
|
||||
.user_music_list
|
||||
.iter()
|
||||
.map(|m| &m.user_music_detail_list)
|
||||
.flatten();
|
||||
|
||||
match (human_readable, format) {
|
||||
(true, _) => {
|
||||
for detail in &music_detail {
|
||||
let mut count = 0;
|
||||
for detail in details {
|
||||
println!("{detail}");
|
||||
println!("----------");
|
||||
count += 1;
|
||||
}
|
||||
|
||||
println!("共查询到 {} 条记录!", music_detail.len());
|
||||
println!("共查询到 {count} 条记录!");
|
||||
}
|
||||
(false, RatingFormat::Origin) => json_display(music_detail)?,
|
||||
(false, RatingFormat::DxRatingNet) => {
|
||||
let dx_export = Vec::from_iter(
|
||||
music_detail
|
||||
.iter()
|
||||
details
|
||||
.map(|music| {
|
||||
DxMusicRecord::try_from(music).inspect_err(|e| {
|
||||
warn!("failed to process {}: {e}", music.music_id)
|
||||
|
||||
Reference in New Issue
Block a user