go基礎筆記----錯誤處理與資源管理

文件目錄結構

test.go

package main

import (
	"bufio"
	"fmt"
	"os"

	"./fib"
)

/*
	go語言資源管理

	1.go使用defer調用來實現資源管理{
		確保調用在函數結束髮生的地方
		參數在defer語句時計算
		defer列表爲先進後出
	}
	2.建了什麼東西 都要想到defer掉{
		open/close
		lock/Unlock
		PrintHeader/PrintFooter
	}

	錯誤處理



*/

func tryDefer() {
	// defer裏面有一個棧 多個語句先進後出 不怕中間出現return || panic
	defer fmt.Println(1)
	defer fmt.Println(2)
	fmt.Println(3)

	for i := 0; i < 100; i++ {
		defer fmt.Println(i)
		if i == 30 {
			panic("printed too many")
		}
	}
}

func writeFile(filename string) {
	file, err := os.Open(filename) // 新建文件要記住關掉 close()
	if err != nil {
		panic(err)
	}
	// err = errors.New("this is a custom error") // 自己新建error
	// if err != nil {
	// 	fmt.Println("Error: ", err)
	// 	if pathError, ok := err.(*os.PathError); !ok {
	// 		panic(err)
	// 	} else {
	// 		fmt.Println(pathError.Op, pathError.Path, pathError.Err)
	// 	}
	// 	return
	// }
	defer file.Close()

	writer := bufio.NewWriter(file) // 要flush
	defer writer.Flush()
	f := fib.Fib()
	for i := 0; i < 20; i++ {
		fmt.Fprintln(writer, f())
	}
}

func main() {
	writeFile("fib.txt")
	// tryDefer()
}

web.go

package main

import (
	"fmt"
	"log"
	"net/http"
	"os"

	"./filelisting"
)

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 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 // 404
			case os.IsPermission(err):
				code = http.StatusForbidden // 403 沒有權限
			default:
				code = http.StatusInternalServerError
			}
			http.Error(writer, http.StatusText(code), code)
		}
	}
}

func tryRecover() {
	defer func() {
		r := recover()
		if err, ok := r.(error); ok {
			fmt.Println("Error occurred:", err)
		} else {
			panic(err)
		}
	}()

	// panic(errors.New("this is an error"))
	// panic(123)
}

type userError interface {
	error
	Message() string
}

// 在url裏面打一個文件地址 把文件內容顯示出來
func main() {
	// tryRecover()
	http.HandleFunc("/",
		errWrapper(filelisting.HandleFileList))

	err := http.ListenAndServe(":8888", nil)
	if err != nil {
		panic(err)
	}
}

/*
	panic 遇到錯誤未知 不建議經常用{
		1.停止當前函數執行
		2.一直向上返回，執行每一層的defer
		3.如果沒有遇見recover，程序退出
	}

	recover{
		僅在defer函數中調用
		獲取panic的值
		如果無法處理，可以重新panic
	}
*/

/*
	error vs panic{
		1.意料之中的用error。如文件打不開
		2.意料之外的使用panic。如數組越界
	}

	案例總合使用{
		defer + panic + recover
		type Assertion
		函數式編程
	}

*/

handle.go

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)
}

// HandleFileList http業務
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
		// panic(err) // 如果出錯 程序並不會掛掉 會被保護起來 recover會保護服務
	}
	defer file.Close()
	all, err := ioutil.ReadAll(file)
	if err != nil {
		return err
	}
	writer.Write(all)

	return nil
}

fib.go

package fib

// Fib 斐波那契數列
func Fib() func() int {
	a, b := 0, 1
	return func() int {
		a, b = b, a+b
		return a
	}
}