Android Kotlin 全局深拷贝的扩展函数

前言

基于Kotlin的扩展函数的特性,我们能很容易封装一个全局的函数来实现某些功能
全局的UI线程回调函数,不管在代码的任何地方,都可以让指定代码块运行到UI线程上

这次就分享一个深拷贝的扩展函数
思路就是通过反射创建一个新的实例,然后读出要拷贝的对象每一个属性的值再重新怼回去
具体可以看代码是怎么实现的

/深拷贝
fun <T: Any> T.deepCopy(): T {

    val copiedObjects = mutableMapOf<Any,Any>()
    var initialObject = getValueFromCopiedCollection(this,copiedObjects)
    initialObject = deepCopy(this, copiedObjects)
    return initialObject
}

private fun<T: Any> deepCopy(obj: T,copiedObjects: MutableMap<Any,Any>): T{

    val objClassJava = obj::class.java
    val objClass = obj::class

    // 基本数据类型直接返回
    if (objClassJava.isPrimitive) {
        return obj
    } else {
        if (objClass.javaPrimitiveType != null) {
            return obj
        }
    }

    when(obj){
        is String -> {
            val newString = String(obj.toCharArray())
            copiedObjects[obj] = newString
            return newString as T
        }
        is Array<*> -> {
            val arrList = ArrayList<Any?>()
            val newArray: Any

            for (elem in obj){
                if (elem == null)
                    arrList.add(elem)
                else
                    arrList.add(getValueFromCopiedCollection(elem, copiedObjects))
            }
            newArray = obj.clone()
            arrList.toArray(newArray)
            copiedObjects[obj] = newArray
            return newArray as T
        }
        is List<*> -> {
            val arrList = ArrayList<Any?>()
            for (elem in obj){
                if (elem == null)
                    arrList.add(elem)
                else
                    arrList.add(getValueFromCopiedCollection(elem, copiedObjects))
                copiedObjects[obj] = arrList
            }
        }
        is Map<*,*> -> {
            val newMap = mutableMapOf<Any?,Any?>()
            for ((key,value) in obj){
                if (key == null){
                    if (value == null)
                        newMap[key] = value
                     else
                        newMap[key] = getValueFromCopiedCollection(value,copiedObjects)
                } else {
                    if (value === null)
                        newMap[getValueFromCopiedCollection(key,copiedObjects)] = value
                    else
                        newMap[getValueFromCopiedCollection(key,copiedObjects)] = getValueFromCopiedCollection(value,copiedObjects)
                }
            }
            copiedObjects[obj] = newMap
            return newMap as T
        }
        is Set<*> -> {
            val newSet = mutableSetOf<Any?>()
            for (elem in obj){
                if (elem == null)
                    newSet.add(elem)
                else
                    newSet.add(getValueFromCopiedCollection(elem, copiedObjects))
            }
            copiedObjects[obj] = newSet
            return newSet as T
        }
        is Date -> {
            return Date(obj.time) as T
        }
        else -> {
            println("deepCopy, class name: ${objClass.simpleName}")
            val properties = objClass.memberProperties
            val newCopy = objClassJava.newInstance()

            properties.forEach {prop ->
                val field = prop.javaField
                if (field != null){
                    field.isAccessible = true
                    val value = field.get(obj)
                    val type = field.type

                    if (value == null)
                        field.set(newCopy, value)
                    else
                        field.set(newCopy, getValueFromCopiedCollection(value,copiedObjects))
                }
            }
            copiedObjects[obj] = newCopy
            return newCopy
        }
    }
    //傻逼kotlin,前面都return完了,还要强行你这样写,虽然可以return when,但是代码不直观
    return obj
}

private fun<T: Any> getValueFromCopiedCollection(value: T,
                                                 copiedObjects: MutableMap<Any,Any>): T{

    if (copiedObjects.containsKey(value)){
        return copiedObjects[value] as T
    }

    var tempValue: Any? = null
    if (!value::class.java.isPrimitive
        && value::class.javaPrimitiveType == null
        && !value::class.java.isArray
        && (value !is Collection<*>)){

        tempValue = value::class.createInstance()
    }

    if (copiedObjects.isNotEmpty()){
        tempValue = deepCopy(value,copiedObjects)
    }

    if (tempValue == null)
        tempValue = value
    else
        copiedObjects[value] = tempValue

    return tempValue as T
}

单元测试(用法)

class DeepCloneTest {

    @Test
    fun testThisClone() {
        val thisCopy = deepCopy()
        assertFalse(thisCopy === this)
    }

    @Test
    fun testStringClone() {
        val string = "水天滑稽天照八野滑稽石"
        val stringClone = string.deepCopy()

        assertNotNull(stringClone)
        assertEquals(string, stringClone)
        assertFalse(string === stringClone)
    }

    @Test
    fun testIntegerPrimitiveClone() {
        val integer = 100500
        val integerClone = integer.deepCopy()

        assertNotNull(integerClone)
        assertTrue(integer == integerClone)
    }

    @Test
    fun testIntegerClone() {
        val integer = Integer(100500)
        val integerClone =  integer.deepCopy()

        println(integerClone.toString())

        assertNotNull(integerClone)
        assertTrue(integer == integerClone)
    }

    @Test
    fun testListClone() {
        val listOfStrings = listOf("a", "b", "c")
        val listOfStringsClone =  listOfStrings.deepCopy() as List<String>

        assertNotNull(listOfStringsClone)
        assertTrue(listOfStringsClone.isNotEmpty())
        assertTrue(listOfStringsClone.size == 3)
    }

    @Test
    fun testSimpleObjectClone() {
        val simpleObject = SimpleObject()
        val simpleObjectClone =  simpleObject.deepCopy()

        assertNotNull(simpleObjectClone)
        assertTrue(simpleObject == simpleObjectClone)
        assertFalse(simpleObject === simpleObjectClone)
    }

    @Test
    fun testComplexObjectClone() {
        val complexObject = ComplexObject()
        val complexObjectClone =  complexObject.deepCopy()

        assertNotNull(complexObjectClone)
        assertFalse(complexObject === complexObjectClone)

    }

}

data class SimpleObject(
    var foo: String = "foo",
    var bar: String = "bar"
)

class ComplexObject {
    val noField: String
        get() {
            return "123"
        }

    var a = 1
        get() = field + 1
        set(value) {
            field = value + 5
        }

    val b = 2

    var listExample = mutableListOf(1, 2, 3)
    var mapExample = mutableMapOf(1 to 1, 2 to 2, 3 to 3)
    var arrExample = arrayOf(arrayOf(1, 2, 3), 2, 3)
    var hashMapExample = hashMapOf(1 to 2, 3 to 2, 2 to 3)
    var setExample = setOf(1, 2, 3)
    var nullableInt: Int? = null
    var date = Date()

    val simpleNestedObject = SimpleObject()

}

全部通过就没事了,具体实现和用法简单看下代码也就明白啦~


©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容