feat(fiber-cfg) Add fiber config wrappers

Trendyol · Sep 11, 2024 · 6571d06 · 6571d06
1 parent 57020b3
commit 6571d06
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 9 deletions.
diff --git a/modules/server/common/types.go b/modules/server/common/types.go
@@ -6,6 +6,7 @@ import (
 )
 
 type (
-	MiddlewareGroup = []fiber.Handler
-	FiberAppWrapper = wrapper.Wrapper[*fiber.App]
+	MiddlewareGroup    = []fiber.Handler
+	FiberAppWrapper    = wrapper.Wrapper[*fiber.App]
+	FiberConfigWrapper = wrapper.Wrapper[fiber.Config]
 )
diff --git a/modules/server/module.go b/modules/server/module.go
@@ -13,11 +13,12 @@ import (
 )
 
 var (
-	asController      = as.Interface[controller.Controller]("servercontrollers")
-	asMiddleware      = as.Struct[fiber.Handler]("servermiddleware")
-	asValidationRule  = as.Interface[validation.Rule]("validationrules")
-	asFiberAppWrapper = as.Struct[common.FiberAppWrapper]("fiberappwrappers")
-	asMiddlewareGroup = as.Struct[common.MiddlewareGroup]("middlewaregroups")
+	asController         = as.Interface[controller.Controller]("servercontrollers")
+	asMiddleware         = as.Struct[fiber.Handler]("servermiddleware")
+	asValidationRule     = as.Interface[validation.Rule]("validationrules")
+	asFiberAppWrapper    = as.Struct[common.FiberAppWrapper]("fiberappwrappers")
+	asMiddlewareGroup    = as.Struct[common.MiddlewareGroup]("middlewaregroups")
+	asFiberConfigWrapper = as.Struct[common.FiberConfigWrapper]("fiberconfigwrappers")
 )
 
 func Module(option ...Option) *module.Module {
@@ -43,6 +44,7 @@ func Module(option ...Option) *module.Module {
 
 		asFiberAppWrapper.Grouper(),
 		asMiddlewareGroup.Grouper(),
+		asFiberConfigWrapper.Grouper(),
 
 		// server
 		defaultFiber,
@@ -75,6 +77,10 @@ func Module(option ...Option) *module.Module {
 			Match: asMiddlewareGroup.Match,
 			Wrap:  asMiddlewareGroup.Value,
 		},
+		module.ProvideHook{
+			Match: asFiberConfigWrapper.Match,
+			Wrap:  asFiberConfigWrapper.Value,
+		},
 	)
 
 	return m

diff --git a/modules/server/server.go b/modules/server/server.go
@@ -81,18 +81,25 @@ func defaultFiber(
 	mws []fiber.Handler,
 	wrappers []common.FiberAppWrapper,
 	groups []common.MiddlewareGroup,
+	configWrappers []common.FiberConfigWrapper,
 	opts *options,
 ) *fiber.App {
 	setDefaultFiberConfigs(cfg)
 	serverCfg := cfg.Of("server")
 
-	app := fiber.New(fiber.Config{
+	fibercfg := fiber.Config{
 		BodyLimit:      serverCfg.GetInt("bodylimit"),
 		ReadBufferSize: serverCfg.GetInt("readbuffersize"),
 		ReadTimeout:    serverCfg.GetDuration("readtimeout"),
 		WriteTimeout:   serverCfg.GetDuration("writetimeout"),
 		ErrorHandler:   opts.errHandler,
-	})
+	}
+
+	for _, fwrapper := range configWrappers {
+		fibercfg = fwrapper(fibercfg)
+	}
+
+	app := fiber.New(fibercfg)
 
 	if serverCfg.Exists("cors") {
 		app.Use(middlewares.CORSMiddleware(serverCfg.Of("cors")))