[add] show tasks and schedules when starting

This commit is contained in:
Max 2022-08-22 15:50:15 +08:00
parent 3d21a9e19e
commit 8bcab3230f
2 changed files with 108 additions and 32 deletions

View file

@ -11,7 +11,9 @@ import (
"github.com/fatih/color"
"github.com/spf13/cobra"
"github.com/yaoapp/gou"
"github.com/yaoapp/gou/task"
"github.com/yaoapp/gou/websocket"
"github.com/yaoapp/kun/log"
"github.com/yaoapp/yao/config"
"github.com/yaoapp/yao/engine"
"github.com/yaoapp/yao/service"
@ -54,34 +56,6 @@ var startCmd = &cobra.Command{
host = "127.0.0.1"
}
if mode == "development" || mode == "production" {
fmt.Println(color.WhiteString("\n---------------------------------"))
fmt.Println(color.WhiteString(L("API List")))
fmt.Println(color.WhiteString("---------------------------------"))
for _, api := range gou.APIs { // API信息
if len(api.HTTP.Paths) <= 0 {
continue
}
fmt.Printf(color.CyanString("\n%s(%d)\n", api.Name, len(api.HTTP.Paths)))
for _, p := range api.HTTP.Paths {
fmt.Println(
colorMehtod(p.Method),
color.WhiteString(filepath.Join("/api", api.HTTP.Group, p.Path)),
"\tprocess:", p.Process)
}
}
fmt.Printf(color.CyanString("\n%s(%d)\n", "WebSocket", len(websocket.Upgraders)))
for name, upgrader := range websocket.Upgraders { // WebSocket
fmt.Println(
colorMehtod("GET"),
color.WhiteString(filepath.Join("/websocket", name)),
"\tprocess:", upgrader.Process)
}
}
fmt.Println(color.WhiteString("\n---------------------------------"))
fmt.Println(color.WhiteString(share.App.Name), color.WhiteString(share.App.Version), mode)
fmt.Println(color.WhiteString("---------------------------------"))
@ -95,12 +69,26 @@ var startCmd = &cobra.Command{
fmt.Println(color.WhiteString(L("API")), color.GreenString(" http://%s%s/api", host, port))
fmt.Println(color.WhiteString(L("Listening")), color.GreenString(" %s:%d", config.Conf.Host, config.Conf.Port))
fmt.Println("")
// 调试模式
// development mode
if mode == "development" {
printApis(false)
printTasks(false)
printSchedules(false)
// Watching
fmt.Println(color.WhiteString("\n---------------------------------"))
fmt.Println(color.WhiteString(L("Watching")))
fmt.Println(color.WhiteString("---------------------------------"))
service.Watch(config.Conf)
}
if mode == "production" {
printApis(true)
printTasks(true)
printSchedules(true)
}
// Start server
go service.Start()
fmt.Println(color.GreenString(L("✨LISTENING✨")))
@ -116,6 +104,94 @@ var startCmd = &cobra.Command{
},
}
func printSchedules(silent bool) {
if silent {
for name, sch := range gou.Schedules {
process := fmt.Sprintf("Process: %s", sch.Process)
if sch.TaskName != "" {
process = fmt.Sprintf("Task: %s", sch.TaskName)
}
log.Info("[Schedule] %s %s %s %s", sch.Schedule, name, sch.Name, process)
}
return
}
fmt.Println(color.WhiteString("\n---------------------------------"))
fmt.Println(color.WhiteString(L("Schedules List (%d)"), len(gou.Schedules)))
fmt.Println(color.WhiteString("---------------------------------"))
for name, sch := range gou.Schedules {
process := fmt.Sprintf("Process: %s", sch.Process)
if sch.TaskName != "" {
process = fmt.Sprintf("Task: %s", sch.TaskName)
}
fmt.Printf(color.CyanString("[Schedule] %s %s", sch.Schedule, name))
fmt.Printf(color.WhiteString("\t%s\t%s\n", sch.Name, process))
}
}
func printTasks(silent bool) {
if silent {
for _, t := range task.Tasks {
log.Info("[Task] %s workers:%d", t.Option.Name, t.Option.WorkerNums)
}
return
}
fmt.Println(color.WhiteString("\n---------------------------------"))
fmt.Println(color.WhiteString(L("Tasks List (%d)"), len(task.Tasks)))
fmt.Println(color.WhiteString("---------------------------------"))
for _, t := range task.Tasks {
fmt.Printf(color.CyanString("[Task] %s", t.Option.Name))
fmt.Printf(color.WhiteString("\t workers: %d\n", t.Option.WorkerNums))
}
}
func printApis(silent bool) {
if silent {
for _, api := range gou.APIs {
if len(api.HTTP.Paths) <= 0 {
continue
}
log.Info("[API] %s(%d)", api.Name, len(api.HTTP.Paths))
for _, p := range api.HTTP.Paths {
log.Info("%s %s %s", p.Method, filepath.Join("/api", api.HTTP.Group, p.Path), p.Process)
}
}
for name, upgrader := range websocket.Upgraders { // WebSocket
log.Info("[WebSocket] GET /websocket/%s process:%s", name, upgrader.Process)
}
return
}
fmt.Println(color.WhiteString("\n---------------------------------"))
fmt.Println(color.WhiteString(L("API List")))
fmt.Println(color.WhiteString("---------------------------------"))
for _, api := range gou.APIs { // API信息
if len(api.HTTP.Paths) <= 0 {
continue
}
fmt.Printf(color.CyanString("\n%s(%d)\n", api.Name, len(api.HTTP.Paths)))
for _, p := range api.HTTP.Paths {
fmt.Println(
colorMehtod(p.Method),
color.WhiteString(filepath.Join("/api", api.HTTP.Group, p.Path)),
"\tprocess:", p.Process)
}
}
fmt.Printf(color.CyanString("\n%s(%d)\n", "WebSocket", len(websocket.Upgraders)))
for name, upgrader := range websocket.Upgraders { // WebSocket
fmt.Println(
colorMehtod("GET"),
color.WhiteString(filepath.Join("/websocket", name)),
"\tprocess:", upgrader.Process)
}
}
func colorMehtod(method string) string {
method = strings.ToUpper(method)
switch method {

View file

@ -28,7 +28,7 @@ function Compile(name, dsl) {
* @param {DSL} dsl
*/
function OnLoad(name, dsl) {
console.log(name, dsl);
log.Info("[Widget] dyform.OnLoad %s %v", name, dsl);
}
/**