checker: fix generic sumtype method (#11565)
parent
95136cbfc7
commit
d6fa6a459c
|
@ -2278,9 +2278,17 @@ pub fn (mut c Checker) method_call(mut node ast.CallExpr) ast.Type {
|
||||||
method = m
|
method = m
|
||||||
has_method = true
|
has_method = true
|
||||||
} else {
|
} else {
|
||||||
if left_type_sym.info is ast.Struct {
|
if left_type_sym.kind in [.struct_, .sum_type, .interface_] {
|
||||||
if left_type_sym.info.parent_type != 0 {
|
mut parent_type := ast.void_type
|
||||||
type_sym := c.table.get_type_symbol(left_type_sym.info.parent_type)
|
if left_type_sym.info is ast.Struct {
|
||||||
|
parent_type = left_type_sym.info.parent_type
|
||||||
|
} else if left_type_sym.info is ast.SumType {
|
||||||
|
parent_type = left_type_sym.info.parent_type
|
||||||
|
} else if left_type_sym.info is ast.Interface {
|
||||||
|
parent_type = left_type_sym.info.parent_type
|
||||||
|
}
|
||||||
|
if parent_type != 0 {
|
||||||
|
type_sym := c.table.get_type_symbol(parent_type)
|
||||||
if m := c.table.type_find_method(type_sym, method_name) {
|
if m := c.table.type_find_method(type_sym, method_name) {
|
||||||
method = m
|
method = m
|
||||||
has_method = true
|
has_method = true
|
||||||
|
|
|
@ -0,0 +1,32 @@
|
||||||
|
struct Empty {}
|
||||||
|
|
||||||
|
struct Node<T> {
|
||||||
|
value T
|
||||||
|
left Leaf<T>
|
||||||
|
right Leaf<T>
|
||||||
|
}
|
||||||
|
|
||||||
|
type Leaf<T> = Empty | Node<T>
|
||||||
|
|
||||||
|
// return size(number of nodes) of BST
|
||||||
|
fn (leaf Leaf<T>) size<T>() int {
|
||||||
|
return match leaf {
|
||||||
|
Empty { 0 }
|
||||||
|
Node<T> { 1 + leaf.left.size() + leaf.right.size() }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn test_generic_sumtype_method() {
|
||||||
|
r := Node<int>{
|
||||||
|
value: 20
|
||||||
|
left: Empty{}
|
||||||
|
right: Empty{}
|
||||||
|
}
|
||||||
|
tree := Leaf<int>(Node<int>{
|
||||||
|
value: 10
|
||||||
|
left: Empty{}
|
||||||
|
right: r
|
||||||
|
})
|
||||||
|
println(tree.size())
|
||||||
|
assert tree.size() == 2
|
||||||
|
}
|
Loading…
Reference in New Issue