断言接口的类型

有志者，事竟成！如果你在学习Golang，那么本文《断言接口的类型》，就很适合你！文章讲解的知识点主要包括，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

在一般情况下，我无法优雅地将图像的像素作为数组获取。

f, err := os.open(imgpath)
check(err)
defer f.close()
img, _, err := image.decode(bufio.newreader(f))
check(err)
pixels, err := getpixels(img)
check(err)
// logic with pixels.

现在函数 getpixels 看起来像这样：

func getpixels(img image.image) ([]uint8, error) {
    if i, ok := img.(*image.nrgba); ok {
        return i.pix, nil
    } else if i, ok := img.(*image.alpha); ok {
        return i.pix, nil
    } else if i, ok := img.(*image.alpha16); ok {
        return i.pix, nil
    } else if i, ok := img.(*image.cmyk); ok {
        return i.pix, nil
    } else if i, ok := img.(*image.gray); ok {
        return i.pix, nil
    } else if i, ok := img.(*image.gray16); ok {
        return i.pix, nil
    } else if i, ok := img.(*image.nrgba64); ok {
        return i.pix, nil
    } else if i, ok := img.(*image.paletted); ok {
        return i.pix, nil
    } else if i, ok := img.(*image.rgba); ok {
        return i.pix, nil
    } else if i, ok := img.(*image.rgba64); ok {
        return i.pix, nil
    }
    return nil, fmt.errorf("unknown image type %t", img)
}

但我认为这很丑陋。 golang 知道图像类型，我更喜欢这样的东西：

func getPixels(img image.Image) ([]uint8, error) {
    if i, ok := img.(eval(fmt.Sprintf("%T", img))); ok {
        return i.Pix, nil
    }
    return nil, fmt.Errorf("unknown image type %T", img)
}

我也无法断言 reflect.typeof(img)。也许有一种方法可以从 reflect.type 接口获取类型？

解决方案

您的大 if ... else 结构可以通过使用 type switch 来简化，如下所示：

func getpixels(img image.image) ([]uint8, error) {
    switch i := img.(type) {
    case *image.nrgba:
        return i.pix, nil
    case *image.alpha:
        return i.pix, nil
    case *image.alpha16:
        return i.pix, nil
    case *image.cmyk:
        return i.pix, nil
        // ...
    }
    return nil, fmt.errorf("unknown image type %t", img)
}

您仍然需要列出所有可能的类型，但这样更好。

由于所有图像实现都是具有名为 pix 字段的结构指针，因此您可以使用反射来获取该字段。此实现将处理未来的图像实现，无需任何更改（如果它们也是带有 pix 字段的结构）。

这就是它的样子：

func getpix(img image.image) ([]uint8, error) {
    v := reflect.valueof(img)
    if v.kind() == reflect.ptr {
        v = v.elem()
    }

    if v.kind() == reflect.struct {
        pv := v.fieldbyname("pix")
        if pv.isvalid() {
            if pix, ok := pv.interface().([]uint8); ok {
                return pix, nil
            }
        }
    }

    return nil, fmt.errorf("unknown image type %t", img)
}

测试它：

fmt.println(getpix(&image.nrgba{}))
fmt.println(getpix(&image.rgba{}))

type unknownimage struct{ image.image }
fmt.println(getpix(unknownimage{}))

输出（在 Go Playground 上尝试）：

[] <nil>
[] <nil>
[] unknown image type main.unknownImage

以上就是《断言接口的类型》的详细内容，更多关于的资料请关注golang学习网公众号！