call/cc variant of define/return

This commit is contained in:
2026-05-11 17:13:09 +02:00
parent 2fb61721f9
commit 0bbda2e6d6
2 changed files with 147 additions and 86 deletions
+99 -39
View File
@@ -2,35 +2,40 @@
@(require (for-label racket/base
racket/contract
(only-in ffi/unsafe cpointer?)
"../contract.rkt"))
@title{define/return/contract}
@title{define/contract/return}
@author[@author+email["Hans Dijkema" "hans@dijkewijk.nl"]]
@defmodule[define-return/contract]
The @racketmodname[define-return] library provides definition forms with an
explicit early return. This is useful in small defensive functions, and
especially around FFI bindings, where null pointers, error codes, unsupported
states, or failed preconditions should leave the function immediately.
The @racketmodname[define-return/contract] library provides
@racket[define/contract/return], a contracted definition form with an
explicit early return. It is useful in small defensive functions, and
especially around FFI bindings, where null pointers, error codes,
unsupported states, or failed preconditions should leave the function
immediately.
The early return is implemented with an internal exception. A @racket[return]
raises that exception, and the definition forms catch it around the function
body. The contracted form additionally checks early-returned values against
the result contract.
The return mechanism is based on @racket[call/cc]. A function definition
captures the continuation around the function body and binds it to the
return identifier supplied by the programmer. Calling that identifier
leaves the body and returns the supplied value as the result of the
function. No exception is raised and no exception handler is installed.
This module provides the contracted version of @racket{define/return}.
The module re-exports @racketmodname[racket/contract], so contracts such as
@racket[->], @racket[->*], @racket[any/c], @racket[or/c],
The module re-exports @racketmodname[racket/contract], so contracts such
as @racket[->], @racket[->*], @racket[any/c], @racket[or/c],
@racket[and/c], and @racket[listof] are available from the same
@racket[require].
@racket[require]. It also re-exports @racketmodname[define-return].
Note. This can lead to clashes of symbol @racket[->] with module @racketmodname[ffi/unsafe].
Note. This can lead to a clash between @racket[->] from
@racketmodname[racket/contract] and @racket[->] from
@racketmodname[ffi/unsafe].
@section{Contracted definitions}
@defform*[
((define/contract/return (name . formals)
((define/contract/return (name . formals) return-id
(contract-part ... result-contract)
body ...)
(define/contract/return name
@@ -38,19 +43,27 @@ Note. This can lead to clashes of symbol @racket[->] with module @racketmodname[
value))
]{
Like @racket[define/contract], but the body may use @racket[return].
Like @racket[define/contract], but the function form gives the body a
local early-return identifier.
The first form defines a contracted function. Ordinary results are checked by
@racket[define/contract]. Early-returned values are checked separately
against @racket[result-contract]. The implementation does this by defining a
small local contracted returner with the same result contract, so the result
contract is passed through Racket's contract machinery.
The first form defines a contracted function. The @racket[return-id]
identifier is part of the surface syntax. It is not a predefined binding
exported by this module. This keeps the form hygienic: the programmer
chooses the name of the local escape continuation, and the body uses that
same name explicitly.
Ordinary results are checked by @racket[define/contract]. Early-returned
values are checked separately against @racket[result-contract]. The
implementation does this by defining a small local contracted returner
with contract @racket[(-> any/c result-contract)]. The early return value
therefore passes through Racket's contract machinery before it leaves the
function body.
The contract must be written inline as a parenthesized contract form. The
last element is used as the early-return result contract.
@racketblock[
(define/contract/return (h x)
(define/contract/return (h x) return
(-> number? (or/c symbol? number?))
(when (< x 0) (return 'x-not-positive))
(let ((y (* x x)))
@@ -60,8 +73,8 @@ last element is used as the early-return result contract.
]
Here the result contract is @racket[(or/c symbol? number?)]. The symbol
returns are accepted, ordinary numeric results are accepted, and the string
@racket["Wrong answer!"] is rejected.
returns are accepted, ordinary numeric results are accepted, and the
string @racket["Wrong answer!"] is rejected.
@codeblock|{
(h -1) ; => 'x-not-positive
@@ -73,23 +86,23 @@ returns are accepted, ordinary numeric results are accepted, and the string
A zero-argument function works in the same way:
@racketblock[
(define/contract/return (z)
(define/contract/return (z) return
(-> symbol?)
(let ((cs (current-seconds)))
(when (= (remainder cs 3) 0) (return "deelbaar door 3"))
(when (= (remainder cs 2) 0) (return 'dividable-by-2))
(when (= (remainder cs 3) 0) (return "divisible by 3"))
(when (= (remainder cs 2) 0) (return 'divisible-by-2))
'yes))
]
The result contract is @racket[symbol?]. Returning
@racket['dividable-by-2] or @racket['yes] is accepted. Returning the string
@racket["deelbaar door 3"] is rejected.
@racket['divisible-by-2] or @racket['yes] is accepted. Returning the
string @racket["divisible by 3"] is rejected.
Rest arguments can be used when the corresponding contract form is accepted
by @racket[define/contract]:
Rest arguments can be used when the corresponding contract form is
accepted by @racket[define/contract]:
@racketblock[
(define/contract/return (sum . xs)
(define/contract/return (sum . xs) return
(->* () #:rest (listof number?) number?)
(when (null? xs) (return 0))
(apply + xs))
@@ -101,18 +114,65 @@ by @racket[define/contract]:
(sum 1 'x) ; contract violation
}|
The second form defines a contracted value. The value expression may use
@racket[return], and the returned value is checked against
@racket[result-contract].
Since @racket[return-id] is a continuation wrapped in a local contracted
procedure, it is called with one value. The continuation should normally
be used only to escape from the dynamic extent of the function body.
The second form defines a contracted value. The value is checked against
@racket[result-contract]. This form has no @racket[return-id] position,
so it does not expose a local early-return identifier. Use the function
form when a body needs an explicit early return.
@racketblock[
(define/contract/return v
(define/contract/return answer
number?
(return 'ss))
42)
]
This definition raises a contract violation, because @racket['ss] does not
satisfy @racket[number?].
This definition succeeds because @racket[42] satisfies
@racket[number?].
@racketblock[
(define/contract/return bad-answer
number?
'not-a-number)
]
This definition raises a contract violation, because
@racket['not-a-number] does not satisfy @racket[number?].
}
@section{FFI-style null pointers}
Many FFI bindings represent a native null pointer as @racket[#f]. The
explicit return identifier makes it easy to leave a wrapper as soon as a
required pointer is missing, while still checking the final result against
the function contract.
@racketblock[
(define/contract/return (make-stream-codec-context stream) return
(-> cpointer? (or/c cpointer? symbol?))
(define codecpar (AVStream-codecpar stream))
(unless codecpar
(return 'missing-codec-parameters))
(define codec
(avcodec-find-decoder
(AVCodecParameters-codec-id codecpar)))
(unless codec
(return 'decoder-not-found))
(define ctx (avcodec-alloc-context3 codec))
(unless ctx
(return 'alloc-codec-context-failed))
ctx)
]
The checks are sequential. The decoder lookup is evaluated only when
@racket[codecpar] is a valid pointer, and the context allocation is
evaluated only when @racket[codec] is a valid pointer. A failed pointer
check returns the corresponding symbol immediately. A successful path
returns @racket[ctx], and both the ordinary result and early-returned
symbols are covered by the result contract.