rogee/atomctl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add cmux

Browse Source
This commit is contained in:
Rogee
2024-12-31 19:23:58 +08:00
parent 64c08249df
commit 498a820a54
4 changed files with 106 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
templates/project/providers/http/engine.go.tpl
View File

@@ -3,6 +3,7 @@ package http
import (
"errors"
"fmt"
"net"
"runtime/debug"
"time"
@@ -29,7 +30,7 @@ type Service struct {
Engine *fiber.App
}
func (svc *Service) Serve() error {
func (svc *Service) listenerConfig() fiber.ListenConfig {
listenConfig := fiber.ListenConfig{
EnablePrintRoutes: true,
OnShutdownSuccess: func() {
@@ -44,7 +45,7 @@ func (svc *Service) Serve() error {
if svc.conf.Tls != nil {
if svc.conf.Tls.Cert == "" || svc.conf.Tls.Key == "" {
return errors.New("tls cert or key is empty")
panic(errors.New("tls cert and key must be set"))
}
listenConfig.CertFile = svc.conf.Tls.Cert
listenConfig.CertKeyFile = svc.conf.Tls.Key
@@ -52,8 +53,15 @@ func (svc *Service) Serve() error {
container.AddCloseAble(func() {
svc.Engine.Shutdown()
})
return listenConfig
}
return svc.Engine.Listen(svc.conf.Address(), listenConfig)
func (svc *Service) Listener(ln net.Listener) error {
return svc.Engine.Listener(ln, svc.listenerConfig())
}
func (svc *Service) Serve() error {
return svc.Engine.Listen(svc.conf.Address(), svc.listenerConfig())
}
func Provide(opts ...opt.Option) error {