v/vlib/builtin/option.v

141 lines
3.1 KiB
V
Raw Normal View History

// Copyright (c) 2019-2021 Alexander Medvednikov. All rights reserved.
2019-06-23 04:21:30 +02:00
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
2019-06-22 20:20:28 +02:00
module builtin
2020-05-31 12:57:26 +02:00
struct OptionBase {
ok bool
is_none bool
error string
ecode int
// Data is trailing after ecode
2020-11-06 15:26:59 +01:00
// and is not included in here but in the
2020-05-31 12:57:26 +02:00
// derived Option_xxx types
}
2019-12-19 21:52:45 +01:00
2020-05-31 12:57:26 +02:00
// `fn foo() ?Foo { return foo }` => `fn foo() ?Foo { return opt_ok(foo); }`
fn opt_ok2(data voidptr, mut option OptionBase, size int) {
2020-05-31 12:57:26 +02:00
unsafe {
*option = OptionBase{
2020-05-31 12:57:26 +02:00
ok: true
}
// use ecode to get the end of OptionBase and then memcpy into it
C.memcpy(byteptr(&option.ecode) + sizeof(int), data, size)
}
2020-05-31 12:57:26 +02:00
}
// Option is the old option type used for bootstrapping
2019-06-22 20:20:28 +02:00
struct Option {
ok bool
is_none bool
2020-05-18 21:38:06 +02:00
error string
ecode int
2019-06-22 20:20:28 +02:00
}
// str returns the string representation of the Option.
pub fn (o Option) str() string {
if o.ok && !o.is_none {
return 'Option{ ok }'
}
if o.is_none {
return 'Option{ none }'
}
return 'Option{ error: "$o.error" }'
}
// opt_none is used internally when returning `none`.
fn opt_none() Option {
2019-12-19 21:52:45 +01:00
return Option{
2020-05-18 21:38:06 +02:00
ok: false
2019-12-19 21:52:45 +01:00
is_none: true
}
}
// error returns an optional containing the error given in `message`.
// `if ouch { return error('an error occurred') }`
pub fn error(message string) Option {
2019-12-19 21:52:45 +01:00
return Option{
2020-05-18 21:38:06 +02:00
ok: false
is_none: false
error: message
}
}
// error_with_code returns an optional containing both error `message` and error `code`.
// `if ouch { return error_with_code('an error occurred',1) }`
pub fn error_with_code(message string, code int) Option {
2019-12-19 21:52:45 +01:00
return Option{
2020-05-18 21:38:06 +02:00
ok: false
is_none: false
error: message
ecode: code
}
2020-05-31 12:57:26 +02:00
}
2021-02-22 17:44:15 +01:00
// Option2 is the base of V's new internal optional return system.
2021-02-22 17:44:15 +01:00
struct Option2 {
state byte
err Error
// Data is trailing after err
// and is not included in here but in the
// derived Option2_xxx types
}
// Error holds information about an error instance
pub struct Error {
pub:
2021-02-22 17:44:15 +01:00
msg string
code int
}
[inline]
fn (e Error) str() string {
// TODO: this should probably have a better str method,
// but this minimizes the amount of broken code after #8924
return e.msg
}
// `fn foo() ?Foo { return foo }` => `fn foo() ?Foo { return opt_ok(foo); }`
fn opt_ok(data voidptr, mut option Option2, size int) {
unsafe {
*option = Option2{}
// use err to get the end of OptionBase and then memcpy into it
C.memcpy(byteptr(&option.err) + sizeof(Error), data, size)
}
}
2021-02-22 17:44:15 +01:00
// /*
pub fn (o Option2) str() string {
if o.state == 0 {
return 'Option2{ ok }'
}
if o.state == 1 {
return 'Option2{ none }'
}
return 'Option2{ err: "$o.err" }'
2021-02-22 17:44:15 +01:00
}
// error returns an optional containing the error given in `message`.
// `if ouch { return error('an error occurred') }`
pub fn error2(message string) Option2 {
return Option2{
state: 2
err: {
msg: message
}
}
}
// error_with_code returns an optional containing both error `message` and error `code`.
// `if ouch { return error_with_code('an error occurred',1) }`
pub fn error_with_code2(message string, code int) Option2 {
return Option2{
state: 2
err: {
msg: message
code: code
}
}
}
// */