在Go中使用反射检查兼容类型



虽然我知道在Go中恐慌可能不是惯用的,但我想测试一下,以确保函数在某些条件下恐慌,而不是在其他条件下恐慌。

函数的示例。

func PanicOnErr(potentialErr error) {
    if potentialErr != nil {
        panic(potentialErr)
    }
}

以下是用于检查函数是否会死机的实现。

func InvocationCausedPanic(f interface{}, params ...interface{}) bool {
    // Obtain the function's signature.
    reflectedFunc := reflect.ValueOf(f)
    funcType := reflect.TypeOf(f)
    if funcType.NumIn() != len(params) {
        panic("InvocationCausedPanic called with a function and an incorrect number of parameter(s).")
    }
    reflectedParams := make([]reflect.Value, len(params))
    for paramIndex, paramValue := range params {
        expectedType := funcType.In(paramIndex)
        actualType := reflect.TypeOf(paramValue)
        if actualType != expectedType {
            errStr := fmt.Sprintf("InvocationCausedPanic called with a mismatched parameter type [parameter #%v: expected %v; got %v].", paramIndex, expectedType, actualType)
            panic(errStr)
        }
        reflectedParams[paramIndex] = reflect.ValueOf(paramValue)
    }
    return invoke(reflectedFunc, reflectedParams)
}
func invoke(reflectedFunc reflect.Value, reflectedParams []reflect.Value) (panicked bool) {
    defer func() {
        if r := recover(); r != nil {
            panicked = true
        }
    }()
    reflectedFunc.Call(reflectedParams)
    return
}

调用以下任一项都将导致类型检查失败。

InvocationCausedPanic(PanicOnErr, errors.New("Some error."))
InvocationCausedPanic(PanicOnErr, nil)

然而,似乎可以使用nil和通过调用errors.New生成的东西(似乎是*errors.errorString类型)来调用PanicOnErr

因此,是否有方法检查某个参数的类型是否适合调用某个函数

虽然我知道可以使用defer和recover来更简单地测试函数,但我很好奇是否可以编写一个可以接受任何函数和参数的通用函数,并确定它是否导致了恐慌(假设函数完成)。

相关围棋游戏:http://play.golang.org/p/qUG7OGuIbD

使用此函数确定参数是否兼容:

func compatible(actual, expected reflect.Type) bool {
  if actual == nil {
    k := expected.Kind()
    return k == reflect.Chan || 
           k == reflect.Func || 
           k == reflect.Interface || 
           k == reflect.Map || 
           k == reflect.Ptr || 
           k == reflect.Slice
  }
  return actual.AssignableTo(expected)
}

操场

相关内容

  • 没有找到相关文章

最新更新