有的同学看到Go和TryCatch一起出现,心里可能会说,难道Go语言升级了,加入了try...catch语句。哈哈,其实Go语言从创建之初就没打算加入try...catch语句,因为创建Go的那帮大爷认为try...catch挺烦人的,如果滥用,会造成程序混乱,所以就不打算加入try...catch(以后加不加入不好说)。

既然Go语言中并没有try...catch语句,那么为何文章标题说要使用TryCatch呢?其实Go语言中只是没有try...catch语句,并不是没有异常处理机制。Go语言中的异常处理机制就是著名的异常三剑客:panic、defer和recover。通过这3个家伙,是完全可以模拟出try...catch语句效果的,对了,后面还应该有个finally。在正式模拟try...catch语句之前,先来回顾下Go语言中的异常处理机制是如何玩的。

Go语言中的异常处理机制

在前面提到,Go语言通过panic、defer和recover来处理异常的,那么这3个东西是什么呢?

不管是什么异常处理机制,核心的原理都是一样的,通常来讲,一个完善的异常处理机制需要由下面3部分组成。

  • 抛出异常
  • 处理异常的代码段
  • 获取异常信息

下面先用Java的异常处理机制来说明这一点。

import java.io.IOException;

public class Main {

    public static void main(String[] args) {
        try
        {
            boolean ioException = false;
            if (ioException) {
                throw new IOException("ioexception");
            } else {
                throw new Exception("exception");
            }
        }
        catch (IOException e) {
            System.err.println(e);
        }
        catch (Exception e) {
            System.out.println(e);
        }
        finally
        {
            System.out.println("finally");
        }
    }
}

上面的代码是标准的Java异常处理机制,try部分的throw用于抛出异常,而catch部分的代码段用于处理特定的异常,通过catch子句的参数e可以获取异常信息。所以对于Java来说,上述的3个异常重要的组成部分都有。

对于Go语言来说,panic、defer和recover也分别对应了这3部分。其中panic是一个函数,用于抛出异常,相当于Java中的throw函数。defer是一个关键字,用于修饰函数,用defer修饰的函数,在抛出异常时会自动调用。recover是一个函数,用于获取异常信息,通常在用defer修饰的函数中使用。

下面是一段用Go语言处理异常的代码。

package main

import "fmt"

func main(){
	//  处理异常的函数
	defer func(){
		fmt.Println("开始处理异常")
		// 获取异常信息
		if err:=recover();err!=nil{
			//  输出异常信息
			fmt.Println("error:",err)
		}
		fmt.Println("结束异常处理")
	}()
	exceptionFun()
}

func exceptionFun(){
	fmt.Println("exceptionFun开始执行")
	panic("异常信息")
	fmt.Println("exceptionFun执行结束")
}

实现Go版的TryCatch

现在已经了解了Go语言的异常处理机制,那么接下来使用异常处理机制来模拟try...catch...finally语句。

现在来分析一下如果模拟。模拟的过程需要完成下面的工作。

  • try、catch和finally这3部分都有各自的代码段,所以为了模拟try...catch...finally,需要用3个Go函数来分别模拟try、catch和finally部分的代码段。这3个Go函数是Try、Catch和Finally。
  • 要确定这3个函数在什么地方调用。Try是正常执行的代码,所以在要首先调用Try函数。而Catch函数只有在抛出异常时调用,所以应该在用defer修饰的函数中调用,而且需要在Catch函数中获取异常信息,所以应该在使用cover函数获取异常信息后再调用Catch函数,通常会将异常信息直接作为参数传递给Catch函数。不管是否抛出异常,Finally函数都必须调用,所以应该用defer修饰Finally函数,而且是第1个用defer修饰的函数。这样,在当前函数结束之前一定刚回调用Finally函数。
  • 触发异常,这就非常简单了,直接用panic函数即可。

上面清楚地描述了用Go语言的异常处理机制模拟try...catch...finally语句的基本原理,下面给出完整的实现代码。

