Add contract for 'use'
#KT-35216 Fixed
This commit is contained in:
committed by
Ilya Gorbunov
parent
043eddb432
commit
941de655c4
@@ -7,6 +7,7 @@
|
||||
package kotlin.io
|
||||
|
||||
import java.io.Closeable
|
||||
import kotlin.contracts.*
|
||||
import kotlin.internal.*
|
||||
|
||||
/**
|
||||
@@ -19,6 +20,9 @@ import kotlin.internal.*
|
||||
@InlineOnly
|
||||
@RequireKotlin("1.2", versionKind = RequireKotlinVersionKind.COMPILER_VERSION, message = "Requires newer compiler version to be inlined correctly.")
|
||||
public inline fun <T : Closeable?, R> T.use(block: (T) -> R): R {
|
||||
contract {
|
||||
callsInPlace(block, InvocationKind.EXACTLY_ONCE)
|
||||
}
|
||||
var exception: Throwable? = null
|
||||
try {
|
||||
return block(this)
|
||||
|
||||
@@ -0,0 +1,20 @@
|
||||
/*
|
||||
* Copyright 2010-2019 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package test.io
|
||||
|
||||
import org.junit.Test
|
||||
import kotlin.test.assertEquals
|
||||
|
||||
class CloseableTest {
|
||||
@Test
|
||||
fun shouldHaveBlockExactlyOnceContract() {
|
||||
val i: Int
|
||||
"".byteInputStream().use {
|
||||
i = 1
|
||||
}
|
||||
assertEquals(1, i)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user