Go语言编程思想3——错误处理和资源管理

Go语言编程思想3——错误处理和资源管理

资源管理：及时关闭文件，及时释放资源，如果打开的文件还未关闭就因为出错而在中间跳出，就无法保证有效的资源管理，因此在这里两者一起进行考虑

一、defer调用

• 调用在函数结束时发生（在return/panic之前执行）
• 参数在defer语句时计算
• defer列表为先进后出（先defer的后执行）

func tryDefer() {
	for i := 0; i < 100; i++ {
		defer fmt.Print(i," ")
		if i == 30 {
			panic("printed too many")
		}
	}
}
//执行结果:30 29 28 27 26 25 ... 3 2 1 0
//退出的时候i是30，但不会全部输出30， i是在执行defer语句时的值

Ex. 将前20个斐波那契数列输出到文件

func writeFile(filename string) {

	file, err := os.Create(filename) //新建文件
	if err != nil {
		panic(err)
	}
	defer file.Close()              //关闭文件
	writer := bufio.NewWriter(file) //新建bufio.Newwriter
	defer writer.Flush()            //writer需要Flush
	//先运行writer.Flush(),再运行file.close()
	f := fib.Fibonacci()
	for i := 0; i < 20; i++ {
		fmt.Fprintln(writer, f())
	}
}

何时使用defer调用

• Open/Close
• Lock/Unlock
• PrintHeader/PrintFooter

二、错误处理

• 尽量用error不用panic
• 意料之中的：使用error。如：文件打不开
• 意料之外的：使用panic。如：数组越界，如开了大小为n的数组，明明循环最大到n，但是结果越界，出现了意料之外的错误，这时用panic

error的定义

type error interface {
	Error() string
}

将error当做普通的值类型来处理即可

panic会把程序挂掉，尽量少用panic，遇到错误时可以输出提示语句后return

func writeFile(filename string) {
	// 注释表明，OpenFile如果出错那么一定是*PathError
	// 所以对Error进行判断，如果不是*PathError那么就报panic
	file, err := os.OpenFile(filename, os.O_EXCL|os.O_CREATE, 0666)
	if err != nil {
		if pathError, ok := err.(*os.PathError); !ok {
			panic(err)
		} else {
			fmt.Printf("%s,%s,%s\n",
				pathError.Op,
				pathError.Path,
				pathError.Err)
		}
		return
	}
	defer file.Close()              //关闭文件
	writer := bufio.NewWriter(file) //新建bufio.Newwriter
	defer writer.Flush()            //writer需要Flush
	//先运行writer.Flush(),再运行file.close()
	f := fib.Fibonacci()
	for i := 0; i < 20; i++ {
		fmt.Fprintln(writer, f())
	}
}

自建error

	err=errors.New("this is a custom error")

服务器统一错误处理

package filelisting

import (
	"io/ioutil"
	"net/http"
	"os"
	"strings"
)

const prefix = "/list/"

//字符串实现接口
type userError string

func (e userError) Error() string {
	return e.Message()
}
func (e userError) Message() string {
	return string(e)
}

func HandleFileList(writer http.ResponseWriter,
	request *http.Request) error {
	if strings.Index(request.URL.Path, prefix) != 0 {

		return userError("path must start " + "with " + prefix)
	}
	path := request.URL.Path[len(prefix):] // /list/fib.txt
	file, err := os.Open(path)
	if err != nil {
		return err
	}
	defer file.Close()

	all, err := ioutil.ReadAll(file)
	if err != nil {
		return err
	}
	writer.Write(all)
	return nil
}

package main

import (
	"learngo/errhandling/filelistingserver/filelisting"
	"log"
	"net/http"
	"os"
)

type appHandler func(writer http.ResponseWriter,
	request *http.Request) error

//函数式编程，将输入的函数包装成输出函数来输出
func errWrapper(
	handler appHandler) func(
	http.ResponseWriter, *http.Request) {
	return func(writer http.ResponseWriter,
		request *http.Request) {

		defer func() {
			if r := recover(); r != nil {
				log.Printf("Panic:%v", r)
				http.Error(writer,
					http.StatusText(http.StatusInternalServerError),
					http.StatusInternalServerError)
			}
		}()

		err := handler(writer, request)
		//错误处理
		if err != nil {
			log.Printf("Error occurred"+
				"handling request:%s",
				err.Error())
			if userErr,ok:=err.(userError);ok{
				http.Error(writer,
					userErr.Message(),
					http.StatusBadRequest)
				return
			}


			code := http.StatusOK
			switch {
			case os.IsNotExist(err):
				code = http.StatusNotFound
			case os.IsPermission(err):
				code = http.StatusForbidden
			default:
				code = http.StatusInternalServerError
			}
			http.Error(writer,
				http.StatusText(code), code)
		}
	}
}

type userError interface {
	error //给系统看的
	Message() string  //给用户看的
}

func main() {
	http.HandleFunc("/",
		errWrapper(filelisting.HandleFileList),
	)
	err := http.ListenAndServe(":8888", nil)
	if err != nil {
		panic(err)
	}
}