242 lines
6.4 KiB
Go
242 lines
6.4 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"crypto/x509"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"nms_nbi/features/dbrest"
|
|
"nms_nbi/features/event"
|
|
"nms_nbi/features/fm"
|
|
"nms_nbi/features/lm"
|
|
"nms_nbi/features/pm"
|
|
"nms_nbi/lib/dborm"
|
|
"nms_nbi/lib/global"
|
|
"nms_nbi/lib/log"
|
|
"nms_nbi/lib/routes"
|
|
"nms_nbi/restagent/config"
|
|
"nms_nbi/src"
|
|
"nms_nbi/src/framework/middleware"
|
|
libSession "nms_nbi/src/lib_features/session"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"golang.org/x/net/http2"
|
|
"golang.org/x/net/http2/h2c"
|
|
)
|
|
|
|
func HttpListen(addr string, router http.Handler) {
|
|
// 创建HTTP服务器
|
|
h2s := &http2.Server{
|
|
// ...
|
|
}
|
|
server := &http.Server{
|
|
Addr: addr,
|
|
Handler: h2c.NewHandler(router, h2s),
|
|
}
|
|
|
|
err := server.ListenAndServe()
|
|
if err != nil {
|
|
fmt.Println("ListenAndServe err:", err)
|
|
os.Exit(12)
|
|
}
|
|
}
|
|
|
|
func HttpListenTLS(addr, caFile, certFile, keyFile string, clientAuthType int, router http.Handler) {
|
|
// 加载根证书
|
|
caCert, err := os.ReadFile(caFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
caCertPool := x509.NewCertPool()
|
|
caCertPool.AppendCertsFromPEM(caCert)
|
|
|
|
// 创建自定义的TLS配置
|
|
tlsConfig := &tls.Config{
|
|
MinVersion: tls.VersionTLS10,
|
|
MaxVersion: tls.VersionTLS13,
|
|
ClientCAs: caCertPool,
|
|
ClientAuth: tls.ClientAuthType(clientAuthType),
|
|
}
|
|
|
|
// 创建HTTP服务器
|
|
server := &http.Server{
|
|
Addr: addr,
|
|
Handler: router,
|
|
TLSConfig: tlsConfig,
|
|
}
|
|
|
|
// support http 2.0 server
|
|
http2.ConfigureServer(server, &http2.Server{})
|
|
if err != nil {
|
|
fmt.Println("ConfigureServer err:", err)
|
|
os.Exit(13)
|
|
}
|
|
err = server.ListenAndServeTLS(certFile, keyFile)
|
|
if err != nil {
|
|
fmt.Println("ListenAndServeTLS err:", err)
|
|
os.Exit(14)
|
|
}
|
|
}
|
|
|
|
func HttpListenWebServerTLS(addr, caFile, certFile, keyFile string, clientAuthType int) {
|
|
// 加载根证书
|
|
caCert, err := os.ReadFile(caFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
caCertPool := x509.NewCertPool()
|
|
caCertPool.AppendCertsFromPEM(caCert)
|
|
|
|
// 创建自定义的TLS配置
|
|
tlsConfig := &tls.Config{
|
|
MinVersion: tls.VersionTLS10,
|
|
MaxVersion: tls.VersionTLS13,
|
|
ClientCAs: caCertPool,
|
|
ClientAuth: tls.ClientAuthType(clientAuthType),
|
|
}
|
|
|
|
// 创建HTTP服务器
|
|
server := &http.Server{
|
|
Addr: addr,
|
|
TLSConfig: tlsConfig,
|
|
}
|
|
|
|
// support http 2.0 server
|
|
http2.ConfigureServer(server, &http2.Server{})
|
|
if err != nil {
|
|
fmt.Println("ConfigureServer err:", err)
|
|
os.Exit(9)
|
|
}
|
|
err = server.ListenAndServeTLS(certFile, keyFile)
|
|
if err != nil {
|
|
fmt.Println("ListenAndServeTLS err:", err)
|
|
os.Exit(10)
|
|
}
|
|
}
|
|
|
|
func HttpListenWebServer(addr string) {
|
|
// 创建HTTP服务器
|
|
server := &http.Server{
|
|
Addr: addr,
|
|
}
|
|
|
|
// support http 2.0 server
|
|
err := http2.ConfigureServer(server, &http2.Server{})
|
|
if err != nil {
|
|
fmt.Println("ConfigureServer err:", err)
|
|
os.Exit(7)
|
|
}
|
|
err = server.ListenAndServe()
|
|
if err != nil {
|
|
fmt.Println("ListenAndServe err:", err)
|
|
os.Exit(8)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
// src 配置中心初始加载
|
|
src.ConfigurationInit()
|
|
app := src.AppEngine()
|
|
|
|
conf := config.GetYamlConfig()
|
|
|
|
log.InitLogger(conf.Logger.File, conf.Logger.Duration, conf.Logger.Count, "omc:restagent", config.GetLogLevel())
|
|
fmt.Printf("OMC restagent version: %s\n", global.Version)
|
|
log.Infof("========================= OMC restagent startup =========================")
|
|
log.Infof("OMC restagent version: %s %s %s", global.Version, global.BuildTime, global.GoVer)
|
|
|
|
err := dborm.InitDbClient(conf.Database.Type, conf.Database.User, conf.Database.Password,
|
|
conf.Database.Host, conf.Database.Port, conf.Database.Name)
|
|
if err != nil {
|
|
fmt.Println("dborm.initDbClient err:", err)
|
|
os.Exit(4)
|
|
}
|
|
err = fm.InitDbClient(conf.Database.Type, conf.Database.User, conf.Database.Password,
|
|
conf.Database.Host, conf.Database.Port, conf.Database.Name)
|
|
if err != nil {
|
|
fmt.Println("dborm.initDbClient err:", err)
|
|
os.Exit(4)
|
|
}
|
|
err = pm.InitDbClient(conf.Database.Type, conf.Database.User, conf.Database.Password,
|
|
conf.Database.Host, conf.Database.Port, conf.Database.Name)
|
|
if err != nil {
|
|
fmt.Println("dborm.initDbClient err:", err)
|
|
os.Exit(4)
|
|
}
|
|
err = dbrest.InitDbClient(conf.Database.Type, conf.Database.User, conf.Database.Password,
|
|
conf.Database.Host, conf.Database.Port, conf.Database.Name)
|
|
if err != nil {
|
|
fmt.Println("dbrest.initDbClient err:", err)
|
|
os.Exit(4)
|
|
}
|
|
err = lm.InitDbClient(conf.Database.Type, conf.Database.User, conf.Database.Password,
|
|
conf.Database.Host, conf.Database.Port, conf.Database.Name)
|
|
if err != nil {
|
|
fmt.Println("lm.initDbClient err:", err)
|
|
os.Exit(4)
|
|
}
|
|
|
|
// 将 mux.Router 注册到 gin.Engine
|
|
|
|
// 默认路由组
|
|
defaultUriGroup := app.Group(config.DefaultUriPrefix)
|
|
defaultUriGroup.Use(middleware.PreAuthorize(nil))
|
|
defaultUriGroup.Use(libSession.SessionHeader())
|
|
defaultUriGroup.Any("/*any", gin.WrapH(routes.NewRouter()))
|
|
// 可配置前缀路由组
|
|
uriGroup := app.Group(config.UriPrefix)
|
|
uriGroup.Use(libSession.SessionHeader())
|
|
uriGroup.Any("/*any", gin.WrapH(routes.NewRouter()))
|
|
// AMF上报的UE事件, 无前缀，暂时特殊处理
|
|
app.POST(event.UriUEEvent, event.PostUEEventFromAMF)
|
|
|
|
for _, rest := range conf.Rest {
|
|
// ipv4 goroutines
|
|
if rest.IPv4 != "" {
|
|
listen := rest.IPv4 + ":" + strconv.Itoa(int(rest.Port))
|
|
if strings.ToLower(rest.Scheme) == "https" {
|
|
go HttpListenTLS(listen, rest.CaFile, rest.CertFile, rest.KeyFile, rest.ClientAuthType, app)
|
|
} else {
|
|
go HttpListen(listen, app)
|
|
}
|
|
}
|
|
if rest.IPv4 != "0.0.0.0" {
|
|
// 默认启动localhost侦听
|
|
listenLocal := "127.0.0.1" + ":" + strconv.Itoa(int(rest.Port))
|
|
if strings.ToLower(rest.Scheme) == "https" {
|
|
go HttpListenTLS(listenLocal, rest.CaFile, rest.CertFile, rest.KeyFile, rest.ClientAuthType, app)
|
|
} else {
|
|
go HttpListen(listenLocal, app)
|
|
}
|
|
}
|
|
// ipv6 goroutines
|
|
if rest.IPv6 != "" {
|
|
listenv6 := "[" + rest.IPv6 + "]" + ":" + strconv.Itoa(int(rest.Port))
|
|
if strings.ToLower(rest.Scheme) == "https" {
|
|
go HttpListenTLS(listenv6, rest.CaFile, rest.CertFile, rest.KeyFile, rest.ClientAuthType, app)
|
|
} else {
|
|
go HttpListen(listenv6, app)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
if conf.WebServer.Enabled {
|
|
fs := http.FileServer(http.Dir(conf.WebServer.RootDir))
|
|
http.Handle("/", fs)
|
|
for _, listen := range conf.WebServer.Listen {
|
|
if strings.ToLower(listen.Scheme) == "https" {
|
|
go HttpListenWebServerTLS(listen.Addr, listen.CaFile, listen.CertFile, listen.KeyFile, listen.ClientAuthType)
|
|
} else {
|
|
go HttpListenWebServer(listen.Addr)
|
|
}
|
|
}
|
|
}
|
|
|
|
select {}
|
|
}
|