64 lines
1.1 KiB
Go
64 lines
1.1 KiB
Go
|
package gateway
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"esway/internal/middleware/analysis"
|
||
|
|
"esway/internal/middleware/logger"
|
||
|
|
"esway/internal/opt"
|
||
|
|
"esway/internal/tool"
|
||
|
|
|
||
|
|
"github.com/loveuer/nf"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Start(ctx context.Context) error {
|
||
|
|
ch := make(chan struct{})
|
||
|
|
app := nf.New(nf.Config{
|
||
|
|
BodyLimit: 4 * 1024 * 1024,
|
||
|
|
DisableLogger: true,
|
||
|
|
DisableBanner: true,
|
||
|
|
DisableMessagePrint: true,
|
||
|
|
})
|
||
|
|
|
||
|
|
h, err := proxy(ctx)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
app.Use(logger.New(logger.Config{IgnoreFn: func(c *nf.Ctx) bool {
|
||
|
|
path := c.Path()
|
||
|
|
if strings.HasPrefix(path, "/.") || strings.HasPrefix(path, "/_") {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
|
||
|
|
return false
|
||
|
|
}}))
|
||
|
|
app.Use(analysis.New())
|
||
|
|
|
||
|
|
app.Any("/*any", h)
|
||
|
|
|
||
|
|
ln, err := net.Listen("tcp", opt.Cfg.Listen.Gateway)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("gateway listen at %s err: %s", opt.Cfg.Listen.Gateway, err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
go func() {
|
||
|
|
ch <- struct{}{}
|
||
|
|
fmt.Printf("esway: gateway listen at %s\n", opt.Cfg.Listen.Gateway)
|
||
|
|
_ = app.RunListener(ln)
|
||
|
|
}()
|
||
|
|
<-ch
|
||
|
|
|
||
|
|
go func() {
|
||
|
|
ch <- struct{}{}
|
||
|
|
<-ctx.Done()
|
||
|
|
_ = app.Shutdown(tool.Timeout(2))
|
||
|
|
}()
|
||
|
|
<-ch
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|