feat: record external generation runs

server-rs/crates/api-server/src/big_fish/formal_assets.rs

@@ -1,4 +1,5 @@
 use super::*;
+use crate::tracking::record_external_generation_run_after_success;
 
 struct BigFishDashScopeSettings {
     base_url: String,
@@ -39,52 +40,99 @@ pub(super) async fn generate_big_fish_formal_asset(
     motion_key: Option<&str>,
     generated_at_micros: i64,
 ) -> Result<String, AppError> {
-    let session = state
-        .spacetime_client()
-        .get_big_fish_session(session_id.to_string(), owner_user_id.to_string())
-        .await
-        .map_err(map_big_fish_client_error)?;
-    let draft = session.draft.as_ref().ok_or_else(|| {
-        AppError::from_status(StatusCode::BAD_REQUEST).with_details(json!({
-            "provider": "big-fish",
-            "message": "玩法草稿尚未编译，不能生成正式图片。",
-        }))
-    })?;
-    let context = build_big_fish_formal_asset_context(
-        &session,
-        draft,
-        asset_kind,
-        level,
-        motion_key,
-        generated_at_micros,
-    )?;
-    let settings = require_big_fish_dashscope_settings(state)?;
-    let http_client = build_big_fish_dashscope_http_client(&settings)?;
-    let generated = create_big_fish_text_to_image_generation(
-        &http_client,
-        &settings,
-        context.prompt.as_str(),
-        context.negative_prompt.as_str(),
-        context.size.as_str(),
-    )
-    .await?;
-    let downloaded = download_big_fish_remote_image(
-        &http_client,
-        generated.image_url.as_str(),
-        "下载 Big Fish 正式图片失败",
-        context.apply_transparent_background_post_process,
-    )
-    .await?;
+    let started_at_micros = current_utc_micros();
+    let request_payload = json!({
+        "assetKind": asset_kind,
+        "level": level,
+        "motionKey": motion_key,
+        "sessionId": session_id,
+        "ownerUserId": owner_user_id,
+    });
+    let outcome = async {
+        let session = state
+            .spacetime_client()
+            .get_big_fish_session(session_id.to_string(), owner_user_id.to_string())
+            .await
+            .map_err(map_big_fish_client_error)?;
+        let draft = session.draft.as_ref().ok_or_else(|| {
+            AppError::from_status(StatusCode::BAD_REQUEST).with_details(json!({
+                "provider": "big-fish",
+                "message": "玩法草稿尚未编译，不能生成正式图片。",
+            }))
+        })?;
+        let context = build_big_fish_formal_asset_context(
+            &session,
+            draft,
+            asset_kind,
+            level,
+            motion_key,
+            generated_at_micros,
+        )?;
+        let settings = require_big_fish_dashscope_settings(state)?;
+        let http_client = build_big_fish_dashscope_http_client(&settings)?;
+        let generated = create_big_fish_text_to_image_generation(
+            &http_client,
+            &settings,
+            context.prompt.as_str(),
+            context.negative_prompt.as_str(),
+            context.size.as_str(),
+        )
+        .await?;
+        let downloaded = download_big_fish_remote_image(
+            &http_client,
+            generated.image_url.as_str(),
+            "下载 Big Fish 正式图片失败",
+            context.apply_transparent_background_post_process,
+        )
+        .await?;
 
-    persist_big_fish_formal_asset(
-        state,
-        owner_user_id,
-        &context,
-        generated,
-        downloaded,
-        generated_at_micros,
-    )
-    .await
+        persist_big_fish_formal_asset(
+            state,
+            owner_user_id,
+            &context,
+            generated,
+            downloaded,
+            generated_at_micros,
+        )
+        .await
+    }
+    .await;
+    match outcome {
+        Ok(value) => {
+            record_external_generation_run_after_success(
+                state,
+                "dashscope",
+                "big_fish_text_to_image",
+                "大鱼正式图片生成",
+                request_payload,
+                started_at_micros,
+                true,
+                None,
+                None,
+                Some(json!({
+                    "legacyPublicPath": value.clone(),
+                })),
+            )
+            .await;
+            Ok(value)
+        }
+        Err(error) => {
+            record_external_generation_run_after_success(
+                state,
+                "dashscope",
+                "big_fish_text_to_image",
+                "大鱼正式图片生成",
+                request_payload,
+                started_at_micros,
+                false,
+                Some(error.to_string()),
+                None,
+                None,
+            )
+            .await;
+            Err(error)
+        }
+    }
 }
 
 fn build_big_fish_formal_asset_context(
@@ -626,6 +674,10 @@ fn map_big_fish_asset_binding_prepare_error(error: AssetObjectFieldError) -> App
     }))
 }
 
+fn current_utc_micros() -> i64 {
+    (time::OffsetDateTime::now_utc().unix_timestamp_nanos() / 1_000) as i64
+}
+
 fn map_big_fish_asset_spacetime_error(error: SpacetimeClientError) -> AppError {
     AppError::from_status(StatusCode::BAD_GATEWAY).with_details(json!({
         "provider": "spacetimedb",