tbphp
5c16ed84b3
* feat(windows): 增加原生服务安装包 * fix(windows): 收紧服务安装与升级流程 * fix(windows): 固定服务安装路径与语言资源 * fix(windows): 等待安装器进程完成 * fix(windows): 清理首次安装失败残留
91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"gpt-load/internal/app"
|
|
"gpt-load/internal/container"
|
|
"gpt-load/internal/platform/config"
|
|
"gpt-load/internal/platform/redact"
|
|
"gpt-load/internal/platform/utils"
|
|
)
|
|
|
|
type managedApplication interface {
|
|
Start() error
|
|
Stop(context.Context) error
|
|
ServeErrors() <-chan error
|
|
}
|
|
|
|
type managedRuntime struct {
|
|
application managedApplication
|
|
shutdownTimeout time.Duration
|
|
}
|
|
|
|
func buildManagedRuntime() (*managedRuntime, error) {
|
|
dependencyContainer, err := container.BuildContainer()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("build dependency container: %w", err)
|
|
}
|
|
|
|
if err := dependencyContainer.Invoke(func(cfg *config.Config, runtimeRedactor *redact.Redactor) {
|
|
utils.SetupLogger(utils.LogConfig{
|
|
Level: cfg.Log.Level,
|
|
Format: cfg.Log.Format,
|
|
})
|
|
logrus.AddHook(redact.NewHook(runtimeRedactor))
|
|
}); err != nil {
|
|
return nil, fmt.Errorf("configure logger: %w", err)
|
|
}
|
|
|
|
var application *app.App
|
|
var cfg *config.Config
|
|
if err := dependencyContainer.Invoke(func(resolvedApp *app.App, resolvedConfig *config.Config) {
|
|
application = resolvedApp
|
|
cfg = resolvedConfig
|
|
}); err != nil {
|
|
return nil, fmt.Errorf("resolve application: %w", err)
|
|
}
|
|
|
|
return &managedRuntime{
|
|
application: application,
|
|
shutdownTimeout: time.Duration(cfg.Server.GracefulShutdownTimeout) * time.Second,
|
|
}, nil
|
|
}
|
|
|
|
func (runtime *managedRuntime) start() error {
|
|
if err := runtime.application.Start(); err != nil {
|
|
cleanupCtx, cancel := context.WithTimeout(context.Background(), time.Second)
|
|
defer cancel()
|
|
_ = runtime.application.Stop(cleanupCtx)
|
|
return fmt.Errorf("start application: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (runtime *managedRuntime) serveErrors() <-chan error {
|
|
return runtime.application.ServeErrors()
|
|
}
|
|
|
|
func (runtime *managedRuntime) stop(force <-chan struct{}) error {
|
|
shutdownCtx, cancel := context.WithTimeout(context.Background(), runtime.shutdownTimeout)
|
|
defer cancel()
|
|
|
|
stopResult := make(chan error, 1)
|
|
go func() { stopResult <- runtime.application.Stop(shutdownCtx) }()
|
|
if force == nil {
|
|
return <-stopResult
|
|
}
|
|
|
|
select {
|
|
case err := <-stopResult:
|
|
return err
|
|
case <-force:
|
|
cancel()
|
|
return <-stopResult
|
|
}
|
|
}
|