ed8715379b85943e4becb2c80d0fb44b1f393cba
[aquarium] / src / main / scala / gr / grnet / aquarium / connector / rabbitmq / service / GenericPayloadHandler.scala
1 /*
2  * Copyright 2011-2012 GRNET S.A. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or
5  * without modification, are permitted provided that the following
6  * conditions are met:
7  *
8  *   1. Redistributions of source code must retain the above
9  *      copyright notice, this list of conditions and the following
10  *      disclaimer.
11  *
12  *   2. Redistributions in binary form must reproduce the above
13  *      copyright notice, this list of conditions and the following
14  *      disclaimer in the documentation and/or other materials
15  *      provided with the distribution.
16  *
17  * THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
18  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
20  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
21  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
22  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
23  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
24  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
25  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
27  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28  * POSSIBILITY OF SUCH DAMAGE.
29  *
30  * The views and conclusions contained in the software and
31  * documentation are those of the authors and should not be
32  * interpreted as representing official policies, either expressed
33  * or implied, of GRNET S.A.
34  */
35
36 package gr.grnet.aquarium.connector.rabbitmq.service
37
38 import com.ckkloverdos.maybe.{Just, Failed, MaybeEither}
39
40 import gr.grnet.aquarium.converter.JsonTextFormat
41 import gr.grnet.aquarium.connector.handler._
42 import gr.grnet.aquarium.event.model.ExternalEventModel
43 import gr.grnet.aquarium.util.safeUnit
44 import gr.grnet.aquarium.service.EventBusService
45 import gr.grnet.aquarium.Aquarium
46
47 /**
48  * Generic handler of events arriving to Aquarium.
49  *
50  * We first parse them to JSON ([[gr.grnet.aquarium.converter.JsonTextFormat]]) and an appropriate event model
51  * (`E <:` [[gr.grnet.aquarium.event.model.ExternalEventModel]]),
52  * then store them to DB
53  * (`S <:` [[gr.grnet.aquarium.event.model.ExternalEventModel]])
54  * and then forward them to business logic.
55  *
56  * All the above actions are given polymorphically via appropriate functions.
57  *
58  * @author Christos KK Loverdos <loverdos@gmail.com>
59  */
60
61 class GenericPayloadHandler[E <: ExternalEventModel, S <: ExternalEventModel]
62     (jsonParser: Array[Byte] ⇒ MaybeEither[JsonTextFormat],
63      jsonParserErrorAction: (Array[Byte], Throwable) ⇒ Unit,
64      eventParser: JsonTextFormat ⇒ E,
65      saveAction: E ⇒ S,
66      forwardAction: S ⇒ Unit) extends PayloadHandler {
67
68   def handlePayload(payload: Array[Byte]): HandlerResult = {
69     // 1. try to parse as json
70     jsonParser(payload) match {
71       case Failed(e) ⇒
72         jsonParserErrorAction(payload, e)
73
74         HandlerResultReject(e.getMessage)
75
76       case Just(jsonTextFormat) ⇒
77         // 2. try to parse as model
78         MaybeEither { eventParser(jsonTextFormat) } match {
79           case Failed(e) ⇒
80             HandlerResultReject(e.getMessage)
81
82           case Just(event) ⇒
83             // 3. try to save to DB
84             MaybeEither { saveAction(event) } match {
85               case Failed(e) ⇒
86                 HandlerResultPanic
87
88               case Just(s) ⇒
89                 // 4. try forward but it's OK if something bad happens here.
90                 safeUnit { forwardAction(s) }
91
92                 HandlerResultSuccess
93             }
94
95         }
96     }
97   }
98 }