给JSONObject添加自定义遍历方法

FunTester

发布于 2021-11-04 11:28:11

1.9K0

发布于 2021-11-04 11:28:11

文章被收录于专栏：FunTester

经过了Groovy动态添加方法和属性及Spock单测的铺垫，今天用fastjson为例，演示如何给JSONObject添加一个自定义的方法，来帮助提高编程效率。

需求

在fastjson的com.alibaba.fastjson.JSONObject其实并有直接给出遍历的方法，如果想在接口测试的过程中，需要打印JSON响应中的部分信息，都需要单独写一个for循环或者Lambda语法表达式，例如：

演示JSON数据

        def params = new JSONObject()
        params.code= 1
        params.msg= "FunTester"
        def data = new JSONObject()
        data.name = "张三"
        data.age = 22
        params.data = []
        params.data << data
        params.data << data.clone()
        params.data << data.clone()

打印内容：

~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~ JSON ~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~
＞  {
＞  ① . "msg":"FunTester",
＞  ① . "code":1,
＞  ① . "data":[
＞  ② . . . {
＞  ② . . . "name":"张三",
＞  ② . . . "age":22
＞  ① . },
＞  ① . {
＞  ② . . . "name":"张三",
＞  ② . . . "age":22
＞  ① . },
＞  ① . {
＞  ② . . . "name":"张三",
＞  ② . . . "age":22
＞  ② . . . }
＞  ① . ]
＞  }
~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~ JSON ~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~

直接实现输出方法

这里我们还是通过groovy.lang.MetaClass来实现。

        JSONObject.metaClass.fun = {
            def array = params.getJSONArray("data")
            array.each {
                sleep(1.0)
                output(Time.getNow(Time.DEFAULT_FORMAT.get()))
                output(it)
            }
        }

        params.fun()

控制台输出：

INFO-> main 当前用户：oker，工作目录：/Users/oker/IdeaProjects/funtester/,系统编码格式:UTF-8,系统Mac OS X版本:10.16
INFO-> main 2021-10-21 19:19:36
INFO-> main 
~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~ JSON ~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~
＞  {
＞  ① . "name":"张三",
＞  ① . "age":22
＞  }
~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~ JSON ~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~
INFO-> main 2021-10-21 19:19:37
INFO-> main 
~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~ JSON ~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~
＞  {
＞  ① . "name":"张三",
＞  ① . "age":22
＞  }
~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~ JSON ~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~
INFO-> main 2021-10-21 19:19:38
INFO-> main 
~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~ JSON ~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~
＞  {
＞  ① . "name":"张三",
＞  ① . "age":22
＞  }
~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~ JSON ~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~~☢~

Process finished with exit code 0

使用闭包灵活实现

这里我使用的是groovy.lang.Closure作为闭包的实现类，毕竟这个功能只能在Groovy脚本中实现。具体的文章有兴趣的可以翻一翻以前的文章。这次我用了两种闭包的实现方式，其中it关键字简直就是灵魂，非常推荐大家学Groovy。脚本代码如下：

        JSONObject.metaClass.fun2 = {f ->

            def array = params.getJSONArray("data")
            array.each {
                f(it)
            }
        }

        params.fun2({x -> output(x.get("name"))})
        params.fun2({output(it.get("age"))})

控制台输出：

INFO-> main 当前用户：oker，工作目录：/Users/oker/IdeaProjects/funtester/,系统编码格式:UTF-8,系统Mac OS X版本:10.16
INFO-> main 张三
INFO-> main 张三
INFO-> main 张三
INFO-> main 22
INFO-> main 22
INFO-> main 22

Process finished with exit code 0

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2021-10-28，如有侵权请联系 cloudcommunity@tencent.com 删除

json