2022-07-28 23:00:05 +08:00
|
|
|
package node
|
2022-06-02 01:35:41 +08:00
|
|
|
|
|
|
|
import (
|
2023-06-09 12:36:49 +08:00
|
|
|
"fmt"
|
2023-07-29 18:47:47 +08:00
|
|
|
|
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"
|
2023-06-08 01:18:56 +08:00
|
|
|
vCore "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
|
|
|
}
|
|
|
|
|
2023-06-08 01:18:56 +08:00
|
|
|
func (n *Node) Start(nodes []*conf.NodeConfig, core vCore.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-07-29 18:47:47 +08:00
|
|
|
n.controllers[i] = NewController(core, p, c.Options)
|
2022-12-18 23:31:06 +08:00
|
|
|
err = n.controllers[i].Start()
|
2022-10-10 11:31:15 +08:00
|
|
|
if err != nil {
|
2023-06-09 12:36:49 +08:00
|
|
|
return fmt.Errorf("start node controller [%s-%s-%d] error: %s",
|
|
|
|
c.ApiConfig.NodeType,
|
|
|
|
c.ApiConfig.APIHost,
|
|
|
|
c.ApiConfig.NodeID,
|
|
|
|
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
|
|
|
}
|