mirror of
https://github.com/lordmathis/llamactl.git
synced 2025-11-05 16:44:22 +00:00
100 lines
2.7 KiB
Go
100 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"llamactl/pkg/config"
|
|
"llamactl/pkg/manager"
|
|
"llamactl/pkg/server"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
)
|
|
|
|
// version is set at build time using -ldflags "-X main.version=1.0.0"
|
|
var version string = "unknown"
|
|
var commitHash string = "unknown"
|
|
var buildTime string = "unknown"
|
|
|
|
// @title llamactl API
|
|
// @version 1.0
|
|
// @description llamactl is a control server for managing Llama Server instances.
|
|
// @license.name MIT License
|
|
// @license.url https://opensource.org/license/mit/
|
|
// @basePath /api/v1
|
|
func main() {
|
|
|
|
// --version flag to print the version
|
|
if len(os.Args) > 1 && os.Args[1] == "--version" {
|
|
fmt.Printf("llamactl version: %s\n", version)
|
|
fmt.Printf("Commit hash: %s\n", commitHash)
|
|
fmt.Printf("Build time: %s\n", buildTime)
|
|
return
|
|
}
|
|
|
|
configPath := os.Getenv("LLAMACTL_CONFIG_PATH")
|
|
cfg, err := config.LoadConfig(configPath)
|
|
if err != nil {
|
|
fmt.Printf("Error loading config: %v\n", err)
|
|
fmt.Println("Using default configuration.")
|
|
}
|
|
|
|
// Set version information
|
|
cfg.Version = version
|
|
cfg.CommitHash = commitHash
|
|
cfg.BuildTime = buildTime
|
|
|
|
// Create the data directory if it doesn't exist
|
|
if cfg.Instances.AutoCreateDirs {
|
|
if err := os.MkdirAll(cfg.Instances.InstancesDir, 0755); err != nil {
|
|
fmt.Printf("Error creating config directory %s: %v\n", cfg.Instances.InstancesDir, err)
|
|
fmt.Println("Persistence will not be available.")
|
|
}
|
|
|
|
if err := os.MkdirAll(cfg.Instances.LogsDir, 0755); err != nil {
|
|
fmt.Printf("Error creating log directory %s: %v\n", cfg.Instances.LogsDir, err)
|
|
fmt.Println("Instance logs will not be available.")
|
|
}
|
|
}
|
|
|
|
// Initialize the instance manager
|
|
instanceManager := manager.NewInstanceManager(cfg.Backends, cfg.Instances, cfg.Nodes)
|
|
|
|
// Create a new handler with the instance manager
|
|
handler := server.NewHandler(instanceManager, cfg)
|
|
|
|
// Setup the router with the handler
|
|
r := server.SetupRouter(handler)
|
|
|
|
// Handle graceful shutdown
|
|
stop := make(chan os.Signal, 1)
|
|
signal.Notify(stop, os.Interrupt, syscall.SIGTERM)
|
|
|
|
server := http.Server{
|
|
Addr: fmt.Sprintf("%s:%d", cfg.Server.Host, cfg.Server.Port),
|
|
Handler: r,
|
|
}
|
|
|
|
go func() {
|
|
fmt.Printf("Llamactl server listening on %s:%d\n", cfg.Server.Host, cfg.Server.Port)
|
|
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
|
fmt.Printf("Error starting server: %v\n", err)
|
|
}
|
|
}()
|
|
|
|
// Wait for shutdown signal
|
|
<-stop
|
|
fmt.Println("Shutting down server...")
|
|
|
|
if err := server.Close(); err != nil {
|
|
fmt.Printf("Error shutting down server: %v\n", err)
|
|
} else {
|
|
fmt.Println("Server shut down gracefully.")
|
|
}
|
|
|
|
// Wait for all instances to stop
|
|
instanceManager.Shutdown()
|
|
|
|
fmt.Println("Exiting llamactl.")
|
|
}
|