84 lines
2.9 KiB
Go
84 lines
2.9 KiB
Go
package router
|
||
|
||
import (
|
||
"electricity_bill_calc/controller"
|
||
"electricity_bill_calc/logger"
|
||
"electricity_bill_calc/security"
|
||
"fmt"
|
||
"runtime"
|
||
"time"
|
||
|
||
"github.com/gofiber/fiber/v2"
|
||
"github.com/gofiber/fiber/v2/middleware/compress"
|
||
"github.com/gofiber/fiber/v2/middleware/recover"
|
||
jsontime "github.com/liamylian/jsontime/v2/v2"
|
||
"go.uber.org/zap"
|
||
)
|
||
|
||
var json = jsontime.ConfigWithCustomTimeFormat
|
||
|
||
func init() {
|
||
timeZoneShanghai, _ := time.LoadLocation("Asia/Shanghai")
|
||
jsontime.AddTimeFormatAlias("simple_datetime", "2006-01-02 15:04:05")
|
||
jsontime.AddTimeFormatAlias("simple_date", "2006-01-02")
|
||
jsontime.AddLocaleAlias("shanghai", timeZoneShanghai)
|
||
}
|
||
|
||
func App() *fiber.App {
|
||
app := fiber.New(fiber.Config{ //创建fiber实例的时候选择配置选项
|
||
BodyLimit: 30 * 1024 * 1024, //设置请求正文允许的最大大小。
|
||
EnablePrintRoutes: true, //自定义方案，用于启动消息
|
||
EnableTrustedProxyCheck: false, //禁用受信代理
|
||
Prefork: false, //禁止预处理（如果要启用预处理则需要通过shell脚本运行）
|
||
ErrorHandler: errorHandler, //相应全局处理错误
|
||
JSONEncoder: json.Marshal, //json编码
|
||
JSONDecoder: json.Unmarshal, //json解码
|
||
})
|
||
app.Use(compress.New()) //压缩中间件
|
||
app.Use(recover.New(recover.Config{
|
||
EnableStackTrace: true,
|
||
StackTraceHandler: stackTraceHandler,
|
||
})) //恢复中间件
|
||
app.Use(logger.NewLogMiddleware(logger.LogMiddlewareConfig{
|
||
Logger: logger.Named("App"),
|
||
})) //日志中间件
|
||
app.Use(security.SessionRecovery) //会话恢复中间件
|
||
|
||
controller.InitializeUserHandlers(app)
|
||
controller.InitializeRegionHandlers(app)
|
||
controller.InitializeChargeHandlers(app)
|
||
controller.InitializeParkHandlers(app)
|
||
controller.InitializeTenementHandler(app)
|
||
controller.InitializeMeterHandlers(app)
|
||
controller.InitializeInvoiceHandler(app)
|
||
controller.InitializeTopUpHandlers(app)
|
||
controller.InitializeReportHandlers(app)
|
||
|
||
controller.InitializeWithdrawHandlers(app) // 公示撤回
|
||
controller.InitializeFoundationHandlers(app) // 基础数据
|
||
controller.InitializeStatisticsController(app) // 首页信息
|
||
controller.InitializeGmController(app) // 天神模式
|
||
|
||
return app
|
||
}
|
||
|
||
// 全局错误处理
|
||
func errorHandler(c *fiber.Ctx, err error) error {
|
||
code := fiber.StatusInternalServerError
|
||
if e, ok := err.(*fiber.Error); ok {
|
||
code = e.Code
|
||
}
|
||
e := c.Status(code).SendString(err.Error())
|
||
if e != nil {
|
||
return c.Status(fiber.StatusInternalServerError).SendString(err.Error())
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 处理Recover中间件输出的栈追踪信息
|
||
func stackTraceHandler(c *fiber.Ctx, e interface{}) {
|
||
buf := make([]byte, 1024)
|
||
buf = buf[:runtime.Stack(buf, false)]
|
||
logger.Named("App", "StackTrace").Warn(fmt.Sprintf("panic: %+v", e), zap.ByteString("trace", buf), zap.Any("origin", e))
|
||
}
|