在Kotlin访问属性委托

时间:2015-06-17 15:43:12

标签: properties delegates kotlin

Kotlin委托了属性,这是一个非常好的功能。但有时get()set()方法还不够。假设我想懒洋洋地创建一个Closeable对象并稍后关闭它。以下是如何实现此类委托属性的示例:

fun <T : Closeable> closeableLazy(initializer: () -> T) =
        CloseableLazyVal(initializer)

class CloseableLazyVal<T : Closeable>(
    private val initializer: () -> T
) : ReadOnlyProperty<Any?, T> {

    private var value: T? = null

    override fun get(thisRef: Any?, desc: PropertyMetadata): T {
        if (value == null) {
            value = initializer()
        }
        return value
    }

    fun close() {
        value?.close()
    }
}

这就是我想用它的方式:

private val stream by closeableLazy { FileOutputStream("/path/to/file") }

fun writeBytes(bytes: ByteArray) {
    stream.write(bytes)
}

override fun close() {
    stream::delegate.close() // This line will not compile
}

不幸的是,这种方法不起作用,因为看起来Kotlin似乎不允许直接访问属性委托。有什么办法可以做我想要的吗?或者是否有任何计划将此类功能添加到Kotlin,因为它将是一个非常简洁的功能。

2 个答案:

答案 0 :(得分:6)

好的,所以我提出了以下解决方案:

fun <T : Closeable> closeableLazy(initializer: () -> T) =
        CloseableLazyVal(initializer)

class CloseableLazyVal<T : Closeable>(
        private val initializer: () -> T
) : ReadOnlyProperty<CloseableDelegateHost, T> {

    private var value: T? = null

    override fun get(thisRef: CloseableDelegateHost, desc: PropertyMetadata): T {
        if (value == null) {
            value = initializer()
            thisRef.registerCloseable(value!!)
        }
        return value!!
    }

}

interface CloseableDelegateHost : Closeable {
    fun registerCloseable(prop : Closeable)
}

class ClosableDelegateHostImpl : CloseableDelegateHost {

    val closeables = arrayListOf<Closeable>()

    override fun registerCloseable(prop: Closeable) {
        closeables.add(prop)
    }

    override fun close() = closeables.forEach { it.close() }
}

class Foo : CloseableDelegateHost by ClosableDelegateHostImpl() {
    private val stream by closeableLazy { FileOutputStream("/path/to/file") }

    fun writeBytes(bytes: ByteArray) {
        stream.write(bytes)
    }

}

注意,属性的get方法有一个参数thisRef。我要求它继承自CloseableDelegateHost,它会在关闭时关闭任何已注册的Closeable。为了简化实现,我将此接口委托给一个简单的基于列表的实现。

更新(从评论中复制): 我意识到,您可以将委托声明为单独的属性,然后将第二个属性委托给它。这样您就可以轻松访问代理本身。

private val streamDelegate = closeableLazy { FileOutputStream("/path/to/file") }
private val stream by streamDelegate

fun writeBytes(bytes: ByteArray) {
    stream.write(bytes)
}

override fun close() {
    streamDelegate.close()
}

答案 1 :(得分:6)

在Kotlin 1.1中(从beta 2开始),可以从属性中检索代理,因此您现在可以编写

override fun close() {
    (::stream.apply { isAccessible = true }.getDelegate() 
        as CloseableLazyVal<*>).close()
}