Объединение / слияние классов данных в Котлине
Есть ли способ объединить классы данных kotlin без указания всех свойств?
data class MyDataClass(val prop1: String, val prop2: Int, ...//many props)
с функцией со следующей подписью:
fun merge(left: MyDataClass, right: MyDataClass): MyDataClass
где эта функция проверяет каждое свойство в обоих классах и где они различаются, использует левый параметр для создания нового MyDataClass.
Возможно ли это с помощью kotlin-рефлекса или других средств?
РЕДАКТИРОВАТЬ: больше ясности
Вот лучшее описание того, что я хочу делать
data class Bob(
val name: String?,
val age: Int?,
val remoteId: String?,
val id: String)
@Test
fun bob(){
val original = Bob(id = "local_id", name = null, age = null, remoteId = null)
val withName = original.copy(name = "Ben")
val withAge = original.copy(age = 1)
val withRemoteId = original.copy(remoteId = "remote_id")
//TODO: merge without accessing all properties
// val result =
assertThat(result).isEqualTo(Bob(id = "local_id", name = "Ben", age=1, remoteId = "remote_id"))
}
5 ответов
infix fun <T : Any> T.merge(mapping: KProperty1<T, *>.() -> Any?): T {
//data class always has primary constructor ---v
val constructor = this::class.primaryConstructor!!
//calculate the property order
val order = constructor.parameters.mapIndexed { index, it -> it.name to index }
.associate { it };
// merge properties
@Suppress("UNCHECKED_CAST")
val merged = (this::class as KClass<T>).declaredMemberProperties
.sortedWith(compareBy{ order[it.name]})
.map { it.mapping() }
.toTypedArray()
return constructor.call(*merged);
}
редактировать
infix fun <T : Any> T.merge(right: T): T {
val left = this;
return left merge mapping@ {
// v--- implement your own merge strategy
return@mapping this.get(left) ?: this.get(right);
};
}
пример
val original = Bob(id = "local_id", name = null, age = null, remoteId = null)
val withName = original.copy(name = "Ben")
val withAge = original.copy(age = 1)
val withRemoteId = original.copy(remoteId = "remote_id")
val result = withName merge withAge merge withRemoteId;
Если вы хотите скопировать значения справа, когда значения слева null
тогда вы можете сделать следующее:
infix inline fun <reified T : Any> T.merge(other: T): T {
val nameToProperty = T::class.declaredMemberProperties.associateBy { it.name }
val primaryConstructor = T::class.primaryConstructor!!
val args = primaryConstructor.parameters.associate { parameter ->
val property = nameToProperty[parameter.name]!!
parameter to (property.get(this) ?: property.get(other))
}
return primaryConstructor.callBy(args)
}
Использование:
data class MyDataClass(val prop1: String?, val prop2: Int?)
val a = MyDataClass(null, 1)
val b = MyDataClass("b", 2)
val c = a merge b // MyDataClass(prop1=b, prop2=1)
Класс-специфический способ объединения классов данных, когда мы можем определить поля, которые мы хотим объединить, будет:
data class SomeData(val dataA: Int?, val dataB: String?, val dataC: Boolean?) {
fun combine(newData: SomeData): SomeData {
//Let values of new data replace corresponding values of this instance, otherwise fall back on the current values.
return this.copy(dataA = newData.dataA ?: dataA,
dataB = newData.dataB ?: dataB,
dataC = newData.dataC ?: dataC)
}
}
Решение @mfulton26 объединяет свойства, которые являются частью только основного конструктора. Я расширил это, чтобы поддержать все свойства
inline infix fun <reified T : Any> T.merge(other: T): T {
val nameToProperty = T::class.declaredMemberProperties.associateBy { it.name }
val primaryConstructor = T::class.primaryConstructor!!
val args = primaryConstructor.parameters.associate { parameter ->
val property = nameToProperty[parameter.name]!!
parameter to (property.get(other) ?: property.get(this))
}
val mergedObject = primaryConstructor.callBy(args)
nameToProperty.values.forEach { it ->
run {
val property = it as KMutableProperty<*>
val value = property.javaGetter!!.invoke(other) ?: property.javaGetter!!.invoke(this)
property.javaSetter!!.invoke(mergedObject, value)
}
}
return mergedObject
}
Ваши требования точно такие же, как при копировании left
значение:
fun merge(left: MyDataClass, right: MyDataClass) = left.copy()
Возможно, один из пользователей неправильно понимает другой. Пожалуйста, уточните, если это не то, что вы хотите.
Обратите внимание, что с right
не используется, вы можете сделать это vararg и "объединить" столько, сколько хотите:)
fun merge(left: MyDataClass, vararg right: MyDataClass) = left.copy()
val totallyNewData = merge(data1, data2, data3, data4, ...)
РЕДАКТИРОВАТЬ
Занятия в Котлине не отслеживают их дельты. Подумайте о том, что вы получаете, проходя этот процесс. После первого изменения у вас есть
current = Bob("Ben", null, null, "local_id")
next = Bob(null, 1, null, "local_id")
Как это должно знать, что вы хотите next
применить изменения к age
но нет name
? Если вы просто обновляете на основе nullability, у @mfulton есть хороший ответ. В противном случае вам необходимо предоставить информацию самостоятельно.