[Golang] 从零开始写Socket Server（6）【完结】：日志模块的设计与定时任务模块模块

好久没写文章啦。。。今天把golang挖的这个坑给补完吧～

作为一个Server，日志（Log）功能是必不可少的，一个设计良好的日志模块，不论是开发Server时的调试，还是运行时候的维护，都是非常有帮助的。

因为这里写的是一个比较简化的Server框架，因此我选择对Golang本身的log库进行扩充，从而实现一个简单的Log模块。

在这里，我将日志的等级大致分为Debug，Operating，Error 3个等级，Debug主要用于存放调试阶段的日志信息，Operateing用于保存Server日常运行时产生的信息，Error则是保存报错信息。

模块代码如下：

func LogErr(v ...interface{}) {

	logfile := os.Stdout
	log.Println(v...)
	logger := log.New(logfile,"\r\n",log.Llongfile|log.Ldate|log.Ltime);
	logger.SetPrefix("[Error]")
	logger.Println(v...)
	defer logfile.Close();
}

func Log(v ...interface{}) {

	logfile := os.Stdout
	log.Println(v...)
	logger := log.New(logfile,"\r\n",log.Ldate|log.Ltime);
	logger.SetPrefix("[Info]")
	logger.Println(v...)
	defer logfile.Close();
}

func LogDebug(v ...interface{}) {
	logfile := os.Stdout
	log.Println(v...)
	logger := log.New(logfile,"\r\n",log.Ldate|log.Ltime);
	logger.SetPrefix("[Debug]")
	logger.Println(v...)
	defer logfile.Close();
}

func CheckError(err error) {
	if err != nil {
		LogErr(os.Stderr, "Fatal error: %s", err.Error())
	}
}

注意这里log的输出我使用的是stdout，因为这样在Server运行的时候可以直接将log重定向到指定的位置，方便整个Server的部署。不过在日常开发的时候，为了方便调试代码，我推荐将log输出到指定文件位置下，这样在调试的时候会方便很多（主要是因为golang的调试实在太麻烦，很多时候都要依靠打log的时候进行步进。便于调试的Log模块代码示意：

func Log(v ...interface{}) {

	logfile := os.OpenFile("server.log",os.O_RDWR|os.O_APPEND|os.O_CREATE,0);
	if err != nil {
		fmt.Fprintf(os.Stderr, "Fatal error: %s", err.Error())
		return    }
	log.Println(v...)
	logger := log.New(logfile,"\r\n",log.Ldate|log.Ltime);
	logger.SetPrefix("[Info]")
	logger.Println(v...)
	defer logfile.Close();
}

然后就是计时循环模块啦，日常运行中，Server经常要执行一些定时任务，比如隔一定时间刷新后台，隔一段时间自动刷新爬虫等等，在这里我设计了一个Task接口，通过类似于TaskList的的方式将所有定时任务注册后统一执行，代码如下：

type DoTask interface {
	Excute()
}

var tasklist []interface{}

func AddTask(controller DoTask) {
	var arr interface{}
	arr = controller
	tasklist = append(tasklist,arr)
	fmt.Println(tasklist)
	}

在这里以一个定时报时任务作为例子，注意，在这里我定义了一个channel用于阻塞main函数，否则main函数这里直接就结束了没法看到结果，实际使用中，通过类似的方法保持server一直运行就可以啦：

var channel = make(chan int, 10)
type Task1 struct {}
func (this * Task1)Excute() {
	for {
		time.Sleep(2 * time.Second)
		fmt.Println("this is task1") 
	}
}

type Task2 struct {}

func (this * Task2)Excute() {
	 for {
		time.Sleep(4 * time.Second)
		fmt.Println("this is task2")
	 }
}

func main(){
	var task1 Task1
	var task2 Task2
	tasklist = make([]interface{} ,0 , 20)
	AddTask(&task1)
	AddTask(&task2)
	m :=0
	for _, _ = range tasklist {
		m = m+1
	}

	for _, v := range tasklist{
		go v.(DoTask).Excute()
	}
	<-channel
}

运行结果如下，2秒间隔的task1执行两次后，正好完成4秒间隔的task2，证明定时接口成功被循环执行：

我已经把SocketServer系列的代码整合到了一起，发布到了我个人的github上：点击链接，希望大家有兴趣的可以学习star一下～

[Golang] 从零开始写Socket Server（6）【完结】：日志模块的设计与定时任务模块模块

猜你喜欢