练习:错误
从先前的练习中复制 Sqrt
函数,并修改使其返回 error
值。
由于不支持复数,当 Sqrt
接收到一个负数时,应当返回一个非 nil 的错误值。
创建一个新类型
type ErrNegativeSqrt float64
为其实现
func (e ErrNegativeSqrt) Error() string
使其成为一个 error
, 该方法就可以让 ErrNegativeSqrt(-2).Error()
返回 `"cannot Sqrt negative number: -2"`。
*注意:* 在 Error
方法内调用 fmt.Sprint(e)
将会让程序陷入死循环。可以通过先转换 e
来避免这个问题:fmt.Sprint(float64(e))
。请思考这是为什么呢?
修改 Sqrt
函数,使其接受一个负数时,返回 ErrNegativeSqrt
值。
参考解法:
package main
import (
"fmt"
)
type ErrNegativeSqrt float64
func (e ErrNegativeSqrt) Error() string {
return fmt.Sprintf("cannot Sqrt negative number: %f", e)
}
func Sqrt(x float64) (float64, error) {
if x < 0 {
return 0, ErrNegativeSqrt(x)
}
z := 1.0
for i := 0; i < 20; i++ {
z = z - (z*z-x)/(2*x)
}
return z, nil
}
func main() {
nums := []float64{4, -6, 8, 10}
for _, v := range nums {
sqrv, e := Sqrt(v)
if e == nil {
fmt.Printf("sqrt(%f)=%f\n", v, sqrv)
} else {
fmt.Println(e)
}
}
}