如何以更简洁的方式编写转换

时间:2020-05-27 17:19:22

标签: kotlin rx-java rx-java2

我有一个方法,它接受对象列表(小部件)-包含一些属性(标题)和嵌套列表(组件)。我想将列表展平为一个列表,并使用以下代码:

@SuppressLint("CheckResult")
    fun flatten(fatList: Single<List<Widget>>) {
        val flatList: MutableList<IUiData> = mutableListOf()

        fatList.map {
            Observable.fromIterable(it).map { widget ->
                if (widget.header.isNotEmpty()) {
                    flatList.add(ProductHeaderUi(widget.header))
                }
                widget.componentList.map { component ->
                    when (component.type) {
                        TILE_TEXT -> {
                            flatList.add(HeaderUi(component))
                        }
                        TILE_IMAGE -> {
                            flatList.add(ImageTileUi(component))
                        }
                        TILE_FOOTER -> {
                            flatList.add(FooterUi(component))
                        }
                        UNKNOWN -> {
                            //Do Nothing 
                        }
                    }
                }
            }
        }
    }

我打算通过此方法返回一个列表:Single<MutableList<IUiData>>,此目的可以立即实现,但我正在寻找一种更简洁的方法

1 个答案:

答案 0 :(得分:2)

您正在以非预期的方式同时使用Rx的Observable map和Kotlin的Iterable map。它们用于将一种类型转换为另一种类型,而不是迭代某些内容。

您还已经在最外面的map函数中嵌套了不必要的Observable迭代器。

您只需要映射Single的输出。在map函数内部,您迭代(而不是map)原始列表以提取MutableList所需的数据。

我是Rx的新手,没有检查此内容,因此对任何语法错误感到抱歉。

fun flatten(fatList: Single<List<Widget>>): Single<MutableList<IUData>> = fatList.map { widgetList ->
    val flatList: MutableList<IUiData> = mutableListOf()
    for (widget in widgetList) {
        if (widget.header.isNotEmpty()) {
            flatList.add(ProductHeaderUi(widget.header))
        }
        for (component in widget.componentList) { 
            when (component.type) {
                TILE_TEXT -> flatList.add(HeaderUi(component))
                TILE_IMAGE -> flatList.add(ImageTileUi(component))
                TILE_FOOTER -> flatList.add(FooterUi(component))
                // Else do nothing
            }
        }
    }
    flatList
}

但是为了与典型的Rx链接语法保持一致,我将其设为扩展函数,因此我必须像这样第一行。然后,您可以将其放在Rx调用链的中间:

fun Single<List<Widget>>.flatten(): Single<MutableList<IUData>> = map { widgetList ->

您还可以通过使用Kotlin的flatMap以更简洁,实用,但效率较低的方式执行此操作:

fun Single<List<Widget>>.flatten(): Single<MutableList<IUData>> = map {
    it.flatMap { widget ->
        listOfNotNull(widget.header.takeIf(Header::isNotEmpty)?.let(::ProductHeaderUi))
        +
        widget.componentList.mapNotNull { component ->            
           when (component.type) {
                TILE_TEXT -> HeaderUi(component)
                TILE_IMAGE -> ImageTileUi(component)
                TILE_FOOTER -> FooterUi(component)
                else -> null
            }
        }.toMutableList()
}

...其中Headerwidget.header使用的类型。