120 lines
3.1 KiB
Go
120 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"path"
|
|
"strconv"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/gorilla/handlers"
|
|
"github.com/gorilla/mux"
|
|
"github.com/ossrs/go-oryx-lib/logger"
|
|
"github.com/ossrs/srs-sip/pkg/api"
|
|
"github.com/ossrs/srs-sip/pkg/config"
|
|
"github.com/ossrs/srs-sip/pkg/service"
|
|
)
|
|
|
|
func WaitTerminationSignal(cancel context.CancelFunc) {
|
|
sigc := make(chan os.Signal, 1)
|
|
signal.Notify(sigc, syscall.SIGINT, syscall.SIGTERM)
|
|
defer signal.Stop(sigc)
|
|
<-sigc
|
|
cancel()
|
|
}
|
|
|
|
func main() {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
|
|
conf, err := config.LoadConfig("conf/config.yaml")
|
|
if err != nil {
|
|
logger.E(nil, "load config failed: %v", err)
|
|
return
|
|
}
|
|
|
|
sipSvr, err := service.NewService(ctx, conf)
|
|
if err != nil {
|
|
logger.Ef("create service failed. err is %v", err.Error())
|
|
return
|
|
}
|
|
|
|
if err := sipSvr.Start(); err != nil {
|
|
logger.Ef("start sip service failed. err is %v", err.Error())
|
|
return
|
|
}
|
|
|
|
// 创建主路由
|
|
router := mux.NewRouter().StrictSlash(true)
|
|
|
|
// CORS配置
|
|
headers := handlers.AllowedHeaders([]string{"X-Requested-With", "Content-Type", "Authorization"})
|
|
methods := handlers.AllowedMethods([]string{"GET", "POST", "PUT", "DELETE", "OPTIONS"})
|
|
origins := handlers.AllowedOrigins([]string{"*"})
|
|
|
|
// 设置API路由 - 需要在静态文件路由之前设置
|
|
apiSvr, err := api.NewHttpApiServer(conf, sipSvr)
|
|
if err != nil {
|
|
logger.Ef("create http service failed. err is %v", err.Error())
|
|
return
|
|
}
|
|
apiSvr.Start(router)
|
|
|
|
// 使用配置中指定的目录，如果不存在则尝试备选目录
|
|
targetDir := conf.Http.Dir
|
|
if _, err := os.Stat(path.Join(targetDir, "index.html")); err != nil {
|
|
backupDirs := []string{"./html", "../web/NextGB/dist"}
|
|
for _, dir := range backupDirs {
|
|
if _, err := os.Stat(path.Join(dir, "index.html")); err == nil {
|
|
targetDir = dir
|
|
break
|
|
}
|
|
}
|
|
}
|
|
if targetDir == "" {
|
|
logger.Ef(ctx, "index.html not found")
|
|
return
|
|
}
|
|
|
|
// 创建文件服务器
|
|
fs := http.FileServer(http.Dir(targetDir))
|
|
|
|
// 添加静态文件处理 - 使用NotFoundHandler来处理未匹配的路由
|
|
router.NotFoundHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
logger.Tf(context.Background(), "Handling request: %s", r.URL.Path)
|
|
|
|
// 检查请求的文件是否存在
|
|
filePath := path.Join(targetDir, r.URL.Path)
|
|
_, err := os.Stat(filePath)
|
|
if os.IsNotExist(err) {
|
|
// 如果文件不存在，返回 index.html
|
|
r.URL.Path = "/"
|
|
}
|
|
fs.ServeHTTP(w, r)
|
|
})
|
|
|
|
// 启动合并后的HTTP服务
|
|
go func() {
|
|
httpPort := strconv.Itoa(conf.Http.Port)
|
|
handler := handlers.CORS(headers, methods, origins)(router)
|
|
server := &http.Server{
|
|
Addr: ":" + httpPort,
|
|
Handler: handler,
|
|
ReadTimeout: 10 * time.Second,
|
|
WriteTimeout: 10 * time.Second,
|
|
IdleTimeout: 30 * time.Second,
|
|
ReadHeaderTimeout: 5 * time.Second,
|
|
}
|
|
logger.Tf(ctx, "http server listen on %s, home is %v", httpPort, targetDir)
|
|
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
logger.Ef(ctx, "listen on %s failed", httpPort)
|
|
}
|
|
}()
|
|
|
|
WaitTerminationSignal(cancel)
|
|
|
|
sipSvr.Stop()
|
|
}
|