code.grnet.gr Git - aquarium/blob - 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] ⇒ JsonTextFormat,
  63      onJsonParserSuccess: (Array[Byte], JsonTextFormat) ⇒ Unit,
  64      onJsonParserError: (Array[Byte], Throwable) ⇒ Unit,
  65      eventParser: JsonTextFormat ⇒ E,
  66      onEventParserSuccess: (Array[Byte], E) ⇒ Unit,
  67      onEventParserError: (Array[Byte], Throwable) ⇒ Unit,
  68      saveAction: E ⇒ S,
  69      forwardAction: S ⇒ Unit) extends PayloadHandler {
  70
  71   def handlePayload(payload: Array[Byte]): HandlerResult = {
  72     // 1. try to parse as json
  73     MaybeEither { jsonParser(payload) } match {
  74       case Failed(e) ⇒
  75         safeUnit(onJsonParserError(payload, e))
  76
  77         HandlerResultReject(e.getMessage)
  78
  79       case Just(jsonTextFormat) ⇒
  80         safeUnit(onJsonParserSuccess(payload, jsonTextFormat))
  81
  82         // 2. try to parse as model
  83         MaybeEither { eventParser(jsonTextFormat) } match {
  84           case Failed(e) ⇒
  85             safeUnit(onEventParserError(payload, e))
  86
  87             HandlerResultReject(e.getMessage)
  88
  89           case Just(event) ⇒
  90             safeUnit(onEventParserSuccess(payload, event))
  91
  92             // 3. try to save to DB
  93             MaybeEither { saveAction(event) } match {
  94               case Failed(e) ⇒
  95                 HandlerResultPanic
  96
  97               case Just(s) ⇒
  98                 // 4. try forward but it's OK if something bad happens here.
  99                 safeUnit { forwardAction(s) }
 100
 101                 HandlerResultSuccess
 102             }
 103
 104         }
 105     }
 106   }
 107 }