如何在Go Rpc中使用自定义错误

type CustomError struct {
    Message string
    Code    int
}

func (e CustomError) Error() string {
    return fmt.Sprintf("Error: %s (Code: %d)", e.Message, e.Code)
}

// 定义自定义错误类型
type CustomError struct {
    Message string
    Code    int
}

// 实现自定义错误类型的Error()方法
func (e CustomError) Error() string {
    return fmt.Sprintf("Error: %s (Code: %d)", e.Message, e.Code)
}

// RPC服务端方法
func (s *MyService) Divide(args *Args, reply *float64) error {
    if args.B == 0 {
        // 返回自定义错误类型的实例
        return CustomError{Message: "Division by zero", Code: 100}
    }
    *reply = args.A / args.B
    return nil
}

// RPC客户端调用
func main() {
    client, err := rpc.Dial("tcp", "localhost:1234")
    if err != nil {
        log.Fatal("dialing:", err)
    }

    args := &Args{A: 10, B: 0}
    var result float64

    // 调用RPC服务端方法
    err = client.Call("MyService.Divide", args, &result)
    if err != nil {
        // 判断返回的错误是否是自定义错误类型
        if customErr, ok := err.(CustomError); ok {
            fmt.Println("Custom Error:", customErr.Message)
            fmt.Println("Error Code:", customErr.Code)
        } else {
            fmt.Println("Error:", err)
        }
    } else {
        fmt.Println("Result:", result)
    }
}