Golang异常处理

Golang中对于异常的处理通常是返回Error的形式。 假设我需要封装http Response返回，代码可能会是如下情况

1type Header struct{
2	Key,Value string
3}
4type Status struct{
5	Code int
6	Message string
7}
8func Response(w io.Writer,status Status, headers []Header, body io.Reader) error {
9	_,err:=fmt.Fprintf(w, "HTTP/1.1 %d %s\r\n", status.Code,status.Message)
10	if err!=nil{
11		return err
12	}
13	for _,header:=range headers{
14		_,err=fmt.Fprintf(w,"%s: %s\r\n",header.Key,header.Value)
15		if err!=nil{
16			return err
17		}
18	}
19	_,err=fmt.Fprint(w,"\r\n")
20	if err!=nil{
21		return err
22	}
23	_,err=io.Copy(w,body)
24	if err!=nil{
25		return err
26	}
27	return nil
28}

可以看到通篇都充斥着if err!=null的判断，看着很不优雅。 对于这种情况可以考虑用以下方式改变代码使其变得优雅。

1type Header struct {
2	Key, Value string
3}
4type Status struct {
5	Code    int
6	Message string
7}
8type errWriter struct {
9	w   io.Writer
10	err error
11}
12
13func (ew *errWriter) Write(bytes []byte) (int, error) {
14	if ew.err != nil {
15		return -1, ew.err
16	}
17	n,err:=ew.w.Write(bytes)
18	ew.err=err
19	return n,err
20}
21
22func Response(w io.Writer, status Status, headers []Header, body io.Reader) error {
23	ew := &errWriter{w, nil}
24	fmt.Fprintf(ew, "HTTP/1.1 %d %s\r\n", status.Code, status.Message)
25	for _, header := range headers {
26		fmt.Fprintf(ew, "%s: %s\r\n", header.Key, header.Value)
27	}
28	io.Copy(ew, body)
29	return ew.err
30}