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