repository: Delete by attribute instead of object

Signed-off-by: Shashank Verma <[email protected]>
CC-MNNIT · Nov 1, 2023 · cb7108b · cb7108b
1 parent 47ae05d
commit cb7108b
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 2 deletions.
diff --git a/src/main/kotlin/com/mnnit/moticlubs/repository/AdminRepository.kt b/src/main/kotlin/com/mnnit/moticlubs/repository/AdminRepository.kt
@@ -58,5 +58,17 @@ class AdminRepository(
         )
 
     @Transactional
-    fun delete(admin: Admin): Mono<Void> = db.delete(admin).then()
+    fun delete(admin: Admin): Mono<Void> = db
+        .delete(
+            Query.query(
+                Criteria
+                    .where(Admin::cid.name)
+                    .`is`(admin.cid)
+                    .and(
+                        Criteria.where(Admin::uid.name).`is`(admin.uid),
+                    ),
+            ),
+            Admin::class.java,
+        )
+        .then()
 }
diff --git a/src/main/kotlin/com/mnnit/moticlubs/repository/MemberRepository.kt b/src/main/kotlin/com/mnnit/moticlubs/repository/MemberRepository.kt
@@ -40,7 +40,19 @@ class MemberRepository(
 
     @Transactional
     fun delete(member: Member): Mono<Void> = exists(member)
-        .flatMap { if (it) db.delete(member) else Mono.just(member) }
+        .flatMap {
+            if (it) {
+                db.delete(
+                    Query.query(
+                        Criteria.where(Member::uid.name).`is`(member.uid)
+                            .and(Criteria.where(Member::chid.name).`is`(member.chid)),
+                    ),
+                    Member::class.java,
+                )
+            } else {
+                Mono.just(member)
+            }
+        }
         .then()
 
     @Transactional