package com.bw.functest

import com.alibaba.fastjson.{JSON, JSONArray, JSONObject}

import scala.collection.JavaConversions._

object JsonFilter {

/**

* 将json字符串中的key全部转为小写

* @param jsonStr 待转json字符串

* @return 转换完成的json对象

*/

def transToLowerObject(jsonStr: String): JSONObject = {

// 用来重装key变小写的json

val obj2 = new JSONObject()

// 将json字符串转为json对象

val obj1 = JSON.parseObject(jsonStr)

// 遍历json中的key

for (key:String

// 获得对应key的value

val obj = obj1.get(key)

// 如果获得的value仍然为json对象，key改小写，value为参数再次回调

if (obj.getClass.toString.endsWith("JSONObject")){

obj2.put(key.toLowerCase(), transToLowerObject(obj.toString))

// 如果获得的value为JSONArray对象，key改小写，调下面的transToArray方法

}else if (obj.getClass.toString.endsWith("JSONArray")){

obj2.put(key.toLowerCase(), transToArray(obj1.getJSONArray(key).toString))

// 如果不是以上两种情况，key改小写，value原样放入

}else{

obj2.put(key.toLowerCase(), obj)

}

}

obj2

}

/**

* 将jsonArray字符串中的key全部转为小写

* @param jsonArr 待转jsonArray字符串

* @return 转换完成的jsonArray对象

*/

def transToArray(jsonArr: String): JSONArray = {

// 用来重装key变小写的jsonArray

val jsonArr2 = new JSONArray()

// 将字符串转为jsonArray对象

val jsonArr1 = JSON.parseArray(jsonArr)

// 遍历jsonArray对象中每个json对象

for (i

// 获取第i个元素

val jArray = jsonArr1.getJSONObject(i)

// 如果第i个元素是json对象，调用上面的transToLowerObject方法

if(jArray.getClass.toString.endsWith("JSONObject")){

jsonArr2.add(transToLowerObject(jArray.toString))

// 如果第i个元素是JSONArray对象，回调transToArray方法

}else if (jArray.getClass.toString.endsWith("JSONArray")){

jsonArr2.add(transToArray(jArray.toString))

}else{

// 否则出现未知情况

println("json中key转小写出现未知情况")

throw new Exception("json中key转小写出现未知情况")

}

}

jsonArr2

}

def main(args: Array[String]): Unit = {

val a = "{'DATA':{'A':'a', 'B':'b', 'C':[{'D':1, 'E':2.00},{'A':'a'}]}, 'MM':[{'A':'a', 'B':'b'}, {'S':{'P':'p', 'Q':'q'}}]}"

// 如果a是json对象的字符串，用transToLowerObject方法，如果a是jsonArray对象的字符串，则直接用transToArray方法

val result = transToLowerObject(a)

println(result)

}

}