golang 延時函數defer 作final處理檔案或異常
defer 函數先進後出原則,在函數內部使用時,會在return之後執行,所以根據這一特性,可以處理因為異常導致檔案流未關閉問題(記憶體流失)和異常捕捉使用,具體如下:
- 檔案流關閉
func CopyFile(dstName, srcName string) (written int64, err error) {src, err := os.Open(srcName)if err != nil {return}defer src.Close() // 此處就算後續因為異常return,也會執行此處的close()dst, err := os.Create(dstName)if err != nil {return}defer dst.Close()return io.Copy(dst, src)}
- 資料庫查詢關閉
resultIterator, err := stub.GetStateByPartialCompositeKey(DT_AGENT_COMMISSION_DATA,[]string{params.Agent_code,params.Period}) if err != nil { return ErrorPbResponse(RESP_CODE_SYSTEM_ERROR,err.Error()) } defer resultIterator.Close() //此處就算return或異常都會執行
- 用作異常捕捉,結合recover()捕獲異常函數使用,類似java的final
func test(){ var exception interface{} temp("hello", func(e interface{}) { exception = e }) if exception != nil { fmt.Println(exception) } return}func temp(msg string, handler func(e interface{})){ defer func() { if err := recover(); err != nil { handler(err) } }() fmt.Println(msg[len(msg)+1])//異常 return}func main() { test()}