Parcelize: Handle class hierarchies of Parcelers (KT-46567)

This commit is contained in:
Steven Schäfer
2021-05-12 14:51:29 +02:00
committed by Alexander Udalov
parent 709c127f1b
commit bf7db84451
11 changed files with 293 additions and 34 deletions
@@ -0,0 +1,64 @@
// WITH_RUNTIME
@file:JvmName("TestKt")
package test
import kotlinx.android.parcel.*
import android.os.Parcel
import android.os.Parcelable
import java.util.Arrays
/**
* Generic pair parceler
* Create concrete object to use (see below)
*/
open class PairParceler<F: Any, S: Any>(private val firstParceler: Parceler<F>, private val secondParceler: Parceler<S>): Parceler<Pair<F, S>> {
/**
* Reads the [T] instance state from the [parcel], constructs the new [T] instance and returns it.
*/
override fun create(parcel: Parcel): Pair<F, S> =
firstParceler.create(parcel) to secondParceler.create(parcel)
/**
* Writes the [T] instance state to the [parcel].
*/
override fun Pair<F, S>.write(parcel: Parcel, flags: Int) {
with(firstParceler) { this@write.first.write(parcel, 0) }
with(secondParceler) { this@write.second.write(parcel, 0) }
}
}
object IntParceler: Parceler<Int> {
/**
* Reads the [T] instance state from the [parcel], constructs the new [T] instance and returns it.
*/
override fun create(parcel: Parcel): Int = parcel.readInt()
/**
* Writes the [T] instance state to the [parcel].
*/
override fun Int.write(parcel: Parcel, flags: Int) {
parcel.writeInt(this)
}
}
/**
* [Int] to [Int] pair parceler
*/
object IntToIntParceler: PairParceler<Int, Int>(IntParceler, IntParceler)
@Parcelize
@TypeParceler<Pair<Int, Int>, IntToIntParceler>
class A(val pair: Pair<Int, Int>): Parcelable
fun box() = parcelTest { parcel ->
val a1 = A(1 to 2)
a1.writeToParcel(parcel, 0)
val bytes = parcel.marshall()
parcel.unmarshall(bytes, 0, bytes.size)
parcel.setDataPosition(0)
val a2 = readFromParcel<A>(parcel)
assert(a1.pair == a2.pair)
}
@@ -0,0 +1,45 @@
// WITH_RUNTIME
// IGNORE_BACKEND: JVM
@file:JvmName("TestKt")
package test
import kotlinx.android.parcel.*
import android.os.Parcel
import android.os.Parcelable
abstract class UserParceler : Parceler<User> {
override fun User.write(parcel: Parcel, flags: Int) {
parcel.writeString(name)
}
override fun newArray(size: Int): Array<User> {
return Array(size + 1) { User(null) }
}
}
@Parcelize
class User(val name: String?) : Parcelable {
companion object : UserParceler() {
override fun create(parcel: Parcel) = User(parcel.readString())
}
}
fun box() = parcelTest { parcel ->
val user = User("John")
val user2 = User("Joe")
val array = arrayOf(user, user2)
parcel.writeTypedArray(array, 0)
val bytes = parcel.marshall()
parcel.unmarshall(bytes, 0, bytes.size)
parcel.setDataPosition(0)
val creator = User::class.java.getDeclaredField("CREATOR").get(null) as Parcelable.Creator<User>
val result = parcel.createTypedArray(creator)
assert(result.size == 3)
assert(result[0].name == user.name)
assert(result[1].name == user2.name)
assert(result[2].name == null)
}
@@ -167,4 +167,4 @@ public final class test/Foo : java/lang/Object, android/os/Parcelable {
RETURN
LABEL (L1)
}
}
}