有没有一种方法可以与golang中的任何指针对象进行比较?

时间:2019-02-06 23:51:07

标签: go

我正在尝试做一个比较器,它使用两个字符串指针或两个整数指针并返回结果。指针中的任何一个都可以为nil,只有当它们具有值并且值相等时,我才希望为真。

我尝试使用

这样的界面
type T *interface{}
func compare(a T,b T) bool

因为将* string强制转换为T时我必须检查nil,所以它是没有用的。

我希望调用

之类的函数
var a *string
var b *string
if compare(a, b){
  // do something
}

var a *string
var b *string
if a.equal(b){ 
  // do sth 
}

1 个答案:

答案 0 :(得分:0)

鉴于您仅使用两种类型,请使用type assertions获取具体的指针类型并进行比较:

func compare(a interface{}, b interface{}) bool {
    switch a := a.(type) {
    case *string:
        b, _ := b.(*string)
        return a != nil && b != nil && *a == *b
    case *int:
        b, _ := b.(*int)
        return a != nil && b != nil && *a == *b
    default:
        return false
    }
}

以下是调用该函数的一些示例:

a := "hello"
b := "world"
c := "hello"
fmt.Println(compare(&a, nil))
fmt.Println(compare(nil, &a))
fmt.Println(compare(&a, &b))
fmt.Println(compare(&a, &a))
fmt.Println(compare(&a, &c))

Run it in the Playground

使用reflect API比较所有可比较的类型:

func compare(a interface{}, b interface{}) bool {
    if a == nil && b == nil {
        return true
    }
    if a == nil || b == nil {
        return false
    }
    va := reflect.ValueOf(a)
    vb := reflect.ValueOf(b)
    if va.Type() != vb.Type() {
        return false
    }
    if va.Kind() == reflect.Ptr {
        if va.IsNil() || vb.IsNil() {
            return false
        }
        va = va.Elem()
        vb = vb.Elem()
    }
    if !va.Type().Comparable() {
        return false
    }
    return va.Interface() == vb.Interface()
}