通用的问题推断类型

时间:2019-10-11 16:23:28

标签: crystal-lang

我有一个绑定的c库,其中有许多结构可容纳不同类型的数据。

@[Link("foo")]
lib LibFoo

  struct IntStruct
    data : LibC::Int
  end

  struct DoubleStruct
    data : LibC::Double
  end

  fun get_int_struct(n : LibC::SizeT) : IntStruct*
  fun get_double_struct(n : LibC::SizeT) : DoubleStruct*

end

我有一个通用类,其中我希望类型T是其中一个结构的类型,基于输入数组。

class GenericClass(T)
    @ptr : Pointer(T)
    @obj : T

    def initialize(arr : Array)
        @ptr = infer_type(arr)
        @obj = @ptr.value
    end
end

infer_type如下所示:

def infer_type(data : Array(Int32))
    return LibFoo.get_int_struct(data.size)
end

def infer_type(data : Array(Float64))
    return LibFoo.get_double_struct(data.size)
end

当我显式设置类型时,代码可以正常工作...

GenericClass(LibFoo::IntStruct).new [1, 2, 3, 4, 5]

...但是我想推断类型。

1 个答案:

答案 0 :(得分:3)

也许提供一种自定义的通用new方法对您有用吗?

struct A    
end

struct B    
end

class Generic(T)
  def self.new(items : Array(A)) forall A
    new(fetch_struct(items))
  end

  private def initialize(@struct : T)    
  end

  private def self.fetch_struct(items : Array(Int32))    
    A.new
  end

  private def self.fetch_struct(items : Array(Float64))
    B.new
  end
end

p Generic.new([1])
p Generic.new([1.0])

https://carc.in/#/r/7s6d