Golang函数执行模板(log,耗时,panic)

前端之家收集整理的这篇文章主要介绍了Golang函数执行模板(log,耗时,panic)前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
有些函数执行前后,需要加日志、记录耗时、还要处理panic,都是些重复性的东西。
所以弄了个模板式的东西,碰到这类需求,直接把相关函数和参数,丢到run函数去就不用管了,省得浪费时间。

先上个调用例子:

//函数执行模板
//author: Xiong Chuan Liang
//date: 2015-3-19

package main

import (
	"fmt"
	"log"
	"os"

	"utils"
)

func main() {
	log.SetOutput(os.Stdout)

	w := utils.NewWorker()
	w.Run("myPrintf()",myPrintf,"aa","bb")

}

func myPrintf(args ...interface{}) (int,error) {
	fmt.Println("myPrintf() begin.")
	for _,arg := range args {
		fmt.Println("args:",arg)
	}
	fmt.Println("myPrintf() end.")
	panic("尝试抛出panic错误")
	return 0,nil
}

/*
运行结果:

2015/03/19 13:49:30 [worker.Run()] 函数: myPrintf()
myPrintf() begin.
args: aa
args: bb
myPrintf() end.
2015/03/19 13:49:30 [worker.Run()] panic:
 myPrintf()
 尝试抛出panic错误
2015/03/19 13:49:30 [worker.Run()] 函数:myPrintf() 耗时:0.004000 秒

*/
调用还是很方便的,传入函数和附带的参数即可.

实现也很简单:

//函数执行模板
//author: Xiong Chuan Liang
//date: 2015-3-19

package utils

import (
	"fmt"
	"log"
	"time"
)

type workerFunc func(...interface{}) (int,error)

type worker struct {
}

func NewWorker() *worker {
	return &worker{}
}

func (w *worker) Run(name string,workerFunc workerFunc,args ...interface{}) (int,error) {

	now := time.Now()
	log.Println("[worker.Run()] 函数:",name)

	defer func() {
		if r := recover(); r != nil {
			log.Println("[worker.Run()] panic:\n",name,"\n",fmt.Sprint(r))
			w.elasped(name,now,time.Now())
		}
	}()

	ret,err := workerFunc(args...)
	w.elasped(name,time.Now())
	return ret,err
}

func (w *worker) elasped(name string,beginTime,endTime time.Time) {
	log.Printf("[worker.Run()] 函数:%s 耗时:%f 秒 \n",endTime.Sub(beginTime).Seconds())
}

没多少东西,我也就处理func(...interface{}) (int,error)这类函数,其它没管了。


BLOG: http://blog.csdn.net/xcl168

MAIL: xcl_168@aliyun.com

猜你在找的Go相关文章