ADTを特別扱いして、problemsに入れないようにする

Author: hotate29

atcoder-problems-backend/src/crawler_utils.rs

@@ -244,25 +244,31 @@ async fn upsert_problems(
     new_problems: Vec<Problem>,
 ) -> Result<usize, DbErr> {
     for problem in &new_problems {
-        // Check if problem already exists; only insert if it doesn't.
-        // This prevents overwriting existing problem metadata (e.g. original contest_id)
-        // when the same problem_id appears in multiple contests (ADT reuses ABC problems).
-        // In other words: this is an ADT-specific countermeasure to keep the first-seen
-        // contest as the canonical one for the problem.
-        let existing = sql_entities::problems::Entity::find_by_id(problem.id.clone())
-            .one(db)
-            .await?;
+        // Insert into problems table
+        let title = problem.title();
+        let model = sql_entities::problems::ActiveModel {
+            id: Set(problem.id.clone()),
+            contest_id: Set(problem.contest_id.clone()),
+            problem_index: Set(problem.problem_index.clone()),
+            name: Set(problem.name.clone()),
+            title: Set(title),
+        };
 
-        if existing.is_none() {
-            let title = problem.title();
-            let model = sql_entities::problems::ActiveModel {
-                id: Set(problem.id.clone()),
-                contest_id: Set(problem.contest_id.clone()),
-                problem_index: Set(problem.problem_index.clone()),
-                name: Set(problem.name.clone()),
-                title: Set(title),
-            };
+        // ADTの問題は全てABCで既出であり、これらは同一視したい。
+        // そのまま通すと問題情報がABCからADTに上書きされてしまうので、ADTの問題はproblemsには入れないようにする。
+        // （ただしcontest_problemには入れる）
+        if !problem.contest_id.starts_with("adt") {
             sql_entities::problems::Entity::insert(model)
+                .on_conflict(
+                    OnConflict::column(sql_entities::problems::Column::Id)
+                        .update_columns([
+                            sql_entities::problems::Column::ContestId,
+                            sql_entities::problems::Column::ProblemIndex,
+                            sql_entities::problems::Column::Name,
+                            sql_entities::problems::Column::Title,
+                        ])
+                        .to_owned(),
+                )
                 .exec(db)
                 .await?;
         }