fe86846e3b
Rename the proxy environment type and generated fake to clarify that the configuration surface belongs to the proxy server. Update proxy bootstrap, load balancers, protocol servers, debug, signal handling, tests, and codebase memory to use ProxyEnvironment.
53 lines
1.0 KiB
Go
53 lines
1.0 KiB
Go
// Copyright (c) 2026 Winlin
|
|
//
|
|
// SPDX-License-Identifier: MIT
|
|
package signal
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"srsx/internal/env"
|
|
"srsx/internal/errors"
|
|
"srsx/internal/logger"
|
|
)
|
|
|
|
// Indirections so tests can substitute signal delivery and process exit.
|
|
var (
|
|
signalNotify = signal.Notify
|
|
osExit = os.Exit
|
|
)
|
|
|
|
func InstallSignals(ctx context.Context, cancel context.CancelFunc) {
|
|
sc := make(chan os.Signal, 1)
|
|
signalNotify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt)
|
|
|
|
go func() {
|
|
for s := range sc {
|
|
logger.Df(ctx, "Got signal %v", s)
|
|
cancel()
|
|
}
|
|
}()
|
|
}
|
|
|
|
func InstallForceQuit(ctx context.Context, environment env.ProxyEnvironment) error {
|
|
var forceTimeout time.Duration
|
|
timeoutStr := environment.ForceQuitTimeout()
|
|
if t, err := time.ParseDuration(timeoutStr); err != nil {
|
|
return errors.Wrapf(err, "parse force timeout %v", timeoutStr)
|
|
} else {
|
|
forceTimeout = t
|
|
}
|
|
|
|
go func() {
|
|
<-ctx.Done()
|
|
time.Sleep(forceTimeout)
|
|
logger.Wf(ctx, "Force to exit by timeout")
|
|
osExit(1)
|
|
}()
|
|
return nil
|
|
}
|