delete relay along study

manohar9999 · Oct 10, 2017 · ad6e31b · ad6e31b
1 parent 4a686bb
commit ad6e31b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/modules/relay/src/main/Env.scala b/modules/relay/src/main/Env.scala
@@ -41,8 +41,9 @@ final class Env(
     import lila.study.actorApi._
     def receive = {
       case lila.study.actorApi.StudyLikes(id, likes) => api.setLikes(Relay.Id(id.value), likes)
+      case lila.hub.actorApi.study.RemoveStudy(studyId, _) => api.onStudyRemove(studyId)
     }
-  })), 'studyLikes)
+  })), 'studyLikes, 'study)
 }
 
 object Env {

diff --git a/modules/relay/src/main/RelayApi.scala b/modules/relay/src/main/RelayApi.scala
@@ -103,6 +103,9 @@ final class RelayApi(
       sendToContributors(id, "relayLog", JsonView.syncLogEventWrites writes event) >>-
       event.error.foreach { err => logger.info(s"$id $err") }
 
+  private[relay] def onStudyRemove(studyId: Study.Id) =
+    coll.remove($id(studyId)).void
+
   private[relay] def publishRelay(relay: Relay): Funit =
     sendToContributors(relay.id, "relayData", JsonView.relayWrites writes relay)