Go 快速入门指南 - 判断是否实现接口

判断是否实现接口

Go 里面没有 implements 关键字来判断一个结构体 (对象) 是否实现了某个接口， 要实现类似的功能， 需要用到 类型断言。

类型断言

// 判断 v 是否实现了接口 i
// 返回值 ok 为 true 表示已实现，ok 为 false 表示未实现
if _, ok := v.(i); ok {
// do something...
} 

例子

判断是否实现接口

package main

import "fmt"

type geometry interface {
    area() float64
    perimeter() float64
}

type rectangle struct {
    width, height float64
}

type circle struct {
    radius float64
}

func (r *rectangle) area() float64 {
    return r.width * r.height
}

func (r *rectangle) perimeter() float64 {
    return (r.width + r.height) * 2
}

func main() {
    var r interface{}

    r = &rectangle{
        width:  10,
        height: 5,
    }
    if v, ok := r.(geometry); ok {
        fmt.Printf("r implements interface geometry, area = %.2f, perimeter = %.2f \n", v.area(), v.perimeter())
    }

    var c interface{}
    c = &circle{
        radius: 10,
    }
    if _, ok := c.(geometry); !ok {
        fmt.Println("c does not implement interface geometry")
    }
}

// $ go run main.go
// 输出如下 
/**
  r implements interface geometry, area = 50.00, perimeter = 30.00
  c does not implement interface geometry
*/

实现系统错误接口

error 数据类型本质上就是一个接口，原型如下:

package builtin

type error interface {
    Error() string
}

例子

实现 error 接口

package main

import (
    "fmt"
)

// 自定义错误结构体
type divideError struct {
    msg string
}

// 实现 error 接口
func (d *divideError) Error() string {
    return d.msg
}

func newDivideError() *divideError {
    return &divideError{
        msg: "divide by zero",
    }
}

// 自定义除法函数
func myDivide(dividend, divisor float64) (float64, error) {
    if divisor == 0 { // 除数不能为 0
        return 0, newDivideError() // 返回一个自定义错误
    }
    return dividend / divisor, nil
}

func main() {
    divide, err := myDivide(100, 0)
    if err != nil {
        fmt.Printf("Error: %s\n", err) // 输出错误信息
    } else {
        fmt.Printf("100 / 0 = %.2f\n", divide) // 代码执行不到这里
    }
}

// $ go run main.go
// 输出如下
/**
  Error: divide by zero
*/

联系我