go-zero 全局异常处理-全局中间件

原创

已于 2024-01-27 11:54:07 修改 · 1.7k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#golang #中间件 #开发语言

于 2024-01-25 20:19:54 首次发布

本文介绍了如何在Go-Zero框架中使用全局中间件捕获并处理异常，包括通过`server.Use(middleware.WithMiddleware)`注册中间件，以及使用`defer`和`recover`进行异常恢复，并通过自定义错误处理函数返回统一的错误响应。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、在主入口新增

package main

import (
	"flag"
	"fmt"
	"go/application/internal/config"
	"go/application/internal/handler"
	"go/application/internal/middleware"
	"go/application/internal/svc"
	manage "go/pkg/manage"

	"github.com/zeromicro/go-zero/core/conf"
	"github.com/zeromicro/go-zero/rest"
	"github.com/zeromicro/go-zero/rest/httpx"
)

var configFile = flag.String("f", "etc/service-api.yaml", "配置文件")

func main() {

	flag.Parse()

	var c config.Config

	// 读取配置文件中的值，将其保存在c变量上  所以这一块传递的是地址
	conf.MustLoad(*configFile, &c)

	// 将全局配置保存在全局变量上
	config.GlobalConfig = c

	// 注册服务，传递服务配置
	server := rest.MustNewServer(c.RestConf)

	// 全局中间件  这儿主要用来捕获一下全局的异常，抛出给前端
	server.Use(middleware.WithMiddleware)

	// 停止服务
	defer server.Stop()

	// 为上下文赋值  将常用的方法绑定在上下文当中(redis、orm等等)
	ctx := svc.NewServiceContext(c)

	// 注册路由
	handler.RegisterHandlers(server, ctx)

	// 统一错误处理
	httpx.SetErrorHandler(manage.ErrHandler)

	fmt.Printf("Starting server at %s:%d...\n", c.Ho