[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一下～