mirror of
https://github.com/SunBK201/UA3F.git
synced 2025-12-18 09:46:58 +00:00
74 lines
1.4 KiB
Go
74 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/sunbk201/ua3f/internal/config"
|
|
"github.com/sunbk201/ua3f/internal/log"
|
|
"github.com/sunbk201/ua3f/internal/rewrite"
|
|
"github.com/sunbk201/ua3f/internal/server"
|
|
"github.com/sunbk201/ua3f/internal/server/netlink"
|
|
"github.com/sunbk201/ua3f/internal/statistics"
|
|
)
|
|
|
|
const version = "1.8.0"
|
|
|
|
func main() {
|
|
cfg, showVer := config.Parse()
|
|
|
|
log.SetLogConf(cfg.LogLevel)
|
|
|
|
if showVer {
|
|
logrus.Infof("UA3F version: %s", version)
|
|
return
|
|
}
|
|
|
|
rw, err := rewrite.New(cfg)
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
|
|
srv, err := server.NewServer(cfg, rw)
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
defer srv.Close()
|
|
|
|
log.LogHeader(version, cfg)
|
|
|
|
helper := netlink.New(cfg)
|
|
defer helper.Close()
|
|
err = helper.Setup()
|
|
if err != nil {
|
|
logrus.Fatal(err)
|
|
}
|
|
|
|
cleanup := make(chan os.Signal, 1)
|
|
signal.Notify(cleanup, syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
go helper.Start()
|
|
go statistics.StartRecorder()
|
|
|
|
go func() {
|
|
<-cleanup
|
|
logrus.Info("Shutting down UA3F...")
|
|
if err := helper.Close(); err != nil {
|
|
logrus.Errorf("Error during helper close: %v", err)
|
|
}
|
|
if err := srv.Close(); err != nil {
|
|
logrus.Errorf("Error during UA3F close: %v", err)
|
|
}
|
|
logrus.Info("UA3F exited gracefully")
|
|
os.Exit(0)
|
|
}()
|
|
|
|
if err := srv.Start(); err != nil {
|
|
helper.Close()
|
|
logrus.Fatal(err)
|
|
}
|
|
}
|