91 lines
2.3 KiB
Go
91 lines
2.3 KiB
Go
|
package route
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"github.com/winc-link/hummingbird/internal/hummingbird/core/container"
|
||
|
|
pkgContainer "github.com/winc-link/hummingbird/internal/pkg/container"
|
||
|
|
"github.com/winc-link/hummingbird/internal/pkg/di"
|
||
|
|
"github.com/winc-link/hummingbird/internal/pkg/startup"
|
||
|
|
"net/http"
|
||
|
|
"strconv"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
)
|
||
|
|
|
||
|
|
///var/bin/cmd/hummingbird-ui/build
|
||
|
|
const WebBuildPath = "./cmd/hummingbird-ui/build"
|
||
|
|
const WebBuildPath2 = "/var/build"
|
||
|
|
|
||
|
|
// WebBootstrap contains references to dependencies required by the BootstrapHandler.
|
||
|
|
type WebBootstrap struct {
|
||
|
|
router *gin.Engine
|
||
|
|
AppMode string
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewWebBootstrap is a factory method that returns an initialized WebBootstrap receiver struct.
|
||
|
|
func NewWebBootstrap() *WebBootstrap {
|
||
|
|
// 不做路由日志输出
|
||
|
|
g := gin.New()
|
||
|
|
g.Use(gin.Recovery(), gin.Logger())
|
||
|
|
return &WebBootstrap{
|
||
|
|
router: g,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// BootstrapHandler fulfills the BootstrapHandler contract and performs initialization needed by the resource service.
|
||
|
|
func (b *WebBootstrap) BootstrapHandler(ctx context.Context, wg *sync.WaitGroup, _ startup.Timer, dic *di.Container) bool {
|
||
|
|
configuration := container.ConfigurationFrom(dic.Get)
|
||
|
|
lc := pkgContainer.LoggingClientFrom(dic.Get)
|
||
|
|
|
||
|
|
lc.Infof("start WebBootstrap BootstrapHandler in...")
|
||
|
|
|
||
|
|
//pwd, _ := os.Getwd()
|
||
|
|
LoadWebProxyRoutes(b.router, WebBuildPath, dic)
|
||
|
|
|
||
|
|
if configuration.WebServer.Host == "" || configuration.WebServer.Port == 0 {
|
||
|
|
lc.Errorf("WebServer Host is null OR port is 0")
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
port := strconv.Itoa(configuration.WebServer.Port)
|
||
|
|
addr := configuration.WebServer.Host + ":" + port
|
||
|
|
timeout := time.Second * time.Duration(configuration.WebServer.Timeout)
|
||
|
|
server := &http.Server{
|
||
|
|
Addr: addr,
|
||
|
|
Handler: b.router,
|
||
|
|
WriteTimeout: timeout,
|
||
|
|
ReadTimeout: timeout,
|
||
|
|
}
|
||
|
|
|
||
|
|
wg.Add(1)
|
||
|
|
go func() {
|
||
|
|
defer wg.Done()
|
||
|
|
|
||
|
|
<-ctx.Done()
|
||
|
|
lc.Info("WebProxy server shutting down")
|
||
|
|
_ = server.Shutdown(context.Background())
|
||
|
|
lc.Info("WebProxy server shut down")
|
||
|
|
}()
|
||
|
|
|
||
|
|
lc.Info("WebProxy server starting (" + addr + ")")
|
||
|
|
|
||
|
|
wg.Add(1)
|
||
|
|
go func() {
|
||
|
|
defer func() {
|
||
|
|
wg.Done()
|
||
|
|
}()
|
||
|
|
|
||
|
|
err := server.ListenAndServe()
|
||
|
|
if err != nil {
|
||
|
|
lc.Errorf("WebProxy server failed: %v", err)
|
||
|
|
cancel := pkgContainer.CancelFuncFrom(dic.Get)
|
||
|
|
cancel() // this will caused the service to stop
|
||
|
|
} else {
|
||
|
|
lc.Info("WebProxy server stopped")
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
return true
|
||
|
|
}
|