go fiber: 把異常資訊寫到錯誤日誌中

刘宏缔的架构森林發表於2024-11-16
Go

一，程式碼:

1,userBusiness.go

package business

import ("fmt")

//得到多個使用者，按分頁返回
func GetUserList(page int ,pageSize int) (string,error) {

    b := 0
    a:=100/b
    fmt.Println(a)

    return "1,2,3",nil
}

程式碼中包含有除0錯，會引發panic

2,userController.go

它用呼叫包含除0錯的userBusiness

package controller

import (
    "github.com/gofiber/fiber/v2"
    //"time"
    "industry/business"

    "fmt"
)

type UserController struct{}

func NewUserController() *UserController {
	return &UserController{}
}


func (dc *UserController) ListUser(c *fiber.Ctx) error {
	// 處理列出使用者的邏輯
    list,_:=business.GetUserList(1,30)
    fmt.Println(list)


	return c.SendString("列出使用者")
}

3,recoverMiddleware.go

參考了fiber官方的recover類，新增了寫日誌功能

package middleware

import (
	"fmt"
	//"os"
	"runtime/debug"
    "time"
	"github.com/gofiber/fiber/v2"
    "industry/config"
)

//在中介軟體中得到panic,獲取堆疊後寫入到日誌中
func NewRecover(c *fiber.Ctx) (err error) { //nolint:nonamedreturns // Uses recover() to overwrite the error

		// Catch panics
		defer func() {

			if r := recover(); r != nil {
               // fmt.Printf("panic: %v\n%s\n", r.(error), debug.Stack())

                timeStr:=time.Now().Format("2006-01-02 15:04:05")
                //得到時間,ip地址,請求的地址，引數，user-agent,
                fullURL := c.OriginalURL()   //url
                clientIP := c.IP()    //ip地址
                method := c.Method() // 獲取請求方法
                userAgent := c.Get("User-Agent")   //user agent
                // 獲取所有查詢引數
                queryParams := fmt.Sprintf("%v",c.Queries())
                fmt.Println("Query Params:", queryParams)

    filename:="/data/logs/gologs/exception-" + time.Now().Format("2006-01-02") + ".log"
    content:="["+timeStr+"] "+r.(error).Error()+"\n"+clientIP+" "+method+" "+fullURL+" "+userAgent+"\n"+queryParams+"\n"+string(debug.Stack())+"\n"
    config.GlobalWriteFile(filename,content)

				var ok bool
				if err, ok = r.(error); !ok {
					// Set error that will call the global error handler
					err = fmt.Errorf("%v", r)
				}

			}
		}()

		// Return err if exist, else move to next handler
		return c.Next()
}

4,routes.go

package routes

import (
	"github.com/gofiber/fiber/v2"
	"industry/controller"
    "runtime/debug"
    "industry/middleware"
    "fmt"
)



func SetupRoutes() *fiber.App {

    app := fiber.New(fiber.Config{
		ErrorHandler: func(c *fiber.Ctx, err error) error {

			// 傳送自定義錯誤頁面
			//return c.Status(code).JSON(config.Error("內部錯誤:"+err.Error()))
            return c.SendString("內部錯誤:"+err.Error())
		},
	})



    app.Use(middleware.NewRecover)

	//使用者模組
	userController := controller.NewUserController()
	user := app.Group("/user")
	user.Get("/info", userController.GetUser)
	user.Post("/", userController.CreateUser)
    user.Get("/list", userController.ListUser)

    return app
}

5,寫日誌的類:

package config

import (
    "os"
    "fmt"
)

// filename:檔案路徑,content:要寫的內容
func GlobalWriteFile(filename,content string) error {
        // 輸出到檔案
		file, err := os.OpenFile(filename, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
		if err != nil {
			fmt.Println("日誌檔案的開啟錯誤 : " + err.Error())
			return err
		}
		defer file.Close()
		if _, err := file.WriteString(content); err != nil {
			fmt.Println("寫入日誌檔案錯誤 : " + err.Error())
            return err
		}

		return nil
}

二，測試效果:

[2024-11-14 17:56:51] runtime error: integer divide by zero
192.168.219.1 POST /user/list?token=111&a=22&c=33 PostmanRuntime/7.42.0
map[a:22 c:33 token:111]
goroutine 21 [running]:
runtime/debug.Stack()
        /usr/local/soft/go/src/runtime/debug/stack.go:26 +0x5e
industry/middleware.NewRecover.func1()
        /data/goapp/industry/middleware/recoverMiddleware.go:37 +0x29a
panic({0x6f9360?, 0xa08960?})
        /usr/local/soft/go/src/runtime/panic.go:785 +0x132
industry/business.GetUserList(0x7f008002000000?, 0xc00005ba80?)
        /data/goapp/industry/business/userBusiness.go:9 +0x9
industry/controller.(*UserController).ListUser(0x467edd?, 0xc00014e008)
        /data/goapp/industry/controller/userController.go:29 +0x26
github.com/gofiber/fiber/v2.(*App).next(0xc0000e4a08, 0xc00014e008)
        /data/gopath/pkg/mod/github.com/gofiber/fiber/v2@v2.52.5/router.go:145 +0x1be
github.com/gofiber/fiber/v2.(*Ctx).Next(0x10?)
        /data/gopath/pkg/mod/github.com/gofiber/fiber/v2@v2.52.5/ctx.go:1034 +0x4d
industry/middleware.NewRecover(0x6f65e0?)
        /data/goapp/industry/middleware/recoverMiddleware.go:50 +0x53
github.com/gofiber/fiber/v2.(*App).next(0xc0000e4a08, 0xc00014e008)
        /data/gopath/pkg/mod/github.com/gofiber/fiber/v2@v2.52.5/router.go:145 +0x1be
github.com/gofiber/fiber/v2.(*App).handler(0xc0000e4a08, 0x4adc0f?)
        /data/gopath/pkg/mod/github.com/gofiber/fiber/v2@v2.52.5/router.go:172 +0x69
github.com/valyala/fasthttp.(*Server).serveConn(0xc0000ec248, {0x7d3198, 0xc0000ac1e0})
        /data/gopath/pkg/mod/github.com/valyala/fasthttp@v1.57.0/server.go:2385 +0xed1
github.com/valyala/fasthttp.(*workerPool).workerFunc(0xc00011a090, 0xc0000e63a0)
        /data/gopath/pkg/mod/github.com/valyala/fasthttp@v1.57.0/workerpool.go:225 +0x92
github.com/valyala/fasthttp.(*workerPool).getCh.func1()
        /data/gopath/pkg/mod/github.com/valyala/fasthttp@v1.57.0/workerpool.go:197 +0x32
created by github.com/valyala/fasthttp.(*workerPool).getCh in goroutine 1
        /data/gopath/pkg/mod/github.com/valyala/fasthttp@v1.57.0/workerpool.go:196 +0x194

相關文章