From 44601131a2d24cbdbf7833f68f9b83b270f686d9 Mon Sep 17 00:00:00 2001 From: Raymond Yang Date: Thu, 6 Aug 2026 21:00:52 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3=E6=96=B0=E5=A2=9E=E5=87=BA?= =?UTF-8?q?=E5=88=80=E7=B4=80=E9=8C=84=E5=9B=9E=E5=82=B3=E7=B5=90=E6=9E=9C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../database/service/BattleRecordService.kt | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/main/kotlin/club/sakunadaisuki/pcrediveclanrecordbackend/database/service/BattleRecordService.kt b/src/main/kotlin/club/sakunadaisuki/pcrediveclanrecordbackend/database/service/BattleRecordService.kt index ec6f818..b0fb6aa 100644 --- a/src/main/kotlin/club/sakunadaisuki/pcrediveclanrecordbackend/database/service/BattleRecordService.kt +++ b/src/main/kotlin/club/sakunadaisuki/pcrediveclanrecordbackend/database/service/BattleRecordService.kt @@ -18,10 +18,10 @@ class BattleRecordService(private val database: MongoDatabase) { } // Create new car - suspend fun create(data: BattleRecordDto): String = withContext(Dispatchers.IO) { + suspend fun create(data: BattleRecordDto): BattleRecordDto = withContext(Dispatchers.IO) { val doc = data.toDocument() collection.insertOne(doc) - doc["uid"].toString() + doc.let(BattleRecordDto::fromDocument) } // Read a car @@ -39,12 +39,13 @@ class BattleRecordService(private val database: MongoDatabase) { } // Update a car - suspend fun update(date: String, data: BattleRecordDto): Document? = withContext(Dispatchers.IO) { + suspend fun update(date: String, data: BattleRecordDto): BattleRecordDto? = withContext(Dispatchers.IO) { val filter = and( Filters.eq("date", date), Filters.eq("uid", data.uid) ) collection.findOneAndReplace(filter, data.toDocument()) + collection.find(filter).first()?.let(BattleRecordDto::fromDocument) } // Delete a car