Go?Web編程添加服務(wù)器錯誤和訪問日志
前言
錯誤日志和訪問日志是一個服務(wù)器必須支持的功能，我們教程里使用的服務(wù)器到目前為止還沒有這兩個功能。正好前兩天也寫了篇介紹logrus日志庫的文章，那么今天的文章里就給我們自己寫的服務(wù)器加上錯誤日志和訪問日志的功能。在介紹添加訪問日志的時候會介紹一種通過編寫中間件獲取HTTP響應(yīng)的StausCode和Body的方法。
Go Web 編程系列的每篇文章的源代碼都打了對應(yīng)版本的軟件包，供大家參考。公眾號中回復(fù)gohttp11獲取本文源代碼
初始化日志記錄器
我們先來做一下初始化工作，在項目里初始化記錄錯誤日志和訪問日志的記錄器Logger。
//?./utils/vlog
package?vlog
import?(
????"github.com/sirupsen/logrus"
????"os"
)
var?ErrorLog?*logrus.Logger
var?AccessLog?*logrus.Logger
var?errorLogFile?=?"./tmp/log/error.log"
var?accessLogFile?=?"./tmp/log/access.log"
func?init?()?{
????initErrorLog()
????initAccessLog()
}
func?initErrorLog()?{
????ErrorLog?=?logrus.New()
????ErrorLog.SetFormatter(&logrus.JSONFormatter{})
????file?,?err?:=?os.OpenFile(errorLogFile,?os.O_RDWR?|?os.O_CREATE?|?os.O_APPEND,?0755)
????if?err?!=?nil?{
????????panic(err)
????}
????ErrorLog.SetOutput(file)
}
func?initAccessLog()?{
????AccessLog?=?logrus.New()
????AccessLog.SetFormatter(&logrus.JSONFormatter{})
????file?,?err?:=?os.OpenFile(accessLogFile,?os.O_RDWR?|?os.O_CREATE?|?os.O_APPEND,?0755)
????if?err?!=?nil?{
????????panic(err)
????}
????AccessLog.SetOutput(file)
}
- 我們新定義一個
package在init函數(shù)中來初始化記錄器，這樣服務(wù)器成功啟動前就會初始化好記錄器。 /tmp/log這個目錄要提前創(chuàng)建好，執(zhí)行init函數(shù)時會自動創(chuàng)建好access.log和error.log。
添加錯誤日志
我們創(chuàng)建服務(wù)器使用的net/http包的Server類型中，有一個ErrorLog字段供開發(fā)者設(shè)置記錄錯誤日志用的記錄器Logger，默認(rèn)使用的是log包默認(rèn)的記錄器（應(yīng)該是系統(tǒng)的標(biāo)準(zhǔn)錯誤）：
type?Server?struct?{
???Addr????string??//?TCP?address?to?listen?on,?":http"?if?empty
???Handler?Handler?//?handler?to?invoke,?http.DefaultServeMux?if?nil
???...
???//?ErrorLog?specifies?an?optional?logger?for?errors?accepting
???//?connections,?unexpected?behavior?from?handlers,?and
???//?underlying?FileSystem?errors.
???//?If?nil,?logging?is?done?via?the?log?package's?standard?logger.
???ErrorLog?*log.Logger
?????...
}
我們之前在創(chuàng)建服務(wù)器的時候自己實現(xiàn)了Server類型的對象，那么現(xiàn)在要做的就是將上面初始化好的錯誤日志的記錄器指定給Server的ErrorLog字段。
func?main()?{
??...
? ??//?將logrus的Logger轉(zhuǎn)換為io.Writer
????errorWriter?:=?vlog.ErrorLog.Writer()
? ??//?記得關(guān)閉io.Writer
????defer?errorWriter.Close()
????server?:=?&http.Server{
????????Addr:????":8080",
????????Handler:?muxRouter,
? ? ? ??//?用記錄器轉(zhuǎn)換成的io.Writer創(chuàng)建log.Logger
????????ErrorLog:?log.New(vlog.ErrorLog.Writer(),?"",?0),
????}
????...
}
添加好錯誤日志的記錄器后，我們找個路由處理函數(shù)，在里面故意制造運(yùn)行時錯誤驗證一下是否能記錄到錯誤。
func?(*HelloHandler)?ServeHTTP(w?http.ResponseWriter,?r?*http.Request)?{
???ints?:=?[]int{0,?1,?2}
???fmt.Fprintf(w,?"%v",?ints[0:5])
}
在上面處理函數(shù)中，通過切片表達(dá)式越界故意制造了一個運(yùn)行時錯誤，打開error.log后能看到文件里已經(jīng)記錄到這個運(yùn)行時錯誤及其Stack trace
添加訪問日志
和Server對象可以設(shè)置錯誤日志的記錄器不一樣，訪問日志只能是我們通過自己編寫中間件的方式來實現(xiàn)了。在記錄訪問日志的中間件里我們會記錄ip，method，path，query，request_body，status和response_body這些個字段的內(nèi)容。
status和response_body兩個字段來自請求對應(yīng)的響應(yīng)。響應(yīng)在net/http包里是用http.ResponseWriter接口表示的
type?ResponseWriter?interface?{
????Header()?Header
????Write([]byte)?(int,?error)
????WriteHeader(statusCode?int)
}
接口本身以及net/http提供的實現(xiàn)都沒有讓我們進(jìn)行讀取的方法，所以在編寫的用于記錄訪問日志的中間件里需要對net/http庫本身實現(xiàn)的ResponseWriter做一層包裝。
利用Go語言結(jié)構(gòu)體類型嵌套匿名類型后，結(jié)構(gòu)體擁有了被嵌套類型的所有導(dǎo)出字段和方法的特性，我們可以很方便地對原來的ResponseWriter做一層包裝，然后只重新實現(xiàn)需要更改的方法即可：
type?ResponseWithRecorder?struct?{
???http.ResponseWriter
???statusCode?int
???body?bytes.Buffer
}
func?(rec?*ResponseWithRecorder)?WriteHeader(statusCode?int)?{
???rec.ResponseWriter.WriteHeader(statusCode)
???rec.statusCode?=?statusCode
}
func?(rec?*ResponseWithRecorder)?Write(d?[]byte)?(n?int,?err?error)?{
???n,?err?=?rec.ResponseWriter.Write(d)
???if?err?!=?nil?{
??????return
???}
???rec.body.Write(d)
???return
}
定義好新的類型后我們重新實現(xiàn)了WriteHeader和Write方法，在向原來的ReponseWriter中寫入后也會向ResponseWriteRecoder.statusCode和ResponseWriteRecoder.body寫入對應(yīng)的數(shù)據(jù)。這樣我們就可以在中間件里通過這兩個字段訪問響應(yīng)碼和響應(yīng)數(shù)據(jù)了。
記錄訪問日志的中間件定義如下：
func?AccessLogging?(f?http.Handler)?http.Handler?{
????//?創(chuàng)建一個新的handler包裝http.HandlerFunc
????return?http.HandlerFunc(func(w?http.ResponseWriter,?r?*http.Request)?{
????????buf?:=?new(bytes.Buffer)
????????buf.ReadFrom(r.Body)
????????logEntry?:=?vlog.AccessLog.WithFields(logrus.Fields{
????????????"ip":?r.RemoteAddr,
????????????"method":?r.Method,
????????????"path":?r.RequestURI,
????????????"query":?r.URL.RawQuery,
????????????"request_body":?buf.String(),
????????})
????????wc?:=?&ResponseWithRecorder{
????????????ResponseWriter:?w,
????????????statusCode:?http.StatusOK,
????????????body:?bytes.Buffer{},
????????}
????????//?調(diào)用下一個中間件或者最終的handler處理程序
????????f.ServeHTTP(wc,?r)
????????defer?logEntry.WithFields(logrus.Fields{
????????????"status":?wc.statusCode,
????????????"response_body":?wc.body.String(),
????????}).Info()
????})
}
在Router上應(yīng)用創(chuàng)建好的AccessLogging中間件后，就可以正常的記錄服務(wù)器的訪問日志了。
//?router/router.go
func?RegisterRoutes(r?*mux.Router)?{
????...
????//?apply?Logging?middleware
????r.Use(middleware.Logging(),?middleware.AccessLogging)
????...
}
不過有兩點需要注意一下
- 這里為了演示獲取響應(yīng)數(shù)據(jù)記錄了
response_body字段，如果是接口響應(yīng)內(nèi)容記錄下還可以，但是如果是HTML還是不記錄的為好。 - 初始化
ResponseWithRecorder時默認(rèn)設(shè)置了statusCode是因為，服務(wù)器正確返回響應(yīng)時不會顯式調(diào)用WriteHeader方法，只有在返回NOT_FOUND之類的錯誤的時候才會調(diào)用WriteHeader方法，針對這種情況需要在初始化的時候把statusCode的默認(rèn)值設(shè)置為200。
現(xiàn)在再訪問服務(wù)器后打開access.log會看到剛剛的訪問日志，就能看到剛剛請求的url，method，客戶端IP等信息了。
{"ip":"......","level":"info","method":"GET","msg":"","path":"/index/","query":"","request_body":"","response_body":"Hello World1","status":200,"time":"2020-03-26T04:21:46Z"}
注意：文章只為說明演示方便，獲取IP的方法無法獲取代理后的真實IP，請悉知。
以上就是Go Web編程添加服務(wù)器錯誤和訪問日志的詳細(xì)內(nèi)容，更多關(guān)于Go Web服務(wù)器錯誤訪問日志的資料請關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
slice（切片）是go中常見和強(qiáng)大的類型，這篇文章不是slice使用簡介，從源碼角度來分析slice的實現(xiàn)，slice的一些迷惑的使用方式，感興趣的朋友跟隨小編一起看看吧2023-06-06
Go指針和C指針在許多方面非常相似，但其中也有一些不同。本文詳細(xì)講解了Go語言指針的用法，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-07-07
go:embed 是 Go 1.16 引入的一個強(qiáng)大功能,它允許在編譯時將外部文件或目錄嵌入到 Go 程序中,下面介紹如何使用 go:embed 來嵌入JAR 文件,感興趣的朋友一起看看吧2025-04-04
Go語言提供了多種同步和通信機(jī)制,可以用于實現(xiàn)阻塞的效果,本文主要介紹了Go 阻塞的實現(xiàn)示例,具有一定的參考價值,感興趣的可以了解一下2024-05-05
淺析Go項目中的依賴包管理與Go?Module常規(guī)操作
這篇文章主要為大家詳細(xì)介紹了Go項目中的依賴包管理與Go?Module常規(guī)操作,文中的示例代碼講解詳細(xì),對我們深入了解Go語言有一定的幫助,需要的可以跟隨小編一起學(xué)習(xí)一下2023-10-10
這篇文章主要介紹了淺談golang slice 切片原理，詳細(xì)的介紹了golang slice 切片的概念和原理，具有一定的參考價值，有興趣的可以了解一下2017-11-11
Go語言并發(fā)編程之控制并發(fā)數(shù)量實現(xiàn)實例
這篇文章主要為大家介紹了Go語言并發(fā)編程之控制并發(fā)數(shù)量實例探究,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2024-01-01
這篇文章主要為大家詳細(xì)介紹了如何利用Go語言實現(xiàn)聊天小工具，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-03-03
本篇給出了4個在運(yùn)維開發(fā)工作中較為常見的且也是比較典型的場景，通過這些場景來帶大家掌握Go中Goroutine和通道是怎么使用的，需要的可以學(xué)習(xí)一下2023-05-05
go語言channel實現(xiàn)多核并行化運(yùn)行的方法
這篇文章主要介紹了go語言channel實現(xiàn)多核并行化運(yùn)行的方法,實例分析了channel實現(xiàn)多核并行化運(yùn)行的技巧,具有一定參考借鑒價值,需要的朋友可以參考下2015-03-03