org.elasticmq.rest.sqs.ChangeMessageVisibilityDirectives.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of elasticmq-rest-sqs_2.11 Show documentation

elasticmq-rest-sqs

The newest version!

package org.elasticmq.rest.sqs

import Constants._
import org.elasticmq.{DeliveryReceipt, MillisVisibilityTimeout}
import akka.actor.ActorRef
import org.elasticmq.actor.reply._
import org.elasticmq.msg.UpdateVisibilityTimeout
import org.elasticmq.rest.sqs.directives.ElasticMQDirectives

trait ChangeMessageVisibilityDirectives { this: ElasticMQDirectives =>
  def changeMessageVisibility(p: AnyParams) = {
    p.action("ChangeMessageVisibility") {
      queueActorFromRequest(p) { queueActor =>
        doChangeMessageVisibility(queueActor, p).map { _ =>
          respondWith {
            
              
                {EmptyRequestId}
              
            
          }
        }
      }
    }
  }

  def doChangeMessageVisibility(queueActor: ActorRef, parameters: AnyParams) = {
    val visibilityTimeout = MillisVisibilityTimeout.fromSeconds(parameters(VisibilityTimeoutParameter).toLong)
    val msgId = DeliveryReceipt(parameters(ReceiptHandleParameter)).extractId

    for {
      updateResult <- queueActor ? UpdateVisibilityTimeout(msgId, visibilityTimeout)
    } yield {
      updateResult match {
        case Left(_)  => throw SQSException.invalidParameterValue
        case Right(_) => // ok
      }
    }
  }
}