com.github.seratch.scalikesolr.response.DIHCommandResponse.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scalikesolr_2.9.2 Show documentation
scalikesolr
There is a newer version: 4.6.0
/*
 * Copyright 2011 Kazuhiro Sera
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
 * either express or implied. See the License for the specific language
 * governing permissions and limitations under the License.
 */

package com.github.seratch.scalikesolr.response

import dih.{ StatusMessages, InitArgs }
import java.lang.UnsupportedOperationException

import parser.ResponseParser
import xml.{ Node, XML }
import reflect.BeanProperty

import common.ResponseHeader
import com.github.seratch.scalikesolr.request.common.WriterType
import com.github.seratch.scalikesolr.{ SolrDocumentValue, SolrDocument }
import util.parsing.json.JSON
import com.github.seratch.scalikesolr.util.JSONUtil

case class DIHCommandResponse(@BeanProperty val writerType: WriterType = WriterType.Standard,
    @BeanProperty val rawBody: String = "") {

  private lazy val jsonMapFromRawBody: Map[String, Option[Any]] = writerType match {
    case WriterType.JSON => JSONUtil.toMap(JSON.parseFull(rawBody))
    case _ => Map()
  }

  @BeanProperty
  lazy val responseHeader: ResponseHeader = ResponseParser.getResponseHeader(
    writerType,
    rawBody
  )

  @BeanProperty
  lazy val initArgs: InitArgs = {
    writerType match {
      case WriterType.Standard =>
        val xml = XML.loadString(rawBody)
        val initArgs = (xml \ "lst").filter(lst => (lst \ "@name").text == "initArgs")(0)
        val docMap = initArgs.child.filter {
          lst => (lst \ "@name").text == "defaults"
        }(0).child.map {
          case elem: Node => ((elem \ "@name").text, SolrDocumentValue(elem.text))
        }.toMap
        new InitArgs(defaults = SolrDocument(map = docMap))
      case WriterType.JSON =>
        val initArgs = (jsonMapFromRawBody.get("initArgs").getOrElse(Nil) match {
          case Nil => Map()
          case list: Seq[_] => Map(list.head -> Some(list.tail.head))
        }).asInstanceOf[Map[String, Option[Any]]]
        val docMap = initArgs.map {
          case (k, v) => (k, new SolrDocumentValue(v.getOrElse("").toString))
        }.toMap
        new InitArgs(defaults = new SolrDocument(writerType = writerType, map = docMap))
      case other =>
        throw new UnsupportedOperationException("\"" + other.wt + "\" is currently not supported.")
    }
  }

  @BeanProperty
  lazy val command: String = {
    writerType match {
      case WriterType.Standard =>
        val xml = XML.loadString(rawBody)
        (xml \ "str").filter(lst => (lst \ "@name").text == "command")(0).text
      case WriterType.JSON =>
        jsonMapFromRawBody.get("command").getOrElse("").toString
      case other =>
        throw new UnsupportedOperationException("\"" + other.wt + "\" is currently not supported.")
    }
  }

  @BeanProperty
  lazy val status: String = {
    writerType match {
      case WriterType.Standard =>
        val xml = XML.loadString(rawBody)
        (xml \ "str").filter(lst => (lst \ "@name").text == "status")(0).text
      case WriterType.JSON =>
        jsonMapFromRawBody.get("status").getOrElse("").toString
      case other =>
        throw new UnsupportedOperationException("\"" + other.wt + "\" is currently not supported.")
    }
  }

  @BeanProperty
  lazy val importResponse: String = {
    writerType match {
      case WriterType.Standard =>
        val xml = XML.loadString(rawBody)
        (xml \ "str").filter(lst => (lst \ "@name").text == "importResponse")(0).text
      case WriterType.JSON =>
        jsonMapFromRawBody.get("importResponse").getOrElse("").toString
      case other =>
        throw new UnsupportedOperationException("\"" + other.wt + "\" is currently not supported.")
    }
  }

  @BeanProperty
  lazy val statusMessages: StatusMessages = {
    writerType match {
      case WriterType.Standard =>
        val xml = XML.loadString(rawBody)
        val statusMessages = (xml \ "lst").filter(lst => (lst \ "@name").text == "statusMessages")(0)
        val docMap = statusMessages.child.map {
          case elem: Node => ((elem \ "@name").text, SolrDocumentValue(elem.text))
        }.toMap
        new StatusMessages(defaults = new SolrDocument(map = docMap))
      case WriterType.JSON =>
        val doc = JSONUtil.toMap(jsonMapFromRawBody.get("statusMessages"))
        val docMap = doc.keys.map {
          case docKey => (docKey, new SolrDocumentValue(doc.getOrElse(docKey, "").toString))
        }.toMap
        new StatusMessages(defaults = new SolrDocument(map = docMap))
      case other =>
        throw new UnsupportedOperationException("\"" + other.wt + "\" is currently not supported.")
    }
  }

}