在实时性要求高的工程项目中，对于方法执行的时间有较高的要求，本示例程序，实现了一种安全的传入任意参数的方法限时调用工具类，能执行方法的最大容忍运行时间，保证服务不超时。
//@description 超时运行方法,限制方法的运行时间 // 需要实现抽象接口AppRuning的Run方法 // 调用时TimeOutRun(timeout, fun, args...) // timeout超时时间(ms),fun实现Run()的回调方法,args...为方法调用参数列表 //@author chenbintao //@data 2017-03-20 19:00 初稿 // 2017-03-20 19:30 调试通过 package main import ( "log" "runtime/debug" "time" ) const ( TIME_OUT_RUN_OK int = 1 //运行完成 TIME_OUT_RUN_OUT int = 0 //超时退出 ) var ( _TIME_OUT_RUN_DEBUG = false ) func SetDebug(debug_mode bool) { //设置调试模式:true调试模式 _TIME_OUT_RUN_DEBUG = debug_mode return } func TimeOutRunApp(timeout uint64, callback func(v ...interface{}), v ...interface{}) { var single chan int = make(chan int, 1) //开启线程运行(参数必须采用v...,这样才是全参数传递,否则会出现[]interface{}问题) go timeOutRunApp(single, safeFunction(callback), v...) //阻塞等待运行结果或超时 select { case <-single: //运行完成退出 if _TIME_OUT_RUN_DEBUG { log.Println("run ok!") } close(single) return case <-time.After(time.Millisecond /*ms*/ * time.Duration(timeout)): //超时退出 if _TIME_OUT_RUN_DEBUG { log.Println("run timeout!") } close(single) return } } func timeOutRunApp(single chan int, callback func(v ...interface{}), v ...interface{}) { //调用执行回调方法,执行完成后,写入chan callback(v...) single <- TIME_OUT_RUN_OK return } func safeFunction(fn func(v ...interface{})) func(v ...interface{}) { //闭包方法安全运行 return func(v ...interface{}) { defer func() { if e, ok := recover().(error); ok { log.Println("WARN: panic in %v - %v", fn, e) log.Println(string(debug.Stack())) } }() fn(v...) } } //=================测试样例 func Run(v ...interface{}) { //用于变量类型转换 var convert interface{} //第一个参数 convert = v[0] var args0 string = convert.(string) //第二个参数 convert = v[1] var args1 string = convert.(string) //循环执行,flag=false:演示超时的情况,flag=true:演示非超时的情况 var count uint32 var flag bool = false for { time.Sleep(time.Millisecond * 1000) log.Println(args0, args1, count) count++ if flag { return } } return } func main() { //设置调用超时5s,传入回调方法类App,及相关参数 TimeOutRunApp(5000, Run, "利用interface实现超时执行某方法:", "count=") }
运行可参见http://studygolang.com/topics/2569
有疑问加站长微信联系（非本文作者）
