2022-07-28 23:00:05 +08:00
|
|
|
package node
|
2022-06-02 01:35:41 +08:00
|
|
|
|
|
|
|
import (
|
2022-08-16 13:04:33 +08:00
|
|
|
"github.com/Yuzuki616/V2bX/api/panel"
|
2022-07-28 23:00:05 +08:00
|
|
|
"github.com/Yuzuki616/V2bX/conf"
|
2022-08-12 18:02:06 +08:00
|
|
|
"github.com/Yuzuki616/V2bX/core"
|
2022-06-02 01:35:41 +08:00
|
|
|
)
|
|
|
|
|
2022-07-28 23:00:05 +08:00
|
|
|
type Node struct {
|
2023-05-16 09:15:29 +08:00
|
|
|
controllers []*Controller
|
2022-06-02 01:35:41 +08:00
|
|
|
}
|
|
|
|
|
2022-10-10 11:31:15 +08:00
|
|
|
func New() *Node {
|
|
|
|
return &Node{}
|
2022-06-02 01:35:41 +08:00
|
|
|
}
|
|
|
|
|
2022-10-10 11:31:15 +08:00
|
|
|
func (n *Node) Start(nodes []*conf.NodeConfig, core *core.Core) error {
|
2023-05-16 09:15:29 +08:00
|
|
|
n.controllers = make([]*Controller, len(nodes))
|
2022-10-10 11:31:15 +08:00
|
|
|
for i, c := range nodes {
|
2022-12-18 23:31:06 +08:00
|
|
|
p, err := panel.New(c.ApiConfig)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2022-10-10 11:31:15 +08:00
|
|
|
// Register controller service
|
2023-05-16 09:15:29 +08:00
|
|
|
n.controllers[i] = NewController(core, p, c.ControllerConfig)
|
2022-12-18 23:31:06 +08:00
|
|
|
err = n.controllers[i].Start()
|
2022-10-10 11:31:15 +08:00
|
|
|
if err != nil {
|
|
|
|
return err
|
2022-09-07 23:02:02 +08:00
|
|
|
}
|
|
|
|
}
|
2022-06-02 01:35:41 +08:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2022-10-10 11:31:15 +08:00
|
|
|
func (n *Node) Close() {
|
|
|
|
for _, c := range n.controllers {
|
|
|
|
err := c.Close()
|
2022-10-10 08:59:00 +08:00
|
|
|
if err != nil {
|
2022-10-10 11:31:15 +08:00
|
|
|
panic(err)
|
2022-10-10 08:59:00 +08:00
|
|
|
}
|
|
|
|
}
|
2022-10-10 11:31:15 +08:00
|
|
|
n.controllers = nil
|
2022-06-02 01:35:41 +08:00
|
|
|
}
|