package main
import (
"fmt"
)
type ExceptionStruct struct {
	Try     func()
	Catch   func(Exception)
	Finally func()
}
type Exception interface{}
func Throw(up Exception) {
	panic(up)
}
func (this ExceptionStruct) Do() {
	if this.Finally != nil {

		defer this.Finally()
	}
	if this.Catch != nil {
		defer func() {
			if e := recover(); e != nil {
				this.Catch(e)
			}
		}()
	}
	this.Try()
}

func main() {
	fmt.Println("开始执行...")
	ExceptionStruct{
		Try: func() {
			fmt.Println("try...")
			Throw("发生了错误")
		},
		Catch: func(e Exception) {
			fmt.Printf("exception %v\n", e)
		},
		Finally: func() {
			fmt.Println("Finally...")
		},
	}.Do()
	fmt.Println("结束运行")
}

上面的代码将Try、Catch、Finally函数都封装在了ExceptionStruct结构体中。然后调用方式就与前面的描述的一致了。执行这段代码,会输出如下图的信息。

image.png

增强版的TryCatch

到现在为止,其实已经完整地实现了try...catch...finally语句,但细心的同学会发现,这个实现有一点小问题。通常的try...catch...finally语句,try部分有且只有1个,finally部分是可选的,但最多只能有1个,而catch部分也是可选的,可以有0到n个,也就是catch部分可以有任意多个。但前面的实现,Catch函数只能指定一个,如果要指定任意多个应该如何做呢?其实很简单,用一个Catch函数集合保存所有指定的Catch函数即可。不过需要快速定位某一个Catch函数。在Java中,是通过异常类型(如IOException、Exception等)定位特定的catch子句的,我们也可以模拟这一过程,通过特定的异常来定位与该异常对应的Catch函数,为了方便,可以用int类型的异常代码。那么在调用Catch函数之前,就需要通过异常代码先定位到某一个Catch函数,然后再调用。下面就是完整的实现代码。

package main

import (
	"log"
)

type Exception struct {
	Id int       // exception id
	Msg string   // exception msg
}

type TryStruct struct {
	catches map[int]ExceptionHandler
	try   func()
}

func Try(tryHandler func()) *TryStruct {
	tryStruct := TryStruct{
		catches: make(map[int]ExceptionHandler),
		try: tryHandler,
	}
	return &tryStruct
}


type ExceptionHandler func(Exception)

func (this *TryStruct) Catch(exceptionId int, catch func(Exception)) *TryStruct {
	this.catches[exceptionId] = catch
	return this
}

func (this *TryStruct) Finally(finally func()) {
	defer func() {
		if e := recover(); nil != e {

			exception := e.(Exception)

			if catch, ok := this.catches[exception.Id]; ok {			
				catch(exception)
			}
			
			finally()
		}
	}()
	
	this.try()
}

func Throw(id int, msg string) Exception {
	panic(Exception{id,msg})
}

func main() {

	exception.Try(func() {
		log.Println("try...")
               //  指定了异常代码为2,错误信息为error2
		exception.Throw(2,"error2")
	}).Catch(1, func(e exception.Exception) {
		log.Println(e.Id,e.Msg)
	}).Catch(2, func(e exception.Exception) {
		log.Println(e.Id,e.Msg)
	}).Finally(func() {
		log.Println("finally")
	})
}

执行结果如下图所示。

image.png

这个实现与Java中的try...catch...finally的唯一区别就是必须要调用Finally函数,因为处理异常的代码都在Finally函数中。不过这并不影响使用,如果finally部分没什么需要处理的,那么就设置一个空函数即可。

为了方便大家,我已经将该实现封装成了函数库,调用代码如下:

package main
import (
	"exception"
	"log"
)

func main() {

	exception.Try(func() {
		log.Println("try...")
		exception.Throw(2,"error2")
	}).Catch(1, func(e exception.Exception) {
		log.Println(e.Id,e.Msg)
	}).Catch(2, func(e exception.Exception) {
		log.Println(e.Id,e.Msg)
	}).Finally(func() {
		log.Println("finally")
	})
}

获得本文源代码,请关注”极客起源“或”欧瑞科技“公众号,并输入308178获得源代码。

image.png