1. 程式人生 > >[Golang] 從零開始寫Socket Server(6)【完結】:日誌模組的設計與定時任務模組模組

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