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

Go to download
package org.elasticmq.rest

import akka.http.scaladsl.server.{MissingQueryParamRejection, Directive1, Directive0}
import akka.http.scaladsl.server.Directives._

package object sqs {
  type AnyParams = Map[String, String]

  implicit class RichAnyParam(p: AnyParams) {
    def action(requiredActionName: String): Directive0 = {
      if (p.get("Action").contains(requiredActionName)) {
        pass
      } else {
        reject
      }
    }

    def requiredParam(n: String): Directive1[String] = {
      p.get(n) match {
        case Some(v) => provide(v)
        case None    => reject(MissingQueryParamRejection(n))
      }
    }
  }
}