mirror of
https://github.com/InazumaV/Ratte.git
synced 2026-02-04 04:30:09 +00:00
feat: use wire
This commit is contained in:
Yuzuki616
215
cmd/ratte/run.go
215
cmd/ratte/run.go
@@ -1,17 +1,11 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/InazumaV/Ratte-Interface/core"
|
||||
"github.com/InazumaV/Ratte-Interface/panel"
|
||||
"github.com/InazumaV/Ratte/acme"
|
||||
"github.com/InazumaV/Ratte/boot"
|
||||
"github.com/InazumaV/Ratte/conf"
|
||||
"github.com/InazumaV/Ratte/handler"
|
||||
"github.com/InazumaV/Ratte/trigger"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"github.com/spf13/cobra"
|
||||
"os"
|
||||
"os/exec"
|
||||
"os/signal"
|
||||
"runtime"
|
||||
"syscall"
|
||||
@@ -32,14 +26,6 @@ func init() {
|
||||
command.AddCommand(&runCommand)
|
||||
}
|
||||
|
||||
var (
|
||||
cores map[string]*core.PluginClient
|
||||
panels map[string]*panel.PluginClient
|
||||
acmes map[string]*acme.Acme
|
||||
handlers []*handler.Handler
|
||||
triggers []*trigger.Trigger
|
||||
)
|
||||
|
||||
func runHandle(_ *cobra.Command, _ []string) {
|
||||
c := conf.New(config)
|
||||
log.WithField("path", config).Info("Load config...")
|
||||
@@ -49,74 +35,36 @@ func runHandle(_ *cobra.Command, _ []string) {
|
||||
}
|
||||
log.WithField("path", config).Info("Loaded.")
|
||||
|
||||
log.Info("Init core plugin...")
|
||||
err = startCores(c.Plugin.Core, c.Core)
|
||||
log.Info("Init...")
|
||||
b, err := boot.InitBoot(c)
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Init core plugin failed")
|
||||
log.WithError(err).Fatal("Init failed")
|
||||
}
|
||||
log.Info("Done.")
|
||||
log.Info("Init done.")
|
||||
|
||||
log.Info("Init panel plugin...")
|
||||
err = startPanel(c.Plugin.Panel)
|
||||
log.Info("Start...")
|
||||
err = b.Start()
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Init panel plugin failed")
|
||||
log.WithError(err).Fatal("Start failed")
|
||||
}
|
||||
log.Info("Done.")
|
||||
|
||||
log.Info("Init acme...")
|
||||
// new acme
|
||||
acmes = make(map[string]*acme.Acme)
|
||||
for _, a := range c.Acme {
|
||||
ac, err := acme.NewAcme(&a)
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("New acme failed")
|
||||
}
|
||||
acmes[a.Name] = ac
|
||||
}
|
||||
log.Info("Done.")
|
||||
|
||||
log.Info("Starting...")
|
||||
// new node
|
||||
err = startTriggerAndHandler(c.Node)
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Start trigger and handler failed")
|
||||
}
|
||||
log.Info("Started.")
|
||||
|
||||
c.SetErrorHandler(func(err error) {
|
||||
log.WithFields(map[string]interface{}{
|
||||
"error": err,
|
||||
"Services": "ConfWatcher",
|
||||
}).Error("")
|
||||
}).Error("watch config error")
|
||||
})
|
||||
c.SetEventHandler(func(event uint, target ...string) {
|
||||
l := log.WithFields(map[string]interface{}{
|
||||
"Service": "ConfWatcher",
|
||||
})
|
||||
l.Info("Config changed, restart...")
|
||||
err := closeTriggerAndHandler()
|
||||
b, err := boot.InitBoot(c)
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Close trigger and handler failed")
|
||||
l.WithError(err).Fatal("Init boot failed")
|
||||
}
|
||||
err = closeCore()
|
||||
err = b.Start()
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Close core failed")
|
||||
}
|
||||
err = closePanel()
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Close panel failed")
|
||||
}
|
||||
err = startCores(c.Plugin.Core, c.Core)
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Start core failed")
|
||||
}
|
||||
err = startPanel(c.Plugin.Panel)
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Start panel failed")
|
||||
}
|
||||
err = startTriggerAndHandler(c.Node)
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Start trigger and handler failed")
|
||||
l.WithError(err).Fatal("Start boot failed")
|
||||
}
|
||||
l.Info("Restart Done.")
|
||||
})
|
||||
@@ -124,145 +72,14 @@ func runHandle(_ *cobra.Command, _ []string) {
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Watch config failed")
|
||||
}
|
||||
log.Info("Start done.")
|
||||
|
||||
runtime.GC()
|
||||
sig := make(chan os.Signal, 0)
|
||||
sig := make(chan os.Signal)
|
||||
signal.Notify(sig, syscall.SIGINT, syscall.SIGKILL, syscall.SIGTERM)
|
||||
<-sig
|
||||
// clear
|
||||
log.Info("Shutting down...")
|
||||
err = closeTriggerAndHandler()
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Close trigger and handler failed")
|
||||
}
|
||||
err = closeCore()
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Close core failed")
|
||||
}
|
||||
err = closePanel()
|
||||
if err != nil {
|
||||
log.WithError(err).Fatal("Close panel failed")
|
||||
}
|
||||
b.Close()
|
||||
log.Info("Done.")
|
||||
}
|
||||
|
||||
func startCores(coresP []conf.Plugin, cc []conf.Core) error {
|
||||
// new cores
|
||||
cores = make(map[string]*core.PluginClient, len(coresP))
|
||||
for _, ccV := range cc {
|
||||
for _, co := range coresP {
|
||||
if co.Name != ccV.Type {
|
||||
continue
|
||||
}
|
||||
c, err := core.NewClient(nil, exec.Command(co.Path))
|
||||
if err != nil {
|
||||
return fmt.Errorf("new core error: %w", err)
|
||||
}
|
||||
err = c.Start(ccV.DataPath, ccV.Config)
|
||||
if err != nil {
|
||||
return fmt.Errorf("start core error: %w", err)
|
||||
}
|
||||
cores[co.Name] = c
|
||||
break
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func closeCore() error {
|
||||
for _, c := range cores {
|
||||
err := c.Close()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func startPanel(panelsP []conf.Plugin) error {
|
||||
panels = make(map[string]*panel.PluginClient, len(panelsP))
|
||||
for _, p := range panelsP {
|
||||
pn, err := panel.NewClient(nil, exec.Command(p.Path))
|
||||
if err != nil {
|
||||
return fmt.Errorf("new panel error: %w", err)
|
||||
}
|
||||
panels[p.Name] = pn
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func closePanel() error {
|
||||
for _, p := range panels {
|
||||
err := p.Close()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func startTriggerAndHandler(c []conf.Node) error {
|
||||
triggers = make([]*trigger.Trigger, 0, len(c))
|
||||
handlers = make([]*handler.Handler, 0, len(c))
|
||||
for _, nd := range c {
|
||||
var co core.Core
|
||||
if c, ok := cores[nd.Options.Core]; ok {
|
||||
co = c
|
||||
} else {
|
||||
return fmt.Errorf("unknown core name: %s", nd.Options.Core)
|
||||
}
|
||||
var pl panel.Panel
|
||||
if p, ok := panels[nd.Options.Panel]; ok {
|
||||
pl = p
|
||||
} else {
|
||||
return fmt.Errorf("")
|
||||
}
|
||||
var ac *acme.Acme
|
||||
if len(acmes) != 0 {
|
||||
if a, ok := acmes[nd.Options.Acme]; ok {
|
||||
ac = a
|
||||
} else {
|
||||
return fmt.Errorf("unknown acme name: %s", nd.Options.Acme)
|
||||
}
|
||||
}
|
||||
|
||||
h := handler.New(co, pl, nd.Name, ac, log.WithFields(
|
||||
map[string]interface{}{
|
||||
"node": nd.Name,
|
||||
"service": "handler",
|
||||
},
|
||||
), &nd.Options)
|
||||
handlers = append(handlers, h)
|
||||
tr, err := trigger.NewTrigger(log.WithFields(
|
||||
map[string]interface{}{
|
||||
"node": nd.Name,
|
||||
"service": "trigger",
|
||||
},
|
||||
), &nd.Trigger, h, pl, &nd.Remote)
|
||||
if err != nil {
|
||||
return fmt.Errorf("new trigger error: %w", err)
|
||||
}
|
||||
triggers = append(triggers, tr)
|
||||
err = tr.Start()
|
||||
if err != nil {
|
||||
return fmt.Errorf("start trigger error: %w", err)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func closeTriggerAndHandler() error {
|
||||
for _, t := range triggers {
|
||||
err := t.Close()
|
||||
if err != nil {
|
||||
return fmt.Errorf("close trigger error: %w", err)
|
||||
}
|
||||
}
|
||||
for _, h := range handlers {
|
||||
err := h.Close()
|
||||
if err != nil {
|
||||
return fmt.Errorf("close handler error: %w", err)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user