com.github.seratch.scalikesolr.HttpSolrClient.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.1 Show documentation
scalikesolr
There is a newer version: 4.0.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

import http.HttpClient
import java.net.URL
import reflect.BeanProperty
import org.slf4j.LoggerFactory
import util.{ Log, XMLStringBuilder }

class HttpSolrClient(@BeanProperty val url: URL,
    @BeanProperty val connectTimeout: Int = HttpClient.DEFAULT_CONNECT_TIMEOUT_MILLIS,
    @BeanProperty val readTimeout: Int = HttpClient.DEFAULT_READ_TIMEOUT_MILLIS,
    @BeanProperty val log: Log = new Log(LoggerFactory.getLogger(classOf[HttpSolrClient].getCanonicalName))) extends SolrClient {

  def this(url: URL, log: Log) = {
    this(
      url = url,
      connectTimeout = HttpClient.DEFAULT_CONNECT_TIMEOUT_MILLIS,
      readTimeout = HttpClient.DEFAULT_READ_TIMEOUT_MILLIS,
      log = log
    )
  }

  private val LOG_PREFIX = "["
  private val LOG_PREFIX_URL = "URL: "
  private val LOG_PREFIX_REQ = "Request Body: "
  private val LOG_PREFIX_RESP = "Response Body: "
  private val COMMA = " , "
  private val LOG_SUFFIX = "]"
  private val CONTENT_TYPE_XML = "text/xml"
  private val UTF8 = "UTF-8"

  private def httpClient: HttpClient = new HttpClient(connectTimeout, readTimeout)

  private def basicUrl(core: SolrCore): String = {
    val coreName = if (core.name.isEmpty) "" else "/" + core.name
    url.getProtocol + "://" + url.getHost + ":" + url.getPort + url.getPath + coreName
  }

  private def logGet(requestUrl: => String): Unit = {
    log.debug(LOG_PREFIX + LOG_PREFIX_URL + requestUrl + LOG_SUFFIX)
  }

  private def logPost(requestUrl: => String, requestBody: => String): Unit = {
    log.debug(LOG_PREFIX + LOG_PREFIX_URL + requestUrl + COMMA + LOG_PREFIX_REQ + requestBody + LOG_SUFFIX)
  }

  private def logResponse(responseBody: => String): Unit = {
    log.debug(LOG_PREFIX + LOG_PREFIX_RESP + responseBody + LOG_SUFFIX)
  }

  override def doQuery(request: QueryRequest): QueryResponse = {
    val queryString = request.queryString
    val requestUrl = basicUrl(request.core) + "/select" + queryString
    logGet(requestUrl)
    request.writerType match {
      case WriterType.JavaBinary => {
        val rawJavaBin = httpClient.getAsJavabin(requestUrl).rawJavaBin
        logResponse(rawJavaBin.toString)
        new QueryResponse(
          writerType = request.writerType,
          rawJavabin = rawJavaBin
        )
      }
      case _ => {
        val responseBody = httpClient.get(requestUrl, UTF8).content
        logResponse(responseBody)
        new QueryResponse(
          writerType = request.writerType,
          rawBody = responseBody
        )
      }
    }
  }

  override def doDIHCommand(request: DIHCommandRequest): DIHCommandResponse = {
    val queryString = request.toQueryString
    val requestUrl = basicUrl(request.core) + "/dataimport" + queryString
    logGet(requestUrl)
    val responseBody = httpClient.get(requestUrl, UTF8).content
    logResponse(responseBody)
    new DIHCommandResponse(
      writerType = request.writerType,
      rawBody = responseBody
    )
  }

  override def doAddDocuments(request: AddRequest): AddResponse = {
    val requestUrl = basicUrl(request.core) + "/update" + request.toQueryString
    val xml = new XMLStringBuilder
    xml.append("")
    request.documents foreach {
      case doc =>
        xml.append("")
        doc.keys map {
          case key => {
            xml.append("")
            xml.appendEscaped(doc.get(key).toString)
            xml.append("")
          }
        }
        xml.append("")
    }
    xml.append("")
    val requestBody = xml.toString
    logPost(requestUrl, requestBody)
    val responseBody = httpClient.post(requestUrl, requestBody, CONTENT_TYPE_XML, UTF8).content
    logResponse(responseBody)
    new AddResponse(
      writerType = request.writerType,
      rawBody = responseBody
    )
  }

  override def doDeleteDocuments(request: DeleteRequest): DeleteResponse = {
    val requestUrl = basicUrl(request.core) + "/update" + request.toQueryString
    val xml = new XMLStringBuilder
    xml.append("")
    request.uniqueKeysToDelete foreach {
      case uniqueKey => {
        xml.append("")
        xml.appendEscaped(uniqueKey)
        xml.append("")
      }
    }
    request.queries foreach {
      case query => {
        xml.append("")
        xml.appendEscaped(query.toString)
        xml.append("")
      }
    }
    xml.append("")
    val requestBody = xml.toString
    logPost(requestUrl, requestBody)
    val responseBody = httpClient.post(requestUrl, xml.toString, CONTENT_TYPE_XML, UTF8).content
    logResponse(responseBody)
    new DeleteResponse(
      writerType = request.writerType,
      rawBody = responseBody
    )
  }

  override def doCommit(request: UpdateRequest): UpdateResponse = {
    val requestUrl = basicUrl(request.core) + "/update" + request.toQueryString
    val requestBody = ""
    logPost(requestUrl, requestBody)
    val responseBody = httpClient.post(requestUrl, requestBody, CONTENT_TYPE_XML, UTF8).content
    logResponse(responseBody)
    new UpdateResponse(
      writerType = request.writerType,
      rawBody = responseBody
    )
  }

  override def doOptimize(request: UpdateRequest): UpdateResponse = {
    val requestUrl = basicUrl(request.core) + "/update" + request.toQueryString
    val requestBody = ""
    logPost(requestUrl, requestBody)
    val responseBody = httpClient.post(requestUrl, requestBody, CONTENT_TYPE_XML, UTF8).content
    logResponse(responseBody)
    new UpdateResponse(
      writerType = request.writerType,
      rawBody = responseBody
    )
  }

  override def doRollback(request: UpdateRequest): UpdateResponse = {
    val requestUrl = basicUrl(request.core) + "/update" + request.toQueryString
    val requestBody = ""
    logPost(requestUrl, requestBody)
    val responseBody = httpClient.post(requestUrl, requestBody, CONTENT_TYPE_XML, UTF8).content
    logResponse(responseBody)
    new UpdateResponse(
      writerType = request.writerType,
      rawBody = responseBody
    )
  }

  override def doPing(request: PingRequest): PingResponse = {
    val queryString = request.queryString
    val requestUrl = basicUrl(request.core) + "/admin/ping" + queryString
    logGet(requestUrl)
    val responseBody = httpClient.get(requestUrl, UTF8).content
    logResponse(responseBody)
    new PingResponse(
      writerType = request.writerType,
      rawBody = responseBody
    )
  }

  override def doAddDocumentsInCSV(request: UpdateRequest): UpdateResponse = {
    val requestUrl = basicUrl(request.core) + "/update/csv" + request.toQueryString
    logPost(requestUrl, request.requestBody)
    val responseBody = httpClient.post(requestUrl, request.requestBody, CONTENT_TYPE_XML, UTF8).content
    logResponse(responseBody)
    new UpdateResponse(
      writerType = request.writerType,
      rawBody = responseBody
    )
  }

  override def doUpdateInXML(request: UpdateRequest): UpdateResponse = {
    val requestUrl = basicUrl(request.core) + "/update" + request.toQueryString
    logPost(requestUrl, request.requestBody)
    val responseBody = httpClient.post(requestUrl, request.requestBody, CONTENT_TYPE_XML, UTF8).content
    logResponse(responseBody)
    new UpdateResponse(
      writerType = request.writerType,
      rawBody = responseBody
    )
  }

  override def doUpdateInJSON(request: UpdateRequest): UpdateResponse = {
    val requestUrl = basicUrl(request.core) + "/update/json" + request.toQueryString
    logPost(requestUrl, request.requestBody)
    val responseBody = httpClient.post(requestUrl, request.requestBody, CONTENT_TYPE_XML, UTF8).content
    logResponse(responseBody)
    new UpdateResponse(
      writerType = request.writerType,
      rawBody = responseBody
    )
  }

}