Add special unsafe calls for DeviceClient to mirror safe device
This commit is contained in:
parent
8bd9bcc6a6
commit
231f1bc858
@ -0,0 +1,79 @@
|
|||||||
|
package space.kscience.controls.client
|
||||||
|
|
||||||
|
import kotlinx.coroutines.CoroutineScope
|
||||||
|
import kotlinx.coroutines.Job
|
||||||
|
import kotlinx.coroutines.flow.*
|
||||||
|
import kotlinx.coroutines.launch
|
||||||
|
import space.kscience.controls.api.PropertyChangedMessage
|
||||||
|
import space.kscience.controls.api.requestProperty
|
||||||
|
import space.kscience.controls.spec.DeviceActionSpec
|
||||||
|
import space.kscience.controls.spec.DevicePropertySpec
|
||||||
|
import space.kscience.controls.spec.MutableDevicePropertySpec
|
||||||
|
import space.kscience.controls.spec.name
|
||||||
|
import space.kscience.dataforge.meta.Meta
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* An accessor that allows DeviceClient to connect to any property without type checks
|
||||||
|
*/
|
||||||
|
public suspend fun <T> DeviceClient.read(propertySpec: DevicePropertySpec<*, T>): T =
|
||||||
|
propertySpec.converter.metaToObject(readProperty(propertySpec.name)) ?: error("Property read result is not valid")
|
||||||
|
|
||||||
|
|
||||||
|
public suspend fun <T> DeviceClient.request(propertySpec: DevicePropertySpec<*, T>): T =
|
||||||
|
propertySpec.converter.metaToObject(requestProperty(propertySpec.name))
|
||||||
|
|
||||||
|
public suspend fun <T> DeviceClient.write(propertySpec: MutableDevicePropertySpec<*, T>, value: T) {
|
||||||
|
writeProperty(propertySpec.name, propertySpec.converter.objectToMeta(value))
|
||||||
|
}
|
||||||
|
|
||||||
|
public fun <T> DeviceClient.writeAsync(propertySpec: MutableDevicePropertySpec<*, T>, value: T): Job = launch {
|
||||||
|
write(propertySpec, value)
|
||||||
|
}
|
||||||
|
|
||||||
|
public fun <T> DeviceClient.propertyFlow(spec: DevicePropertySpec<*, T>): Flow<T> = messageFlow
|
||||||
|
.filterIsInstance<PropertyChangedMessage>()
|
||||||
|
.filter { it.property == spec.name }
|
||||||
|
.mapNotNull { spec.converter.metaToObject(it.value) }
|
||||||
|
|
||||||
|
public fun <T> DeviceClient.onPropertyChange(
|
||||||
|
spec: DevicePropertySpec<*, T>,
|
||||||
|
scope: CoroutineScope = this,
|
||||||
|
callback: suspend PropertyChangedMessage.(T) -> Unit,
|
||||||
|
): Job = messageFlow
|
||||||
|
.filterIsInstance<PropertyChangedMessage>()
|
||||||
|
.filter { it.property == spec.name }
|
||||||
|
.onEach { change ->
|
||||||
|
val newValue = spec.converter.metaToObject(change.value)
|
||||||
|
if (newValue != null) {
|
||||||
|
change.callback(newValue)
|
||||||
|
}
|
||||||
|
}.launchIn(scope)
|
||||||
|
|
||||||
|
public fun <T> DeviceClient.useProperty(
|
||||||
|
spec: DevicePropertySpec<*, T>,
|
||||||
|
scope: CoroutineScope = this,
|
||||||
|
callback: suspend (T) -> Unit,
|
||||||
|
): Job = scope.launch {
|
||||||
|
callback(read(spec))
|
||||||
|
messageFlow
|
||||||
|
.filterIsInstance<PropertyChangedMessage>()
|
||||||
|
.filter { it.property == spec.name }
|
||||||
|
.collect { change ->
|
||||||
|
val newValue = spec.converter.metaToObject(change.value)
|
||||||
|
if (newValue != null) {
|
||||||
|
callback(newValue)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public suspend fun <I, O> DeviceClient.execute(actionSpec: DeviceActionSpec<*, I, O>, input: I): O {
|
||||||
|
val inputMeta = actionSpec.inputConverter.objectToMeta(input)
|
||||||
|
val res = execute(actionSpec.name, inputMeta)
|
||||||
|
return actionSpec.outputConverter.metaToObject(res ?: Meta.EMPTY)
|
||||||
|
}
|
||||||
|
|
||||||
|
public suspend fun <O> DeviceClient.execute(actionSpec: DeviceActionSpec<*, Unit, O>): O {
|
||||||
|
val res = execute(actionSpec.name, Meta.EMPTY)
|
||||||
|
return actionSpec.outputConverter.metaToObject(res ?: Meta.EMPTY)
|
||||||
|
}
|
@ -4,7 +4,6 @@ import kotlinx.coroutines.flow.Flow
|
|||||||
import kotlinx.coroutines.flow.map
|
import kotlinx.coroutines.flow.map
|
||||||
import kotlinx.coroutines.test.runTest
|
import kotlinx.coroutines.test.runTest
|
||||||
import kotlinx.serialization.json.Json
|
import kotlinx.serialization.json.Json
|
||||||
import space.kscience.controls.api.Device
|
|
||||||
import space.kscience.controls.manager.DeviceManager
|
import space.kscience.controls.manager.DeviceManager
|
||||||
import space.kscience.controls.manager.install
|
import space.kscience.controls.manager.install
|
||||||
import space.kscience.controls.manager.respondMessage
|
import space.kscience.controls.manager.respondMessage
|
||||||
@ -25,9 +24,6 @@ import kotlin.test.assertContains
|
|||||||
import kotlin.time.Duration.Companion.milliseconds
|
import kotlin.time.Duration.Companion.milliseconds
|
||||||
|
|
||||||
|
|
||||||
public suspend fun <T> Device.readUnsafe(propertySpec: DevicePropertySpec<*, T>): T =
|
|
||||||
propertySpec.converter.metaToObject(readProperty(propertySpec.name)) ?: error("Property read result is not valid")
|
|
||||||
|
|
||||||
internal class RemoteDeviceConnect {
|
internal class RemoteDeviceConnect {
|
||||||
|
|
||||||
class TestDevice(context: Context, meta: Meta) : DeviceBySpec<TestDevice>(TestDevice, context, meta) {
|
class TestDevice(context: Context, meta: Meta) : DeviceBySpec<TestDevice>(TestDevice, context, meta) {
|
||||||
@ -82,6 +78,6 @@ internal class RemoteDeviceConnect {
|
|||||||
|
|
||||||
val remoteDevice = virtualMagixEndpoint.remoteDevice(context, "test", Name.EMPTY)
|
val remoteDevice = virtualMagixEndpoint.remoteDevice(context, "test", Name.EMPTY)
|
||||||
|
|
||||||
assertContains(0.0..1.0, remoteDevice.readUnsafe(TestDevice.value))
|
assertContains(0.0..1.0, remoteDevice.read(TestDevice.value))
|
||||||
}
|
}
|
||||||
}
|
}
|
Loading…
Reference in New Issue
Block a user