将接口断言为其类型

Question

在一般情况下，我无法优雅地将图像的像素作为数组获取。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界面获取类型？

Answer 1

你的大if ... else结构可以通过使用这样的类型开关来简化：

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

由于所有图像实现都是具有名为 的字段的结构指针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