67 lines
1.7 KiB
Go
67 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/engigu/taskpool/cmd"
|
|
"github.com/engigu/taskpool/internal/bootstrap"
|
|
"github.com/engigu/taskpool/internal/constant"
|
|
)
|
|
|
|
// @title TaskPool API
|
|
// @version 1.0
|
|
// @description TaskPool OpenAPI Server documentation.
|
|
// @termsOfService http://swagger.io/terms/
|
|
// @contact.name API Support
|
|
// @contact.url http://www.swagger.io/support
|
|
// @contact.email support@swagger.io
|
|
// @license.name Apache 2.0
|
|
// @license.url http://www.apache.org/licenses/LICENSE-2.0.html
|
|
// @host localhost:8052
|
|
// @BasePath /open2api/v1
|
|
// @query.collection.format multi
|
|
// @securityDefinitions.apikey BearerAuth
|
|
// @in header
|
|
// @name Authorization
|
|
// @description Type "Bearer" followed by a space and the API token.
|
|
|
|
func printHelp() {
|
|
fmt.Println("\nTaskPool - 极致轻量、高性能的自动化任务调度平台")
|
|
fmt.Println("用法:")
|
|
fmt.Println(" taskpool <命令> [参数]")
|
|
fmt.Println("可用命令:")
|
|
for _, info := range constant.Commands {
|
|
fmt.Printf(" %-15s %s\n", info.Name, info.Description)
|
|
}
|
|
fmt.Println("\n使用 'taskpool <命令> --help' 查看具体命令的参数说明。")
|
|
fmt.Println()
|
|
}
|
|
|
|
func main() {
|
|
if len(os.Args) < 2 {
|
|
printHelp()
|
|
os.Exit(1)
|
|
}
|
|
|
|
commandName := os.Args[1]
|
|
|
|
if commandName == "server" {
|
|
bootstrap.New().Run()
|
|
return
|
|
}
|
|
|
|
if handler, ok := cmd.Handlers[commandName]; ok {
|
|
// mcp 是独立 stdio 进程，禁止初始化 DB/日志，避免污染协议流
|
|
if commandName != "-v" && commandName != "-V" && commandName != "version" && commandName != "mcp" {
|
|
bootstrap.InitBasicForCmd() // 专为命令行工具定制启动基础环境，屏蔽后台启动刷屏日志
|
|
}
|
|
handler(os.Args[2:])
|
|
return
|
|
}
|
|
|
|
fmt.Printf("Unknown command: %s\n", commandName)
|
|
printHelp()
|
|
os.Exit(1)
|
|
}
|