com.knockdata.spark.highcharts.base.JsonConversion.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spark-highcharts Show documentation

Provide Highcharts Support for Spark either in Zepplin or spark-shell

There is a newer version: 0.6.6

package com.knockdata.spark.highcharts.base

import com.knockdata.spark.highcharts.model._

import net.liftweb.json.JsonAST._

import scala.language.implicitConversions

/**
  * Created by Rockie Yang on 2016/06/01.
  */
object JsonConversion {
  def stringToOptionString(s: String): Option[String] = Some(s)

  def isInt(value: Double): Boolean = {
    value == Math.floor(value) && !java.lang.Double.isInfinite(value)
  }

  def valueToJValue(value: Any): JValue = {
    value match {
      case v: Int =>
        JInt(v)
      case v: Long =>
        JInt(v)
      case v: Double =>
        if (isInt(v))
          JInt(v.toInt)
        else
          JDouble(v)

      case v: String =>
        JString(v)
      case v: Boolean =>
        JBool(v)
      case null =>
        JNull
      case v: JValue =>
        v
      case v: Code =>
        throw new Exception("it should not arrive here, BaseModel need be enhanced to support more fieldValue")
      case v: List[_] =>
        listToJArray(v)
      case v: Map[String @unchecked, _] =>
        mapToJObject(v)
      case x: Any =>
        JString(x.toString)
    }
  }

  def mapToJObject(vs: Map[String, Any]): JObject = {
    val vss = vs.map { case (name, value) =>
      JField(name, valueToJValue(value))
    }
    JObject(vss.toList)
  }

  //  def pairToJArray(vs: Tuple2): JArray = {
  //
  //  }

  def toJValue(value: Any): JValue = {
    value match {
      case v: List[_] =>
        listToJArray(v)

      case v: Map[String @unchecked, _] =>
        mapToJObject(v)
      case null =>
        JNull
      case v: Any =>
        valueToJValue(v)

    }
  }

  def listToJArray(vs: List[Any]): JArray = {
    val ar = vs.map {
      case v: JValue =>
        v
      case v: BaseModel =>
        v.result
      case v: (_, _) =>
        listToJArray(v.productIterator.toList)
      case v: List[_] =>
        if (v.size == 1)
          toJValue(v.head)
        else
          listToJArray(v)
      case v: Seq[_] =>
        if (v.size == 1)
          toJValue(v.head)
        else
          listToJArray(v.toList)
      case v: Map[String@unchecked, _] =>
        mapToJObject(v)
      case v: Array[_] =>
        listToJArray(v.toList)
      case value =>
        valueToJValue(value)
    }

    JArray(ar)
  }

//
//  def jvalueToString(jvalue: JValue): String = {
//    pretty(render(jvalue))
//  }
}