| ... | @@ -2033,7 +2033,27 @@ test "union variant switch" { | ... | @@ -2033,7 +2033,27 @@ test "union variant switch" { |
| 2033 | assert(mem.eql(u8, what_is_it, "this is a number")); | 2033 | assert(mem.eql(u8, what_is_it, "this is a number")); |
| 2034 | } | 2034 | } |
| 2035 | | 2035 | |
| 2036 | // TODO union methods | 2036 | // Unions can have methods just like structs and enums: |
| | 2037 | |
| | 2038 | const Variant = union(enum) { |
| | 2039 | Int: i32, |
| | 2040 | Bool: bool, |
| | 2041 | |
| | 2042 | fn truthy(self: &const Variant) bool { |
| | 2043 | return switch (*self) { |
| | 2044 | Variant.Int => |x_int| x_int != 0, |
| | 2045 | Variant.Bool => |x_bool| x_bool, |
| | 2046 | }; |
| | 2047 | } |
| | 2048 | }; |
| | 2049 | |
| | 2050 | test "union method" { |
| | 2051 | var v1 = Variant { .Int = 1 }; |
| | 2052 | var v2 = Variant { .Bool = false }; |
| | 2053 | |
| | 2054 | assert(v1.truthy()); |
| | 2055 | assert(!v2.truthy()); |
| | 2056 | } |
| 2037 | | 2057 | |
| 2038 | | 2058 | |
| 2039 | const Small = union { | 2059 | const Small = union { |
Andrew Kelley <andrew@ziglang.org> 2018-04-08 18:03:02-04:00