1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2018 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.api.idea
21 
22 import akka.Done
23 import akka.actor.typed.{ActorSystem, Behavior}
24 import akka.util.Timeout
25 import grizzled.slf4j.Logging
26 import org.make.api.technical.KafkaConsumerBehavior
27 import org.make.api.technical.KafkaConsumerBehavior.Protocol
28 import org.make.core.idea.IdeaId
29 import org.make.core.idea.indexed.IndexedIdea
30 
31 import scala.concurrent.ExecutionContext.Implicits.global
32 import scala.concurrent.Future
33 import scala.concurrent.duration.DurationInt
34 
35 class IdeaConsumerBehavior(ideaService: IdeaService, elasticsearchIdeaAPI: IdeaSearchEngine)
36     extends KafkaConsumerBehavior[IdeaEventWrapper]
37     with Logging {
38 
39   override protected val topicKey: String = IdeaProducerBehavior.topicKey
40   implicit val timeout: Timeout = Timeout(5.seconds)
41 
42   override def handleMessage(message: IdeaEventWrapper)(system: ActorSystem[_]): Future[_] = {
43     message.event match {
44       case event: IdeaCreatedEvent => onCreateOrUpdate(event)
45       case event: IdeaUpdatedEvent => onCreateOrUpdate(event)
46     }
47   }
48 
49   def onCreateOrUpdate(event: IdeaEvent): Future[Done] = {
50     retrieveAndShapeIdea(event.ideaId).flatMap(indexOrUpdate)
51   }
52 
53   def indexOrUpdate(idea: IndexedIdea): Future[Done] = {
54     debug(s"Indexing $idea")
55     elasticsearchIdeaAPI
56       .findIdeaById(idea.ideaId)
57       .flatMap {
58         case None        => elasticsearchIdeaAPI.indexIdea(idea)
59         case Some(found) => elasticsearchIdeaAPI.updateIdea(idea.copy(proposalsCount = found.proposalsCount))
60       }
61   }
62 
63   private def retrieveAndShapeIdea(id: IdeaId): Future[IndexedIdea] = {
64     ideaService.fetchOne(id).flatMap {
65       case None       => Future.failed(new IllegalArgumentException(s"Idea ${id.value} doesn't exist"))
66       case Some(idea) => Future.successful(IndexedIdea.createFromIdea(idea, proposalsCount = 0))
67     }
68   }
69 
70   override val groupId = "idea-consumer"
71 }
72 
73 object IdeaConsumerBehavior {
74   def apply(ideaService: IdeaService, elasticsearchIdeaAPI: IdeaSearchEngine): Behavior[Protocol] =
75     new IdeaConsumerBehavior(ideaService, elasticsearchIdeaAPI).createBehavior(name)
76   val name: String = "idea-consumer"
77 }