com.github.seratch.scalikesolr.response.parser.ResponseParser.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.parser

import com.github.seratch.scalikesolr.request.common.WriterType
import util.parsing.json.JSON
import com.github.seratch.scalikesolr.util.JSONUtil
import com.github.seratch.scalikesolr.response.common.ResponseHeader
import com.github.seratch.scalikesolr.{ SolrDocument, SolrDocumentValue }
import xml.XML
import org.apache.solr.common.util.{ SimpleOrderedMap, NamedList }

object ResponseParser {

  def getResponseHeader(writerType: WriterType,
    rawBody: String,
    rawJavaBin: NamedList[Any] = null): ResponseHeader = {
    writerType match {
      case WriterType.Standard => {
        val xml = XML.loadString(rawBody)
        val intItems = xml \ "lst" \ "int"
        val paramsList = (xml \ "lst" \ "list").filter(item => (item \ "@name").text == "params")
        paramsList.size match {
          case 0 =>
            new ResponseHeader(
              intItems.filter(item => (item \ "@name").text == "status").head.text.toInt,
              intItems.filter(item => (item \ "@name").text == "QTime").head.text.toInt
            )
          case _ =>
            val params = paramsList(0)
            new ResponseHeader(
              intItems.filter(item => (item \ "@name").text == "status").head.text.toInt,
              intItems.filter(item => (item \ "@name").text == "QTime").head.text.toInt,
              SolrDocument(writerType = writerType, rawBody = params.toString)
            )
        }
      }
      case WriterType.JSON => {
        val jsonMap = JSONUtil.toMap(JSON.parseFull(rawBody))
        val responseHeader = JSONUtil.toMap(jsonMap.get("responseHeader"))
        val status = JSONUtil.normalizeNum(responseHeader.get("status").getOrElse("0").toString).toInt
        val qTime = JSONUtil.normalizeNum(responseHeader.get("QTime").getOrElse("0").toString).toInt
        val params = JSONUtil.toMap(responseHeader.get("params"))
        val docMap = params.keys.map {
          case key => (key, new SolrDocumentValue(params.getOrElse(key, "").toString))
        }.toMap
        new ResponseHeader(status, qTime, new SolrDocument(writerType = writerType, map = docMap))
      }
      case WriterType.JavaBinary => {
        val responseHeader = rawJavaBin.get("responseHeader").asInstanceOf[NamedList[Any]]
        import collection.JavaConverters._
        val docMap = responseHeader.get("params").asInstanceOf[SimpleOrderedMap[Any]].asScala.map {
          case e: java.util.Map.Entry[_, _] => (e.getKey.toString, new SolrDocumentValue(e.getValue.toString))
        }.toMap
        new ResponseHeader(
          responseHeader.get("status").toString.toInt,
          responseHeader.get("QTime").toString.toInt,
          new SolrDocument(writerType = writerType, map = docMap)
        )
      }
      case other => throw new UnsupportedOperationException("\"" + other.wt + "\" is currently not supported.")
    }
  }

}