采用interface实现的限时调用方法-博文

在实时性要求高的工程项目中，对于方法执行的时间有较高的要求，本示例程序，实现了一种安全的传入任意参数的方法限时调用工具类，能执行方法的最大容忍运行时间，保证服务不超时。

//@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

有疑问加站长微信联系（非本文作者）