[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经常要执行一些定时任务，比如隔一定时间刷新后台，隔一段时间自动刷新爬虫等等，也懒得单门设计一个接口了，这里我比较粗暴的直接使用了time.sleep来把时间睡过去，其实更好的办法是使用一个channel进行时间的控制和防范一直阻塞，不过懒得改了 (O w O)

在这里以一个定时爬虫作为例子：

func GetSomething()error{

		resp, err := http.Get("http://www.myurl.com")
		if err != nil {
			Log(err)
		}
		fetch, err := ioutil.ReadAll(resp.Body)
		defer resp.Body.Close()

		if err != nil {
			Log(err)
		}
		
		if string(fetch) == ""{
			return errors.New("Get a blank url")
		} 		 																	return nil
	}

	func Mytask(gap) {
		
		if  GetSomething()!=nil { 
		 GetSomething()
		}
		time.Sleep(time.Duration(gap) * time.Second)
		Log("Awake! Here we go the next place")
		Mytask(gap)
	}

需要注意的是，因为这里使用的Time.sleep来阻塞线程，所以在调用MyTask的时候需要使用go进行并发操作，否则整个服务器在阻塞期间啥都不能干啦。。

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