mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-03 04:50:52 +08:00
5a50856fc1
- Introduced a keep-alive endpoint to monitor service activity. - Added timeout-specific shutdown functionality when the endpoint is idle. - Implemented password-protected access for the keep-alive endpoint. - Updated server startup to support configurable keep-alive options.
56 lines
1.8 KiB
Go
56 lines
1.8 KiB
Go
// Package cmd provides command-line interface functionality for the CLI Proxy API server.
|
|
// It includes authentication flows for various AI service providers, service startup,
|
|
// and other command-line operations.
|
|
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/router-for-me/CLIProxyAPI/v6/internal/api"
|
|
"github.com/router-for-me/CLIProxyAPI/v6/internal/config"
|
|
"github.com/router-for-me/CLIProxyAPI/v6/sdk/cliproxy"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// StartService builds and runs the proxy service using the exported SDK.
|
|
// It creates a new proxy service instance, sets up signal handling for graceful shutdown,
|
|
// and starts the service with the provided configuration.
|
|
//
|
|
// Parameters:
|
|
// - cfg: The application configuration
|
|
// - configPath: The path to the configuration file
|
|
// - localPassword: Optional password accepted for local management requests
|
|
func StartService(cfg *config.Config, configPath string, localPassword string) {
|
|
builder := cliproxy.NewBuilder().
|
|
WithConfig(cfg).
|
|
WithConfigPath(configPath).
|
|
WithLocalManagementPassword(localPassword)
|
|
|
|
ctxSignal, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer cancel()
|
|
|
|
runCtx := ctxSignal
|
|
if localPassword != "" {
|
|
var keepAliveCancel context.CancelFunc
|
|
runCtx, keepAliveCancel = context.WithCancel(ctxSignal)
|
|
builder = builder.WithServerOptions(api.WithKeepAliveEndpoint(10*time.Second, func() {
|
|
log.Warn("keep-alive endpoint idle for 10s, shutting down")
|
|
keepAliveCancel()
|
|
}))
|
|
}
|
|
|
|
service, err := builder.Build()
|
|
if err != nil {
|
|
log.Fatalf("failed to build proxy service: %v", err)
|
|
}
|
|
|
|
err = service.Run(runCtx)
|
|
if err != nil && !errors.Is(err, context.Canceled) {
|
|
log.Fatalf("proxy service exited with error: %v", err)
|
|
}
|
|
}
|