这个是A Tour of Go这个教程里面的第六个练习,原题如下:
从之前的练习中复制
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
值。
我们只需要修改一下之前的实现方法,加一个条件判断,当小于0时调用一下错误函数,返回错误信息即可,实现如下:
package main
import (
"fmt"
)
type ErrNegativeSqrt float64
func (e ErrNegativeSqrt) Error() string {
return fmt.Sprintf("cannot Sqrt negative number: %g", e)
}
func Sqrt(x float64) (float64, error) {
if x < 0 {
return x, ErrNegativeSqrt(x)
}
z := 1.0
for i := 0; i < 10; i++ {
z -= (z*z -x) / (2*z)
}
return z, nil
}
func main() {
fmt.Println(Sqrt(0))
fmt.Println(Sqrt(-2.2))
}