mirror of
https://github.com/oarkflow/mq.git
synced 2025-11-01 05:32:42 +08:00
37 lines
1.1 KiB
Go
37 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/oarkflow/cli"
|
|
"github.com/oarkflow/cli/console"
|
|
"github.com/oarkflow/cli/contracts"
|
|
|
|
"github.com/oarkflow/mq"
|
|
"github.com/oarkflow/mq/dag"
|
|
"github.com/oarkflow/mq/handlers"
|
|
"github.com/oarkflow/mq/services"
|
|
dagConsole "github.com/oarkflow/mq/services/console"
|
|
)
|
|
|
|
func main() {
|
|
handlers.Init()
|
|
brokerAddr := ":5051"
|
|
loader := services.NewLoader("config")
|
|
loader.Load()
|
|
serverApp := fiber.New(fiber.Config{EnablePrintRoutes: true})
|
|
services.Setup(loader, serverApp, brokerAddr)
|
|
cli.Run("mq", "v0.0.1", func(client contracts.Cli) []contracts.Command {
|
|
return []contracts.Command{
|
|
console.NewListCommand(client),
|
|
dagConsole.NewRunHandler(loader.UserConfig, loader.ParsedPath, brokerAddr),
|
|
dagConsole.NewRunServer(serverApp),
|
|
}
|
|
})
|
|
}
|
|
|
|
func init() {
|
|
dag.AddHandler("render-html", func(id string) mq.Processor { return handlers.NewRenderHTMLNode(id) })
|
|
dag.AddHandler("condition", func(id string) mq.Processor { return handlers.NewCondition(id) })
|
|
dag.AddHandler("output", func(id string) mq.Processor { return handlers.NewOutputHandler(id) })
|
|
}
|